ToolSub.py 30 KB

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