ToolCopperThieving.py 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771
  1. # ##########################################################
  2. # FlatCAM: 2D Post-processing for Manufacturing #
  3. # File Author: Marius Adrian Stanciu (c) #
  4. # Date: 10/25/2019 #
  5. # MIT Licence #
  6. # ##########################################################
  7. from PyQt5 import QtWidgets, QtCore
  8. import FlatCAMApp
  9. from FlatCAMTool import FlatCAMTool
  10. from flatcamGUI.GUIElements import FCDoubleSpinner, RadioSet
  11. from FlatCAMObj import FlatCAMGerber, FlatCAMGeometry, FlatCAMExcellon
  12. import shapely.geometry.base as base
  13. from shapely.ops import cascaded_union, unary_union
  14. from shapely.geometry import Polygon, MultiPolygon
  15. from shapely.geometry import box as box
  16. import logging
  17. from copy import deepcopy
  18. import numpy as np
  19. from collections import Iterable
  20. import gettext
  21. import FlatCAMTranslation as fcTranslate
  22. import builtins
  23. fcTranslate.apply_language('strings')
  24. if '_' not in builtins.__dict__:
  25. _ = gettext.gettext
  26. log = logging.getLogger('base')
  27. class ToolCopperThieving(FlatCAMTool):
  28. toolName = _("Copper Thieving Tool")
  29. def __init__(self, app):
  30. FlatCAMTool.__init__(self, app)
  31. self.app = app
  32. self.canvas = self.app.plotcanvas
  33. self.decimals = 4
  34. self.units = ''
  35. # ## Title
  36. title_label = QtWidgets.QLabel("%s" % self.toolName)
  37. title_label.setStyleSheet("""
  38. QLabel
  39. {
  40. font-size: 16px;
  41. font-weight: bold;
  42. }
  43. """)
  44. self.layout.addWidget(title_label)
  45. self.layout.addWidget(QtWidgets.QLabel(''))
  46. # ## Grid Layout
  47. i_grid_lay = QtWidgets.QGridLayout()
  48. self.layout.addLayout(i_grid_lay)
  49. i_grid_lay.setColumnStretch(0, 0)
  50. i_grid_lay.setColumnStretch(1, 1)
  51. self.grb_object_combo = QtWidgets.QComboBox()
  52. self.grb_object_combo.setModel(self.app.collection)
  53. self.grb_object_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  54. self.grb_object_combo.setCurrentIndex(1)
  55. self.grbobj_label = QtWidgets.QLabel("<b>%s:</b>" % _("GERBER"))
  56. self.grbobj_label.setToolTip(
  57. _("Gerber Object to which will be added a copper thieving.")
  58. )
  59. i_grid_lay.addWidget(self.grbobj_label, 0, 0)
  60. i_grid_lay.addWidget(self.grb_object_combo, 0, 1, 1, 2)
  61. i_grid_lay.addWidget(QtWidgets.QLabel(''), 1, 0)
  62. # ## Grid Layout
  63. grid_lay = QtWidgets.QGridLayout()
  64. self.layout.addLayout(grid_lay)
  65. grid_lay.setColumnStretch(0, 0)
  66. grid_lay.setColumnStretch(1, 1)
  67. self.copper_fill_label = QtWidgets.QLabel('<b>%s</b>' % _('Parameters'))
  68. self.copper_fill_label.setToolTip(
  69. _("Parameters used for this tool.")
  70. )
  71. grid_lay.addWidget(self.copper_fill_label, 0, 0, 1, 2)
  72. # CLEARANCE #
  73. self.clearance_label = QtWidgets.QLabel('%s:' % _("Clearance"))
  74. self.clearance_label.setToolTip(
  75. _("This set the distance between the copper thieving components\n"
  76. "(the polygon fill may be split in multiple polygons)\n"
  77. "and the copper traces in the Gerber file.")
  78. )
  79. self.clearance_entry = FCDoubleSpinner()
  80. self.clearance_entry.setMinimum(0.00001)
  81. self.clearance_entry.set_precision(self.decimals)
  82. self.clearance_entry.setSingleStep(0.1)
  83. grid_lay.addWidget(self.clearance_label, 1, 0)
  84. grid_lay.addWidget(self.clearance_entry, 1, 1)
  85. # MARGIN #
  86. self.margin_label = QtWidgets.QLabel('%s:' % _("Margin"))
  87. self.margin_label.setToolTip(
  88. _("Bounding box margin.")
  89. )
  90. self.margin_entry = FCDoubleSpinner()
  91. self.margin_entry.setMinimum(0.0)
  92. self.margin_entry.set_precision(self.decimals)
  93. self.margin_entry.setSingleStep(0.1)
  94. grid_lay.addWidget(self.margin_label, 2, 0)
  95. grid_lay.addWidget(self.margin_entry, 2, 1)
  96. # Reference #
  97. self.reference_radio = RadioSet([
  98. {'label': _('Itself'), 'value': 'itself'},
  99. {"label": _("Area Selection"), "value": "area"},
  100. {'label': _("Reference Object"), 'value': 'box'}
  101. ], orientation='vertical', stretch=False)
  102. self.reference_label = QtWidgets.QLabel(_("Reference:"))
  103. self.reference_label.setToolTip(
  104. _("- 'Itself' - the copper thieving extent is based on the object that is copper cleared.\n "
  105. "- 'Area Selection' - left mouse click to start selection of the area to be filled.\n"
  106. "- 'Reference Object' - will do copper thieving within the area specified by another object.")
  107. )
  108. grid_lay.addWidget(self.reference_label, 3, 0)
  109. grid_lay.addWidget(self.reference_radio, 3, 1)
  110. self.box_combo_type_label = QtWidgets.QLabel('%s:' % _("Ref. Type"))
  111. self.box_combo_type_label.setToolTip(
  112. _("The type of FlatCAM object to be used as copper thieving reference.\n"
  113. "It can be Gerber, Excellon or Geometry.")
  114. )
  115. self.box_combo_type = QtWidgets.QComboBox()
  116. self.box_combo_type.addItem(_("Reference Gerber"))
  117. self.box_combo_type.addItem(_("Reference Excellon"))
  118. self.box_combo_type.addItem(_("Reference Geometry"))
  119. grid_lay.addWidget(self.box_combo_type_label, 4, 0)
  120. grid_lay.addWidget(self.box_combo_type, 4, 1)
  121. self.box_combo_label = QtWidgets.QLabel('%s:' % _("Ref. Object"))
  122. self.box_combo_label.setToolTip(
  123. _("The FlatCAM object to be used as non copper clearing reference.")
  124. )
  125. self.box_combo = QtWidgets.QComboBox()
  126. self.box_combo.setModel(self.app.collection)
  127. self.box_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  128. self.box_combo.setCurrentIndex(1)
  129. grid_lay.addWidget(self.box_combo_label, 5, 0)
  130. grid_lay.addWidget(self.box_combo, 5, 1)
  131. self.box_combo.hide()
  132. self.box_combo_label.hide()
  133. self.box_combo_type.hide()
  134. self.box_combo_type_label.hide()
  135. # Bounding Box Type #
  136. self.bbox_type_radio = RadioSet([
  137. {'label': _('Rectangular'), 'value': 'rect'},
  138. {"label": _("Minimal"), "value": "min"}
  139. ], stretch=False)
  140. self.bbox_type_label = QtWidgets.QLabel(_("Box Type:"))
  141. self.bbox_type_label.setToolTip(
  142. _("- 'Rectangular' - the bounding box will be of rectangular shape.\n "
  143. "- 'Minimal' - the bounding box will be the convex hull shape.")
  144. )
  145. grid_lay.addWidget(self.bbox_type_label, 6, 0)
  146. grid_lay.addWidget(self.bbox_type_radio, 6, 1)
  147. self.bbox_type_label.hide()
  148. self.bbox_type_radio.hide()
  149. separator_line = QtWidgets.QFrame()
  150. separator_line.setFrameShape(QtWidgets.QFrame.HLine)
  151. separator_line.setFrameShadow(QtWidgets.QFrame.Sunken)
  152. grid_lay.addWidget(separator_line, 7, 0, 1, 2)
  153. # Fill Type
  154. self.fill_type_radio = RadioSet([
  155. {'label': _('Solid'), 'value': 'solid'},
  156. {"label": _("Dots Grid"), "value": "dot"},
  157. {"label": _("Squares Grid"), "value": "square"},
  158. {"label": _("Lines Grid"), "value": "line"}
  159. ], orientation='vertical', stretch=False)
  160. self.fill_type_label = QtWidgets.QLabel(_("Fill Type:"))
  161. self.fill_type_label.setToolTip(
  162. _("- 'Solid' - copper thieving will be a solid polygon.\n "
  163. "- 'Dots Grid' - the empty area will be filled with a pattern of dots.\n"
  164. "- 'Squares Grid' - the empty area will be filled with a pattern of squares.\n"
  165. "- 'Lines Grid' - the empty area will be filled with a pattern of lines.")
  166. )
  167. grid_lay.addWidget(self.fill_type_label, 8, 0)
  168. grid_lay.addWidget(self.fill_type_radio, 8, 1)
  169. # ## Insert Copper Thieving
  170. self.fill_button = QtWidgets.QPushButton(_("Insert Copper thieving"))
  171. self.fill_button.setToolTip(
  172. _("Will add a polygon (may be split in multiple parts)\n"
  173. "that will surround the actual Gerber traces at a certain distance.")
  174. )
  175. self.layout.addWidget(self.fill_button)
  176. self.layout.addStretch()
  177. # Objects involved in Copper thieving
  178. self.grb_object = None
  179. self.ref_obj = None
  180. self.sel_rect = list()
  181. # Events ID
  182. self.mr = None
  183. self.mm = None
  184. # Mouse cursor positions
  185. self.mouse_is_dragging = False
  186. self.cursor_pos = (0, 0)
  187. self.first_click = False
  188. self.area_method = False
  189. # Tool properties
  190. self.clearance_val = None
  191. self.margin_val = None
  192. self.geo_steps_per_circle = 128
  193. # SIGNALS
  194. self.fill_button.clicked.connect(self.execute)
  195. self.box_combo_type.currentIndexChanged.connect(self.on_combo_box_type)
  196. self.reference_radio.group_toggle_fn = self.on_toggle_reference
  197. def run(self, toggle=True):
  198. self.app.report_usage("ToolCopperThieving()")
  199. if toggle:
  200. # if the splitter is hidden, display it, else hide it but only if the current widget is the same
  201. if self.app.ui.splitter.sizes()[0] == 0:
  202. self.app.ui.splitter.setSizes([1, 1])
  203. else:
  204. try:
  205. if self.app.ui.tool_scroll_area.widget().objectName() == self.toolName:
  206. # if tab is populated with the tool but it does not have the focus, focus on it
  207. if not self.app.ui.notebook.currentWidget() is self.app.ui.tool_tab:
  208. # focus on Tool Tab
  209. self.app.ui.notebook.setCurrentWidget(self.app.ui.tool_tab)
  210. else:
  211. self.app.ui.splitter.setSizes([0, 1])
  212. except AttributeError:
  213. pass
  214. else:
  215. if self.app.ui.splitter.sizes()[0] == 0:
  216. self.app.ui.splitter.setSizes([1, 1])
  217. FlatCAMTool.run(self)
  218. self.set_tool_ui()
  219. self.app.ui.notebook.setTabText(2, _("Copper Thieving Tool"))
  220. def install(self, icon=None, separator=None, **kwargs):
  221. FlatCAMTool.install(self, icon, separator, shortcut='ALT+F', **kwargs)
  222. def set_tool_ui(self):
  223. self.units = self.app.ui.general_defaults_form.general_app_group.units_radio.get_value()
  224. self.clearance_entry.set_value(float(self.app.defaults["tools_copper_thieving_clearance"]))
  225. self.margin_entry.set_value(float(self.app.defaults["tools_copper_thieving_margin"]))
  226. self.reference_radio.set_value(self.app.defaults["tools_copper_thieving_reference"])
  227. self.bbox_type_radio.set_value(self.app.defaults["tools_copper_thieving_box_type"])
  228. self.fill_type_radio.set_value(self.app.defaults["tools_copper_thieving_fill_type"])
  229. self.geo_steps_per_circle = int(self.app.defaults["tools_copper_thieving_circle_steps"])
  230. self.area_method = False
  231. def on_combo_box_type(self):
  232. obj_type = self.box_combo_type.currentIndex()
  233. self.box_combo.setRootModelIndex(self.app.collection.index(obj_type, 0, QtCore.QModelIndex()))
  234. self.box_combo.setCurrentIndex(0)
  235. def on_toggle_reference(self):
  236. if self.reference_radio.get_value() == "itself" or self.reference_radio.get_value() == "area":
  237. self.box_combo.hide()
  238. self.box_combo_label.hide()
  239. self.box_combo_type.hide()
  240. self.box_combo_type_label.hide()
  241. else:
  242. self.box_combo.show()
  243. self.box_combo_label.show()
  244. self.box_combo_type.show()
  245. self.box_combo_type_label.show()
  246. if self.reference_radio.get_value() == "itself":
  247. self.bbox_type_label.show()
  248. self.bbox_type_radio.show()
  249. else:
  250. self.bbox_type_label.hide()
  251. self.bbox_type_radio.hide()
  252. def execute(self):
  253. self.app.call_source = "copper_thieving_tool"
  254. self.clearance_val = self.clearance_entry.get_value()
  255. self.margin_val = self.margin_entry.get_value()
  256. reference_method = self.reference_radio.get_value()
  257. # get the Gerber object on which the Copper thieving will be inserted
  258. selection_index = self.grb_object_combo.currentIndex()
  259. model_index = self.app.collection.index(selection_index, 0, self.grb_object_combo.rootModelIndex())
  260. try:
  261. self.grb_object = model_index.internalPointer().obj
  262. except Exception as e:
  263. log.debug("ToolCopperThieving.execute() --> %s" % str(e))
  264. self.app.inform.emit('[WARNING_NOTCL] %s' % _("There is no Gerber object loaded ..."))
  265. return 'fail'
  266. if reference_method == 'itself':
  267. bound_obj_name = self.grb_object_combo.currentText()
  268. # Get reference object.
  269. try:
  270. self.ref_obj = self.app.collection.get_by_name(bound_obj_name)
  271. except Exception as e:
  272. self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve object"), str(e)))
  273. return "Could not retrieve object: %s" % self.obj_name
  274. self.on_copper_fill(
  275. fill_obj=self.grb_object,
  276. c_val=self.clearance_val,
  277. margin=self.margin_val
  278. )
  279. elif reference_method == 'area':
  280. self.app.inform.emit('[WARNING_NOTCL] %s' % _("Click the start point of the area."))
  281. self.area_method = True
  282. if self.app.is_legacy is False:
  283. self.app.plotcanvas.graph_event_disconnect('mouse_press', self.app.on_mouse_click_over_plot)
  284. self.app.plotcanvas.graph_event_disconnect('mouse_move', self.app.on_mouse_move_over_plot)
  285. self.app.plotcanvas.graph_event_disconnect('mouse_release', self.app.on_mouse_click_release_over_plot)
  286. else:
  287. self.app.plotcanvas.graph_event_disconnect(self.app.mp)
  288. self.app.plotcanvas.graph_event_disconnect(self.app.mm)
  289. self.app.plotcanvas.graph_event_disconnect(self.app.mr)
  290. self.mr = self.app.plotcanvas.graph_event_connect('mouse_release', self.on_mouse_release)
  291. self.mm = self.app.plotcanvas.graph_event_connect('mouse_move', self.on_mouse_move)
  292. elif reference_method == 'box':
  293. bound_obj_name = self.box_combo.currentText()
  294. # Get reference object.
  295. try:
  296. self.ref_obj = self.app.collection.get_by_name(bound_obj_name)
  297. except Exception as e:
  298. self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve object"), bound_obj_name))
  299. return "Could not retrieve object: %s. Error: %s" % (bound_obj_name, str(e))
  300. self.on_copper_fill(
  301. fill_obj=self.grb_object,
  302. ref_obj=self.ref_obj,
  303. c_val=self.clearance_val,
  304. margin=self.margin_val
  305. )
  306. # To be called after clicking on the plot.
  307. def on_mouse_release(self, event):
  308. if self.app.is_legacy is False:
  309. event_pos = event.pos
  310. # event_is_dragging = event.is_dragging
  311. right_button = 2
  312. else:
  313. event_pos = (event.xdata, event.ydata)
  314. # event_is_dragging = self.app.plotcanvas.is_dragging
  315. right_button = 3
  316. event_pos = self.app.plotcanvas.translate_coords(event_pos)
  317. # do clear area only for left mouse clicks
  318. if event.button == 1:
  319. if self.first_click is False:
  320. self.first_click = True
  321. self.app.inform.emit('[WARNING_NOTCL] %s' % _("Click the end point of the filling area."))
  322. self.cursor_pos = self.app.plotcanvas.translate_coords(event_pos)
  323. if self.app.grid_status() is True:
  324. self.cursor_pos = self.app.geo_editor.snap(event_pos[0], event_pos[1])
  325. else:
  326. self.app.inform.emit(_("Zone added. Click to start adding next zone or right click to finish."))
  327. self.app.delete_selection_shape()
  328. if self.app.grid_status() is True:
  329. curr_pos = self.app.geo_editor.snap(event_pos[0], event_pos[1])
  330. else:
  331. curr_pos = (event_pos[0], event_pos[1])
  332. x0, y0 = self.cursor_pos[0], self.cursor_pos[1]
  333. x1, y1 = curr_pos[0], curr_pos[1]
  334. pt1 = (x0, y0)
  335. pt2 = (x1, y0)
  336. pt3 = (x1, y1)
  337. pt4 = (x0, y1)
  338. new_rectangle = Polygon([pt1, pt2, pt3, pt4])
  339. self.sel_rect.append(new_rectangle)
  340. # add a temporary shape on canvas
  341. self.draw_tool_selection_shape(old_coords=(x0, y0), coords=(x1, y1))
  342. self.first_click = False
  343. return
  344. elif event.button == right_button and self.mouse_is_dragging is False:
  345. self.area_method = False
  346. self.first_click = False
  347. self.delete_tool_selection_shape()
  348. if self.app.is_legacy is False:
  349. self.app.plotcanvas.graph_event_disconnect('mouse_release', self.on_mouse_release)
  350. self.app.plotcanvas.graph_event_disconnect('mouse_move', self.on_mouse_move)
  351. else:
  352. self.app.plotcanvas.graph_event_disconnect(self.mr)
  353. self.app.plotcanvas.graph_event_disconnect(self.mm)
  354. self.app.mp = self.app.plotcanvas.graph_event_connect('mouse_press',
  355. self.app.on_mouse_click_over_plot)
  356. self.app.mm = self.app.plotcanvas.graph_event_connect('mouse_move',
  357. self.app.on_mouse_move_over_plot)
  358. self.app.mr = self.app.plotcanvas.graph_event_connect('mouse_release',
  359. self.app.on_mouse_click_release_over_plot)
  360. if len(self.sel_rect) == 0:
  361. return
  362. self.sel_rect = cascaded_union(self.sel_rect)
  363. if not isinstance(self.sel_rect, Iterable):
  364. self.sel_rect = [self.sel_rect]
  365. self.on_copper_fill(
  366. fill_obj=self.grb_object,
  367. ref_obj=self.sel_rect,
  368. c_val=self.clearance_val,
  369. margin=self.margin_val
  370. )
  371. # called on mouse move
  372. def on_mouse_move(self, event):
  373. if self.app.is_legacy is False:
  374. event_pos = event.pos
  375. event_is_dragging = event.is_dragging
  376. # right_button = 2
  377. else:
  378. event_pos = (event.xdata, event.ydata)
  379. event_is_dragging = self.app.plotcanvas.is_dragging
  380. # right_button = 3
  381. curr_pos = self.app.plotcanvas.translate_coords(event_pos)
  382. # detect mouse dragging motion
  383. if event_is_dragging is True:
  384. self.mouse_is_dragging = True
  385. else:
  386. self.mouse_is_dragging = False
  387. # update the cursor position
  388. if self.app.grid_status() is True:
  389. # Update cursor
  390. curr_pos = self.app.geo_editor.snap(curr_pos[0], curr_pos[1])
  391. self.app.app_cursor.set_data(np.asarray([(curr_pos[0], curr_pos[1])]),
  392. symbol='++', edge_color=self.app.cursor_color_3D,
  393. size=self.app.defaults["global_cursor_size"])
  394. # update the positions on status bar
  395. self.app.ui.position_label.setText("&nbsp;&nbsp;&nbsp;&nbsp;<b>X</b>: %.4f&nbsp;&nbsp; "
  396. "<b>Y</b>: %.4f" % (curr_pos[0], curr_pos[1]))
  397. if self.cursor_pos is None:
  398. self.cursor_pos = (0, 0)
  399. dx = curr_pos[0] - float(self.cursor_pos[0])
  400. dy = curr_pos[1] - float(self.cursor_pos[1])
  401. self.app.ui.rel_position_label.setText("<b>Dx</b>: %.4f&nbsp;&nbsp; <b>Dy</b>: "
  402. "%.4f&nbsp;&nbsp;&nbsp;&nbsp;" % (dx, dy))
  403. # draw the utility geometry
  404. if self.first_click:
  405. self.app.delete_selection_shape()
  406. self.app.draw_moving_selection_shape(old_coords=(self.cursor_pos[0], self.cursor_pos[1]),
  407. coords=(curr_pos[0], curr_pos[1]))
  408. def on_copper_fill(self, fill_obj, ref_obj=None, c_val=None, margin=None, run_threaded=True):
  409. """
  410. :param fill_obj:
  411. :param ref_obj:
  412. :param c_val:
  413. :param margin:
  414. :param run_threaded:
  415. :return:
  416. """
  417. if run_threaded:
  418. proc = self.app.proc_container.new('%s ...' % _("Copper thieving"))
  419. else:
  420. self.app.proc_container.view.set_busy('%s ...' % _("Copper thieving"))
  421. QtWidgets.QApplication.processEvents()
  422. # #####################################################################
  423. # ####### Read the parameters #########################################
  424. # #####################################################################
  425. log.debug("Copper Thieving Tool started. Reading parameters.")
  426. self.app.inform.emit(_("Copper Thieving Tool started. Reading parameters."))
  427. ref_selected = self.reference_radio.get_value()
  428. if c_val is None:
  429. c_val = float(self.app.defaults["tools_copperfill_clearance"])
  430. if margin is None:
  431. margin = float(self.app.defaults["tools_copperfill_margin"])
  432. # make sure that the source object solid geometry is an Iterable
  433. if not isinstance(self.grb_object.solid_geometry, Iterable):
  434. self.grb_object.solid_geometry = [self.grb_object.solid_geometry]
  435. # #########################################################################################
  436. # Prepare isolation polygon. This will create the clearance over the Gerber features ######
  437. # #########################################################################################
  438. log.debug("Copper Thieving Tool. Preparing isolation polygons.")
  439. self.app.inform.emit(_("Copper Thieving Tool. Preparing isolation polygons."))
  440. # variables to display the percentage of work done
  441. geo_len = 0
  442. try:
  443. for pol in self.grb_object.solid_geometry:
  444. geo_len += 1
  445. except TypeError:
  446. geo_len = 1
  447. old_disp_number = 0
  448. pol_nr = 0
  449. clearance_geometry = []
  450. try:
  451. for pol in self.grb_object.solid_geometry:
  452. if self.app.abort_flag:
  453. # graceful abort requested by the user
  454. raise FlatCAMApp.GracefulException
  455. clearance_geometry.append(
  456. pol.buffer(c_val, int(int(self.geo_steps_per_circle) / 4))
  457. )
  458. pol_nr += 1
  459. disp_number = int(np.interp(pol_nr, [0, geo_len], [0, 100]))
  460. if old_disp_number < disp_number <= 100:
  461. self.app.proc_container.update_view_text(' %s ... %d%%' %
  462. (_("Buffering"), int(disp_number)))
  463. old_disp_number = disp_number
  464. except TypeError:
  465. # taking care of the case when the self.solid_geometry is just a single Polygon, not a list or a
  466. # MultiPolygon (not an iterable)
  467. clearance_geometry.append(
  468. self.grb_object.solid_geometry.buffer(c_val, int(int(self.geo_steps_per_circle) / 4))
  469. )
  470. self.app.proc_container.update_view_text(' %s' % _("Buffering"))
  471. clearance_geometry = unary_union(clearance_geometry)
  472. # #########################################################################################
  473. # Prepare the area to fill with copper. ###################################################
  474. # #########################################################################################
  475. log.debug("Copper Thieving Tool. Preparing areas to fill with copper.")
  476. self.app.inform.emit(_("Copper Thieving Tool. Preparing areas to fill with copper."))
  477. try:
  478. if ref_obj is None or ref_obj == 'itself':
  479. working_obj = fill_obj
  480. else:
  481. working_obj = ref_obj
  482. except Exception as e:
  483. log.debug("ToolCopperThieving.on_copper_fill() --> %s" % str(e))
  484. return 'fail'
  485. bounding_box = None
  486. if ref_selected == 'itself':
  487. geo_n = working_obj.solid_geometry
  488. try:
  489. if self.bbox_type_radio.get_value() == 'min':
  490. if isinstance(geo_n, MultiPolygon):
  491. env_obj = geo_n.convex_hull
  492. elif (isinstance(geo_n, MultiPolygon) and len(geo_n) == 1) or \
  493. (isinstance(geo_n, list) and len(geo_n) == 1) and isinstance(geo_n[0], Polygon):
  494. env_obj = cascaded_union(geo_n)
  495. else:
  496. env_obj = cascaded_union(geo_n)
  497. env_obj = env_obj.convex_hull
  498. else:
  499. if isinstance(geo_n, Polygon) or \
  500. (isinstance(geo_n, list) and len(geo_n) == 1) or \
  501. (isinstance(geo_n, MultiPolygon) and len(geo_n) == 1):
  502. env_obj = geo_n.buffer(0, join_style=base.JOIN_STYLE.mitre).exterior
  503. elif isinstance(geo_n, MultiPolygon):
  504. x0, y0, x1, y1 = geo_n.bounds
  505. geo = box(x0, y0, x1, y1)
  506. env_obj = geo.buffer(0, join_style=base.JOIN_STYLE.mitre)
  507. else:
  508. self.app.inform.emit(
  509. '[ERROR_NOTCL] %s: %s' % (_("Geometry not supported for bounding box"), type(geo_n))
  510. )
  511. return 'fail'
  512. bounding_box = env_obj.buffer(distance=margin, join_style=base.JOIN_STYLE.mitre)
  513. except Exception as e:
  514. log.debug("ToolCopperFIll.on_copper_fill() 'itself' --> %s" % str(e))
  515. self.app.inform.emit('[ERROR_NOTCL] %s' % _("No object available."))
  516. return 'fail'
  517. elif ref_selected == 'area':
  518. geo_n = cascaded_union(working_obj)
  519. try:
  520. __ = iter(geo_n)
  521. except Exception as e:
  522. log.debug("ToolCopperFIll.on_copper_fill() 'area' --> %s" % str(e))
  523. geo_n = [geo_n]
  524. geo_buff_list = []
  525. for poly in geo_n:
  526. if self.app.abort_flag:
  527. # graceful abort requested by the user
  528. raise FlatCAMApp.GracefulException
  529. geo_buff_list.append(poly.buffer(distance=margin, join_style=base.JOIN_STYLE.mitre))
  530. bounding_box = cascaded_union(geo_buff_list)
  531. elif ref_selected == 'box':
  532. geo_n = working_obj.solid_geometry
  533. if isinstance(working_obj, FlatCAMGeometry):
  534. try:
  535. __ = iter(geo_n)
  536. except Exception as e:
  537. log.debug("ToolCopperFIll.on_copper_fill() 'box' --> %s" % str(e))
  538. geo_n = [geo_n]
  539. geo_buff_list = []
  540. for poly in geo_n:
  541. if self.app.abort_flag:
  542. # graceful abort requested by the user
  543. raise FlatCAMApp.GracefulException
  544. geo_buff_list.append(poly.buffer(distance=margin, join_style=base.JOIN_STYLE.mitre))
  545. bounding_box = cascaded_union(geo_buff_list)
  546. elif isinstance(working_obj, FlatCAMGerber):
  547. geo_n = cascaded_union(geo_n).convex_hull
  548. bounding_box = cascaded_union(self.ncc_obj.solid_geometry).convex_hull.intersection(geo_n)
  549. bounding_box = bounding_box.buffer(distance=margin, join_style=base.JOIN_STYLE.mitre)
  550. else:
  551. self.app.inform.emit('[ERROR_NOTCL] %s' % _("The reference object type is not supported."))
  552. return 'fail'
  553. log.debug("Copper Thieving Tool. Finished creating areas to fill with copper.")
  554. self.app.inform.emit(_("Copper Thieving Tool. Appending new geometry and buffering."))
  555. new_solid_geometry = bounding_box.difference(clearance_geometry)
  556. geo_list = self.grb_object.solid_geometry
  557. if isinstance(self.grb_object.solid_geometry, MultiPolygon):
  558. geo_list = list(self.grb_object.solid_geometry.geoms)
  559. if '0' not in self.grb_object.apertures:
  560. self.grb_object.apertures['0'] = dict()
  561. self.grb_object.apertures['0']['geometry'] = list()
  562. self.grb_object.apertures['0']['type'] = 'REG'
  563. self.grb_object.apertures['0']['size'] = 0.0
  564. try:
  565. for poly in new_solid_geometry:
  566. # append to the new solid geometry
  567. geo_list.append(poly)
  568. # append into the '0' aperture
  569. geo_elem = dict()
  570. geo_elem['solid'] = poly
  571. geo_elem['follow'] = poly.exterior
  572. self.grb_object.apertures['0']['geometry'].append(deepcopy(geo_elem))
  573. except TypeError:
  574. # append to the new solid geometry
  575. geo_list.append(new_solid_geometry)
  576. # append into the '0' aperture
  577. geo_elem = dict()
  578. geo_elem['solid'] = new_solid_geometry
  579. geo_elem['follow'] = new_solid_geometry.exterior
  580. self.grb_object.apertures['0']['geometry'].append(deepcopy(geo_elem))
  581. self.grb_object.solid_geometry = MultiPolygon(geo_list).buffer(0.0000001).buffer(-0.0000001)
  582. # update the source file with the new geometry:
  583. self.grb_object.source_file = self.app.export_gerber(obj_name=self.grb_object.options['name'], filename=None,
  584. local_use=self.grb_object, use_thread=False)
  585. self.on_exit()
  586. self.app.inform.emit('[success] %s' % _("Copper Thieving Tool done."))
  587. def replot(self, obj):
  588. def worker_task():
  589. with self.app.proc_container.new('%s...' % _("Plotting")):
  590. obj.plot()
  591. self.app.worker_task.emit({'fcn': worker_task, 'params': []})
  592. def on_exit(self):
  593. # plot the object
  594. self.replot(obj=self.grb_object)
  595. # update the bounding box values
  596. try:
  597. a, b, c, d = self.grb_object.bounds()
  598. self.grb_object.options['xmin'] = a
  599. self.grb_object.options['ymin'] = b
  600. self.grb_object.options['xmax'] = c
  601. self.grb_object.options['ymax'] = d
  602. except Exception as e:
  603. log.debug("ToolCopperThieving.on_exit() bounds error --> %s" % str(e))
  604. # reset the variables
  605. self.grb_object = None
  606. self.ref_obj = None
  607. self.sel_rect = list()
  608. # Events ID
  609. self.mr = None
  610. self.mm = None
  611. # Mouse cursor positions
  612. self.mouse_is_dragging = False
  613. self.cursor_pos = (0, 0)
  614. self.first_click = False
  615. # if True it means we exited from tool in the middle of area adding therefore disconnect the events
  616. if self.area_method is True:
  617. self.app.delete_selection_shape()
  618. self.area_method = False
  619. if self.app.is_legacy is False:
  620. self.app.plotcanvas.graph_event_disconnect('mouse_release', self.on_mouse_release)
  621. self.app.plotcanvas.graph_event_disconnect('mouse_move', self.on_mouse_move)
  622. else:
  623. self.app.plotcanvas.graph_event_disconnect(self.mr)
  624. self.app.plotcanvas.graph_event_disconnect(self.mm)
  625. self.app.mp = self.app.plotcanvas.graph_event_connect('mouse_press',
  626. self.app.on_mouse_click_over_plot)
  627. self.app.mm = self.app.plotcanvas.graph_event_connect('mouse_move',
  628. self.app.on_mouse_move_over_plot)
  629. self.app.mr = self.app.plotcanvas.graph_event_connect('mouse_release',
  630. self.app.on_mouse_click_release_over_plot)
  631. self.app.call_source = "app"
  632. self.app.inform.emit('[success] %s' % _("Copper Thieving Tool exit."))