ToolSub.py 32 KB

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