FlatCAMGrbEditor.py 81 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022
  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.plot_all()
  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]['solid_geometry']:
  229. if Point(pos).within(shape.geo):
  230. self.sel_storage.append(shape)
  231. xmin, ymin, xmax, ymax = self.bounds(self.sel_storage)
  232. if pos[0] < xmin or pos[0] > xmax or pos[1] < ymin or pos[1] > ymax:
  233. self.grb_editor_app.selected = []
  234. else:
  235. key_modifier = QtWidgets.QApplication.keyboardModifiers()
  236. if self.grb_editor_app.app.defaults["global_mselect_key"] == 'Control':
  237. # if CONTROL key is pressed then we add to the selected list the current shape but if it's already
  238. # in the selected list, we removed it. Therefore first click selects, second deselects.
  239. if key_modifier == Qt.ControlModifier:
  240. if closest_shape in self.grb_editor_app.selected:
  241. self.grb_editor_app.selected.remove(closest_shape)
  242. else:
  243. self.grb_editor_app.selected.append(closest_shape)
  244. else:
  245. self.grb_editor_app.selected = []
  246. self.grb_editor_app.selected.append(closest_shape)
  247. else:
  248. if key_modifier == Qt.ShiftModifier:
  249. if closest_shape in self.grb_editor_app.selected:
  250. self.grb_editor_app.selected.remove(closest_shape)
  251. else:
  252. self.grb_editor_app.selected.append(closest_shape)
  253. else:
  254. self.grb_editor_app.selected = []
  255. self.grb_editor_app.selected.append(closest_shape)
  256. # select the aperture of the selected shape in the tool table
  257. for storage in self.grb_editor_app.storage_dict:
  258. for shape_s in self.grb_editor_app.selected:
  259. if shape_s in self.grb_editor_app.storage_dict[storage]:
  260. for key in self.grb_editor_app.tool2tooldia:
  261. if self.grb_editor_app.tool2tooldia[key] == storage:
  262. item = self.grb_editor_app.apertures_table.item((key - 1), 1)
  263. self.grb_editor_app.apertures_table.setCurrentItem(item)
  264. # item.setSelected(True)
  265. # self.grb_editor_app.apertures_table.selectItem(key - 1)
  266. # midx = self.grb_editor_app.apertures_table.model().index((key - 1), 0)
  267. # self.grb_editor_app.apertures_table.setCurrentIndex(midx)
  268. self.draw_app.last_tool_selected = key
  269. # delete whatever is in selection storage, there is no longer need for those shapes
  270. self.sel_storage = []
  271. return ""
  272. class FlatCAMGrbEditor(QtCore.QObject):
  273. draw_shape_idx = -1
  274. def __init__(self, app):
  275. assert isinstance(app, FlatCAMApp.App), \
  276. "Expected the app to be a FlatCAMApp.App, got %s" % type(app)
  277. super(FlatCAMGrbEditor, self).__init__()
  278. self.app = app
  279. self.canvas = self.app.plotcanvas
  280. ## Current application units in Upper Case
  281. self.units = self.app.ui.general_defaults_form.general_app_group.units_radio.get_value().upper()
  282. self.grb_edit_widget = QtWidgets.QWidget()
  283. layout = QtWidgets.QVBoxLayout()
  284. self.grb_edit_widget.setLayout(layout)
  285. ## Page Title box (spacing between children)
  286. self.title_box = QtWidgets.QHBoxLayout()
  287. layout.addLayout(self.title_box)
  288. ## Page Title icon
  289. pixmap = QtGui.QPixmap('share/flatcam_icon32.png')
  290. self.icon = QtWidgets.QLabel()
  291. self.icon.setPixmap(pixmap)
  292. self.title_box.addWidget(self.icon, stretch=0)
  293. ## Title label
  294. self.title_label = QtWidgets.QLabel("<font size=5><b>%s</b></font>" % _('Gerber Editor'))
  295. self.title_label.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter)
  296. self.title_box.addWidget(self.title_label, stretch=1)
  297. ## Object name
  298. self.name_box = QtWidgets.QHBoxLayout()
  299. layout.addLayout(self.name_box)
  300. name_label = QtWidgets.QLabel(_("Name:"))
  301. self.name_box.addWidget(name_label)
  302. self.name_entry = FCEntry()
  303. self.name_box.addWidget(self.name_entry)
  304. ## Box for custom widgets
  305. # This gets populated in offspring implementations.
  306. self.custom_box = QtWidgets.QVBoxLayout()
  307. layout.addLayout(self.custom_box)
  308. # add a frame and inside add a vertical box layout. Inside this vbox layout I add all the Drills widgets
  309. # this way I can hide/show the frame
  310. self.apertures_frame = QtWidgets.QFrame()
  311. self.apertures_frame.setContentsMargins(0, 0, 0, 0)
  312. self.custom_box.addWidget(self.apertures_frame)
  313. self.apertures_box = QtWidgets.QVBoxLayout()
  314. self.apertures_box.setContentsMargins(0, 0, 0, 0)
  315. self.apertures_frame.setLayout(self.apertures_box)
  316. #### Gerber Apertures ####
  317. self.apertures_table_label = QtWidgets.QLabel(_('<b>Apertures:</b>'))
  318. self.apertures_table_label.setToolTip(
  319. _("Apertures Table for the Gerber Object.")
  320. )
  321. self.apertures_box.addWidget(self.apertures_table_label)
  322. self.apertures_table = FCTable()
  323. # delegate = SpinBoxDelegate(units=self.units)
  324. # self.apertures_table.setItemDelegateForColumn(1, delegate)
  325. self.apertures_box.addWidget(self.apertures_table)
  326. self.apertures_table.setColumnCount(5)
  327. self.apertures_table.setHorizontalHeaderLabels(['#', _('Code'), _('Type'), _('Size'), _('Dim')])
  328. self.apertures_table.setSortingEnabled(False)
  329. self.apertures_table.horizontalHeaderItem(0).setToolTip(
  330. _("Index"))
  331. self.apertures_table.horizontalHeaderItem(1).setToolTip(
  332. _("Aperture Code"))
  333. self.apertures_table.horizontalHeaderItem(2).setToolTip(
  334. _("Type of aperture: circular, rectangle, macros etc"))
  335. self.apertures_table.horizontalHeaderItem(4).setToolTip(
  336. _("Aperture Size:"))
  337. self.apertures_table.horizontalHeaderItem(4).setToolTip(
  338. _("Aperture Dimensions:\n"
  339. " - (width, height) for R, O type.\n"
  340. " - (dia, nVertices) for P type"))
  341. self.empty_label = QtWidgets.QLabel('')
  342. self.apertures_box.addWidget(self.empty_label)
  343. #### Add a new Tool ####
  344. self.addaperture_label = QtWidgets.QLabel('<b>%s</b>' % _('Add/Delete Aperture'))
  345. self.addaperture_label.setToolTip(
  346. _("Add/Delete an aperture to the aperture list")
  347. )
  348. self.apertures_box.addWidget(self.addaperture_label)
  349. grid1 = QtWidgets.QGridLayout()
  350. self.apertures_box.addLayout(grid1)
  351. addaperture_entry_lbl = QtWidgets.QLabel(_('Aperture Size:'))
  352. addaperture_entry_lbl.setToolTip(
  353. _("Size for the new aperture")
  354. )
  355. grid1.addWidget(addaperture_entry_lbl, 0, 0)
  356. hlay = QtWidgets.QHBoxLayout()
  357. self.addtool_entry = FCEntry()
  358. self.addtool_entry.setValidator(QtGui.QDoubleValidator(0.0001, 99.9999, 4))
  359. hlay.addWidget(self.addtool_entry)
  360. self.addaperture_btn = QtWidgets.QPushButton(_('Add Aperture'))
  361. self.addaperture_btn.setToolTip(
  362. _( "Add a new aperture to the aperture list")
  363. )
  364. self.addaperture_btn.setFixedWidth(80)
  365. hlay.addWidget(self.addaperture_btn)
  366. grid1.addLayout(hlay, 0, 1)
  367. grid2 = QtWidgets.QGridLayout()
  368. self.apertures_box.addLayout(grid2)
  369. self.delaperture_btn = QtWidgets.QPushButton(_('Delete Aperture'))
  370. self.delaperture_btn.setToolTip(
  371. _( "Delete a aperture in the aperture list")
  372. )
  373. grid2.addWidget(self.delaperture_btn, 0, 1)
  374. # add a frame and inside add a vertical box layout. Inside this vbox layout I add all the aperture widgets
  375. # this way I can hide/show the frame
  376. self.resize_frame = QtWidgets.QFrame()
  377. self.resize_frame.setContentsMargins(0, 0, 0, 0)
  378. self.apertures_box.addWidget(self.resize_frame)
  379. self.resize_box = QtWidgets.QVBoxLayout()
  380. self.resize_box.setContentsMargins(0, 0, 0, 0)
  381. self.resize_frame.setLayout(self.resize_box)
  382. #### Resize a aperture ####
  383. self.emptyresize_label = QtWidgets.QLabel('')
  384. self.resize_box.addWidget(self.emptyresize_label)
  385. self.apertureresize_label = QtWidgets.QLabel('<b>%s</b>' % _("Resize Aperture"))
  386. self.apertureresize_label.setToolTip(
  387. _("Resize a aperture or a selection of apertures.")
  388. )
  389. self.resize_box.addWidget(self.apertureresize_label)
  390. grid3 = QtWidgets.QGridLayout()
  391. self.resize_box.addLayout(grid3)
  392. res_entry_lbl = QtWidgets.QLabel(_('Resize Dia:'))
  393. res_entry_lbl.setToolTip(
  394. _( "Size to resize to.")
  395. )
  396. grid3.addWidget(res_entry_lbl, 0, 0)
  397. hlay2 = QtWidgets.QHBoxLayout()
  398. self.resdrill_entry = LengthEntry()
  399. hlay2.addWidget(self.resdrill_entry)
  400. self.resize_btn = QtWidgets.QPushButton(_('Resize'))
  401. self.resize_btn.setToolTip(
  402. _("Resize drill(s)")
  403. )
  404. self.resize_btn.setFixedWidth(80)
  405. hlay2.addWidget(self.resize_btn)
  406. grid3.addLayout(hlay2, 0, 1)
  407. self.resize_frame.hide()
  408. # add a frame and inside add a vertical box layout. Inside this vbox layout I add
  409. # all the add drill array widgets
  410. # this way I can hide/show the frame
  411. self.array_frame = QtWidgets.QFrame()
  412. self.array_frame.setContentsMargins(0, 0, 0, 0)
  413. self.apertures_box.addWidget(self.array_frame)
  414. self.array_box = QtWidgets.QVBoxLayout()
  415. self.array_box.setContentsMargins(0, 0, 0, 0)
  416. self.array_frame.setLayout(self.array_box)
  417. #### Add DRILL Array ####
  418. self.emptyarray_label = QtWidgets.QLabel('')
  419. self.array_box.addWidget(self.emptyarray_label)
  420. self.drillarray_label = QtWidgets.QLabel('<b>%s</b>' % _("Add Drill Array"))
  421. self.drillarray_label.setToolTip(
  422. _("Add an array of drills (linear or circular array)")
  423. )
  424. self.array_box.addWidget(self.drillarray_label)
  425. self.array_type_combo = FCComboBox()
  426. self.array_type_combo.setToolTip(
  427. _( "Select the type of drills array to create.\n"
  428. "It can be Linear X(Y) or Circular")
  429. )
  430. self.array_type_combo.addItem(_("Linear"))
  431. self.array_type_combo.addItem(_("Circular"))
  432. self.array_box.addWidget(self.array_type_combo)
  433. self.array_form = QtWidgets.QFormLayout()
  434. self.array_box.addLayout(self.array_form)
  435. self.drill_array_size_label = QtWidgets.QLabel(_('Nr of drills:'))
  436. self.drill_array_size_label.setToolTip(
  437. _("Specify how many drills to be in the array.")
  438. )
  439. self.drill_array_size_label.setFixedWidth(100)
  440. self.drill_array_size_entry = LengthEntry()
  441. self.array_form.addRow(self.drill_array_size_label, self.drill_array_size_entry)
  442. self.array_linear_frame = QtWidgets.QFrame()
  443. self.array_linear_frame.setContentsMargins(0, 0, 0, 0)
  444. self.array_box.addWidget(self.array_linear_frame)
  445. self.linear_box = QtWidgets.QVBoxLayout()
  446. self.linear_box.setContentsMargins(0, 0, 0, 0)
  447. self.array_linear_frame.setLayout(self.linear_box)
  448. self.linear_form = QtWidgets.QFormLayout()
  449. self.linear_box.addLayout(self.linear_form)
  450. self.drill_axis_label = QtWidgets.QLabel(_('Direction:'))
  451. self.drill_axis_label.setToolTip(
  452. _("Direction on which the linear array is oriented:\n"
  453. "- 'X' - horizontal axis \n"
  454. "- 'Y' - vertical axis or \n"
  455. "- 'Angle' - a custom angle for the array inclination")
  456. )
  457. self.drill_axis_label.setFixedWidth(100)
  458. self.drill_axis_radio = RadioSet([{'label': 'X', 'value': 'X'},
  459. {'label': 'Y', 'value': 'Y'},
  460. {'label': _('Angle'), 'value': 'A'}])
  461. self.drill_axis_radio.set_value('X')
  462. self.linear_form.addRow(self.drill_axis_label, self.drill_axis_radio)
  463. self.drill_pitch_label = QtWidgets.QLabel(_('Pitch:'))
  464. self.drill_pitch_label.setToolTip(
  465. _("Pitch = Distance between elements of the array.")
  466. )
  467. self.drill_pitch_label.setFixedWidth(100)
  468. self.drill_pitch_entry = LengthEntry()
  469. self.linear_form.addRow(self.drill_pitch_label, self.drill_pitch_entry)
  470. self.linear_angle_label = QtWidgets.QLabel(_('Angle:'))
  471. self.linear_angle_label.setToolTip(
  472. _( "Angle at which the linear array is placed.\n"
  473. "The precision is of max 2 decimals.\n"
  474. "Min value is: -359.99 degrees.\n"
  475. "Max value is: 360.00 degrees.")
  476. )
  477. self.linear_angle_label.setFixedWidth(100)
  478. self.linear_angle_spinner = FCDoubleSpinner()
  479. self.linear_angle_spinner.set_precision(2)
  480. self.linear_angle_spinner.setRange(-359.99, 360.00)
  481. self.linear_form.addRow(self.linear_angle_label, self.linear_angle_spinner)
  482. self.array_circular_frame = QtWidgets.QFrame()
  483. self.array_circular_frame.setContentsMargins(0, 0, 0, 0)
  484. self.array_box.addWidget(self.array_circular_frame)
  485. self.circular_box = QtWidgets.QVBoxLayout()
  486. self.circular_box.setContentsMargins(0, 0, 0, 0)
  487. self.array_circular_frame.setLayout(self.circular_box)
  488. self.drill_direction_label = QtWidgets.QLabel(_('Direction:'))
  489. self.drill_direction_label.setToolTip(
  490. _( "Direction for circular array."
  491. "Can be CW = clockwise or CCW = counter clockwise.")
  492. )
  493. self.drill_direction_label.setFixedWidth(100)
  494. self.circular_form = QtWidgets.QFormLayout()
  495. self.circular_box.addLayout(self.circular_form)
  496. self.drill_direction_radio = RadioSet([{'label': 'CW', 'value': 'CW'},
  497. {'label': 'CCW.', 'value': 'CCW'}])
  498. self.drill_direction_radio.set_value('CW')
  499. self.circular_form.addRow(self.drill_direction_label, self.drill_direction_radio)
  500. self.drill_angle_label = QtWidgets.QLabel(_('Angle:'))
  501. self.drill_angle_label.setToolTip(
  502. _("Angle at which each element in circular array is placed.")
  503. )
  504. self.drill_angle_label.setFixedWidth(100)
  505. self.drill_angle_entry = LengthEntry()
  506. self.circular_form.addRow(self.drill_angle_label, self.drill_angle_entry)
  507. self.array_circular_frame.hide()
  508. self.linear_angle_spinner.hide()
  509. self.linear_angle_label.hide()
  510. self.array_frame.hide()
  511. self.apertures_box.addStretch()
  512. ## Toolbar events and properties
  513. self.tools_gerber = {
  514. "select": {"button": self.app.ui.select_drill_btn,
  515. "constructor": FCApertureSelect},
  516. "drill_resize": {"button": self.app.ui.resize_drill_btn,
  517. "constructor": FCApertureResize},
  518. "drill_copy": {"button": self.app.ui.copy_drill_btn,
  519. "constructor": FCApertureCopy},
  520. "drill_move": {"button": self.app.ui.move_drill_btn,
  521. "constructor": FCApertureMove},
  522. }
  523. ### Data
  524. self.active_tool = None
  525. self.storage_dict = {}
  526. self.current_storage = []
  527. # build the data from the Excellon point into a dictionary
  528. # {tool_dia: [geometry_in_points]}
  529. self.points_edit = {}
  530. self.sorted_apid =[]
  531. self.new_apertures = {}
  532. self.new_aperture_macros = {}
  533. # dictionary to store the tool_row and diameters in Tool_table
  534. # it will be updated everytime self.build_ui() is called
  535. self.olddia_newdia = {}
  536. self.tool2tooldia = {}
  537. # this will store the value for the last selected tool, for use after clicking on canvas when the selection
  538. # is cleared but as a side effect also the selected tool is cleared
  539. self.last_aperture_selected = None
  540. self.utility = []
  541. # this will flag if the Editor "tools" are launched from key shortcuts (True) or from menu toolbar (False)
  542. self.launched_from_shortcuts = False
  543. # this var will store the state of the toolbar before starting the editor
  544. self.toolbar_old_state = False
  545. self.app.ui.delete_drill_btn.triggered.connect(self.on_delete_btn)
  546. self.name_entry.returnPressed.connect(self.on_name_activate)
  547. self.addaperture_btn.clicked.connect(self.on_tool_add)
  548. # self.addtool_entry.editingFinished.connect(self.on_tool_add)
  549. self.delaperture_btn.clicked.connect(self.on_tool_delete)
  550. self.apertures_table.selectionModel().currentChanged.connect(self.on_row_selected)
  551. self.array_type_combo.currentIndexChanged.connect(self.on_array_type_combo)
  552. self.drill_axis_radio.activated_custom.connect(self.on_linear_angle_radio)
  553. self.app.ui.exc_resize_drill_menuitem.triggered.connect(self.exc_resize_drills)
  554. self.app.ui.exc_copy_drill_menuitem.triggered.connect(self.exc_copy_drills)
  555. self.app.ui.exc_delete_drill_menuitem.triggered.connect(self.on_delete_btn)
  556. self.app.ui.exc_move_drill_menuitem.triggered.connect(self.exc_move_drills)
  557. # Init GUI
  558. self.drill_array_size_entry.set_value(5)
  559. self.drill_pitch_entry.set_value(2.54)
  560. self.drill_angle_entry.set_value(12)
  561. self.drill_direction_radio.set_value('CW')
  562. self.drill_axis_radio.set_value('X')
  563. self.gerber_obj = None
  564. # VisPy Visuals
  565. self.shapes = self.app.plotcanvas.new_shape_collection(layers=1)
  566. self.tool_shape = self.app.plotcanvas.new_shape_collection(layers=1)
  567. self.app.pool_recreated.connect(self.pool_recreated)
  568. # Remove from scene
  569. self.shapes.enabled = False
  570. self.tool_shape.enabled = False
  571. ## List of selected shapes.
  572. self.selected = []
  573. self.move_timer = QtCore.QTimer()
  574. self.move_timer.setSingleShot(True)
  575. self.key = None # Currently pressed key
  576. self.modifiers = None
  577. self.x = None # Current mouse cursor pos
  578. self.y = None
  579. # Current snapped mouse pos
  580. self.snap_x = None
  581. self.snap_y = None
  582. self.pos = None
  583. def make_callback(thetool):
  584. def f():
  585. self.on_tool_select(thetool)
  586. return f
  587. for tool in self.tools_gerber:
  588. self.tools_gerber[tool]["button"].triggered.connect(make_callback(tool)) # Events
  589. self.tools_gerber[tool]["button"].setCheckable(True) # Checkable
  590. self.options = {
  591. "global_gridx": 0.1,
  592. "global_gridy": 0.1,
  593. "snap_max": 0.05,
  594. "grid_snap": True,
  595. "corner_snap": False,
  596. "grid_gap_link": True
  597. }
  598. self.app.options_read_form()
  599. for option in self.options:
  600. if option in self.app.options:
  601. self.options[option] = self.app.options[option]
  602. # flag to show if the object was modified
  603. self.is_modified = False
  604. self.edited_obj_name = ""
  605. self.tool_row = 0
  606. # store the status of the editor so the Delete at object level will not work until the edit is finished
  607. self.editor_active = False
  608. def entry2option(option, entry):
  609. self.options[option] = float(entry.text())
  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 pool_recreated(self, pool):
  613. self.shapes.pool = pool
  614. self.tool_shape.pool = pool
  615. def set_ui(self):
  616. # updated units
  617. self.units = self.app.ui.general_defaults_form.general_app_group.units_radio.get_value().upper()
  618. self.olddia_newdia.clear()
  619. self.tool2tooldia.clear()
  620. # update the olddia_newdia dict to make sure we have an updated state of the tool_table
  621. for key in self.storage_dict:
  622. self.olddia_newdia[key] = key
  623. sort_temp = []
  624. for aperture in self.olddia_newdia:
  625. sort_temp.append(float(aperture))
  626. self.sorted_apid = sorted(sort_temp)
  627. # populate self.intial_table_rows dict with the tool number as keys and tool diameters as values
  628. for i in range(len(self.sorted_apid)):
  629. tt_aperture = self.sorted_apid[i]
  630. self.tool2tooldia[i + 1] = tt_aperture
  631. def build_ui(self):
  632. try:
  633. # if connected, disconnect the signal from the slot on item_changed as it creates issues
  634. self.apertures_table.itemChanged.disconnect()
  635. except:
  636. pass
  637. # updated units
  638. self.units = self.app.ui.general_defaults_form.general_app_group.units_radio.get_value().upper()
  639. # make a new name for the new Excellon object (the one with edited content)
  640. self.edited_obj_name = self.gerber_obj.options['name']
  641. self.name_entry.set_value(self.edited_obj_name)
  642. if self.units == "IN":
  643. self.addtool_entry.set_value(0.039)
  644. else:
  645. self.addtool_entry.set_value(1.00)
  646. self.apertures_row = 0
  647. aper_no = self.apertures_row + 1
  648. sort = []
  649. for k, v in list(self.gerber_obj.apertures.items()):
  650. sort.append(int(k))
  651. sorted_apertures = sorted(sort)
  652. sort = []
  653. for k, v in list(self.gerber_obj.aperture_macros.items()):
  654. sort.append(k)
  655. sorted_macros = sorted(sort)
  656. n = len(sorted_apertures) + len(sorted_macros)
  657. self.apertures_table.setRowCount(n)
  658. for ap_code in sorted_apertures:
  659. ap_code = str(ap_code)
  660. ap_id_item = QtWidgets.QTableWidgetItem('%d' % int(self.apertures_row + 1))
  661. ap_id_item.setFlags(QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
  662. self.apertures_table.setItem(self.apertures_row, 0, ap_id_item) # Tool name/id
  663. ap_code_item = QtWidgets.QTableWidgetItem(ap_code)
  664. ap_code_item.setFlags(QtCore.Qt.ItemIsEnabled)
  665. ap_type_item = QtWidgets.QTableWidgetItem(str(self.gerber_obj.apertures[ap_code]['type']))
  666. ap_type_item.setFlags(QtCore.Qt.ItemIsEnabled)
  667. if str(self.gerber_obj.apertures[ap_code]['type']) == 'R' or str(self.gerber_obj.apertures[ap_code]['type']) == 'O':
  668. ap_dim_item = QtWidgets.QTableWidgetItem(
  669. '%.4f, %.4f' % (self.gerber_obj.apertures[ap_code]['width'] * self.gerber_obj.file_units_factor,
  670. self.gerber_obj.apertures[ap_code]['height'] * self.gerber_obj.file_units_factor
  671. )
  672. )
  673. ap_dim_item.setFlags(QtCore.Qt.ItemIsEnabled)
  674. elif str(self.gerber_obj.apertures[ap_code]['type']) == 'P':
  675. ap_dim_item = QtWidgets.QTableWidgetItem(
  676. '%.4f, %.4f' % (self.gerber_obj.apertures[ap_code]['diam'] * self.gerber_obj.file_units_factor,
  677. self.gerber_obj.apertures[ap_code]['nVertices'] * self.gerber_obj.file_units_factor)
  678. )
  679. ap_dim_item.setFlags(QtCore.Qt.ItemIsEnabled)
  680. else:
  681. ap_dim_item = QtWidgets.QTableWidgetItem('')
  682. ap_dim_item.setFlags(QtCore.Qt.ItemIsEnabled)
  683. try:
  684. if self.gerber_obj.apertures[ap_code]['size'] is not None:
  685. ap_size_item = QtWidgets.QTableWidgetItem('%.4f' %
  686. float(self.gerber_obj.apertures[ap_code]['size'] *
  687. self.gerber_obj.file_units_factor))
  688. else:
  689. ap_size_item = QtWidgets.QTableWidgetItem('')
  690. except KeyError:
  691. ap_size_item = QtWidgets.QTableWidgetItem('')
  692. ap_size_item.setFlags(QtCore.Qt.ItemIsEnabled)
  693. self.apertures_table.setItem(self.apertures_row, 1, ap_code_item) # Aperture Code
  694. self.apertures_table.setItem(self.apertures_row, 2, ap_type_item) # Aperture Type
  695. self.apertures_table.setItem(self.apertures_row, 3, ap_size_item) # Aperture Dimensions
  696. self.apertures_table.setItem(self.apertures_row, 4, ap_dim_item) # Aperture Dimensions
  697. self.apertures_row += 1
  698. for ap_code in sorted_macros:
  699. ap_code = str(ap_code)
  700. ap_id_item = QtWidgets.QTableWidgetItem('%d' % int(self.apertures_row + 1))
  701. ap_id_item.setFlags(QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
  702. self.apertures_table.setItem(self.apertures_row, 0, ap_id_item) # Tool name/id
  703. ap_code_item = QtWidgets.QTableWidgetItem(ap_code)
  704. ap_type_item = QtWidgets.QTableWidgetItem('AM')
  705. ap_type_item.setFlags(QtCore.Qt.ItemIsEnabled)
  706. self.apertures_table.setItem(self.apertures_row, 1, ap_code_item) # Aperture Code
  707. self.apertures_table.setItem(self.apertures_row, 2, ap_type_item) # Aperture Type
  708. self.apertures_row += 1
  709. self.apertures_table.selectColumn(0)
  710. self.apertures_table.resizeColumnsToContents()
  711. self.apertures_table.resizeRowsToContents()
  712. vertical_header = self.apertures_table.verticalHeader()
  713. # vertical_header.setSectionResizeMode(QtWidgets.QHeaderView.ResizeToContents)
  714. vertical_header.hide()
  715. self.apertures_table.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
  716. horizontal_header = self.apertures_table.horizontalHeader()
  717. horizontal_header.setMinimumSectionSize(10)
  718. horizontal_header.setDefaultSectionSize(70)
  719. horizontal_header.setSectionResizeMode(0, QtWidgets.QHeaderView.Fixed)
  720. horizontal_header.resizeSection(0, 20)
  721. horizontal_header.setSectionResizeMode(1, QtWidgets.QHeaderView.ResizeToContents)
  722. horizontal_header.setSectionResizeMode(2, QtWidgets.QHeaderView.ResizeToContents)
  723. horizontal_header.setSectionResizeMode(3, QtWidgets.QHeaderView.ResizeToContents)
  724. horizontal_header.setSectionResizeMode(4, QtWidgets.QHeaderView.Stretch)
  725. self.apertures_table.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
  726. self.apertures_table.setSortingEnabled(False)
  727. self.apertures_table.setMinimumHeight(self.apertures_table.getHeight())
  728. self.apertures_table.setMaximumHeight(self.apertures_table.getHeight())
  729. # make sure no rows are selected so the user have to click the correct row, meaning selecting the correct tool
  730. self.apertures_table.clearSelection()
  731. # Remove anything else in the GUI Selected Tab
  732. self.app.ui.selected_scroll_area.takeWidget()
  733. # Put ourself in the GUI Selected Tab
  734. self.app.ui.selected_scroll_area.setWidget(self.grb_edit_widget)
  735. # Switch notebook to Selected page
  736. self.app.ui.notebook.setCurrentWidget(self.app.ui.selected_tab)
  737. # we reactivate the signals after the after the tool adding as we don't need to see the tool been populated
  738. self.apertures_table.itemChanged.connect(self.on_tool_edit)
  739. def on_tool_add(self, tooldia=None):
  740. self.is_modified = True
  741. if tooldia:
  742. tool_dia = tooldia
  743. else:
  744. try:
  745. tool_dia = float(self.addtool_entry.get_value())
  746. except ValueError:
  747. # try to convert comma to decimal point. if it's still not working error message and return
  748. try:
  749. tool_dia = float(self.addtool_entry.get_value().replace(',', '.'))
  750. except ValueError:
  751. self.app.inform.emit(_("[ERROR_NOTCL] Wrong value format entered, "
  752. "use a number.")
  753. )
  754. return
  755. if tool_dia not in self.olddia_newdia:
  756. storage_elem = FlatCAMGeoEditor.make_storage()
  757. self.storage_dict[tool_dia] = storage_elem
  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[tool_dia] = tool_dia
  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)] = tool_dia
  769. self.app.inform.emit(_("[success] Added new tool with dia: {dia} {units}").format(dia=str(tool_dia), units=str(self.units)))
  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] == tool_dia:
  775. row_to_be_selected = int(key) - 1
  776. break
  777. self.apertures_table.selectRow(row_to_be_selected)
  778. def on_tool_delete(self, dia=None):
  779. self.is_modified = True
  780. deleted_tool_dia_list = []
  781. deleted_tool_offset_list = []
  782. try:
  783. if dia is None or dia 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(float(self.apertures_table.item(row, 1).text()))
  788. else:
  789. if isinstance(dia, list):
  790. for dd in dia:
  791. deleted_tool_dia_list.append(float('%.4f' % dd))
  792. else:
  793. deleted_tool_dia_list.append(float('%.4f' % dia))
  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 de tool offset
  799. self.gerber_obj.tool_offset.pop(float(deleted_tool_dia), None)
  800. # delete the storage used for that tool
  801. storage_elem = FlatCAMGeoEditor.make_storage()
  802. self.storage_dict[deleted_tool_dia] = storage_elem
  803. self.storage_dict.pop(deleted_tool_dia, None)
  804. # I've added this flag_del variable because dictionary don't like
  805. # having keys deleted while iterating through them
  806. flag_del = []
  807. # self.points_edit.pop(deleted_tool_dia, None)
  808. for deleted_tool in self.tool2tooldia:
  809. if self.tool2tooldia[deleted_tool] == deleted_tool_dia:
  810. flag_del.append(deleted_tool)
  811. if flag_del:
  812. for tool_to_be_deleted in flag_del:
  813. # delete the tool
  814. self.tool2tooldia.pop(tool_to_be_deleted, None)
  815. # delete also the drills from points_edit dict just in case we add the tool again, we don't want to show the
  816. # number of drills from before was deleter
  817. self.points_edit[deleted_tool_dia] = []
  818. flag_del = []
  819. self.olddia_newdia.pop(deleted_tool_dia, None)
  820. self.app.inform.emit(_("[success] Deleted tool with dia: {del_dia} {units}").format(del_dia=str(deleted_tool_dia), units=str(self.units)))
  821. self.plot_all()
  822. # self.app.inform.emit("Could not delete selected tool")
  823. self.build_ui()
  824. def on_tool_edit(self, item_changed):
  825. # if connected, disconnect the signal from the slot on item_changed as it creates issues
  826. self.apertures_table.itemChanged.disconnect()
  827. # self.apertures_table.selectionModel().currentChanged.disconnect()
  828. self.is_modified = True
  829. geometry = []
  830. current_table_dia_edited = None
  831. if self.apertures_table.currentItem() is not None:
  832. try:
  833. current_table_dia_edited = float(self.apertures_table.currentItem().text())
  834. except ValueError as e:
  835. log.debug("FlatCAMExcEditor.on_tool_edit() --> %s" % str(e))
  836. self.apertures_table.setCurrentItem(None)
  837. return
  838. row_of_item_changed = self.apertures_table.currentRow()
  839. # rows start with 0, tools start with 1 so we adjust the value by 1
  840. key_in_tool2tooldia = row_of_item_changed + 1
  841. dia_changed = self.tool2tooldia[key_in_tool2tooldia]
  842. # tool diameter is not used so we create a new tool with the desired diameter
  843. if current_table_dia_edited not in self.olddia_newdia.values():
  844. # update the dict that holds as keys our initial diameters and as values the edited diameters
  845. self.olddia_newdia[dia_changed] = current_table_dia_edited
  846. # update the dict that holds tool_no as key and tool_dia as value
  847. self.tool2tooldia[key_in_tool2tooldia] = current_table_dia_edited
  848. # update the tool offset
  849. modified_offset = self.gerber_obj.tool_offset.pop(dia_changed)
  850. self.gerber_obj.tool_offset[current_table_dia_edited] = modified_offset
  851. self.plot_all()
  852. else:
  853. # tool diameter is already in use so we move the drills from the prior tool to the new tool
  854. factor = current_table_dia_edited / dia_changed
  855. for shape in self.storage_dict[dia_changed].get_objects():
  856. geometry.append(DrawToolShape(
  857. MultiLineString([affinity.scale(subgeo, xfact=factor, yfact=factor) for subgeo in shape.geo])))
  858. self.points_edit[current_table_dia_edited].append((0, 0))
  859. self.add_gerber_shape(geometry, self.storage_dict[current_table_dia_edited])
  860. self.on_tool_delete(dia=dia_changed)
  861. # delete the tool offset
  862. self.gerber_obj.tool_offset.pop(dia_changed, None)
  863. # we reactivate the signals after the after the tool editing
  864. self.apertures_table.itemChanged.connect(self.on_tool_edit)
  865. # self.apertures_table.selectionModel().currentChanged.connect(self.on_row_selected)
  866. def on_name_activate(self):
  867. self.edited_obj_name = self.name_entry.get_value()
  868. def activate(self):
  869. self.connect_canvas_event_handlers()
  870. # self.app.collection.view.keyPressed.connect(self.on_canvas_key)
  871. self.shapes.enabled = True
  872. self.tool_shape.enabled = True
  873. # self.app.app_cursor.enabled = True
  874. self.app.ui.snap_max_dist_entry.setEnabled(True)
  875. self.app.ui.corner_snap_btn.setEnabled(True)
  876. self.app.ui.snap_magnet.setVisible(True)
  877. self.app.ui.corner_snap_btn.setVisible(True)
  878. self.app.ui.grb_editor_menu.setDisabled(False)
  879. self.app.ui.grb_editor_menu.menuAction().setVisible(True)
  880. self.app.ui.update_obj_btn.setEnabled(True)
  881. self.app.ui.grb_editor_cmenu.setEnabled(True)
  882. self.app.ui.grb_edit_toolbar.setDisabled(False)
  883. self.app.ui.grb_edit_toolbar.setVisible(True)
  884. # self.app.ui.snap_toolbar.setDisabled(False)
  885. # start with GRID toolbar activated
  886. if self.app.ui.grid_snap_btn.isChecked() is False:
  887. self.app.ui.grid_snap_btn.trigger()
  888. # Tell the App that the editor is active
  889. self.editor_active = True
  890. def deactivate(self):
  891. self.disconnect_canvas_event_handlers()
  892. self.clear()
  893. self.app.ui.grb_edit_toolbar.setDisabled(True)
  894. settings = QSettings("Open Source", "FlatCAM")
  895. if settings.contains("layout"):
  896. layout = settings.value('layout', type=str)
  897. if layout == 'standard':
  898. # self.app.ui.exc_edit_toolbar.setVisible(False)
  899. self.app.ui.snap_max_dist_entry.setEnabled(False)
  900. self.app.ui.corner_snap_btn.setEnabled(False)
  901. self.app.ui.snap_magnet.setVisible(False)
  902. self.app.ui.corner_snap_btn.setVisible(False)
  903. elif layout == 'compact':
  904. # self.app.ui.exc_edit_toolbar.setVisible(True)
  905. self.app.ui.snap_max_dist_entry.setEnabled(False)
  906. self.app.ui.corner_snap_btn.setEnabled(False)
  907. self.app.ui.snap_magnet.setVisible(True)
  908. self.app.ui.corner_snap_btn.setVisible(True)
  909. else:
  910. # self.app.ui.exc_edit_toolbar.setVisible(False)
  911. self.app.ui.snap_max_dist_entry.setEnabled(False)
  912. self.app.ui.corner_snap_btn.setEnabled(False)
  913. self.app.ui.snap_magnet.setVisible(False)
  914. self.app.ui.corner_snap_btn.setVisible(False)
  915. # set the Editor Toolbar visibility to what was before entering in the Editor
  916. self.app.ui.grb_edit_toolbar.setVisible(False) if self.toolbar_old_state is False \
  917. else self.app.ui.grb_edit_toolbar.setVisible(True)
  918. # Disable visuals
  919. self.shapes.enabled = False
  920. self.tool_shape.enabled = False
  921. # self.app.app_cursor.enabled = False
  922. # Tell the app that the editor is no longer active
  923. self.editor_active = False
  924. self.app.ui.grb_editor_menu.setDisabled(True)
  925. self.app.ui.grb_editor_menu.menuAction().setVisible(False)
  926. self.app.ui.update_obj_btn.setEnabled(False)
  927. self.app.ui.g_editor_cmenu.setEnabled(False)
  928. self.app.ui.grb_editor_cmenu.setEnabled(False)
  929. self.app.ui.e_editor_cmenu.setEnabled(False)
  930. # Show original geometry
  931. if self.gerber_obj:
  932. self.gerber_obj.visible = True
  933. def connect_canvas_event_handlers(self):
  934. ## Canvas events
  935. # make sure that the shortcuts key and mouse events will no longer be linked to the methods from FlatCAMApp
  936. # but those from FlatCAMGeoEditor
  937. self.app.plotcanvas.vis_disconnect('mouse_press', self.app.on_mouse_click_over_plot)
  938. self.app.plotcanvas.vis_disconnect('mouse_move', self.app.on_mouse_move_over_plot)
  939. self.app.plotcanvas.vis_disconnect('mouse_release', self.app.on_mouse_click_release_over_plot)
  940. self.app.plotcanvas.vis_disconnect('mouse_double_click', self.app.on_double_click_over_plot)
  941. self.app.collection.view.clicked.disconnect()
  942. self.canvas.vis_connect('mouse_press', self.on_canvas_click)
  943. self.canvas.vis_connect('mouse_move', self.on_canvas_move)
  944. self.canvas.vis_connect('mouse_release', self.on_canvas_click_release)
  945. def disconnect_canvas_event_handlers(self):
  946. self.canvas.vis_disconnect('mouse_press', self.on_canvas_click)
  947. self.canvas.vis_disconnect('mouse_move', self.on_canvas_move)
  948. self.canvas.vis_disconnect('mouse_release', self.on_canvas_click_release)
  949. # we restore the key and mouse control to FlatCAMApp method
  950. self.app.plotcanvas.vis_connect('mouse_press', self.app.on_mouse_click_over_plot)
  951. self.app.plotcanvas.vis_connect('mouse_move', self.app.on_mouse_move_over_plot)
  952. self.app.plotcanvas.vis_connect('mouse_release', self.app.on_mouse_click_release_over_plot)
  953. self.app.plotcanvas.vis_connect('mouse_double_click', self.app.on_double_click_over_plot)
  954. self.app.collection.view.clicked.connect(self.app.collection.on_mouse_down)
  955. def clear(self):
  956. self.active_tool = None
  957. # self.shape_buffer = []
  958. self.selected = []
  959. self.storage_dict = {}
  960. self.shapes.clear(update=True)
  961. self.tool_shape.clear(update=True)
  962. # self.storage = FlatCAMExcEditor.make_storage()
  963. self.plot_all()
  964. def edit_fcgerber(self, exc_obj):
  965. """
  966. Imports the geometry found in self.apertures from the given FlatCAM Gerber object
  967. into the editor.
  968. :param fcgeometry: FlatCAMExcellon
  969. :return: None
  970. """
  971. assert isinstance(exc_obj, Gerber), \
  972. "Expected an Excellon Object, got %s" % type(exc_obj)
  973. self.deactivate()
  974. self.activate()
  975. # Hide original geometry
  976. self.gerber_obj = exc_obj
  977. exc_obj.visible = False
  978. # Set selection tolerance
  979. # DrawToolShape.tolerance = fc_excellon.drawing_tolerance * 10
  980. self.select_tool("select")
  981. self.set_ui()
  982. # now that we hava data, create the GUI interface and add it to the Tool Tab
  983. self.build_ui()
  984. # we activate this after the initial build as we don't need to see the tool been populated
  985. self.apertures_table.itemChanged.connect(self.on_tool_edit)
  986. # build the geometry for each tool-diameter, each drill will be represented by a '+' symbol
  987. # and then add it to the storage elements (each storage elements is a member of a list
  988. def job_thread(apid):
  989. with self.app.proc_container.new(_("Adding aperture: %s geo ...") % str(apid)):
  990. solid_storage_elem = []
  991. follow_storage_elem = []
  992. self.storage_dict[apid] = {}
  993. for k, v in self.gerber_obj.apertures[apid].items():
  994. if k == 'solid_geometry':
  995. for geo in v:
  996. if geo is not None:
  997. self.add_gerber_shape(DrawToolShape(geo), solid_storage_elem)
  998. self.storage_dict[apid][k] = solid_storage_elem
  999. elif k == 'follow_geometry':
  1000. for geo in v:
  1001. if geo is not None:
  1002. self.add_gerber_shape(DrawToolShape(geo), follow_storage_elem)
  1003. self.storage_dict[apid][k] = follow_storage_elem
  1004. else:
  1005. self.storage_dict[apid][k] = v
  1006. apid_promise = apid
  1007. # Check promises and clear if exists
  1008. self.app.collection.plot_remove_promise(apid_promise)
  1009. # if apid_promise in self.app.collection.plot_promises:
  1010. # self.app.collection.plot_promises.remove(apid_promise)
  1011. for apid in self.gerber_obj.apertures:
  1012. self.app.worker_task.emit({'fcn': job_thread, 'params': [apid]})
  1013. self.app.collection.plot_promise(apid)
  1014. self.start_delayed_plot(check_period=500)
  1015. def update_fcgerber(self, grb_obj):
  1016. """
  1017. Create a new Gerber object that contain the edited content of the source Gerber object
  1018. :param grb_obj: FlatCAMGerber
  1019. :return: None
  1020. """
  1021. # if the 'delayed plot' malfunctioned stop the QTimer
  1022. try:
  1023. self.plot_thread.stop()
  1024. except:
  1025. pass
  1026. if "_edit" in self.edited_obj_name:
  1027. try:
  1028. id = int(self.edited_obj_name[-1]) + 1
  1029. self.edited_obj_name = self.edited_obj_name[:-1] + str(id)
  1030. except ValueError:
  1031. self.edited_obj_name += "_1"
  1032. else:
  1033. self.edited_obj_name += "_edit"
  1034. self.app.worker_task.emit({'fcn': self.new_edited_gerber,
  1035. 'params': [self.edited_obj_name]})
  1036. # reset the tool table
  1037. self.apertures_table.clear()
  1038. self.apertures_table.setHorizontalHeaderLabels(['#', _('Code'), _('Type'), _('Size'), _('Dim')])
  1039. self.last_aperture_selected = None
  1040. # restore GUI to the Selected TAB
  1041. # Remove anything else in the GUI
  1042. self.app.ui.selected_scroll_area.takeWidget()
  1043. # Switch notebook to Selected page
  1044. self.app.ui.notebook.setCurrentWidget(self.app.ui.selected_tab)
  1045. def update_options(self, obj):
  1046. try:
  1047. if not obj.options:
  1048. obj.options = {}
  1049. obj.options['xmin'] = 0
  1050. obj.options['ymin'] = 0
  1051. obj.options['xmax'] = 0
  1052. obj.options['ymax'] = 0
  1053. return True
  1054. else:
  1055. return False
  1056. except AttributeError:
  1057. obj.options = {}
  1058. return True
  1059. def new_edited_gerber(self, outname):
  1060. """
  1061. Creates a new Gerber object for the edited Gerber. Thread-safe.
  1062. :param outname: Name of the resulting object. None causes the name to be that of the file.
  1063. :type outname: str
  1064. :return: None
  1065. """
  1066. self.app.log.debug("Update the Gerber object with edited content. Source is %s" %
  1067. self.gerber_obj.options['name'])
  1068. # How the object should be initialized
  1069. def obj_init(grb_obj, app_obj):
  1070. poly_buffer = []
  1071. follow_buffer = []
  1072. for storage_apid, storage_val in self.storage_dict.items():
  1073. grb_obj.apertures[storage_apid] = {}
  1074. for k, v in storage_val.items():
  1075. if k == 'solid_geometry':
  1076. grb_obj.apertures[storage_apid][k] = []
  1077. for geo in v:
  1078. grb_obj.apertures[storage_apid][k].append(deepcopy(geo.geo))
  1079. poly_buffer.append(deepcopy(geo.geo))
  1080. if k == 'follow_geometry':
  1081. grb_obj.apertures[storage_apid][k] = []
  1082. for geo in v:
  1083. grb_obj.apertures[storage_apid][k].append(deepcopy(geo.geo))
  1084. follow_buffer.append(deepcopy(geo.geo))
  1085. else:
  1086. grb_obj.apertures[storage_apid][k] = v
  1087. grb_obj.aperture_macros = deepcopy(self.gerber_obj.aperture_macros)
  1088. new_poly = MultiPolygon(poly_buffer)
  1089. new_poly = new_poly.buffer(0.00000001)
  1090. new_poly = new_poly.buffer(-0.00000001)
  1091. grb_obj.solid_geometry = new_poly
  1092. grb_obj.follow_geometry = deepcopy(follow_buffer)
  1093. grb_obj.options = self.gerber_obj.options.copy()
  1094. grb_obj.options['name'] = outname
  1095. try:
  1096. grb_obj.create_geometry()
  1097. except KeyError:
  1098. self.app.inform.emit(
  1099. _( "[ERROR_NOTCL] There are no Aperture definitions in the file. Aborting Gerber creation.")
  1100. )
  1101. except:
  1102. msg = _("[ERROR] An internal error has ocurred. See shell.\n")
  1103. msg += traceback.format_exc()
  1104. app_obj.inform.emit(msg)
  1105. raise
  1106. # raise
  1107. with self.app.proc_container.new(_("Creating Gerber.")):
  1108. try:
  1109. self.app.new_object("gerber", outname, obj_init)
  1110. except Exception as e:
  1111. log.error("Error on object creation: %s" % str(e))
  1112. self.app.progress.emit(100)
  1113. return
  1114. self.app.inform.emit(_("[success] Gerber editing finished."))
  1115. # self.progress.emit(100)
  1116. def on_tool_select(self, tool):
  1117. """
  1118. Behavior of the toolbar. Tool initialization.
  1119. :rtype : None
  1120. """
  1121. current_tool = tool
  1122. self.app.log.debug("on_tool_select('%s')" % tool)
  1123. if self.last_aperture_selected is None and current_tool is not 'select':
  1124. # self.draw_app.select_tool('select')
  1125. self.complete = True
  1126. current_tool = 'select'
  1127. self.app.inform.emit(_("[WARNING_NOTCL] Cancelled. There is no Tool/Drill selected"))
  1128. # This is to make the group behave as radio group
  1129. if current_tool in self.tools_gerber:
  1130. if self.tools_gerber[current_tool]["button"].isChecked():
  1131. self.app.log.debug("%s is checked." % current_tool)
  1132. for t in self.tools_gerber:
  1133. if t != current_tool:
  1134. self.tools_gerber[t]["button"].setChecked(False)
  1135. # this is where the Editor toolbar classes (button's) are instantiated
  1136. self.active_tool = self.tools_gerber[current_tool]["constructor"](self)
  1137. # self.app.inform.emit(self.active_tool.start_msg)
  1138. else:
  1139. self.app.log.debug("%s is NOT checked." % current_tool)
  1140. for t in self.tools_gerber:
  1141. self.tools_gerber[t]["button"].setChecked(False)
  1142. self.active_tool = None
  1143. def on_row_selected(self):
  1144. self.selected = []
  1145. try:
  1146. selected_dia = self.tool2tooldia[self.apertures_table.currentRow() + 1]
  1147. self.last_aperture_selected = self.apertures_table.currentRow() + 1
  1148. for obj in self.storage_dict[selected_dia].get_objects():
  1149. self.selected.append(obj)
  1150. except Exception as e:
  1151. self.app.log.debug(str(e))
  1152. self.plot_all()
  1153. def toolbar_tool_toggle(self, key):
  1154. self.options[key] = self.sender().isChecked()
  1155. if self.options[key] == True:
  1156. return 1
  1157. else:
  1158. return 0
  1159. def on_canvas_click(self, event):
  1160. """
  1161. event.x and .y have canvas coordinates
  1162. event.xdaya and .ydata have plot coordinates
  1163. :param event: Event object dispatched by Matplotlib
  1164. :return: None
  1165. """
  1166. if event.button is 1:
  1167. self.app.ui.rel_position_label.setText("<b>Dx</b>: %.4f&nbsp;&nbsp; <b>Dy</b>: "
  1168. "%.4f&nbsp;&nbsp;&nbsp;&nbsp;" % (0, 0))
  1169. self.pos = self.canvas.vispy_canvas.translate_coords(event.pos)
  1170. ### Snap coordinates
  1171. x, y = self.app.geo_editor.snap(self.pos[0], self.pos[1])
  1172. self.pos = (x, y)
  1173. # print(self.active_tool)
  1174. # Selection with left mouse button
  1175. if self.active_tool is not None and event.button is 1:
  1176. # Dispatch event to active_tool
  1177. # msg = self.active_tool.click(self.app.geo_editor.snap(event.xdata, event.ydata))
  1178. msg = self.active_tool.click(self.app.geo_editor.snap(self.pos[0], self.pos[1]))
  1179. # If it is a shape generating tool
  1180. if isinstance(self.active_tool, FCShapeTool) and self.active_tool.complete:
  1181. if self.current_storage is not None:
  1182. self.on_exc_shape_complete(self.current_storage)
  1183. self.build_ui()
  1184. # MS: always return to the Select Tool if modifier key is not pressed
  1185. # else return to the current tool
  1186. key_modifier = QtWidgets.QApplication.keyboardModifiers()
  1187. if self.app.defaults["global_mselect_key"] == 'Control':
  1188. modifier_to_use = Qt.ControlModifier
  1189. else:
  1190. modifier_to_use = Qt.ShiftModifier
  1191. # if modifier key is pressed then we add to the selected list the current shape but if it's already
  1192. # in the selected list, we removed it. Therefore first click selects, second deselects.
  1193. if key_modifier == modifier_to_use:
  1194. self.select_tool(self.active_tool.name)
  1195. else:
  1196. self.select_tool("select")
  1197. return
  1198. if isinstance(self.active_tool, FCApertureSelect):
  1199. # self.app.log.debug("Replotting after click.")
  1200. self.plot_all()
  1201. else:
  1202. self.app.log.debug("No active tool to respond to click!")
  1203. def on_exc_shape_complete(self, storage):
  1204. self.app.log.debug("on_shape_complete()")
  1205. # Add shape
  1206. if type(storage) is list:
  1207. for item_storage in storage:
  1208. self.add_gerber_shape(self.active_tool.geometry, item_storage)
  1209. else:
  1210. self.add_gerber_shape(self.active_tool.geometry, storage)
  1211. # Remove any utility shapes
  1212. self.delete_utility_geometry()
  1213. self.tool_shape.clear(update=True)
  1214. # Replot and reset tool.
  1215. self.plot_all()
  1216. # self.active_tool = type(self.active_tool)(self)
  1217. def add_gerber_shape(self, shape, storage):
  1218. """
  1219. Adds a shape to the shape storage.
  1220. :param shape: Shape to be added.
  1221. :type shape: DrawToolShape
  1222. :return: None
  1223. """
  1224. # List of DrawToolShape?
  1225. if isinstance(shape, list):
  1226. for subshape in shape:
  1227. self.add_gerber_shape(subshape, storage)
  1228. return
  1229. assert isinstance(shape, DrawToolShape), \
  1230. "Expected a DrawToolShape, got %s" % str(type(shape))
  1231. assert shape.geo is not None, \
  1232. "Shape object has empty geometry (None)"
  1233. assert (isinstance(shape.geo, list) and len(shape.geo) > 0) or \
  1234. not isinstance(shape.geo, list), \
  1235. "Shape objects has empty geometry ([])"
  1236. if isinstance(shape, DrawToolUtilityShape):
  1237. self.utility.append(shape)
  1238. else:
  1239. storage.append(shape) # TODO: Check performance
  1240. def add_shape(self, shape):
  1241. """
  1242. Adds a shape to the shape storage.
  1243. :param shape: Shape to be added.
  1244. :type shape: DrawToolShape
  1245. :return: None
  1246. """
  1247. # List of DrawToolShape?
  1248. if isinstance(shape, list):
  1249. for subshape in shape:
  1250. self.add_shape(subshape)
  1251. return
  1252. assert isinstance(shape, DrawToolShape), \
  1253. "Expected a DrawToolShape, got %s" % type(shape)
  1254. assert shape.geo is not None, \
  1255. "Shape object has empty geometry (None)"
  1256. assert (isinstance(shape.geo, list) and len(shape.geo) > 0) or \
  1257. not isinstance(shape.geo, list), \
  1258. "Shape objects has empty geometry ([])"
  1259. if isinstance(shape, DrawToolUtilityShape):
  1260. self.utility.append(shape)
  1261. else:
  1262. self.storage.insert(shape) # TODO: Check performance
  1263. def on_canvas_click_release(self, event):
  1264. pos_canvas = self.canvas.vispy_canvas.translate_coords(event.pos)
  1265. self.modifiers = QtWidgets.QApplication.keyboardModifiers()
  1266. if self.app.grid_status():
  1267. pos = self.app.geo_editor.snap(pos_canvas[0], pos_canvas[1])
  1268. else:
  1269. pos = (pos_canvas[0], pos_canvas[1])
  1270. # if the released mouse button was RMB then test if it was a panning motion or not, if not it was a context
  1271. # canvas menu
  1272. try:
  1273. if event.button == 2: # right click
  1274. if self.app.panning_action is True:
  1275. self.app.panning_action = False
  1276. else:
  1277. self.app.cursor = QtGui.QCursor()
  1278. self.app.ui.popMenu.popup(self.app.cursor.pos())
  1279. except Exception as e:
  1280. log.warning("Error: %s" % str(e))
  1281. raise
  1282. # if the released mouse button was LMB then test if we had a right-to-left selection or a left-to-right
  1283. # selection and then select a type of selection ("enclosing" or "touching")
  1284. try:
  1285. if event.button == 1: # left click
  1286. if self.app.selection_type is not None:
  1287. self.draw_selection_area_handler(self.pos, pos, self.app.selection_type)
  1288. self.app.selection_type = None
  1289. elif isinstance(self.active_tool, FCApertureSelect):
  1290. # Dispatch event to active_tool
  1291. # msg = self.active_tool.click(self.app.geo_editor.snap(event.xdata, event.ydata))
  1292. # msg = self.active_tool.click_release((self.pos[0], self.pos[1]))
  1293. # self.app.inform.emit(msg)
  1294. self.active_tool.click_release((self.pos[0], self.pos[1]))
  1295. self.plot_all()
  1296. except Exception as e:
  1297. log.warning("Error: %s" % str(e))
  1298. raise
  1299. def draw_selection_area_handler(self, start_pos, end_pos, sel_type):
  1300. """
  1301. :param start_pos: mouse position when the selection LMB click was done
  1302. :param end_pos: mouse position when the left mouse button is released
  1303. :param sel_type: if True it's a left to right selection (enclosure), if False it's a 'touch' selection
  1304. :type Bool
  1305. :return:
  1306. """
  1307. poly_selection = Polygon([start_pos, (end_pos[0], start_pos[1]), end_pos, (start_pos[0], end_pos[1])])
  1308. self.app.delete_selection_shape()
  1309. for storage in self.storage_dict:
  1310. for obj in self.storage_dict[storage].get_objects():
  1311. if (sel_type is True and poly_selection.contains(obj.geo)) or \
  1312. (sel_type is False and poly_selection.intersects(obj.geo)):
  1313. if self.key == self.app.defaults["global_mselect_key"]:
  1314. if obj in self.selected:
  1315. self.selected.remove(obj)
  1316. else:
  1317. # add the object to the selected shapes
  1318. self.selected.append(obj)
  1319. else:
  1320. self.selected.append(obj)
  1321. # select the diameter of the selected shape in the tool table
  1322. for storage in self.storage_dict:
  1323. for shape_s in self.selected:
  1324. if shape_s in self.storage_dict[storage].get_objects():
  1325. for key in self.tool2tooldia:
  1326. if self.tool2tooldia[key] == storage:
  1327. item = self.apertures_table.item((key - 1), 1)
  1328. self.apertures_table.setCurrentItem(item)
  1329. self.last_aperture_selected = key
  1330. # item.setSelected(True)
  1331. # self.grb_editor_app.apertures_table.selectItem(key - 1)
  1332. self.plot_all()
  1333. def on_canvas_move(self, event):
  1334. """
  1335. Called on 'mouse_move' event
  1336. event.pos have canvas screen coordinates
  1337. :param event: Event object dispatched by VisPy SceneCavas
  1338. :return: None
  1339. """
  1340. pos = self.canvas.vispy_canvas.translate_coords(event.pos)
  1341. event.xdata, event.ydata = pos[0], pos[1]
  1342. self.x = event.xdata
  1343. self.y = event.ydata
  1344. # Prevent updates on pan
  1345. # if len(event.buttons) > 0:
  1346. # return
  1347. # if the RMB is clicked and mouse is moving over plot then 'panning_action' is True
  1348. if event.button == 2:
  1349. self.app.panning_action = True
  1350. return
  1351. else:
  1352. self.app.panning_action = False
  1353. try:
  1354. x = float(event.xdata)
  1355. y = float(event.ydata)
  1356. except TypeError:
  1357. return
  1358. if self.active_tool is None:
  1359. return
  1360. ### Snap coordinates
  1361. x, y = self.app.geo_editor.app.geo_editor.snap(x, y)
  1362. self.snap_x = x
  1363. self.snap_y = y
  1364. # update the position label in the infobar since the APP mouse event handlers are disconnected
  1365. self.app.ui.position_label.setText("&nbsp;&nbsp;&nbsp;&nbsp;<b>X</b>: %.4f&nbsp;&nbsp; "
  1366. "<b>Y</b>: %.4f" % (x, y))
  1367. if self.pos is None:
  1368. self.pos = (0, 0)
  1369. dx = x - self.pos[0]
  1370. dy = y - self.pos[1]
  1371. # update the reference position label in the infobar since the APP mouse event handlers are disconnected
  1372. self.app.ui.rel_position_label.setText("<b>Dx</b>: %.4f&nbsp;&nbsp; <b>Dy</b>: "
  1373. "%.4f&nbsp;&nbsp;&nbsp;&nbsp;" % (dx, dy))
  1374. ### Utility geometry (animated)
  1375. geo = self.active_tool.utility_geometry(data=(x, y))
  1376. if isinstance(geo, DrawToolShape) and geo.geo is not None:
  1377. # Remove any previous utility shape
  1378. self.tool_shape.clear(update=True)
  1379. self.draw_utility_geometry(geo=geo)
  1380. ### Selection area on canvas section ###
  1381. dx = pos[0] - self.pos[0]
  1382. if event.is_dragging == 1 and event.button == 1:
  1383. self.app.delete_selection_shape()
  1384. if dx < 0:
  1385. self.app.draw_moving_selection_shape((self.pos[0], self.pos[1]), (x,y),
  1386. color=self.app.defaults["global_alt_sel_line"],
  1387. face_color=self.app.defaults['global_alt_sel_fill'])
  1388. self.app.selection_type = False
  1389. else:
  1390. self.app.draw_moving_selection_shape((self.pos[0], self.pos[1]), (x,y))
  1391. self.app.selection_type = True
  1392. else:
  1393. self.app.selection_type = None
  1394. # Update cursor
  1395. self.app.app_cursor.set_data(np.asarray([(x, y)]), symbol='++', edge_color='black', size=20)
  1396. def on_canvas_key_release(self, event):
  1397. self.key = None
  1398. def draw_utility_geometry(self, geo):
  1399. # Add the new utility shape
  1400. try:
  1401. # this case is for the Font Parse
  1402. for el in list(geo.geo):
  1403. if type(el) == MultiPolygon:
  1404. for poly in el:
  1405. self.tool_shape.add(
  1406. shape=poly,
  1407. color=(self.app.defaults["global_draw_color"] + '80'),
  1408. update=False,
  1409. layer=0,
  1410. tolerance=None
  1411. )
  1412. elif type(el) == MultiLineString:
  1413. for linestring in el:
  1414. self.tool_shape.add(
  1415. shape=linestring,
  1416. color=(self.app.defaults["global_draw_color"] + '80'),
  1417. update=False,
  1418. layer=0,
  1419. tolerance=None
  1420. )
  1421. else:
  1422. self.tool_shape.add(
  1423. shape=el,
  1424. color=(self.app.defaults["global_draw_color"] + '80'),
  1425. update=False,
  1426. layer=0,
  1427. tolerance=None
  1428. )
  1429. except TypeError:
  1430. self.tool_shape.add(
  1431. shape=geo.geo, color=(self.app.defaults["global_draw_color"] + '80'),
  1432. update=False, layer=0, tolerance=None)
  1433. self.tool_shape.redraw()
  1434. def plot_all(self):
  1435. """
  1436. Plots all shapes in the editor.
  1437. :return: None
  1438. :rtype: None
  1439. """
  1440. with self.app.proc_container.new("Plotting"):
  1441. # self.app.log.debug("plot_all()")
  1442. self.shapes.clear(update=True)
  1443. for storage in self.storage_dict:
  1444. for shape in self.storage_dict[storage]['solid_geometry']:
  1445. if shape.geo is None:
  1446. continue
  1447. if shape in self.selected:
  1448. self.plot_shape(geometry=shape.geo, color=self.app.defaults['global_sel_draw_color'],
  1449. linewidth=2)
  1450. continue
  1451. self.plot_shape(geometry=shape.geo, color=self.app.defaults['global_draw_color'])
  1452. for shape in self.utility:
  1453. self.plot_shape(geometry=shape.geo, linewidth=1)
  1454. continue
  1455. self.shapes.redraw()
  1456. def start_delayed_plot(self, check_period):
  1457. # self.plot_thread = threading.Thread(target=lambda: self.check_plot_finished(check_period))
  1458. # self.plot_thread.start()
  1459. log.debug("FlatCAMGrbEditor --> Delayed Plot started.")
  1460. self.plot_thread = QtCore.QTimer()
  1461. self.plot_thread.setInterval(check_period)
  1462. self.plot_thread.timeout.connect(self.check_plot_finished)
  1463. self.plot_thread.start()
  1464. def check_plot_finished(self):
  1465. print(self.app.collection.plot_promises)
  1466. try:
  1467. has_promise = self.app.collection.has_plot_promises()
  1468. if has_promise == False:
  1469. self.plot_thread.stop()
  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