ToolSub.py 34 KB

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