ToolSub.py 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752
  1. # ##########################################################
  2. # FlatCAM: 2D Post-processing for Manufacturing #
  3. # File Author: Marius Adrian Stanciu (c) #
  4. # Date: 4/24/2019 #
  5. # MIT Licence #
  6. # ##########################################################
  7. from PyQt5 import QtWidgets, QtCore
  8. from FlatCAMTool import FlatCAMTool
  9. from flatcamGUI.GUIElements import FCCheckBox, FCButton
  10. from shapely.geometry import Polygon, MultiPolygon, MultiLineString, LineString
  11. from shapely.ops import cascaded_union
  12. import traceback
  13. from copy import deepcopy
  14. import time
  15. import logging
  16. import gettext
  17. import FlatCAMTranslation as fcTranslate
  18. import builtins
  19. fcTranslate.apply_language('strings')
  20. if '_' not in builtins.__dict__:
  21. _ = gettext.gettext
  22. log = logging.getLogger('base')
  23. class ToolSub(FlatCAMTool):
  24. job_finished = QtCore.pyqtSignal(bool)
  25. toolName = _("Subtract Tool")
  26. def __init__(self, app):
  27. self.app = app
  28. self.decimals = self.app.decimals
  29. FlatCAMTool.__init__(self, app)
  30. self.tools_frame = QtWidgets.QFrame()
  31. self.tools_frame.setContentsMargins(0, 0, 0, 0)
  32. self.layout.addWidget(self.tools_frame)
  33. self.tools_box = QtWidgets.QVBoxLayout()
  34. self.tools_box.setContentsMargins(0, 0, 0, 0)
  35. self.tools_frame.setLayout(self.tools_box)
  36. # Title
  37. title_label = QtWidgets.QLabel("%s" % self.toolName)
  38. title_label.setStyleSheet("""
  39. QLabel
  40. {
  41. font-size: 16px;
  42. font-weight: bold;
  43. }
  44. """)
  45. self.tools_box.addWidget(title_label)
  46. # Form Layout
  47. form_layout = QtWidgets.QFormLayout()
  48. self.tools_box.addLayout(form_layout)
  49. self.gerber_title = QtWidgets.QLabel("<b>%s</b>" % _("Gerber Objects"))
  50. form_layout.addRow(self.gerber_title)
  51. # Target Gerber Object
  52. self.target_gerber_combo = QtWidgets.QComboBox()
  53. self.target_gerber_combo.setModel(self.app.collection)
  54. self.target_gerber_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  55. self.target_gerber_combo.setCurrentIndex(1)
  56. self.target_gerber_label = QtWidgets.QLabel('%s:' % _("Target"))
  57. self.target_gerber_label.setToolTip(
  58. _("Gerber object from which to subtract\n"
  59. "the subtractor Gerber object.")
  60. )
  61. form_layout.addRow(self.target_gerber_label, self.target_gerber_combo)
  62. # Substractor Gerber Object
  63. self.sub_gerber_combo = QtWidgets.QComboBox()
  64. self.sub_gerber_combo.setModel(self.app.collection)
  65. self.sub_gerber_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  66. self.sub_gerber_combo.setCurrentIndex(1)
  67. self.sub_gerber_label = QtWidgets.QLabel('%s:' % _("Subtractor"))
  68. self.sub_gerber_label.setToolTip(
  69. _("Gerber object that will be subtracted\n"
  70. "from the target Gerber object.")
  71. )
  72. e_lab_1 = QtWidgets.QLabel('')
  73. form_layout.addRow(self.sub_gerber_label, self.sub_gerber_combo)
  74. self.intersect_btn = FCButton(_('Substract Gerber'))
  75. self.intersect_btn.setToolTip(
  76. _("Will remove the area occupied by the subtractor\n"
  77. "Gerber from the Target Gerber.\n"
  78. "Can be used to remove the overlapping silkscreen\n"
  79. "over the soldermask.")
  80. )
  81. self.intersect_btn.setStyleSheet("""
  82. QPushButton
  83. {
  84. font-weight: bold;
  85. }
  86. """)
  87. self.tools_box.addWidget(self.intersect_btn)
  88. self.tools_box.addWidget(e_lab_1)
  89. # Form Layout
  90. form_geo_layout = QtWidgets.QFormLayout()
  91. self.tools_box.addLayout(form_geo_layout)
  92. self.geo_title = QtWidgets.QLabel("<b>%s</b>" % _("Geometry Objects"))
  93. form_geo_layout.addRow(self.geo_title)
  94. # Target Geometry Object
  95. self.target_geo_combo = QtWidgets.QComboBox()
  96. self.target_geo_combo.setModel(self.app.collection)
  97. self.target_geo_combo.setRootModelIndex(self.app.collection.index(2, 0, QtCore.QModelIndex()))
  98. self.target_geo_combo.setCurrentIndex(1)
  99. self.target_geo_label = QtWidgets.QLabel('%s:' % _("Target"))
  100. self.target_geo_label.setToolTip(
  101. _("Geometry object from which to subtract\n"
  102. "the subtractor Geometry object.")
  103. )
  104. form_geo_layout.addRow(self.target_geo_label, self.target_geo_combo)
  105. # Substractor Geometry Object
  106. self.sub_geo_combo = QtWidgets.QComboBox()
  107. self.sub_geo_combo.setModel(self.app.collection)
  108. self.sub_geo_combo.setRootModelIndex(self.app.collection.index(2, 0, QtCore.QModelIndex()))
  109. self.sub_geo_combo.setCurrentIndex(1)
  110. self.sub_geo_label = QtWidgets.QLabel('%s:' % _("Subtractor"))
  111. self.sub_geo_label.setToolTip(
  112. _("Geometry object that will be subtracted\n"
  113. "from the target Geometry object.")
  114. )
  115. e_lab_1 = QtWidgets.QLabel('')
  116. form_geo_layout.addRow(self.sub_geo_label, self.sub_geo_combo)
  117. self.close_paths_cb = FCCheckBox(_("Close paths"))
  118. self.close_paths_cb.setToolTip(_("Checking this will close the paths cut by the Geometry subtractor object."))
  119. self.tools_box.addWidget(self.close_paths_cb)
  120. self.intersect_geo_btn = FCButton(_('Subtract Geometry'))
  121. self.intersect_geo_btn.setToolTip(
  122. _("Will remove the area occupied by the subtractor\n"
  123. "Geometry from the Target Geometry.")
  124. )
  125. self.intersect_geo_btn.setStyleSheet("""
  126. QPushButton
  127. {
  128. font-weight: bold;
  129. }
  130. """)
  131. self.tools_box.addWidget(self.intersect_geo_btn)
  132. self.tools_box.addWidget(e_lab_1)
  133. self.tools_box.addStretch()
  134. # ## Reset Tool
  135. self.reset_button = QtWidgets.QPushButton(_("Reset Tool"))
  136. self.reset_button.setToolTip(
  137. _("Will reset the tool parameters.")
  138. )
  139. self.reset_button.setStyleSheet("""
  140. QPushButton
  141. {
  142. font-weight: bold;
  143. }
  144. """)
  145. self.tools_box.addWidget(self.reset_button)
  146. # QTimer for periodic check
  147. self.check_thread = QtCore.QTimer()
  148. # Every time an intersection job is started we add a promise; every time an intersection job is finished
  149. # we remove a promise.
  150. # When empty we start the layer rendering
  151. self.promises = []
  152. self.new_apertures = {}
  153. self.new_tools = {}
  154. self.new_solid_geometry = []
  155. self.sub_solid_union = None
  156. self.sub_follow_union = None
  157. self.sub_clear_union = None
  158. self.sub_grb_obj = None
  159. self.sub_grb_obj_name = None
  160. self.target_grb_obj = None
  161. self.target_grb_obj_name = None
  162. self.sub_geo_obj = None
  163. self.sub_geo_obj_name = None
  164. self.target_geo_obj = None
  165. self.target_geo_obj_name = None
  166. # signal which type of substraction to do: "geo" or "gerber"
  167. self.sub_type = None
  168. # store here the options from target_obj
  169. self.target_options = {}
  170. self.sub_union = []
  171. try:
  172. self.intersect_btn.clicked.disconnect(self.on_grb_intersection_click)
  173. except (TypeError, AttributeError):
  174. pass
  175. self.intersect_btn.clicked.connect(self.on_grb_intersection_click)
  176. try:
  177. self.intersect_geo_btn.clicked.disconnect()
  178. except (TypeError, AttributeError):
  179. pass
  180. self.intersect_geo_btn.clicked.connect(self.on_geo_intersection_click)
  181. self.job_finished.connect(self.on_job_finished)
  182. self.reset_button.clicked.connect(self.set_tool_ui)
  183. def install(self, icon=None, separator=None, **kwargs):
  184. FlatCAMTool.install(self, icon, separator, shortcut='ALT+W', **kwargs)
  185. def run(self, toggle=True):
  186. self.app.report_usage("ToolSub()")
  187. if toggle:
  188. # if the splitter is hidden, display it, else hide it but only if the current widget is the same
  189. if self.app.ui.splitter.sizes()[0] == 0:
  190. self.app.ui.splitter.setSizes([1, 1])
  191. else:
  192. try:
  193. if self.app.ui.tool_scroll_area.widget().objectName() == self.toolName:
  194. # if tab is populated with the tool but it does not have the focus, focus on it
  195. if not self.app.ui.notebook.currentWidget() is self.app.ui.tool_tab:
  196. # focus on Tool Tab
  197. self.app.ui.notebook.setCurrentWidget(self.app.ui.tool_tab)
  198. else:
  199. self.app.ui.splitter.setSizes([0, 1])
  200. except AttributeError:
  201. pass
  202. else:
  203. if self.app.ui.splitter.sizes()[0] == 0:
  204. self.app.ui.splitter.setSizes([1, 1])
  205. FlatCAMTool.run(self)
  206. self.set_tool_ui()
  207. self.new_apertures.clear()
  208. self.new_tools.clear()
  209. self.new_solid_geometry = []
  210. self.target_options.clear()
  211. self.app.ui.notebook.setTabText(2, _("Sub Tool"))
  212. def set_tool_ui(self):
  213. self.tools_frame.show()
  214. self.close_paths_cb.setChecked(self.app.defaults["tools_sub_close_paths"])
  215. def on_grb_intersection_click(self):
  216. # reset previous values
  217. self.new_apertures.clear()
  218. self.new_solid_geometry = []
  219. self.sub_union = []
  220. self.sub_type = "gerber"
  221. self.target_grb_obj_name = self.target_gerber_combo.currentText()
  222. if self.target_grb_obj_name == '':
  223. self.app.inform.emit('[ERROR_NOTCL] %s' % _("No Target object loaded."))
  224. return
  225. self.app.inform.emit('%s' % _("Loading geometry from Gerber objects."))
  226. # Get target object.
  227. try:
  228. self.target_grb_obj = self.app.collection.get_by_name(self.target_grb_obj_name)
  229. except Exception as e:
  230. log.debug("ToolSub.on_grb_intersection_click() --> %s" % str(e))
  231. self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve object"), self.obj_name))
  232. return "Could not retrieve object: %s" % self.target_grb_obj_name
  233. self.sub_grb_obj_name = self.sub_gerber_combo.currentText()
  234. if self.sub_grb_obj_name == '':
  235. self.app.inform.emit('[ERROR_NOTCL] %s' % _("No Subtractor object loaded."))
  236. return
  237. # Get substractor object.
  238. try:
  239. self.sub_grb_obj = self.app.collection.get_by_name(self.sub_grb_obj_name)
  240. except Exception as e:
  241. log.debug("ToolSub.on_grb_intersection_click() --> %s" % str(e))
  242. self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve object"), self.obj_name))
  243. return "Could not retrieve object: %s" % self.sub_grb_obj_name
  244. # crate the new_apertures dict structure
  245. for apid in self.target_grb_obj.apertures:
  246. self.new_apertures[apid] = dict()
  247. self.new_apertures[apid]['type'] = 'C'
  248. self.new_apertures[apid]['size'] = self.target_grb_obj.apertures[apid]['size']
  249. self.new_apertures[apid]['geometry'] = list()
  250. geo_solid_union_list = list()
  251. geo_follow_union_list = list()
  252. geo_clear_union_list = list()
  253. for apid1 in self.sub_grb_obj.apertures:
  254. if 'geometry' in self.sub_grb_obj.apertures[apid1]:
  255. for elem in self.sub_grb_obj.apertures[apid1]['geometry']:
  256. if 'solid' in elem:
  257. geo_solid_union_list.append(elem['solid'])
  258. if 'follow' in elem:
  259. geo_follow_union_list.append(elem['follow'])
  260. if 'clear' in elem:
  261. geo_clear_union_list.append(elem['clear'])
  262. self.app.inform.emit('%s' % _("Processing geometry from Subtractor Gerber object."))
  263. self.sub_solid_union = cascaded_union(geo_solid_union_list)
  264. self.sub_follow_union = cascaded_union(geo_follow_union_list)
  265. self.sub_clear_union = cascaded_union(geo_clear_union_list)
  266. # add the promises
  267. for apid in self.target_grb_obj.apertures:
  268. self.promises.append(apid)
  269. # start the QTimer to check for promises with 0.5 second period check
  270. self.periodic_check(500, reset=True)
  271. for apid in self.target_grb_obj.apertures:
  272. geo = self.target_grb_obj.apertures[apid]['geometry']
  273. self.app.worker_task.emit({'fcn': self.aperture_intersection, 'params': [apid, geo]})
  274. def aperture_intersection(self, apid, geo):
  275. new_geometry = list()
  276. log.debug("Working on promise: %s" % str(apid))
  277. with self.app.proc_container.new('%s: %s...' % (_("Parsing geometry for aperture"), str(apid))):
  278. for geo_el in geo:
  279. new_el = dict()
  280. if 'solid' in geo_el:
  281. work_geo = geo_el['solid']
  282. if self.sub_solid_union:
  283. if work_geo.intersects(self.sub_solid_union):
  284. new_geo = work_geo.difference(self.sub_solid_union)
  285. new_geo = new_geo.buffer(0)
  286. if new_geo:
  287. if not new_geo.is_empty:
  288. new_el['solid'] = new_geo
  289. else:
  290. new_el['solid'] = work_geo
  291. else:
  292. new_el['solid'] = work_geo
  293. else:
  294. new_el['solid'] = work_geo
  295. else:
  296. new_el['solid'] = work_geo
  297. if 'follow' in geo_el:
  298. work_geo = geo_el['follow']
  299. if self.sub_follow_union:
  300. if work_geo.intersects(self.sub_follow_union):
  301. new_geo = work_geo.difference(self.sub_follow_union)
  302. new_geo = new_geo.buffer(0)
  303. if new_geo:
  304. if not new_geo.is_empty:
  305. new_el['follow'] = new_geo
  306. else:
  307. new_el['follow'] = work_geo
  308. else:
  309. new_el['follow'] = work_geo
  310. else:
  311. new_el['follow'] = work_geo
  312. else:
  313. new_el['follow'] = work_geo
  314. if 'clear' in geo_el:
  315. work_geo = geo_el['clear']
  316. if self.sub_clear_union:
  317. if work_geo.intersects(self.sub_clear_union):
  318. new_geo = work_geo.difference(self.sub_clear_union)
  319. new_geo = new_geo.buffer(0)
  320. if new_geo:
  321. if not new_geo.is_empty:
  322. new_el['clear'] = new_geo
  323. else:
  324. new_el['clear'] = work_geo
  325. else:
  326. new_el['clear'] = work_geo
  327. else:
  328. new_el['clear'] = work_geo
  329. else:
  330. new_el['clear'] = work_geo
  331. new_geometry.append(deepcopy(new_el))
  332. self.app.inform.emit('%s: %s...' % (_("Finished parsing geometry for aperture"), str(apid)))
  333. if new_geometry:
  334. while not self.new_apertures[apid]['geometry']:
  335. self.new_apertures[apid]['geometry'] = deepcopy(new_geometry)
  336. time.sleep(0.5)
  337. while True:
  338. # removal from list is done in a multithreaded way therefore not always the removal can be done
  339. # so we keep trying until it's done
  340. if apid not in self.promises:
  341. break
  342. self.promises.remove(apid)
  343. time.sleep(0.5)
  344. log.debug("Promise fulfilled: %s" % str(apid))
  345. def new_gerber_object(self, outname):
  346. def obj_init(grb_obj, app_obj):
  347. grb_obj.apertures = deepcopy(self.new_apertures)
  348. poly_buff = []
  349. follow_buff = []
  350. for ap in self.new_apertures:
  351. for elem in self.new_apertures[ap]['geometry']:
  352. poly_buff.append(elem['solid'])
  353. follow_buff.append(elem['follow'])
  354. work_poly_buff = cascaded_union(poly_buff)
  355. try:
  356. poly_buff = work_poly_buff.buffer(0.0000001)
  357. except ValueError:
  358. pass
  359. try:
  360. poly_buff = poly_buff.buffer(-0.0000001)
  361. except ValueError:
  362. pass
  363. grb_obj.solid_geometry = deepcopy(poly_buff)
  364. grb_obj.follow_geometry = deepcopy(follow_buff)
  365. with self.app.proc_container.new(_("Generating new object ...")):
  366. ret = self.app.new_object('gerber', outname, obj_init, autoselected=False)
  367. if ret == 'fail':
  368. self.app.inform.emit('[ERROR_NOTCL] %s' %
  369. _('Generating new object failed.'))
  370. return
  371. # GUI feedback
  372. self.app.inform.emit('[success] %s: %s' %
  373. (_("Created"), outname))
  374. # cleanup
  375. self.new_apertures.clear()
  376. self.new_solid_geometry[:] = []
  377. try:
  378. self.sub_union[:] = []
  379. except TypeError:
  380. self.sub_union = []
  381. def on_geo_intersection_click(self):
  382. # reset previous values
  383. self.new_tools.clear()
  384. self.target_options.clear()
  385. self.new_solid_geometry = []
  386. self.sub_union = []
  387. self.sub_type = "geo"
  388. self.target_geo_obj_name = self.target_geo_combo.currentText()
  389. if self.target_geo_obj_name == '':
  390. self.app.inform.emit('[ERROR_NOTCL] %s' %
  391. _("No Target object loaded."))
  392. return
  393. # Get target object.
  394. try:
  395. self.target_geo_obj = self.app.collection.get_by_name(self.target_geo_obj_name)
  396. except Exception as e:
  397. log.debug("ToolSub.on_geo_intersection_click() --> %s" % str(e))
  398. self.app.inform.emit('[ERROR_NOTCL] %s: %s' %
  399. (_("Could not retrieve object"), self.target_geo_obj_name))
  400. return "Could not retrieve object: %s" % self.target_grb_obj_name
  401. self.sub_geo_obj_name = self.sub_geo_combo.currentText()
  402. if self.sub_geo_obj_name == '':
  403. self.app.inform.emit('[ERROR_NOTCL] %s' %
  404. _("No Subtractor object loaded."))
  405. return
  406. # Get substractor object.
  407. try:
  408. self.sub_geo_obj = self.app.collection.get_by_name(self.sub_geo_obj_name)
  409. except Exception as e:
  410. log.debug("ToolSub.on_geo_intersection_click() --> %s" % str(e))
  411. self.app.inform.emit('[ERROR_NOTCL] %s: %s' %
  412. (_("Could not retrieve object"), self.sub_geo_obj_name))
  413. return "Could not retrieve object: %s" % self.sub_geo_obj_name
  414. if self.sub_geo_obj.multigeo:
  415. self.app.inform.emit('[ERROR_NOTCL] %s' %
  416. _("Currently, the Subtractor geometry cannot be of type Multigeo."))
  417. return
  418. # create the target_options obj
  419. # self.target_options = dict()
  420. # for k, v in self.target_geo_obj.options.items():
  421. # if k != 'name':
  422. # self.target_options[k] = v
  423. # crate the new_tools dict structure
  424. for tool in self.target_geo_obj.tools:
  425. self.new_tools[tool] = dict()
  426. for key in self.target_geo_obj.tools[tool]:
  427. if key == 'solid_geometry':
  428. self.new_tools[tool][key] = []
  429. else:
  430. self.new_tools[tool][key] = deepcopy(self.target_geo_obj.tools[tool][key])
  431. # add the promises
  432. if self.target_geo_obj.multigeo:
  433. for tool in self.target_geo_obj.tools:
  434. self.promises.append(tool)
  435. else:
  436. self.promises.append("single")
  437. self.sub_union = cascaded_union(self.sub_geo_obj.solid_geometry)
  438. # start the QTimer to check for promises with 0.5 second period check
  439. self.periodic_check(500, reset=True)
  440. if self.target_geo_obj.multigeo:
  441. for tool in self.target_geo_obj.tools:
  442. geo = self.target_geo_obj.tools[tool]['solid_geometry']
  443. self.app.worker_task.emit({'fcn': self.toolgeo_intersection,
  444. 'params': [tool, geo]})
  445. else:
  446. geo = self.target_geo_obj.solid_geometry
  447. self.app.worker_task.emit({'fcn': self.toolgeo_intersection,
  448. 'params': ["single", geo]})
  449. def toolgeo_intersection(self, tool, geo):
  450. new_geometry = []
  451. log.debug("Working on promise: %s" % str(tool))
  452. if tool == "single":
  453. text = _("Parsing solid_geometry ...")
  454. else:
  455. text = '%s: %s...' % (_("Parsing solid_geometry for tool"), str(tool))
  456. with self.app.proc_container.new(text):
  457. # resulting paths are closed resulting into Polygons
  458. if self.close_paths_cb.isChecked():
  459. new_geo = (cascaded_union(geo)).difference(self.sub_union)
  460. if new_geo:
  461. if not new_geo.is_empty:
  462. new_geometry.append(new_geo)
  463. # resulting paths are unclosed resulting in a multitude of rings
  464. else:
  465. try:
  466. for geo_elem in geo:
  467. if isinstance(geo_elem, Polygon):
  468. for ring in self.poly2rings(geo_elem):
  469. new_geo = ring.difference(self.sub_union)
  470. if new_geo and not new_geo.is_empty:
  471. new_geometry.append(new_geo)
  472. elif isinstance(geo_elem, MultiPolygon):
  473. for poly in geo_elem:
  474. for ring in self.poly2rings(poly):
  475. new_geo = ring.difference(self.sub_union)
  476. if new_geo and not new_geo.is_empty:
  477. new_geometry.append(new_geo)
  478. elif isinstance(geo_elem, LineString):
  479. new_geo = geo_elem.difference(self.sub_union)
  480. if new_geo:
  481. if not new_geo.is_empty:
  482. new_geometry.append(new_geo)
  483. elif isinstance(geo_elem, MultiLineString):
  484. for line_elem in geo_elem:
  485. new_geo = line_elem.difference(self.sub_union)
  486. if new_geo and not new_geo.is_empty:
  487. new_geometry.append(new_geo)
  488. except TypeError:
  489. if isinstance(geo, Polygon):
  490. for ring in self.poly2rings(geo):
  491. new_geo = ring.difference(self.sub_union)
  492. if new_geo:
  493. if not new_geo.is_empty:
  494. new_geometry.append(new_geo)
  495. elif isinstance(geo, LineString):
  496. new_geo = geo.difference(self.sub_union)
  497. if new_geo and not new_geo.is_empty:
  498. new_geometry.append(new_geo)
  499. elif isinstance(geo, MultiLineString):
  500. for line_elem in geo:
  501. new_geo = line_elem.difference(self.sub_union)
  502. if new_geo and not new_geo.is_empty:
  503. new_geometry.append(new_geo)
  504. if new_geometry:
  505. if tool == "single":
  506. while not self.new_solid_geometry:
  507. self.new_solid_geometry = deepcopy(new_geometry)
  508. time.sleep(0.5)
  509. else:
  510. while not self.new_tools[tool]['solid_geometry']:
  511. self.new_tools[tool]['solid_geometry'] = deepcopy(new_geometry)
  512. time.sleep(0.5)
  513. while True:
  514. # removal from list is done in a multithreaded way therefore not always the removal can be done
  515. # so we keep trying until it's done
  516. if tool not in self.promises:
  517. break
  518. self.promises.remove(tool)
  519. time.sleep(0.5)
  520. log.debug("Promise fulfilled: %s" % str(tool))
  521. def new_geo_object(self, outname):
  522. geo_name = outname
  523. def obj_init(geo_obj, app_obj):
  524. # geo_obj.options = self.target_options
  525. # create the target_options obj
  526. for k, v in self.target_geo_obj.options.items():
  527. geo_obj.options[k] = v
  528. geo_obj.options['name'] = geo_name
  529. if self.target_geo_obj.multigeo:
  530. geo_obj.tools = deepcopy(self.new_tools)
  531. # this turn on the FlatCAMCNCJob plot for multiple tools
  532. geo_obj.multigeo = True
  533. geo_obj.multitool = True
  534. else:
  535. geo_obj.solid_geometry = deepcopy(self.new_solid_geometry)
  536. try:
  537. geo_obj.tools = deepcopy(self.new_tools)
  538. for tool in geo_obj.tools:
  539. geo_obj.tools[tool]['solid_geometry'] = deepcopy(self.new_solid_geometry)
  540. except Exception as e:
  541. log.debug("ToolSub.new_geo_object() --> %s" % str(e))
  542. geo_obj.multigeo = False
  543. with self.app.proc_container.new(_("Generating new object ...")):
  544. ret = self.app.new_object('geometry', outname, obj_init, autoselected=False)
  545. if ret == 'fail':
  546. self.app.inform.emit('[ERROR_NOTCL] %s' %
  547. _('Generating new object failed.'))
  548. return
  549. # Register recent file
  550. self.app.file_opened.emit('geometry', outname)
  551. # GUI feedback
  552. self.app.inform.emit('[success] %s: %s' %
  553. (_("Created"), outname))
  554. # cleanup
  555. self.new_tools.clear()
  556. self.new_solid_geometry[:] = []
  557. self.sub_union = []
  558. def periodic_check(self, check_period, reset=False):
  559. """
  560. This function starts an QTimer and it will periodically check if intersections are done
  561. :param check_period: time at which to check periodically
  562. :param reset: will reset the timer
  563. :return:
  564. """
  565. log.debug("ToolSub --> Periodic Check started.")
  566. try:
  567. self.check_thread.stop()
  568. except (TypeError, AttributeError):
  569. pass
  570. if reset:
  571. self.check_thread.setInterval(check_period)
  572. try:
  573. self.check_thread.timeout.disconnect(self.periodic_check_handler)
  574. except (TypeError, AttributeError):
  575. pass
  576. self.check_thread.timeout.connect(self.periodic_check_handler)
  577. self.check_thread.start(QtCore.QThread.HighPriority)
  578. def periodic_check_handler(self):
  579. """
  580. If the intersections workers finished then start creating the solid_geometry
  581. :return:
  582. """
  583. # log.debug("checking parsing --> %s" % str(self.parsing_promises))
  584. try:
  585. if not self.promises:
  586. self.check_thread.stop()
  587. self.job_finished.emit(True)
  588. # reset the type of substraction for next time
  589. self.sub_type = None
  590. log.debug("ToolSub --> Periodic check finished.")
  591. except Exception as e:
  592. self.job_finished.emit(False)
  593. log.debug("ToolSub().periodic_check_handler() --> %s" % str(e))
  594. traceback.print_exc()
  595. def on_job_finished(self, succcess):
  596. """
  597. :param succcess: boolean, this parameter signal if all the apertures were processed
  598. :return: None
  599. """
  600. if succcess is True:
  601. if self.sub_type == "gerber":
  602. outname = self.target_gerber_combo.currentText() + '_sub'
  603. # intersection jobs finished, start the creation of solid_geometry
  604. self.app.worker_task.emit({'fcn': self.new_gerber_object,
  605. 'params': [outname]})
  606. else:
  607. outname = self.target_geo_combo.currentText() + '_sub'
  608. # intersection jobs finished, start the creation of solid_geometry
  609. self.app.worker_task.emit({'fcn': self.new_geo_object,
  610. 'params': [outname]})
  611. else:
  612. self.app.inform.emit('[ERROR_NOTCL] %s' %
  613. _('Generating new object failed.'))
  614. def reset_fields(self):
  615. self.target_gerber_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  616. self.sub_gerber_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  617. self.target_geo_combo.setRootModelIndex(self.app.collection.index(2, 0, QtCore.QModelIndex()))
  618. self.sub_geo_combo.setRootModelIndex(self.app.collection.index(2, 0, QtCore.QModelIndex()))
  619. @staticmethod
  620. def poly2rings(poly):
  621. return [poly.exterior] + [interior for interior in poly.interiors]
  622. # end of file