FlatCAMGrbEditor.py 81 KB

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