ToolSub.py 29 KB

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