ToolSub.py 28 KB

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