FlatCAMGrbEditor.py 80 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024
  1. from PyQt5 import QtGui, QtCore, QtWidgets
  2. from PyQt5.QtCore import Qt, QSettings
  3. from shapely.geometry import LineString, LinearRing, MultiLineString
  4. from shapely.ops import cascaded_union
  5. import shapely.affinity as affinity
  6. from numpy import arctan2, Inf, array, sqrt, sign, dot
  7. from rtree import index as rtindex
  8. import threading, time
  9. import copy
  10. from camlib import *
  11. from flatcamGUI.GUIElements import FCEntry, FCComboBox, FCTable, FCDoubleSpinner, LengthEntry, RadioSet, SpinBoxDelegate
  12. from flatcamEditors.FlatCAMGeoEditor import FCShapeTool, DrawTool, DrawToolShape, DrawToolUtilityShape, FlatCAMGeoEditor
  13. from FlatCAMObj import FlatCAMGerber
  14. import gettext
  15. import FlatCAMTranslation as fcTranslate
  16. fcTranslate.apply_language('strings')
  17. import builtins
  18. if '_' not in builtins.__dict__:
  19. _ = gettext.gettext
  20. class FCApertureResize(FCShapeTool):
  21. def __init__(self, draw_app):
  22. DrawTool.__init__(self, draw_app)
  23. self.name = 'aperture_resize'
  24. self.draw_app.app.inform.emit(_("Click on the Apertures to resize ..."))
  25. self.resize_dia = None
  26. self.draw_app.resize_frame.show()
  27. self.points = None
  28. self.selected_dia_list = []
  29. self.current_storage = None
  30. self.geometry = []
  31. self.destination_storage = None
  32. self.draw_app.resize_btn.clicked.connect(self.make)
  33. # Switch notebook to Selected page
  34. self.draw_app.app.ui.notebook.setCurrentWidget(self.draw_app.app.ui.selected_tab)
  35. def make(self):
  36. self.draw_app.is_modified = True
  37. try:
  38. new_dia = self.draw_app.resdrill_entry.get_value()
  39. except:
  40. self.draw_app.app.inform.emit(_("[ERROR_NOTCL] Resize drill(s) failed. Please enter a diameter for resize."))
  41. return
  42. if new_dia not in self.draw_app.olddia_newdia:
  43. self.destination_storage = FlatCAMGeoEditor.make_storage()
  44. self.draw_app.storage_dict[new_dia] = self.destination_storage
  45. # self.olddia_newdia dict keeps the evidence on current tools diameters as keys and gets updated on values
  46. # each time a tool diameter is edited or added
  47. self.draw_app.olddia_newdia[new_dia] = new_dia
  48. else:
  49. self.destination_storage = self.draw_app.storage_dict[new_dia]
  50. for index in self.draw_app.apertures_table.selectedIndexes():
  51. row = index.row()
  52. # on column 1 in tool tables we hold the diameters, and we retrieve them as strings
  53. # therefore below we convert to float
  54. dia_on_row = self.draw_app.apertures_table.item(row, 1).text()
  55. self.selected_dia_list.append(float(dia_on_row))
  56. # since we add a new tool, we update also the intial state of the tool_table through it's dictionary
  57. # we add a new entry in the tool2tooldia dict
  58. self.draw_app.tool2tooldia[len(self.draw_app.olddia_newdia)] = new_dia
  59. sel_shapes_to_be_deleted = []
  60. for sel_dia in self.selected_dia_list:
  61. self.current_storage = self.draw_app.storage_dict[sel_dia]
  62. for select_shape in self.draw_app.get_selected():
  63. if select_shape in self.current_storage.get_objects():
  64. factor = new_dia / sel_dia
  65. self.geometry.append(
  66. DrawToolShape(affinity.scale(select_shape.geo, xfact=factor, yfact=factor, origin='center'))
  67. )
  68. self.current_storage.remove(select_shape)
  69. # a hack to make the tool_table display less drills per diameter when shape(drill) is deleted
  70. # self.points_edit it's only useful first time when we load the data into the storage
  71. # but is still used as reference when building tool_table in self.build_ui()
  72. # the number of drills displayed in column 2 is just a len(self.points_edit) therefore
  73. # deleting self.points_edit elements (doesn't matter who but just the number)
  74. # solved the display issue.
  75. del self.draw_app.points_edit[sel_dia][0]
  76. sel_shapes_to_be_deleted.append(select_shape)
  77. self.draw_app.on_exc_shape_complete(self.destination_storage)
  78. # a hack to make the tool_table display more drills per diameter when shape(drill) is added
  79. # self.points_edit it's only useful first time when we load the data into the storage
  80. # but is still used as reference when building tool_table in self.build_ui()
  81. # the number of drills displayed in column 2 is just a len(self.points_edit) therefore
  82. # deleting self.points_edit elements (doesn't matter who but just the number)
  83. # solved the display issue.
  84. if new_dia not in self.draw_app.points_edit:
  85. self.draw_app.points_edit[new_dia] = [(0, 0)]
  86. else:
  87. self.draw_app.points_edit[new_dia].append((0,0))
  88. self.geometry = []
  89. # if following the resize of the drills there will be no more drills for the selected tool then
  90. # delete that tool
  91. if not self.draw_app.points_edit[sel_dia]:
  92. self.draw_app.on_aperture_delete(sel_dia)
  93. for shp in sel_shapes_to_be_deleted:
  94. self.draw_app.selected.remove(shp)
  95. sel_shapes_to_be_deleted = []
  96. self.draw_app.build_ui()
  97. self.draw_app.plot_all()
  98. self.draw_app.resize_frame.hide()
  99. self.complete = True
  100. self.draw_app.app.inform.emit(_("[success] Done. Drill Resize completed."))
  101. # MS: always return to the Select Tool
  102. self.draw_app.select_tool("select")
  103. class FCApertureMove(FCShapeTool):
  104. def __init__(self, draw_app):
  105. DrawTool.__init__(self, draw_app)
  106. self.name = 'aperture_move'
  107. # self.shape_buffer = self.draw_app.shape_buffer
  108. self.origin = None
  109. self.destination = None
  110. self.selected_dia_list = []
  111. if self.draw_app.launched_from_shortcuts is True:
  112. self.draw_app.launched_from_shortcuts = False
  113. self.draw_app.app.inform.emit(_("Click on target location ..."))
  114. else:
  115. self.draw_app.app.inform.emit(_("Click on reference location ..."))
  116. self.current_storage = None
  117. self.geometry = []
  118. for index in self.draw_app.apertures_table.selectedIndexes():
  119. row = index.row()
  120. # on column 1 in tool tables we hold the diameters, and we retrieve them as strings
  121. # therefore below we convert to float
  122. dia_on_row = self.draw_app.apertures_table.item(row, 1).text()
  123. self.selected_dia_list.append(float(dia_on_row))
  124. # Switch notebook to Selected page
  125. self.draw_app.app.ui.notebook.setCurrentWidget(self.draw_app.app.ui.selected_tab)
  126. def set_origin(self, origin):
  127. self.origin = origin
  128. def click(self, point):
  129. if len(self.draw_app.get_selected()) == 0:
  130. return "Nothing to move."
  131. if self.origin is None:
  132. self.set_origin(point)
  133. self.draw_app.app.inform.emit(_("Click on target location ..."))
  134. return
  135. else:
  136. self.destination = point
  137. self.make()
  138. # MS: always return to the Select Tool
  139. self.draw_app.select_tool("select")
  140. return
  141. def make(self):
  142. # Create new geometry
  143. dx = self.destination[0] - self.origin[0]
  144. dy = self.destination[1] - self.origin[1]
  145. sel_shapes_to_be_deleted = []
  146. for sel_dia in self.selected_dia_list:
  147. self.current_storage = self.draw_app.storage_dict[sel_dia]
  148. for select_shape in self.draw_app.get_selected():
  149. if select_shape in self.current_storage.get_objects():
  150. self.geometry.append(DrawToolShape(affinity.translate(select_shape.geo, xoff=dx, yoff=dy)))
  151. self.current_storage.remove(select_shape)
  152. sel_shapes_to_be_deleted.append(select_shape)
  153. self.draw_app.on_exc_shape_complete(self.current_storage)
  154. self.geometry = []
  155. for shp in sel_shapes_to_be_deleted:
  156. self.draw_app.selected.remove(shp)
  157. sel_shapes_to_be_deleted = []
  158. self.draw_app.build_ui()
  159. self.draw_app.app.inform.emit(_("[success] Done. Drill(s) Move completed."))
  160. def utility_geometry(self, data=None):
  161. """
  162. Temporary geometry on screen while using this tool.
  163. :param data:
  164. :return:
  165. """
  166. geo_list = []
  167. if self.origin is None:
  168. return None
  169. if len(self.draw_app.get_selected()) == 0:
  170. return None
  171. dx = data[0] - self.origin[0]
  172. dy = data[1] - self.origin[1]
  173. for geom in self.draw_app.get_selected():
  174. geo_list.append(affinity.translate(geom.geo, xoff=dx, yoff=dy))
  175. return DrawToolUtilityShape(geo_list)
  176. class FCApertureCopy(FCApertureMove):
  177. def __init__(self, draw_app):
  178. FCApertureMove.__init__(self, draw_app)
  179. self.name = 'aperture_copy'
  180. def make(self):
  181. # Create new geometry
  182. dx = self.destination[0] - self.origin[0]
  183. dy = self.destination[1] - self.origin[1]
  184. sel_shapes_to_be_deleted = []
  185. for sel_dia in self.selected_dia_list:
  186. self.current_storage = self.draw_app.storage_dict[sel_dia]
  187. for select_shape in self.draw_app.get_selected():
  188. if select_shape in self.current_storage.get_objects():
  189. self.geometry.append(DrawToolShape(affinity.translate(select_shape.geo, xoff=dx, yoff=dy)))
  190. # add some fake drills into the self.draw_app.points_edit to update the drill count in tool table
  191. self.draw_app.points_edit[sel_dia].append((0, 0))
  192. sel_shapes_to_be_deleted.append(select_shape)
  193. self.draw_app.on_exc_shape_complete(self.current_storage)
  194. self.geometry = []
  195. for shp in sel_shapes_to_be_deleted:
  196. self.draw_app.selected.remove(shp)
  197. sel_shapes_to_be_deleted = []
  198. self.draw_app.build_ui()
  199. self.draw_app.app.inform.emit(_("[success] Done. Drill(s) copied."))
  200. class FCApertureSelect(DrawTool):
  201. def __init__(self, grb_editor_app):
  202. DrawTool.__init__(self, grb_editor_app)
  203. self.name = 'drill_select'
  204. self.grb_editor_app = grb_editor_app
  205. self.storage = self.grb_editor_app.storage_dict
  206. # self.selected = self.grb_editor_app.selected
  207. # here we store all shapes that were selected
  208. self.sel_storage = []
  209. self.grb_editor_app.resize_frame.hide()
  210. self.grb_editor_app.array_frame.hide()
  211. def click(self, point):
  212. key_modifier = QtWidgets.QApplication.keyboardModifiers()
  213. if self.grb_editor_app.app.defaults["global_mselect_key"] == 'Control':
  214. if key_modifier == Qt.ControlModifier:
  215. pass
  216. else:
  217. self.grb_editor_app.selected = []
  218. else:
  219. if key_modifier == Qt.ShiftModifier:
  220. pass
  221. else:
  222. self.grb_editor_app.selected = []
  223. def click_release(self, point):
  224. self.select_shapes(point)
  225. return ""
  226. def select_shapes(self, pos):
  227. self.grb_editor_app.apertures_table.clearSelection()
  228. for storage in self.grb_editor_app.storage_dict:
  229. for shape in self.grb_editor_app.storage_dict[storage]['solid_geometry']:
  230. if Point(pos).within(shape.geo):
  231. self.sel_storage.append(shape)
  232. xmin, ymin, xmax, ymax = self.bounds(self.sel_storage)
  233. if pos[0] < xmin or pos[0] > xmax or pos[1] < ymin or pos[1] > ymax:
  234. self.grb_editor_app.selected = []
  235. else:
  236. key_modifier = QtWidgets.QApplication.keyboardModifiers()
  237. if self.grb_editor_app.app.defaults["global_mselect_key"] == 'Control':
  238. # if CONTROL key is pressed then we add to the selected list the current shape but if it's already
  239. # in the selected list, we removed it. Therefore first click selects, second deselects.
  240. if key_modifier == Qt.ControlModifier:
  241. if closest_shape in self.grb_editor_app.selected:
  242. self.grb_editor_app.selected.remove(closest_shape)
  243. else:
  244. self.grb_editor_app.selected.append(closest_shape)
  245. else:
  246. self.grb_editor_app.selected = []
  247. self.grb_editor_app.selected.append(closest_shape)
  248. else:
  249. if key_modifier == Qt.ShiftModifier:
  250. if closest_shape in self.grb_editor_app.selected:
  251. self.grb_editor_app.selected.remove(closest_shape)
  252. else:
  253. self.grb_editor_app.selected.append(closest_shape)
  254. else:
  255. self.grb_editor_app.selected = []
  256. self.grb_editor_app.selected.append(closest_shape)
  257. # select the aperture of the selected shape in the tool table
  258. for storage in self.grb_editor_app.storage_dict:
  259. for shape_s in self.grb_editor_app.selected:
  260. if shape_s in self.grb_editor_app.storage_dict[storage]:
  261. for key in self.grb_editor_app.tool2tooldia:
  262. if self.grb_editor_app.tool2tooldia[key] == storage:
  263. item = self.grb_editor_app.apertures_table.item((key - 1), 1)
  264. self.grb_editor_app.apertures_table.setCurrentItem(item)
  265. # item.setSelected(True)
  266. # self.grb_editor_app.apertures_table.selectItem(key - 1)
  267. # midx = self.grb_editor_app.apertures_table.model().index((key - 1), 0)
  268. # self.grb_editor_app.apertures_table.setCurrentIndex(midx)
  269. self.draw_app.last_tool_selected = key
  270. # delete whatever is in selection storage, there is no longer need for those shapes
  271. self.sel_storage = []
  272. return ""
  273. class FlatCAMGrbEditor(QtCore.QObject):
  274. draw_shape_idx = -1
  275. def __init__(self, app):
  276. assert isinstance(app, FlatCAMApp.App), \
  277. "Expected the app to be a FlatCAMApp.App, got %s" % type(app)
  278. super(FlatCAMGrbEditor, self).__init__()
  279. self.app = app
  280. self.canvas = self.app.plotcanvas
  281. ## Current application units in Upper Case
  282. self.units = self.app.ui.general_defaults_form.general_app_group.units_radio.get_value().upper()
  283. self.grb_edit_widget = QtWidgets.QWidget()
  284. layout = QtWidgets.QVBoxLayout()
  285. self.grb_edit_widget.setLayout(layout)
  286. ## Page Title box (spacing between children)
  287. self.title_box = QtWidgets.QHBoxLayout()
  288. layout.addLayout(self.title_box)
  289. ## Page Title icon
  290. pixmap = QtGui.QPixmap('share/flatcam_icon32.png')
  291. self.icon = QtWidgets.QLabel()
  292. self.icon.setPixmap(pixmap)
  293. self.title_box.addWidget(self.icon, stretch=0)
  294. ## Title label
  295. self.title_label = QtWidgets.QLabel("<font size=5><b>%s</b></font>" % _('Gerber Editor'))
  296. self.title_label.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter)
  297. self.title_box.addWidget(self.title_label, stretch=1)
  298. ## Object name
  299. self.name_box = QtWidgets.QHBoxLayout()
  300. layout.addLayout(self.name_box)
  301. name_label = QtWidgets.QLabel(_("Name:"))
  302. self.name_box.addWidget(name_label)
  303. self.name_entry = FCEntry()
  304. self.name_box.addWidget(self.name_entry)
  305. ## Box for custom widgets
  306. # This gets populated in offspring implementations.
  307. self.custom_box = QtWidgets.QVBoxLayout()
  308. layout.addLayout(self.custom_box)
  309. # add a frame and inside add a vertical box layout. Inside this vbox layout I add all the Drills widgets
  310. # this way I can hide/show the frame
  311. self.apertures_frame = QtWidgets.QFrame()
  312. self.apertures_frame.setContentsMargins(0, 0, 0, 0)
  313. self.custom_box.addWidget(self.apertures_frame)
  314. self.apertures_box = QtWidgets.QVBoxLayout()
  315. self.apertures_box.setContentsMargins(0, 0, 0, 0)
  316. self.apertures_frame.setLayout(self.apertures_box)
  317. #### Gerber Apertures ####
  318. self.apertures_table_label = QtWidgets.QLabel(_('<b>Apertures:</b>'))
  319. self.apertures_table_label.setToolTip(
  320. _("Apertures Table for the Gerber Object.")
  321. )
  322. self.apertures_box.addWidget(self.apertures_table_label)
  323. self.apertures_table = FCTable()
  324. # delegate = SpinBoxDelegate(units=self.units)
  325. # self.apertures_table.setItemDelegateForColumn(1, delegate)
  326. self.apertures_box.addWidget(self.apertures_table)
  327. self.apertures_table.setColumnCount(5)
  328. self.apertures_table.setHorizontalHeaderLabels(['#', _('Code'), _('Type'), _('Size'), _('Dim')])
  329. self.apertures_table.setSortingEnabled(False)
  330. self.apertures_table.horizontalHeaderItem(0).setToolTip(
  331. _("Index"))
  332. self.apertures_table.horizontalHeaderItem(1).setToolTip(
  333. _("Aperture Code"))
  334. self.apertures_table.horizontalHeaderItem(2).setToolTip(
  335. _("Type of aperture: circular, rectangle, macros etc"))
  336. self.apertures_table.horizontalHeaderItem(4).setToolTip(
  337. _("Aperture Size:"))
  338. self.apertures_table.horizontalHeaderItem(4).setToolTip(
  339. _("Aperture Dimensions:\n"
  340. " - (width, height) for R, O type.\n"
  341. " - (dia, nVertices) for P type"))
  342. self.empty_label = QtWidgets.QLabel('')
  343. self.apertures_box.addWidget(self.empty_label)
  344. #### Add a new Tool ####
  345. self.addaperture_label = QtWidgets.QLabel('<b>%s</b>' % _('Add/Delete Aperture'))
  346. self.addaperture_label.setToolTip(
  347. _("Add/Delete an aperture to the aperture list")
  348. )
  349. self.apertures_box.addWidget(self.addaperture_label)
  350. grid1 = QtWidgets.QGridLayout()
  351. self.apertures_box.addLayout(grid1)
  352. addaperture_entry_lbl = QtWidgets.QLabel(_('Aperture Size:'))
  353. addaperture_entry_lbl.setToolTip(
  354. _("Size for the new aperture")
  355. )
  356. grid1.addWidget(addaperture_entry_lbl, 0, 0)
  357. hlay = QtWidgets.QHBoxLayout()
  358. self.addtool_entry = FCEntry()
  359. self.addtool_entry.setValidator(QtGui.QDoubleValidator(0.0001, 99.9999, 4))
  360. hlay.addWidget(self.addtool_entry)
  361. self.addaperture_btn = QtWidgets.QPushButton(_('Add Aperture'))
  362. self.addaperture_btn.setToolTip(
  363. _( "Add a new aperture to the aperture list")
  364. )
  365. self.addaperture_btn.setFixedWidth(80)
  366. hlay.addWidget(self.addaperture_btn)
  367. grid1.addLayout(hlay, 0, 1)
  368. grid2 = QtWidgets.QGridLayout()
  369. self.apertures_box.addLayout(grid2)
  370. self.delaperture_btn = QtWidgets.QPushButton(_('Delete Aperture'))
  371. self.delaperture_btn.setToolTip(
  372. _( "Delete a aperture in the aperture list")
  373. )
  374. grid2.addWidget(self.delaperture_btn, 0, 1)
  375. # add a frame and inside add a vertical box layout. Inside this vbox layout I add all the aperture widgets
  376. # this way I can hide/show the frame
  377. self.resize_frame = QtWidgets.QFrame()
  378. self.resize_frame.setContentsMargins(0, 0, 0, 0)
  379. self.apertures_box.addWidget(self.resize_frame)
  380. self.resize_box = QtWidgets.QVBoxLayout()
  381. self.resize_box.setContentsMargins(0, 0, 0, 0)
  382. self.resize_frame.setLayout(self.resize_box)
  383. #### Resize a aperture ####
  384. self.emptyresize_label = QtWidgets.QLabel('')
  385. self.resize_box.addWidget(self.emptyresize_label)
  386. self.apertureresize_label = QtWidgets.QLabel('<b>%s</b>' % _("Resize Aperture"))
  387. self.apertureresize_label.setToolTip(
  388. _("Resize a aperture or a selection of apertures.")
  389. )
  390. self.resize_box.addWidget(self.apertureresize_label)
  391. grid3 = QtWidgets.QGridLayout()
  392. self.resize_box.addLayout(grid3)
  393. res_entry_lbl = QtWidgets.QLabel(_('Resize Dia:'))
  394. res_entry_lbl.setToolTip(
  395. _( "Size to resize to.")
  396. )
  397. grid3.addWidget(res_entry_lbl, 0, 0)
  398. hlay2 = QtWidgets.QHBoxLayout()
  399. self.resdrill_entry = LengthEntry()
  400. hlay2.addWidget(self.resdrill_entry)
  401. self.resize_btn = QtWidgets.QPushButton(_('Resize'))
  402. self.resize_btn.setToolTip(
  403. _("Resize drill(s)")
  404. )
  405. self.resize_btn.setFixedWidth(80)
  406. hlay2.addWidget(self.resize_btn)
  407. grid3.addLayout(hlay2, 0, 1)
  408. self.resize_frame.hide()
  409. # add a frame and inside add a vertical box layout. Inside this vbox layout I add
  410. # all the add drill array widgets
  411. # this way I can hide/show the frame
  412. self.array_frame = QtWidgets.QFrame()
  413. self.array_frame.setContentsMargins(0, 0, 0, 0)
  414. self.apertures_box.addWidget(self.array_frame)
  415. self.array_box = QtWidgets.QVBoxLayout()
  416. self.array_box.setContentsMargins(0, 0, 0, 0)
  417. self.array_frame.setLayout(self.array_box)
  418. #### Add DRILL Array ####
  419. self.emptyarray_label = QtWidgets.QLabel('')
  420. self.array_box.addWidget(self.emptyarray_label)
  421. self.drillarray_label = QtWidgets.QLabel('<b>%s</b>' % _("Add Drill Array"))
  422. self.drillarray_label.setToolTip(
  423. _("Add an array of drills (linear or circular array)")
  424. )
  425. self.array_box.addWidget(self.drillarray_label)
  426. self.array_type_combo = FCComboBox()
  427. self.array_type_combo.setToolTip(
  428. _( "Select the type of drills array to create.\n"
  429. "It can be Linear X(Y) or Circular")
  430. )
  431. self.array_type_combo.addItem(_("Linear"))
  432. self.array_type_combo.addItem(_("Circular"))
  433. self.array_box.addWidget(self.array_type_combo)
  434. self.array_form = QtWidgets.QFormLayout()
  435. self.array_box.addLayout(self.array_form)
  436. self.drill_array_size_label = QtWidgets.QLabel(_('Nr of drills:'))
  437. self.drill_array_size_label.setToolTip(
  438. _("Specify how many drills to be in the array.")
  439. )
  440. self.drill_array_size_label.setFixedWidth(100)
  441. self.drill_array_size_entry = LengthEntry()
  442. self.array_form.addRow(self.drill_array_size_label, self.drill_array_size_entry)
  443. self.array_linear_frame = QtWidgets.QFrame()
  444. self.array_linear_frame.setContentsMargins(0, 0, 0, 0)
  445. self.array_box.addWidget(self.array_linear_frame)
  446. self.linear_box = QtWidgets.QVBoxLayout()
  447. self.linear_box.setContentsMargins(0, 0, 0, 0)
  448. self.array_linear_frame.setLayout(self.linear_box)
  449. self.linear_form = QtWidgets.QFormLayout()
  450. self.linear_box.addLayout(self.linear_form)
  451. self.drill_axis_label = QtWidgets.QLabel(_('Direction:'))
  452. self.drill_axis_label.setToolTip(
  453. _("Direction on which the linear array is oriented:\n"
  454. "- 'X' - horizontal axis \n"
  455. "- 'Y' - vertical axis or \n"
  456. "- 'Angle' - a custom angle for the array inclination")
  457. )
  458. self.drill_axis_label.setFixedWidth(100)
  459. self.drill_axis_radio = RadioSet([{'label': 'X', 'value': 'X'},
  460. {'label': 'Y', 'value': 'Y'},
  461. {'label': _('Angle'), 'value': 'A'}])
  462. self.drill_axis_radio.set_value('X')
  463. self.linear_form.addRow(self.drill_axis_label, self.drill_axis_radio)
  464. self.drill_pitch_label = QtWidgets.QLabel(_('Pitch:'))
  465. self.drill_pitch_label.setToolTip(
  466. _("Pitch = Distance between elements of the array.")
  467. )
  468. self.drill_pitch_label.setFixedWidth(100)
  469. self.drill_pitch_entry = LengthEntry()
  470. self.linear_form.addRow(self.drill_pitch_label, self.drill_pitch_entry)
  471. self.linear_angle_label = QtWidgets.QLabel(_('Angle:'))
  472. self.linear_angle_label.setToolTip(
  473. _( "Angle at which the linear array is placed.\n"
  474. "The precision is of max 2 decimals.\n"
  475. "Min value is: -359.99 degrees.\n"
  476. "Max value is: 360.00 degrees.")
  477. )
  478. self.linear_angle_label.setFixedWidth(100)
  479. self.linear_angle_spinner = FCDoubleSpinner()
  480. self.linear_angle_spinner.set_precision(2)
  481. self.linear_angle_spinner.setRange(-359.99, 360.00)
  482. self.linear_form.addRow(self.linear_angle_label, self.linear_angle_spinner)
  483. self.array_circular_frame = QtWidgets.QFrame()
  484. self.array_circular_frame.setContentsMargins(0, 0, 0, 0)
  485. self.array_box.addWidget(self.array_circular_frame)
  486. self.circular_box = QtWidgets.QVBoxLayout()
  487. self.circular_box.setContentsMargins(0, 0, 0, 0)
  488. self.array_circular_frame.setLayout(self.circular_box)
  489. self.drill_direction_label = QtWidgets.QLabel(_('Direction:'))
  490. self.drill_direction_label.setToolTip(
  491. _( "Direction for circular array."
  492. "Can be CW = clockwise or CCW = counter clockwise.")
  493. )
  494. self.drill_direction_label.setFixedWidth(100)
  495. self.circular_form = QtWidgets.QFormLayout()
  496. self.circular_box.addLayout(self.circular_form)
  497. self.drill_direction_radio = RadioSet([{'label': 'CW', 'value': 'CW'},
  498. {'label': 'CCW.', 'value': 'CCW'}])
  499. self.drill_direction_radio.set_value('CW')
  500. self.circular_form.addRow(self.drill_direction_label, self.drill_direction_radio)
  501. self.drill_angle_label = QtWidgets.QLabel(_('Angle:'))
  502. self.drill_angle_label.setToolTip(
  503. _("Angle at which each element in circular array is placed.")
  504. )
  505. self.drill_angle_label.setFixedWidth(100)
  506. self.drill_angle_entry = LengthEntry()
  507. self.circular_form.addRow(self.drill_angle_label, self.drill_angle_entry)
  508. self.array_circular_frame.hide()
  509. self.linear_angle_spinner.hide()
  510. self.linear_angle_label.hide()
  511. self.array_frame.hide()
  512. self.apertures_box.addStretch()
  513. ## Toolbar events and properties
  514. self.tools_gerber = {
  515. "select": {"button": self.app.ui.select_drill_btn,
  516. "constructor": FCApertureSelect},
  517. "drill_resize": {"button": self.app.ui.resize_drill_btn,
  518. "constructor": FCApertureResize},
  519. "drill_copy": {"button": self.app.ui.copy_drill_btn,
  520. "constructor": FCApertureCopy},
  521. "drill_move": {"button": self.app.ui.move_drill_btn,
  522. "constructor": FCApertureMove},
  523. }
  524. ### Data
  525. self.active_tool = None
  526. self.storage_dict = {}
  527. self.current_storage = []
  528. # build the data from the Excellon point into a dictionary
  529. # {tool_dia: [geometry_in_points]}
  530. self.points_edit = {}
  531. self.sorted_apid =[]
  532. self.new_apertures = {}
  533. self.new_aperture_macros = {}
  534. # store here the plot promises, if empty the delayed plot will be activated
  535. self.grb_plot_promises = []
  536. # dictionary to store the tool_row and diameters in Tool_table
  537. # it will be updated everytime self.build_ui() is called
  538. self.olddia_newdia = {}
  539. self.tool2tooldia = {}
  540. # this will store the value for the last selected tool, for use after clicking on canvas when the selection
  541. # is cleared but as a side effect also the selected tool is cleared
  542. self.last_aperture_selected = None
  543. self.utility = []
  544. # this will flag if the Editor "tools" are launched from key shortcuts (True) or from menu toolbar (False)
  545. self.launched_from_shortcuts = False
  546. # this var will store the state of the toolbar before starting the editor
  547. self.toolbar_old_state = False
  548. self.app.ui.delete_drill_btn.triggered.connect(self.on_delete_btn)
  549. self.name_entry.returnPressed.connect(self.on_name_activate)
  550. self.addaperture_btn.clicked.connect(self.on_aperture_add)
  551. # self.addtool_entry.editingFinished.connect(self.on_tool_add)
  552. self.delaperture_btn.clicked.connect(self.on_aperture_delete)
  553. self.apertures_table.selectionModel().currentChanged.connect(self.on_row_selected)
  554. self.array_type_combo.currentIndexChanged.connect(self.on_array_type_combo)
  555. self.drill_axis_radio.activated_custom.connect(self.on_linear_angle_radio)
  556. self.app.ui.exc_resize_drill_menuitem.triggered.connect(self.exc_resize_drills)
  557. self.app.ui.exc_copy_drill_menuitem.triggered.connect(self.exc_copy_drills)
  558. self.app.ui.exc_delete_drill_menuitem.triggered.connect(self.on_delete_btn)
  559. self.app.ui.exc_move_drill_menuitem.triggered.connect(self.exc_move_drills)
  560. # Init GUI
  561. self.drill_array_size_entry.set_value(5)
  562. self.drill_pitch_entry.set_value(2.54)
  563. self.drill_angle_entry.set_value(12)
  564. self.drill_direction_radio.set_value('CW')
  565. self.drill_axis_radio.set_value('X')
  566. self.gerber_obj = None
  567. self.gerber_obj_options = {}
  568. # VisPy Visuals
  569. self.shapes = self.app.plotcanvas.new_shape_collection(layers=1)
  570. self.tool_shape = self.app.plotcanvas.new_shape_collection(layers=1)
  571. self.app.pool_recreated.connect(self.pool_recreated)
  572. # Remove from scene
  573. self.shapes.enabled = False
  574. self.tool_shape.enabled = False
  575. ## List of selected shapes.
  576. self.selected = []
  577. self.move_timer = QtCore.QTimer()
  578. self.move_timer.setSingleShot(True)
  579. self.key = None # Currently pressed key
  580. self.modifiers = None
  581. self.x = None # Current mouse cursor pos
  582. self.y = None
  583. # Current snapped mouse pos
  584. self.snap_x = None
  585. self.snap_y = None
  586. self.pos = None
  587. def make_callback(thetool):
  588. def f():
  589. self.on_tool_select(thetool)
  590. return f
  591. for tool in self.tools_gerber:
  592. self.tools_gerber[tool]["button"].triggered.connect(make_callback(tool)) # Events
  593. self.tools_gerber[tool]["button"].setCheckable(True) # Checkable
  594. self.options = {
  595. "global_gridx": 0.1,
  596. "global_gridy": 0.1,
  597. "snap_max": 0.05,
  598. "grid_snap": True,
  599. "corner_snap": False,
  600. "grid_gap_link": True
  601. }
  602. self.app.options_read_form()
  603. for option in self.options:
  604. if option in self.app.options:
  605. self.options[option] = self.app.options[option]
  606. # flag to show if the object was modified
  607. self.is_modified = False
  608. self.edited_obj_name = ""
  609. self.tool_row = 0
  610. # store the status of the editor so the Delete at object level will not work until the edit is finished
  611. self.editor_active = False
  612. def entry2option(option, entry):
  613. self.options[option] = float(entry.text())
  614. # store the status of the editor so the Delete at object level will not work until the edit is finished
  615. self.editor_active = False
  616. def pool_recreated(self, pool):
  617. self.shapes.pool = pool
  618. self.tool_shape.pool = pool
  619. def set_ui(self):
  620. # updated units
  621. self.units = self.app.ui.general_defaults_form.general_app_group.units_radio.get_value().upper()
  622. self.olddia_newdia.clear()
  623. self.tool2tooldia.clear()
  624. # update the olddia_newdia dict to make sure we have an updated state of the tool_table
  625. for key in self.storage_dict:
  626. self.olddia_newdia[key] = key
  627. sort_temp = []
  628. for aperture in self.olddia_newdia:
  629. sort_temp.append(int(aperture))
  630. self.sorted_apid = sorted(sort_temp)
  631. # populate self.intial_table_rows dict with the tool number as keys and tool diameters as values
  632. for i in range(len(self.sorted_apid)):
  633. tt_aperture = self.sorted_apid[i]
  634. self.tool2tooldia[i + 1] = tt_aperture
  635. def build_ui(self):
  636. try:
  637. # if connected, disconnect the signal from the slot on item_changed as it creates issues
  638. self.apertures_table.itemChanged.disconnect()
  639. except:
  640. pass
  641. # updated units
  642. self.units = self.app.ui.general_defaults_form.general_app_group.units_radio.get_value().upper()
  643. # make a new name for the new Excellon object (the one with edited content)
  644. self.edited_obj_name = self.gerber_obj.options['name']
  645. self.name_entry.set_value(self.edited_obj_name)
  646. if self.units == "IN":
  647. self.addtool_entry.set_value(0.039)
  648. else:
  649. self.addtool_entry.set_value(1.00)
  650. self.apertures_row = 0
  651. aper_no = self.apertures_row + 1
  652. sort = []
  653. for k, v in list(self.storage_dict.items()):
  654. sort.append(int(k))
  655. sorted_apertures = sorted(sort)
  656. sort = []
  657. for k, v in list(self.gerber_obj.aperture_macros.items()):
  658. sort.append(k)
  659. sorted_macros = sorted(sort)
  660. n = len(sorted_apertures) + len(sorted_macros)
  661. self.apertures_table.setRowCount(n)
  662. for ap_code in sorted_apertures:
  663. ap_code = str(ap_code)
  664. ap_id_item = QtWidgets.QTableWidgetItem('%d' % int(self.apertures_row + 1))
  665. ap_id_item.setFlags(QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
  666. self.apertures_table.setItem(self.apertures_row, 0, ap_id_item) # Tool name/id
  667. ap_code_item = QtWidgets.QTableWidgetItem(ap_code)
  668. ap_code_item.setFlags(QtCore.Qt.ItemIsEnabled)
  669. ap_type_item = QtWidgets.QTableWidgetItem(str(self.storage_dict[ap_code]['type']))
  670. ap_type_item.setFlags(QtCore.Qt.ItemIsEnabled)
  671. if str(self.storage_dict[ap_code]['type']) == 'R' or str(self.storage_dict[ap_code]['type']) == 'O':
  672. ap_dim_item = QtWidgets.QTableWidgetItem(
  673. '%.4f, %.4f' % (self.storage_dict[ap_code]['width'] * self.gerber_obj.file_units_factor,
  674. self.storage_dict[ap_code]['height'] * self.gerber_obj.file_units_factor
  675. )
  676. )
  677. ap_dim_item.setFlags(QtCore.Qt.ItemIsEnabled)
  678. elif str(self.storage_dict[ap_code]['type']) == 'P':
  679. ap_dim_item = QtWidgets.QTableWidgetItem(
  680. '%.4f, %.4f' % (self.storage_dict[ap_code]['diam'] * self.gerber_obj.file_units_factor,
  681. self.storage_dict[ap_code]['nVertices'] * self.gerber_obj.file_units_factor)
  682. )
  683. ap_dim_item.setFlags(QtCore.Qt.ItemIsEnabled)
  684. else:
  685. ap_dim_item = QtWidgets.QTableWidgetItem('')
  686. ap_dim_item.setFlags(QtCore.Qt.ItemIsEnabled)
  687. try:
  688. if self.storage_dict[ap_code]['size'] is not None:
  689. ap_size_item = QtWidgets.QTableWidgetItem('%.4f' %
  690. float(self.storage_dict[ap_code]['size'] *
  691. self.gerber_obj.file_units_factor))
  692. else:
  693. ap_size_item = QtWidgets.QTableWidgetItem('')
  694. except KeyError:
  695. ap_size_item = QtWidgets.QTableWidgetItem('')
  696. ap_size_item.setFlags(QtCore.Qt.ItemIsEnabled)
  697. self.apertures_table.setItem(self.apertures_row, 1, ap_code_item) # Aperture Code
  698. self.apertures_table.setItem(self.apertures_row, 2, ap_type_item) # Aperture Type
  699. self.apertures_table.setItem(self.apertures_row, 3, ap_size_item) # Aperture Dimensions
  700. self.apertures_table.setItem(self.apertures_row, 4, ap_dim_item) # Aperture Dimensions
  701. self.apertures_row += 1
  702. for ap_code in sorted_macros:
  703. ap_code = str(ap_code)
  704. ap_id_item = QtWidgets.QTableWidgetItem('%d' % int(self.apertures_row + 1))
  705. ap_id_item.setFlags(QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
  706. self.apertures_table.setItem(self.apertures_row, 0, ap_id_item) # Tool name/id
  707. ap_code_item = QtWidgets.QTableWidgetItem(ap_code)
  708. ap_type_item = QtWidgets.QTableWidgetItem('AM')
  709. ap_type_item.setFlags(QtCore.Qt.ItemIsEnabled)
  710. self.apertures_table.setItem(self.apertures_row, 1, ap_code_item) # Aperture Code
  711. self.apertures_table.setItem(self.apertures_row, 2, ap_type_item) # Aperture Type
  712. self.apertures_row += 1
  713. self.apertures_table.selectColumn(0)
  714. self.apertures_table.resizeColumnsToContents()
  715. self.apertures_table.resizeRowsToContents()
  716. vertical_header = self.apertures_table.verticalHeader()
  717. # vertical_header.setSectionResizeMode(QtWidgets.QHeaderView.ResizeToContents)
  718. vertical_header.hide()
  719. self.apertures_table.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
  720. horizontal_header = self.apertures_table.horizontalHeader()
  721. horizontal_header.setMinimumSectionSize(10)
  722. horizontal_header.setDefaultSectionSize(70)
  723. horizontal_header.setSectionResizeMode(0, QtWidgets.QHeaderView.Fixed)
  724. horizontal_header.resizeSection(0, 20)
  725. horizontal_header.setSectionResizeMode(1, QtWidgets.QHeaderView.ResizeToContents)
  726. horizontal_header.setSectionResizeMode(2, QtWidgets.QHeaderView.ResizeToContents)
  727. horizontal_header.setSectionResizeMode(3, QtWidgets.QHeaderView.ResizeToContents)
  728. horizontal_header.setSectionResizeMode(4, QtWidgets.QHeaderView.Stretch)
  729. self.apertures_table.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
  730. self.apertures_table.setSortingEnabled(False)
  731. self.apertures_table.setMinimumHeight(self.apertures_table.getHeight())
  732. self.apertures_table.setMaximumHeight(self.apertures_table.getHeight())
  733. # make sure no rows are selected so the user have to click the correct row, meaning selecting the correct tool
  734. self.apertures_table.clearSelection()
  735. # Remove anything else in the GUI Selected Tab
  736. self.app.ui.selected_scroll_area.takeWidget()
  737. # Put ourself in the GUI Selected Tab
  738. self.app.ui.selected_scroll_area.setWidget(self.grb_edit_widget)
  739. # Switch notebook to Selected page
  740. self.app.ui.notebook.setCurrentWidget(self.app.ui.selected_tab)
  741. # we reactivate the signals after the after the tool adding as we don't need to see the tool been populated
  742. self.apertures_table.itemChanged.connect(self.on_tool_edit)
  743. def on_aperture_add(self, apid=None):
  744. self.is_modified = True
  745. if apid:
  746. ap_id = apid
  747. else:
  748. try:
  749. ap_id = str(self.addtool_entry.get_value())
  750. except ValueError:
  751. return
  752. if ap_id not in self.olddia_newdia:
  753. self.storage_dict[ap_id] = {}
  754. self.storage_dict[ap_id]['type'] = 'C'
  755. self.storage_dict[ap_id]['size'] = 1
  756. self.storage_dict[ap_id]['solid_geometry'] = []
  757. self.storage_dict[ap_id]['follow_geometry'] = []
  758. # self.olddia_newdia dict keeps the evidence on current tools diameters as keys and gets updated on values
  759. # each time a tool diameter is edited or added
  760. self.olddia_newdia[ap_id] = ap_id
  761. else:
  762. self.app.inform.emit(_("[WARNING_NOTCL] Tool already in the original or actual tool list.\n"
  763. "Save and reedit Excellon if you need to add this tool. ")
  764. )
  765. return
  766. # since we add a new tool, we update also the initial state of the tool_table through it's dictionary
  767. # we add a new entry in the tool2tooldia dict
  768. self.tool2tooldia[len(self.olddia_newdia)] = ap_id
  769. self.app.inform.emit(_("[success] Added new tool with dia: {apid}").format(apid=str(ap_id)))
  770. self.build_ui()
  771. # make a quick sort through the tool2tooldia dict so we find which row to select
  772. row_to_be_selected = None
  773. for key in sorted(self.tool2tooldia):
  774. if self.tool2tooldia[key] == ap_id:
  775. row_to_be_selected = int(key) - 1
  776. break
  777. self.apertures_table.selectRow(row_to_be_selected)
  778. def on_aperture_delete(self, apid=None):
  779. self.is_modified = True
  780. deleted_tool_dia_list = []
  781. deleted_tool_offset_list = []
  782. try:
  783. if apid is None or apid is False:
  784. # deleted_tool_dia = float(self.apertures_table.item(self.apertures_table.currentRow(), 1).text())
  785. for index in self.apertures_table.selectionModel().selectedRows():
  786. row = index.row()
  787. deleted_tool_dia_list.append(self.apertures_table.item(row, 1).text())
  788. else:
  789. if isinstance(apid, list):
  790. for dd in apid:
  791. deleted_tool_dia_list.append(dd)
  792. else:
  793. deleted_tool_dia_list.append(apid)
  794. except:
  795. self.app.inform.emit(_("[WARNING_NOTCL] Select a tool in Tool Table"))
  796. return
  797. for deleted_tool_dia in deleted_tool_dia_list:
  798. # delete the storage used for that tool
  799. self.storage_dict.pop(deleted_tool_dia, None)
  800. # I've added this flag_del variable because dictionary don't like
  801. # having keys deleted while iterating through them
  802. flag_del = []
  803. for deleted_tool in self.tool2tooldia:
  804. if self.tool2tooldia[deleted_tool] == deleted_tool_dia:
  805. flag_del.append(deleted_tool)
  806. if flag_del:
  807. for tool_to_be_deleted in flag_del:
  808. # delete the tool
  809. self.tool2tooldia.pop(tool_to_be_deleted, None)
  810. flag_del = []
  811. self.olddia_newdia.pop(deleted_tool_dia, None)
  812. self.app.inform.emit(_("[success] Deleted aperture with code: {del_dia}").format(del_dia=str(deleted_tool_dia)))
  813. self.plot_all()
  814. self.build_ui()
  815. def on_tool_edit(self, item_changed):
  816. # if connected, disconnect the signal from the slot on item_changed as it creates issues
  817. self.apertures_table.itemChanged.disconnect()
  818. # self.apertures_table.selectionModel().currentChanged.disconnect()
  819. self.is_modified = True
  820. geometry = []
  821. current_table_dia_edited = None
  822. if self.apertures_table.currentItem() is not None:
  823. try:
  824. current_table_dia_edited = float(self.apertures_table.currentItem().text())
  825. except ValueError as e:
  826. log.debug("FlatCAMExcEditor.on_tool_edit() --> %s" % str(e))
  827. self.apertures_table.setCurrentItem(None)
  828. return
  829. row_of_item_changed = self.apertures_table.currentRow()
  830. # rows start with 0, tools start with 1 so we adjust the value by 1
  831. key_in_tool2tooldia = row_of_item_changed + 1
  832. dia_changed = self.tool2tooldia[key_in_tool2tooldia]
  833. # tool diameter is not used so we create a new tool with the desired diameter
  834. if current_table_dia_edited not in self.olddia_newdia.values():
  835. # update the dict that holds as keys our initial diameters and as values the edited diameters
  836. self.olddia_newdia[dia_changed] = current_table_dia_edited
  837. # update the dict that holds tool_no as key and tool_dia as value
  838. self.tool2tooldia[key_in_tool2tooldia] = current_table_dia_edited
  839. # update the tool offset
  840. modified_offset = self.gerber_obj.tool_offset.pop(dia_changed)
  841. self.gerber_obj.tool_offset[current_table_dia_edited] = modified_offset
  842. self.plot_all()
  843. else:
  844. # tool diameter is already in use so we move the drills from the prior tool to the new tool
  845. factor = current_table_dia_edited / dia_changed
  846. for shape in self.storage_dict[dia_changed].get_objects():
  847. geometry.append(DrawToolShape(
  848. MultiLineString([affinity.scale(subgeo, xfact=factor, yfact=factor) for subgeo in shape.geo])))
  849. self.points_edit[current_table_dia_edited].append((0, 0))
  850. self.add_gerber_shape(geometry, self.storage_dict[current_table_dia_edited])
  851. self.on_aperture_delete(apid=dia_changed)
  852. # delete the tool offset
  853. self.gerber_obj.tool_offset.pop(dia_changed, None)
  854. # we reactivate the signals after the after the tool editing
  855. self.apertures_table.itemChanged.connect(self.on_tool_edit)
  856. # self.apertures_table.selectionModel().currentChanged.connect(self.on_row_selected)
  857. def on_name_activate(self):
  858. self.edited_obj_name = self.name_entry.get_value()
  859. def activate(self):
  860. self.connect_canvas_event_handlers()
  861. # self.app.collection.view.keyPressed.connect(self.on_canvas_key)
  862. self.shapes.enabled = True
  863. self.tool_shape.enabled = True
  864. # self.app.app_cursor.enabled = True
  865. self.app.ui.snap_max_dist_entry.setEnabled(True)
  866. self.app.ui.corner_snap_btn.setEnabled(True)
  867. self.app.ui.snap_magnet.setVisible(True)
  868. self.app.ui.corner_snap_btn.setVisible(True)
  869. self.app.ui.grb_editor_menu.setDisabled(False)
  870. self.app.ui.grb_editor_menu.menuAction().setVisible(True)
  871. self.app.ui.update_obj_btn.setEnabled(True)
  872. self.app.ui.grb_editor_cmenu.setEnabled(True)
  873. self.app.ui.grb_edit_toolbar.setDisabled(False)
  874. self.app.ui.grb_edit_toolbar.setVisible(True)
  875. # self.app.ui.snap_toolbar.setDisabled(False)
  876. # start with GRID toolbar activated
  877. if self.app.ui.grid_snap_btn.isChecked() is False:
  878. self.app.ui.grid_snap_btn.trigger()
  879. # Tell the App that the editor is active
  880. self.editor_active = True
  881. def deactivate(self):
  882. self.disconnect_canvas_event_handlers()
  883. self.clear()
  884. self.app.ui.grb_edit_toolbar.setDisabled(True)
  885. settings = QSettings("Open Source", "FlatCAM")
  886. if settings.contains("layout"):
  887. layout = settings.value('layout', type=str)
  888. if layout == 'standard':
  889. # self.app.ui.exc_edit_toolbar.setVisible(False)
  890. self.app.ui.snap_max_dist_entry.setEnabled(False)
  891. self.app.ui.corner_snap_btn.setEnabled(False)
  892. self.app.ui.snap_magnet.setVisible(False)
  893. self.app.ui.corner_snap_btn.setVisible(False)
  894. elif layout == 'compact':
  895. # self.app.ui.exc_edit_toolbar.setVisible(True)
  896. self.app.ui.snap_max_dist_entry.setEnabled(False)
  897. self.app.ui.corner_snap_btn.setEnabled(False)
  898. self.app.ui.snap_magnet.setVisible(True)
  899. self.app.ui.corner_snap_btn.setVisible(True)
  900. else:
  901. # self.app.ui.exc_edit_toolbar.setVisible(False)
  902. self.app.ui.snap_max_dist_entry.setEnabled(False)
  903. self.app.ui.corner_snap_btn.setEnabled(False)
  904. self.app.ui.snap_magnet.setVisible(False)
  905. self.app.ui.corner_snap_btn.setVisible(False)
  906. # set the Editor Toolbar visibility to what was before entering in the Editor
  907. self.app.ui.grb_edit_toolbar.setVisible(False) if self.toolbar_old_state is False \
  908. else self.app.ui.grb_edit_toolbar.setVisible(True)
  909. # Disable visuals
  910. self.shapes.enabled = False
  911. self.tool_shape.enabled = False
  912. # self.app.app_cursor.enabled = False
  913. # Tell the app that the editor is no longer active
  914. self.editor_active = False
  915. self.app.ui.grb_editor_menu.setDisabled(True)
  916. self.app.ui.grb_editor_menu.menuAction().setVisible(False)
  917. self.app.ui.update_obj_btn.setEnabled(False)
  918. self.app.ui.g_editor_cmenu.setEnabled(False)
  919. self.app.ui.grb_editor_cmenu.setEnabled(False)
  920. self.app.ui.e_editor_cmenu.setEnabled(False)
  921. # Show original geometry
  922. if self.gerber_obj:
  923. self.gerber_obj.visible = True
  924. def connect_canvas_event_handlers(self):
  925. ## Canvas events
  926. # make sure that the shortcuts key and mouse events will no longer be linked to the methods from FlatCAMApp
  927. # but those from FlatCAMGeoEditor
  928. self.app.plotcanvas.vis_disconnect('mouse_press', self.app.on_mouse_click_over_plot)
  929. self.app.plotcanvas.vis_disconnect('mouse_move', self.app.on_mouse_move_over_plot)
  930. self.app.plotcanvas.vis_disconnect('mouse_release', self.app.on_mouse_click_release_over_plot)
  931. self.app.plotcanvas.vis_disconnect('mouse_double_click', self.app.on_double_click_over_plot)
  932. self.app.collection.view.clicked.disconnect()
  933. self.canvas.vis_connect('mouse_press', self.on_canvas_click)
  934. self.canvas.vis_connect('mouse_move', self.on_canvas_move)
  935. self.canvas.vis_connect('mouse_release', self.on_canvas_click_release)
  936. def disconnect_canvas_event_handlers(self):
  937. self.canvas.vis_disconnect('mouse_press', self.on_canvas_click)
  938. self.canvas.vis_disconnect('mouse_move', self.on_canvas_move)
  939. self.canvas.vis_disconnect('mouse_release', self.on_canvas_click_release)
  940. # we restore the key and mouse control to FlatCAMApp method
  941. self.app.plotcanvas.vis_connect('mouse_press', self.app.on_mouse_click_over_plot)
  942. self.app.plotcanvas.vis_connect('mouse_move', self.app.on_mouse_move_over_plot)
  943. self.app.plotcanvas.vis_connect('mouse_release', self.app.on_mouse_click_release_over_plot)
  944. self.app.plotcanvas.vis_connect('mouse_double_click', self.app.on_double_click_over_plot)
  945. self.app.collection.view.clicked.connect(self.app.collection.on_mouse_down)
  946. def clear(self):
  947. self.active_tool = None
  948. # self.shape_buffer = []
  949. self.selected = []
  950. self.storage_dict = {}
  951. self.shapes.clear(update=True)
  952. self.tool_shape.clear(update=True)
  953. # self.storage = FlatCAMExcEditor.make_storage()
  954. self.plot_all()
  955. def edit_fcgerber(self, orig_grb_obj):
  956. """
  957. Imports the geometry found in self.apertures from the given FlatCAM Gerber object
  958. into the editor.
  959. :param fcgeometry: FlatCAMExcellon
  960. :return: None
  961. """
  962. self.deactivate()
  963. self.activate()
  964. # create a reference to the source object
  965. self.gerber_obj = orig_grb_obj
  966. self.gerber_obj_options = orig_grb_obj.options
  967. # Hide original geometry
  968. orig_grb_obj.visible = False
  969. # Set selection tolerance
  970. # DrawToolShape.tolerance = fc_excellon.drawing_tolerance * 10
  971. self.select_tool("select")
  972. # we activate this after the initial build as we don't need to see the tool been populated
  973. self.apertures_table.itemChanged.connect(self.on_tool_edit)
  974. # build the geometry for each tool-diameter, each drill will be represented by a '+' symbol
  975. # and then add it to the storage elements (each storage elements is a member of a list
  976. def job_thread(self, apid):
  977. with self.app.proc_container.new(_("Adding aperture: %s geo ...") % str(apid)):
  978. solid_storage_elem = []
  979. follow_storage_elem = []
  980. self.storage_dict[apid] = {}
  981. for k, v in self.gerber_obj.apertures[apid].items():
  982. if k == 'solid_geometry':
  983. for geo in v:
  984. if geo is not None:
  985. self.add_gerber_shape(DrawToolShape(geo), solid_storage_elem)
  986. self.storage_dict[apid][k] = solid_storage_elem
  987. elif k == 'follow_geometry':
  988. for geo in v:
  989. if geo is not None:
  990. self.add_gerber_shape(DrawToolShape(geo), follow_storage_elem)
  991. self.storage_dict[apid][k] = follow_storage_elem
  992. else:
  993. self.storage_dict[apid][k] = v
  994. # Check promises and clear if exists
  995. while True:
  996. try:
  997. self.grb_plot_promises.remove(apid)
  998. time.sleep(0.5)
  999. except ValueError:
  1000. break
  1001. for apid in self.gerber_obj.apertures:
  1002. self.grb_plot_promises.append(apid)
  1003. self.app.worker_task.emit({'fcn': job_thread, 'params': [self, apid]})
  1004. self.start_delayed_plot(check_period=1000)
  1005. def update_fcgerber(self, grb_obj):
  1006. """
  1007. Create a new Gerber object that contain the edited content of the source Gerber object
  1008. :param grb_obj: FlatCAMGerber
  1009. :return: None
  1010. """
  1011. new_grb_name = self.edited_obj_name
  1012. # if the 'delayed plot' malfunctioned stop the QTimer
  1013. try:
  1014. self.plot_thread.stop()
  1015. except:
  1016. pass
  1017. if "_edit" in self.edited_obj_name:
  1018. try:
  1019. id = int(self.edited_obj_name[-1]) + 1
  1020. new_grb_name= self.edited_obj_name[:-1] + str(id)
  1021. except ValueError:
  1022. new_grb_name += "_1"
  1023. else:
  1024. new_grb_name = self.edited_obj_name + "_edit"
  1025. self.app.worker_task.emit({'fcn': self.new_edited_gerber,
  1026. 'params': [new_grb_name]})
  1027. # reset the tool table
  1028. self.apertures_table.clear()
  1029. self.apertures_table.setHorizontalHeaderLabels(['#', _('Code'), _('Type'), _('Size'), _('Dim')])
  1030. self.last_aperture_selected = None
  1031. # restore GUI to the Selected TAB
  1032. # Remove anything else in the GUI
  1033. self.app.ui.selected_scroll_area.takeWidget()
  1034. # Switch notebook to Selected page
  1035. self.app.ui.notebook.setCurrentWidget(self.app.ui.selected_tab)
  1036. def update_options(self, obj):
  1037. try:
  1038. if not obj.options:
  1039. obj.options = {}
  1040. obj.options['xmin'] = 0
  1041. obj.options['ymin'] = 0
  1042. obj.options['xmax'] = 0
  1043. obj.options['ymax'] = 0
  1044. return True
  1045. else:
  1046. return False
  1047. except AttributeError:
  1048. obj.options = {}
  1049. return True
  1050. def new_edited_gerber(self, outname):
  1051. """
  1052. Creates a new Gerber object for the edited Gerber. Thread-safe.
  1053. :param outname: Name of the resulting object. None causes the name to be that of the file.
  1054. :type outname: str
  1055. :return: None
  1056. """
  1057. self.app.log.debug("Update the Gerber object with edited content. Source is: %s" %
  1058. self.gerber_obj.options['name'].upper())
  1059. out_name = outname
  1060. # How the object should be initialized
  1061. def obj_init(grb_obj, app_obj):
  1062. poly_buffer = []
  1063. follow_buffer = []
  1064. new_geo = []
  1065. for storage_apid, storage_val in self.storage_dict.items():
  1066. grb_obj.apertures[storage_apid] = {}
  1067. for k, v in storage_val.items():
  1068. if k == 'solid_geometry':
  1069. grb_obj.apertures[storage_apid][k] = []
  1070. for geo in v:
  1071. new_geo = deepcopy(geo.geo)
  1072. grb_obj.apertures[storage_apid][k].append(new_geo)
  1073. poly_buffer.append(new_geo)
  1074. elif k == 'follow_geometry':
  1075. grb_obj.apertures[storage_apid][k] = []
  1076. for geo in v:
  1077. new_geo = deepcopy(geo.geo)
  1078. grb_obj.apertures[storage_apid][k].append(new_geo)
  1079. follow_buffer.append(new_geo)
  1080. else:
  1081. grb_obj.apertures[storage_apid][k] = deepcopy(v)
  1082. grb_obj.aperture_macros = deepcopy(self.gerber_obj.aperture_macros)
  1083. new_poly = MultiPolygon(poly_buffer)
  1084. new_poly = new_poly.buffer(0.00000001)
  1085. new_poly = new_poly.buffer(-0.00000001)
  1086. grb_obj.solid_geometry = new_poly
  1087. grb_obj.follow_geometry = deepcopy(follow_buffer)
  1088. for k, v in self.gerber_obj_options.items():
  1089. if k == 'name':
  1090. grb_obj.options[k] = out_name
  1091. else:
  1092. grb_obj.options[k] = deepcopy(v)
  1093. try:
  1094. grb_obj.create_geometry()
  1095. except KeyError:
  1096. self.app.inform.emit(
  1097. _( "[ERROR_NOTCL] There are no Aperture definitions in the file. Aborting Gerber creation.")
  1098. )
  1099. except:
  1100. msg = _("[ERROR] An internal error has ocurred. See shell.\n")
  1101. msg += traceback.format_exc()
  1102. app_obj.inform.emit(msg)
  1103. raise
  1104. # raise
  1105. with self.app.proc_container.new(_("Creating Gerber.")):
  1106. try:
  1107. self.app.new_object("gerber", outname, obj_init)
  1108. except Exception as e:
  1109. log.error("Error on object creation: %s" % str(e))
  1110. self.app.progress.emit(100)
  1111. return
  1112. self.app.inform.emit(_("[success] Gerber editing finished."))
  1113. # self.progress.emit(100)
  1114. def on_tool_select(self, tool):
  1115. """
  1116. Behavior of the toolbar. Tool initialization.
  1117. :rtype : None
  1118. """
  1119. current_tool = tool
  1120. self.app.log.debug("on_tool_select('%s')" % tool)
  1121. if self.last_aperture_selected is None and current_tool is not 'select':
  1122. # self.draw_app.select_tool('select')
  1123. self.complete = True
  1124. current_tool = 'select'
  1125. self.app.inform.emit(_("[WARNING_NOTCL] Cancelled. There is no Tool/Drill selected"))
  1126. # This is to make the group behave as radio group
  1127. if current_tool in self.tools_gerber:
  1128. if self.tools_gerber[current_tool]["button"].isChecked():
  1129. self.app.log.debug("%s is checked." % current_tool)
  1130. for t in self.tools_gerber:
  1131. if t != current_tool:
  1132. self.tools_gerber[t]["button"].setChecked(False)
  1133. # this is where the Editor toolbar classes (button's) are instantiated
  1134. self.active_tool = self.tools_gerber[current_tool]["constructor"](self)
  1135. # self.app.inform.emit(self.active_tool.start_msg)
  1136. else:
  1137. self.app.log.debug("%s is NOT checked." % current_tool)
  1138. for t in self.tools_gerber:
  1139. self.tools_gerber[t]["button"].setChecked(False)
  1140. self.active_tool = None
  1141. def on_row_selected(self):
  1142. self.selected = []
  1143. try:
  1144. selected_dia = self.tool2tooldia[self.apertures_table.currentRow() + 1]
  1145. self.last_aperture_selected = self.apertures_table.currentRow() + 1
  1146. for obj in self.storage_dict[selected_dia].get_objects():
  1147. self.selected.append(obj)
  1148. except Exception as e:
  1149. self.app.log.debug(str(e))
  1150. self.plot_all()
  1151. def toolbar_tool_toggle(self, key):
  1152. self.options[key] = self.sender().isChecked()
  1153. if self.options[key] == True:
  1154. return 1
  1155. else:
  1156. return 0
  1157. def on_canvas_click(self, event):
  1158. """
  1159. event.x and .y have canvas coordinates
  1160. event.xdaya and .ydata have plot coordinates
  1161. :param event: Event object dispatched by Matplotlib
  1162. :return: None
  1163. """
  1164. if event.button is 1:
  1165. self.app.ui.rel_position_label.setText("<b>Dx</b>: %.4f&nbsp;&nbsp; <b>Dy</b>: "
  1166. "%.4f&nbsp;&nbsp;&nbsp;&nbsp;" % (0, 0))
  1167. self.pos = self.canvas.vispy_canvas.translate_coords(event.pos)
  1168. ### Snap coordinates
  1169. x, y = self.app.geo_editor.snap(self.pos[0], self.pos[1])
  1170. self.pos = (x, y)
  1171. # print(self.active_tool)
  1172. # Selection with left mouse button
  1173. if self.active_tool is not None and event.button is 1:
  1174. # Dispatch event to active_tool
  1175. # msg = self.active_tool.click(self.app.geo_editor.snap(event.xdata, event.ydata))
  1176. msg = self.active_tool.click(self.app.geo_editor.snap(self.pos[0], self.pos[1]))
  1177. # If it is a shape generating tool
  1178. if isinstance(self.active_tool, FCShapeTool) and self.active_tool.complete:
  1179. if self.current_storage is not None:
  1180. self.on_exc_shape_complete(self.current_storage)
  1181. self.build_ui()
  1182. # MS: always return to the Select Tool if modifier key is not pressed
  1183. # else return to the current tool
  1184. key_modifier = QtWidgets.QApplication.keyboardModifiers()
  1185. if self.app.defaults["global_mselect_key"] == 'Control':
  1186. modifier_to_use = Qt.ControlModifier
  1187. else:
  1188. modifier_to_use = Qt.ShiftModifier
  1189. # if modifier key is pressed then we add to the selected list the current shape but if it's already
  1190. # in the selected list, we removed it. Therefore first click selects, second deselects.
  1191. if key_modifier == modifier_to_use:
  1192. self.select_tool(self.active_tool.name)
  1193. else:
  1194. self.select_tool("select")
  1195. return
  1196. if isinstance(self.active_tool, FCApertureSelect):
  1197. # self.app.log.debug("Replotting after click.")
  1198. self.plot_all()
  1199. else:
  1200. self.app.log.debug("No active tool to respond to click!")
  1201. def on_exc_shape_complete(self, storage):
  1202. self.app.log.debug("on_shape_complete()")
  1203. # Add shape
  1204. if type(storage) is list:
  1205. for item_storage in storage:
  1206. self.add_gerber_shape(self.active_tool.geometry, item_storage)
  1207. else:
  1208. self.add_gerber_shape(self.active_tool.geometry, storage)
  1209. # Remove any utility shapes
  1210. self.delete_utility_geometry()
  1211. self.tool_shape.clear(update=True)
  1212. # Replot and reset tool.
  1213. self.plot_all()
  1214. # self.active_tool = type(self.active_tool)(self)
  1215. def add_gerber_shape(self, shape, storage):
  1216. """
  1217. Adds a shape to the shape storage.
  1218. :param shape: Shape to be added.
  1219. :type shape: DrawToolShape
  1220. :return: None
  1221. """
  1222. # List of DrawToolShape?
  1223. if isinstance(shape, list):
  1224. for subshape in shape:
  1225. self.add_gerber_shape(subshape, storage)
  1226. return
  1227. assert isinstance(shape, DrawToolShape), \
  1228. "Expected a DrawToolShape, got %s" % str(type(shape))
  1229. assert shape.geo is not None, \
  1230. "Shape object has empty geometry (None)"
  1231. assert (isinstance(shape.geo, list) and len(shape.geo) > 0) or \
  1232. not isinstance(shape.geo, list), \
  1233. "Shape objects has empty geometry ([])"
  1234. if isinstance(shape, DrawToolUtilityShape):
  1235. self.utility.append(shape)
  1236. else:
  1237. storage.append(shape) # TODO: Check performance
  1238. def add_shape(self, shape):
  1239. """
  1240. Adds a shape to the shape storage.
  1241. :param shape: Shape to be added.
  1242. :type shape: DrawToolShape
  1243. :return: None
  1244. """
  1245. # List of DrawToolShape?
  1246. if isinstance(shape, list):
  1247. for subshape in shape:
  1248. self.add_shape(subshape)
  1249. return
  1250. assert isinstance(shape, DrawToolShape), \
  1251. "Expected a DrawToolShape, got %s" % type(shape)
  1252. assert shape.geo is not None, \
  1253. "Shape object has empty geometry (None)"
  1254. assert (isinstance(shape.geo, list) and len(shape.geo) > 0) or \
  1255. not isinstance(shape.geo, list), \
  1256. "Shape objects has empty geometry ([])"
  1257. if isinstance(shape, DrawToolUtilityShape):
  1258. self.utility.append(shape)
  1259. else:
  1260. self.storage.insert(shape) # TODO: Check performance
  1261. def on_canvas_click_release(self, event):
  1262. pos_canvas = self.canvas.vispy_canvas.translate_coords(event.pos)
  1263. self.modifiers = QtWidgets.QApplication.keyboardModifiers()
  1264. if self.app.grid_status():
  1265. pos = self.app.geo_editor.snap(pos_canvas[0], pos_canvas[1])
  1266. else:
  1267. pos = (pos_canvas[0], pos_canvas[1])
  1268. # if the released mouse button was RMB then test if it was a panning motion or not, if not it was a context
  1269. # canvas menu
  1270. try:
  1271. if event.button == 2: # right click
  1272. if self.app.panning_action is True:
  1273. self.app.panning_action = False
  1274. else:
  1275. self.app.cursor = QtGui.QCursor()
  1276. self.app.ui.popMenu.popup(self.app.cursor.pos())
  1277. except Exception as e:
  1278. log.warning("Error: %s" % str(e))
  1279. raise
  1280. # if the released mouse button was LMB then test if we had a right-to-left selection or a left-to-right
  1281. # selection and then select a type of selection ("enclosing" or "touching")
  1282. try:
  1283. if event.button == 1: # left click
  1284. if self.app.selection_type is not None:
  1285. self.draw_selection_area_handler(self.pos, pos, self.app.selection_type)
  1286. self.app.selection_type = None
  1287. elif isinstance(self.active_tool, FCApertureSelect):
  1288. # Dispatch event to active_tool
  1289. # msg = self.active_tool.click(self.app.geo_editor.snap(event.xdata, event.ydata))
  1290. # msg = self.active_tool.click_release((self.pos[0], self.pos[1]))
  1291. # self.app.inform.emit(msg)
  1292. self.active_tool.click_release((self.pos[0], self.pos[1]))
  1293. self.plot_all()
  1294. except Exception as e:
  1295. log.warning("Error: %s" % str(e))
  1296. raise
  1297. def draw_selection_area_handler(self, start_pos, end_pos, sel_type):
  1298. """
  1299. :param start_pos: mouse position when the selection LMB click was done
  1300. :param end_pos: mouse position when the left mouse button is released
  1301. :param sel_type: if True it's a left to right selection (enclosure), if False it's a 'touch' selection
  1302. :type Bool
  1303. :return:
  1304. """
  1305. poly_selection = Polygon([start_pos, (end_pos[0], start_pos[1]), end_pos, (start_pos[0], end_pos[1])])
  1306. self.app.delete_selection_shape()
  1307. for storage in self.storage_dict:
  1308. for obj in self.storage_dict[storage].get_objects():
  1309. if (sel_type is True and poly_selection.contains(obj.geo)) or \
  1310. (sel_type is False and poly_selection.intersects(obj.geo)):
  1311. if self.key == self.app.defaults["global_mselect_key"]:
  1312. if obj in self.selected:
  1313. self.selected.remove(obj)
  1314. else:
  1315. # add the object to the selected shapes
  1316. self.selected.append(obj)
  1317. else:
  1318. self.selected.append(obj)
  1319. # select the diameter of the selected shape in the tool table
  1320. for storage in self.storage_dict:
  1321. for shape_s in self.selected:
  1322. if shape_s in self.storage_dict[storage].get_objects():
  1323. for key in self.tool2tooldia:
  1324. if self.tool2tooldia[key] == storage:
  1325. item = self.apertures_table.item((key - 1), 1)
  1326. self.apertures_table.setCurrentItem(item)
  1327. self.last_aperture_selected = key
  1328. # item.setSelected(True)
  1329. # self.grb_editor_app.apertures_table.selectItem(key - 1)
  1330. self.plot_all()
  1331. def on_canvas_move(self, event):
  1332. """
  1333. Called on 'mouse_move' event
  1334. event.pos have canvas screen coordinates
  1335. :param event: Event object dispatched by VisPy SceneCavas
  1336. :return: None
  1337. """
  1338. pos = self.canvas.vispy_canvas.translate_coords(event.pos)
  1339. event.xdata, event.ydata = pos[0], pos[1]
  1340. self.x = event.xdata
  1341. self.y = event.ydata
  1342. # Prevent updates on pan
  1343. # if len(event.buttons) > 0:
  1344. # return
  1345. # if the RMB is clicked and mouse is moving over plot then 'panning_action' is True
  1346. if event.button == 2:
  1347. self.app.panning_action = True
  1348. return
  1349. else:
  1350. self.app.panning_action = False
  1351. try:
  1352. x = float(event.xdata)
  1353. y = float(event.ydata)
  1354. except TypeError:
  1355. return
  1356. if self.active_tool is None:
  1357. return
  1358. ### Snap coordinates
  1359. x, y = self.app.geo_editor.app.geo_editor.snap(x, y)
  1360. self.snap_x = x
  1361. self.snap_y = y
  1362. # update the position label in the infobar since the APP mouse event handlers are disconnected
  1363. self.app.ui.position_label.setText("&nbsp;&nbsp;&nbsp;&nbsp;<b>X</b>: %.4f&nbsp;&nbsp; "
  1364. "<b>Y</b>: %.4f" % (x, y))
  1365. if self.pos is None:
  1366. self.pos = (0, 0)
  1367. dx = x - self.pos[0]
  1368. dy = y - self.pos[1]
  1369. # update the reference position label in the infobar since the APP mouse event handlers are disconnected
  1370. self.app.ui.rel_position_label.setText("<b>Dx</b>: %.4f&nbsp;&nbsp; <b>Dy</b>: "
  1371. "%.4f&nbsp;&nbsp;&nbsp;&nbsp;" % (dx, dy))
  1372. ### Utility geometry (animated)
  1373. geo = self.active_tool.utility_geometry(data=(x, y))
  1374. if isinstance(geo, DrawToolShape) and geo.geo is not None:
  1375. # Remove any previous utility shape
  1376. self.tool_shape.clear(update=True)
  1377. self.draw_utility_geometry(geo=geo)
  1378. ### Selection area on canvas section ###
  1379. dx = pos[0] - self.pos[0]
  1380. if event.is_dragging == 1 and event.button == 1:
  1381. self.app.delete_selection_shape()
  1382. if dx < 0:
  1383. self.app.draw_moving_selection_shape((self.pos[0], self.pos[1]), (x,y),
  1384. color=self.app.defaults["global_alt_sel_line"],
  1385. face_color=self.app.defaults['global_alt_sel_fill'])
  1386. self.app.selection_type = False
  1387. else:
  1388. self.app.draw_moving_selection_shape((self.pos[0], self.pos[1]), (x,y))
  1389. self.app.selection_type = True
  1390. else:
  1391. self.app.selection_type = None
  1392. # Update cursor
  1393. self.app.app_cursor.set_data(np.asarray([(x, y)]), symbol='++', edge_color='black', size=20)
  1394. def on_canvas_key_release(self, event):
  1395. self.key = None
  1396. def draw_utility_geometry(self, geo):
  1397. # Add the new utility shape
  1398. try:
  1399. # this case is for the Font Parse
  1400. for el in list(geo.geo):
  1401. if type(el) == MultiPolygon:
  1402. for poly in el:
  1403. self.tool_shape.add(
  1404. shape=poly,
  1405. color=(self.app.defaults["global_draw_color"] + '80'),
  1406. update=False,
  1407. layer=0,
  1408. tolerance=None
  1409. )
  1410. elif type(el) == MultiLineString:
  1411. for linestring in el:
  1412. self.tool_shape.add(
  1413. shape=linestring,
  1414. color=(self.app.defaults["global_draw_color"] + '80'),
  1415. update=False,
  1416. layer=0,
  1417. tolerance=None
  1418. )
  1419. else:
  1420. self.tool_shape.add(
  1421. shape=el,
  1422. color=(self.app.defaults["global_draw_color"] + '80'),
  1423. update=False,
  1424. layer=0,
  1425. tolerance=None
  1426. )
  1427. except TypeError:
  1428. self.tool_shape.add(
  1429. shape=geo.geo, color=(self.app.defaults["global_draw_color"] + '80'),
  1430. update=False, layer=0, tolerance=None)
  1431. self.tool_shape.redraw()
  1432. def plot_all(self):
  1433. """
  1434. Plots all shapes in the editor.
  1435. :return: None
  1436. :rtype: None
  1437. """
  1438. with self.app.proc_container.new("Plotting"):
  1439. # self.app.log.debug("plot_all()")
  1440. self.shapes.clear(update=True)
  1441. for storage in self.storage_dict:
  1442. for shape in self.storage_dict[storage]['solid_geometry']:
  1443. if shape.geo is None:
  1444. continue
  1445. if shape in self.selected:
  1446. self.plot_shape(geometry=shape.geo, color=self.app.defaults['global_sel_draw_color'],
  1447. linewidth=2)
  1448. continue
  1449. self.plot_shape(geometry=shape.geo, color=self.app.defaults['global_draw_color'])
  1450. for shape in self.utility:
  1451. self.plot_shape(geometry=shape.geo, linewidth=1)
  1452. continue
  1453. self.shapes.redraw()
  1454. def start_delayed_plot(self, check_period):
  1455. # self.plot_thread = threading.Thread(target=lambda: self.check_plot_finished(check_period))
  1456. # self.plot_thread.start()
  1457. log.debug("FlatCAMGrbEditor --> Delayed Plot started.")
  1458. self.plot_thread = QtCore.QTimer()
  1459. self.plot_thread.setInterval(check_period)
  1460. self.plot_thread.timeout.connect(self.check_plot_finished)
  1461. self.plot_thread.start()
  1462. def check_plot_finished(self):
  1463. # print(self.grb_plot_promises)
  1464. try:
  1465. if not self.grb_plot_promises:
  1466. self.plot_thread.stop()
  1467. self.set_ui()
  1468. # now that we hava data, create the GUI interface and add it to the Tool Tab
  1469. self.build_ui()
  1470. self.plot_all()
  1471. log.debug("FlatCAMGrbEditor --> delayed_plot finished")
  1472. except Exception:
  1473. traceback.print_exc()
  1474. # def stop_delayed_plot(self):
  1475. # self.plot_thread.exit()
  1476. # # self.plot_thread.join()
  1477. # def check_plot_finished(self, delay):
  1478. # """
  1479. # Using Alfe's answer from here:
  1480. # https://stackoverflow.com/questions/474528/what-is-the-best-way-to-repeatedly-execute-a-function-every-x-seconds-in-python
  1481. #
  1482. # :param delay: period of checking if project file size is more than zero; in seconds
  1483. # :param filename: the name of the project file to be checked for size more than zero
  1484. # :return:
  1485. # """
  1486. # next_time = time.time() + delay
  1487. # while True:
  1488. # time.sleep(max(0, next_time - time.time()))
  1489. # try:
  1490. # if self.app.collection.has_plot_promises() is False:
  1491. # self.plot_all()
  1492. # break
  1493. # except Exception:
  1494. # traceback.print_exc()
  1495. #
  1496. # # skip tasks if we are behind schedule:
  1497. # next_time += (time.time() - next_time) // delay * delay + delay
  1498. def plot_shape(self, geometry=None, color='black', linewidth=1):
  1499. """
  1500. Plots a geometric object or list of objects without rendering. Plotted objects
  1501. are returned as a list. This allows for efficient/animated rendering.
  1502. :param geometry: Geometry to be plotted (Any Shapely.geom kind or list of such)
  1503. :param color: Shape color
  1504. :param linewidth: Width of lines in # of pixels.
  1505. :return: List of plotted elements.
  1506. """
  1507. # plot_elements = []
  1508. if geometry is None:
  1509. geometry = self.active_tool.geometry
  1510. try:
  1511. self.shapes.add(shape=geometry.geo, color=color, face_color=color, layer=0)
  1512. except AttributeError:
  1513. if type(geometry) == Point:
  1514. return
  1515. self.shapes.add(shape=geometry, color=color, face_color=color+'AF', layer=0)
  1516. def on_shape_complete(self):
  1517. self.app.log.debug("on_shape_complete()")
  1518. # Add shape
  1519. self.add_shape(self.active_tool.geometry)
  1520. # Remove any utility shapes
  1521. self.delete_utility_geometry()
  1522. self.tool_shape.clear(update=True)
  1523. # Replot and reset tool.
  1524. self.plot_all()
  1525. # self.active_tool = type(self.active_tool)(self)
  1526. def get_selected(self):
  1527. """
  1528. Returns list of shapes that are selected in the editor.
  1529. :return: List of shapes.
  1530. """
  1531. # return [shape for shape in self.shape_buffer if shape["selected"]]
  1532. return self.selected
  1533. def delete_selected(self):
  1534. temp_ref = [s for s in self.selected]
  1535. for shape_sel in temp_ref:
  1536. self.delete_shape(shape_sel)
  1537. self.selected = []
  1538. self.build_ui()
  1539. self.app.inform.emit(_("[success] Done. Drill(s) deleted."))
  1540. def delete_shape(self, shape):
  1541. self.is_modified = True
  1542. if shape in self.utility:
  1543. self.utility.remove(shape)
  1544. return
  1545. for storage in self.storage_dict:
  1546. # try:
  1547. # self.storage_dict[storage].remove(shape)
  1548. # except:
  1549. # pass
  1550. if shape in self.storage_dict[storage].get_objects():
  1551. self.storage_dict[storage].remove(shape)
  1552. # a hack to make the tool_table display less drills per diameter
  1553. # self.points_edit it's only useful first time when we load the data into the storage
  1554. # but is still used as referecen when building tool_table in self.build_ui()
  1555. # the number of drills displayed in column 2 is just a len(self.points_edit) therefore
  1556. # deleting self.points_edit elements (doesn't matter who but just the number) solved the display issue.
  1557. del self.points_edit[storage][0]
  1558. if shape in self.selected:
  1559. self.selected.remove(shape) # TODO: Check performance
  1560. def delete_utility_geometry(self):
  1561. # for_deletion = [shape for shape in self.shape_buffer if shape.utility]
  1562. # for_deletion = [shape for shape in self.storage.get_objects() if shape.utility]
  1563. for_deletion = [shape for shape in self.utility]
  1564. for shape in for_deletion:
  1565. self.delete_shape(shape)
  1566. self.tool_shape.clear(update=True)
  1567. self.tool_shape.redraw()
  1568. def on_delete_btn(self):
  1569. self.delete_selected()
  1570. self.plot_all()
  1571. def select_tool(self, toolname):
  1572. """
  1573. Selects a drawing tool. Impacts the object and GUI.
  1574. :param toolname: Name of the tool.
  1575. :return: None
  1576. """
  1577. self.tools_gerber[toolname]["button"].setChecked(True)
  1578. self.on_tool_select(toolname)
  1579. def set_selected(self, shape):
  1580. # Remove and add to the end.
  1581. if shape in self.selected:
  1582. self.selected.remove(shape)
  1583. self.selected.append(shape)
  1584. def set_unselected(self, shape):
  1585. if shape in self.selected:
  1586. self.selected.remove(shape)
  1587. def on_array_type_combo(self):
  1588. if self.array_type_combo.currentIndex() == 0:
  1589. self.array_circular_frame.hide()
  1590. self.array_linear_frame.show()
  1591. else:
  1592. self.delete_utility_geometry()
  1593. self.array_circular_frame.show()
  1594. self.array_linear_frame.hide()
  1595. self.app.inform.emit(_("Click on the circular array Center position"))
  1596. def on_linear_angle_radio(self):
  1597. val = self.drill_axis_radio.get_value()
  1598. if val == 'A':
  1599. self.linear_angle_spinner.show()
  1600. self.linear_angle_label.show()
  1601. else:
  1602. self.linear_angle_spinner.hide()
  1603. self.linear_angle_label.hide()
  1604. def exc_add_drill(self):
  1605. self.select_tool('add')
  1606. return
  1607. def exc_add_drill_array(self):
  1608. self.select_tool('add_array')
  1609. return
  1610. def exc_resize_drills(self):
  1611. self.select_tool('resize')
  1612. return
  1613. def exc_copy_drills(self):
  1614. self.select_tool('copy')
  1615. return
  1616. def exc_move_drills(self):
  1617. self.select_tool('move')
  1618. return