ToolCutOut.py 68 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600
  1. # ##########################################################
  2. # FlatCAM: 2D Post-processing for Manufacturing #
  3. # File Author: Marius Adrian Stanciu (c) #
  4. # Date: 3/10/2019 #
  5. # MIT Licence #
  6. # ##########################################################
  7. from PyQt5 import QtWidgets, QtGui, QtCore
  8. from appTool import AppTool
  9. from appGUI.GUIElements import FCDoubleSpinner, FCCheckBox, RadioSet, FCComboBox, OptionalInputSection, FCButton
  10. from shapely.geometry import box, MultiPolygon, Polygon, LineString, LinearRing, MultiLineString
  11. from shapely.ops import cascaded_union, unary_union, linemerge
  12. import shapely.affinity as affinity
  13. from matplotlib.backend_bases import KeyEvent as mpl_key_event
  14. from numpy import Inf
  15. from copy import deepcopy
  16. import math
  17. import logging
  18. import gettext
  19. import appTranslation as fcTranslate
  20. import builtins
  21. fcTranslate.apply_language('strings')
  22. if '_' not in builtins.__dict__:
  23. _ = gettext.gettext
  24. log = logging.getLogger('base')
  25. settings = QtCore.QSettings("Open Source", "FlatCAM")
  26. if settings.contains("machinist"):
  27. machinist_setting = settings.value('machinist', type=int)
  28. else:
  29. machinist_setting = 0
  30. class CutOut(AppTool):
  31. def __init__(self, app):
  32. AppTool.__init__(self, app)
  33. self.app = app
  34. self.canvas = app.plotcanvas
  35. self.decimals = self.app.decimals
  36. # #############################################################################
  37. # ######################### Tool GUI ##########################################
  38. # #############################################################################
  39. self.ui = CutoutUI(layout=self.layout, app=self.app)
  40. self.toolName = self.ui.toolName
  41. self.cutting_gapsize = 0.0
  42. self.cutting_dia = 0.0
  43. # true if we want to repeat the gap without clicking again on the button
  44. self.repeat_gap = False
  45. self.flat_geometry = []
  46. # this is the Geometry object generated in this class to be used for adding manual gaps
  47. self.man_cutout_obj = None
  48. # if mouse is dragging set the object True
  49. self.mouse_is_dragging = False
  50. # if mouse events are bound to local methods
  51. self.mouse_events_connected = False
  52. # event handlers references
  53. self.kp = None
  54. self.mm = None
  55. self.mr = None
  56. # hold the mouse position here
  57. self.x_pos = None
  58. self.y_pos = None
  59. # store the default data for the resulting Geometry Object
  60. self.default_data = {}
  61. # store the current cursor type to be restored after manual geo
  62. self.old_cursor_type = self.app.defaults["global_cursor_type"]
  63. # store the current selection shape status to be restored after manual geo
  64. self.old_selection_state = self.app.defaults['global_selection_shape']
  65. # Signals
  66. self.ui.ff_cutout_object_btn.clicked.connect(self.on_freeform_cutout)
  67. self.ui.rect_cutout_object_btn.clicked.connect(self.on_rectangular_cutout)
  68. self.ui.type_obj_radio.activated_custom.connect(self.on_type_obj_changed)
  69. self.ui.man_geo_creation_btn.clicked.connect(self.on_manual_geo)
  70. self.ui.man_gaps_creation_btn.clicked.connect(self.on_manual_gap_click)
  71. self.ui.reset_button.clicked.connect(self.set_tool_ui)
  72. def on_type_obj_changed(self, val):
  73. obj_type = {'grb': 0, 'geo': 2}[val]
  74. self.ui.obj_combo.setRootModelIndex(self.app.collection.index(obj_type, 0, QtCore.QModelIndex()))
  75. self.ui.obj_combo.setCurrentIndex(0)
  76. self.ui.obj_combo.obj_type = {"grb": "Gerber", "geo": "Geometry"}[val]
  77. def run(self, toggle=True):
  78. self.app.defaults.report_usage("ToolCutOut()")
  79. if toggle:
  80. # if the splitter is hidden, display it, else hide it but only if the current widget is the same
  81. if self.app.ui.splitter.sizes()[0] == 0:
  82. self.app.ui.splitter.setSizes([1, 1])
  83. else:
  84. try:
  85. if self.app.ui.tool_scroll_area.widget().objectName() == self.toolName:
  86. # if tab is populated with the tool but it does not have the focus, focus on it
  87. if not self.app.ui.notebook.currentWidget() is self.app.ui.tool_tab:
  88. # focus on Tool Tab
  89. self.app.ui.notebook.setCurrentWidget(self.app.ui.tool_tab)
  90. else:
  91. self.app.ui.splitter.setSizes([0, 1])
  92. except AttributeError:
  93. pass
  94. else:
  95. if self.app.ui.splitter.sizes()[0] == 0:
  96. self.app.ui.splitter.setSizes([1, 1])
  97. AppTool.run(self)
  98. self.set_tool_ui()
  99. self.app.ui.notebook.setTabText(2, _("Cutout Tool"))
  100. def install(self, icon=None, separator=None, **kwargs):
  101. AppTool.install(self, icon, separator, shortcut='Alt+X', **kwargs)
  102. def set_tool_ui(self):
  103. self.reset_fields()
  104. self.ui.dia.set_value(float(self.app.defaults["tools_cutouttooldia"]))
  105. self.ui.obj_kind_combo.set_value(self.app.defaults["tools_cutoutkind"])
  106. self.ui.margin.set_value(float(self.app.defaults["tools_cutoutmargin"]))
  107. self.ui.cutz_entry.set_value(float(self.app.defaults["tools_cutout_z"]))
  108. self.ui.mpass_cb.set_value(float(self.app.defaults["tools_cutout_mdepth"]))
  109. self.ui.maxdepth_entry.set_value(float(self.app.defaults["tools_cutout_depthperpass"]))
  110. self.ui.gapsize.set_value(float(self.app.defaults["tools_cutoutgapsize"]))
  111. self.ui.gaps.set_value(self.app.defaults["tools_gaps_ff"])
  112. self.ui.convex_box.set_value(self.app.defaults['tools_cutout_convexshape'])
  113. self.ui.big_cursor_cb.set_value(self.app.defaults['tools_cutout_big_cursor'])
  114. # use the current selected object and make it visible in the Paint object combobox
  115. sel_list = self.app.collection.get_selected()
  116. if len(sel_list) == 1:
  117. active = self.app.collection.get_active()
  118. kind = active.kind
  119. if kind == 'gerber':
  120. self.ui.type_obj_radio.set_value('grb')
  121. else:
  122. self.ui.type_obj_radio.set_value('geo')
  123. # run those once so the obj_type attribute is updated for the FCComboboxes
  124. # so the last loaded object is displayed
  125. if kind == 'gerber':
  126. self.on_type_obj_changed(val='grb')
  127. else:
  128. self.on_type_obj_changed(val='geo')
  129. self.ui.obj_combo.set_value(active.options['name'])
  130. else:
  131. kind = 'gerber'
  132. self.ui.type_obj_radio.set_value('grb')
  133. # run those once so the obj_type attribute is updated for the FCComboboxes
  134. # so the last loaded object is displayed
  135. if kind == 'gerber':
  136. self.on_type_obj_changed(val='grb')
  137. else:
  138. self.on_type_obj_changed(val='geo')
  139. # self.type_obj_radio.set_value('grb')
  140. self.default_data.update({
  141. "plot": True,
  142. "cutz": float(self.app.defaults["geometry_cutz"]),
  143. "multidepth": self.app.defaults["geometry_multidepth"],
  144. "depthperpass": float(self.app.defaults["geometry_depthperpass"]),
  145. "vtipdia": float(self.app.defaults["geometry_vtipdia"]),
  146. "vtipangle": float(self.app.defaults["geometry_vtipangle"]),
  147. "travelz": float(self.app.defaults["geometry_travelz"]),
  148. "feedrate": float(self.app.defaults["geometry_feedrate"]),
  149. "feedrate_z": float(self.app.defaults["geometry_feedrate_z"]),
  150. "feedrate_rapid": float(self.app.defaults["geometry_feedrate_rapid"]),
  151. "spindlespeed": self.app.defaults["geometry_spindlespeed"],
  152. "dwell": self.app.defaults["geometry_dwell"],
  153. "dwelltime": float(self.app.defaults["geometry_dwelltime"]),
  154. "ppname_g": self.app.defaults["geometry_ppname_g"],
  155. "extracut": self.app.defaults["geometry_extracut"],
  156. "extracut_length": float(self.app.defaults["geometry_extracut_length"]),
  157. "toolchange": self.app.defaults["geometry_toolchange"],
  158. "toolchangexy": self.app.defaults["geometry_toolchangexy"],
  159. "toolchangez": float(self.app.defaults["geometry_toolchangez"]),
  160. "startz": self.app.defaults["geometry_startz"],
  161. "endz": float(self.app.defaults["geometry_endz"]),
  162. "area_exclusion": self.app.defaults["geometry_area_exclusion"],
  163. "area_shape": self.app.defaults["geometry_area_shape"],
  164. "area_strategy": self.app.defaults["geometry_area_strategy"],
  165. "area_overz": float(self.app.defaults["geometry_area_overz"]),
  166. # NCC
  167. "tools_nccoperation": self.app.defaults["tools_nccoperation"],
  168. "tools_nccmilling_type": self.app.defaults["tools_nccmilling_type"],
  169. "tools_nccoverlap": float(self.app.defaults["tools_nccoverlap"]),
  170. "tools_nccmargin": float(self.app.defaults["tools_nccmargin"]),
  171. "tools_nccmethod": self.app.defaults["tools_nccmethod"],
  172. "tools_nccconnect": self.app.defaults["tools_nccconnect"],
  173. "tools_ncccontour": self.app.defaults["tools_ncccontour"],
  174. "tools_ncc_offset_choice": self.app.defaults["tools_ncc_offset_choice"],
  175. "tools_ncc_offset_value": float(self.app.defaults["tools_ncc_offset_value"]),
  176. # Paint
  177. "tools_paintoverlap": float(self.app.defaults["tools_paintoverlap"]),
  178. "tools_paintoffset": float(self.app.defaults["tools_paintoffset"]),
  179. "tools_paintmethod": self.app.defaults["tools_paintmethod"],
  180. "tools_pathconnect": self.app.defaults["tools_pathconnect"],
  181. "tools_paintcontour": self.app.defaults["tools_paintcontour"],
  182. # Isolation Tool
  183. "tools_iso_passes": self.app.defaults["tools_iso_passes"],
  184. "tools_iso_overlap": self.app.defaults["tools_iso_overlap"],
  185. "tools_iso_milling_type": self.app.defaults["tools_iso_milling_type"],
  186. "tools_iso_follow": self.app.defaults["tools_iso_follow"],
  187. "tools_iso_isotype": self.app.defaults["tools_iso_isotype"],
  188. })
  189. def on_freeform_cutout(self):
  190. log.debug("Cutout.on_freeform_cutout() was launched ...")
  191. # def subtract_rectangle(obj_, x0, y0, x1, y1):
  192. # pts = [(x0, y0), (x1, y0), (x1, y1), (x0, y1)]
  193. # obj_.subtract_polygon(pts)
  194. name = self.ui.obj_combo.currentText()
  195. # Get source object.
  196. try:
  197. cutout_obj = self.app.collection.get_by_name(str(name))
  198. except Exception as e:
  199. log.debug("CutOut.on_freeform_cutout() --> %s" % str(e))
  200. self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve object"), name))
  201. return "Could not retrieve object: %s" % name
  202. if cutout_obj is None:
  203. self.app.inform.emit('[ERROR_NOTCL] %s' %
  204. _("There is no object selected for Cutout.\nSelect one and try again."))
  205. return
  206. dia = float(self.ui.dia.get_value())
  207. if 0 in {dia}:
  208. self.app.inform.emit('[WARNING_NOTCL] %s' %
  209. _("Tool Diameter is zero value. Change it to a positive real number."))
  210. return "Tool Diameter is zero value. Change it to a positive real number."
  211. try:
  212. kind = self.ui.obj_kind_combo.get_value()
  213. except ValueError:
  214. return
  215. margin = float(self.ui.margin.get_value())
  216. gapsize = float(self.ui.gapsize.get_value())
  217. try:
  218. gaps = self.ui.gaps.get_value()
  219. except TypeError:
  220. self.app.inform.emit('[WARNING_NOTCL] %s' % _("Number of gaps value is missing. Add it and retry."))
  221. return
  222. if gaps not in ['None', 'LR', 'TB', '2LR', '2TB', '4', '8']:
  223. self.app.inform.emit('[WARNING_NOTCL] %s' %
  224. _("Gaps value can be only one of: 'None', 'lr', 'tb', '2lr', '2tb', 4 or 8. "
  225. "Fill in a correct value and retry. "))
  226. return
  227. if cutout_obj.multigeo is True:
  228. self.app.inform.emit('[ERROR] %s' % _("Cutout operation cannot be done on a multi-geo Geometry.\n"
  229. "Optionally, this Multi-geo Geometry can be converted to "
  230. "Single-geo Geometry,\n"
  231. "and after that perform Cutout."))
  232. return
  233. convex_box = self.ui.convex_box.get_value()
  234. gapsize = gapsize / 2 + (dia / 2)
  235. def geo_init(geo_obj, app_obj):
  236. solid_geo = []
  237. if cutout_obj.kind == 'gerber':
  238. if isinstance(cutout_obj.solid_geometry, list):
  239. cutout_obj.solid_geometry = MultiPolygon(cutout_obj.solid_geometry)
  240. try:
  241. if convex_box:
  242. object_geo = cutout_obj.solid_geometry.convex_hull
  243. else:
  244. object_geo = cutout_obj.solid_geometry
  245. except Exception as err:
  246. log.debug("CutOut.on_freeform_cutout().geo_init() --> %s" % str(err))
  247. object_geo = cutout_obj.solid_geometry
  248. else:
  249. object_geo = cutout_obj.solid_geometry
  250. def cutout_handler(geom):
  251. # Get min and max data for each object as we just cut rectangles across X or Y
  252. xxmin, yymin, xxmax, yymax = CutOut.recursive_bounds(geom)
  253. px = 0.5 * (xxmin + xxmax) + margin
  254. py = 0.5 * (yymin + yymax) + margin
  255. lenx = (xxmax - xxmin) + (margin * 2)
  256. leny = (yymax - yymin) + (margin * 2)
  257. proc_geometry = []
  258. if gaps == 'None':
  259. pass
  260. else:
  261. if gaps == '8' or gaps == '2LR':
  262. geom = self.subtract_poly_from_geo(geom,
  263. xxmin - gapsize, # botleft_x
  264. py - gapsize + leny / 4, # botleft_y
  265. xxmax + gapsize, # topright_x
  266. py + gapsize + leny / 4) # topright_y
  267. geom = self.subtract_poly_from_geo(geom,
  268. xxmin - gapsize,
  269. py - gapsize - leny / 4,
  270. xxmax + gapsize,
  271. py + gapsize - leny / 4)
  272. if gaps == '8' or gaps == '2TB':
  273. geom = self.subtract_poly_from_geo(geom,
  274. px - gapsize + lenx / 4,
  275. yymin - gapsize,
  276. px + gapsize + lenx / 4,
  277. yymax + gapsize)
  278. geom = self.subtract_poly_from_geo(geom,
  279. px - gapsize - lenx / 4,
  280. yymin - gapsize,
  281. px + gapsize - lenx / 4,
  282. yymax + gapsize)
  283. if gaps == '4' or gaps == 'LR':
  284. geom = self.subtract_poly_from_geo(geom,
  285. xxmin - gapsize,
  286. py - gapsize,
  287. xxmax + gapsize,
  288. py + gapsize)
  289. if gaps == '4' or gaps == 'TB':
  290. geom = self.subtract_poly_from_geo(geom,
  291. px - gapsize,
  292. yymin - gapsize,
  293. px + gapsize,
  294. yymax + gapsize)
  295. try:
  296. for g in geom:
  297. if g and not g.is_empty:
  298. proc_geometry.append(g)
  299. except TypeError:
  300. if geom and not geom.is_empty:
  301. proc_geometry.append(geom)
  302. return proc_geometry
  303. if kind == 'single':
  304. object_geo = unary_union(object_geo)
  305. # for geo in object_geo:
  306. if cutout_obj.kind == 'gerber':
  307. if isinstance(object_geo, MultiPolygon):
  308. x0, y0, x1, y1 = object_geo.bounds
  309. object_geo = box(x0, y0, x1, y1)
  310. if margin >= 0:
  311. geo_buf = object_geo.buffer(margin + abs(dia / 2))
  312. else:
  313. geo_buf = object_geo.buffer(margin - abs(dia / 2))
  314. geo = geo_buf.exterior
  315. else:
  316. geo = object_geo
  317. solid_geo = cutout_handler(geom=geo)
  318. else:
  319. try:
  320. __ = iter(object_geo)
  321. except TypeError:
  322. object_geo = [object_geo]
  323. for geom_struct in object_geo:
  324. if cutout_obj.kind == 'gerber':
  325. if margin >= 0:
  326. geom_struct = (geom_struct.buffer(margin + abs(dia / 2))).exterior
  327. else:
  328. geom_struct_buff = geom_struct.buffer(-margin + abs(dia / 2))
  329. geom_struct = geom_struct_buff.interiors
  330. solid_geo += cutout_handler(geom=geom_struct)
  331. if not solid_geo:
  332. app_obj.inform.emit('[ERROR_NOTCL] %s' % _("Failed."))
  333. return "fail"
  334. solid_geo = linemerge(solid_geo)
  335. geo_obj.solid_geometry = deepcopy(solid_geo)
  336. xmin, ymin, xmax, ymax = CutOut.recursive_bounds(geo_obj.solid_geometry)
  337. geo_obj.options['xmin'] = xmin
  338. geo_obj.options['ymin'] = ymin
  339. geo_obj.options['xmax'] = xmax
  340. geo_obj.options['ymax'] = ymax
  341. geo_obj.options['cnctooldia'] = str(dia)
  342. geo_obj.options['cutz'] = self.ui.cutz_entry.get_value()
  343. geo_obj.options['multidepth'] = self.ui.mpass_cb.get_value()
  344. geo_obj.options['depthperpass'] = self.ui.maxdepth_entry.get_value()
  345. geo_obj.tools.update({
  346. 1: {
  347. 'tooldia': str(dia),
  348. 'offset': 'Path',
  349. 'offset_value': 0.0,
  350. 'type': _('Rough'),
  351. 'tool_type': 'C1',
  352. 'data': self.default_data,
  353. 'solid_geometry': geo_obj.solid_geometry
  354. }
  355. })
  356. geo_obj.multigeo = True
  357. geo_obj.tools[1]['data']['name'] = outname
  358. geo_obj.tools[1]['data']['cutz'] = self.ui.cutz_entry.get_value()
  359. geo_obj.tools[1]['data']['multidepth'] = self.ui.mpass_cb.get_value()
  360. geo_obj.tools[1]['data']['depthperpass'] = self.ui.maxdepth_entry.get_value()
  361. outname = cutout_obj.options["name"] + "_cutout"
  362. ret = self.app.app_obj.new_object('geometry', outname, geo_init)
  363. if ret == 'fail':
  364. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Failed."))
  365. return
  366. cutout_obj.plot()
  367. self.app.inform.emit('[success] %s' % _("Any form CutOut operation finished."))
  368. # self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
  369. self.app.should_we_save = True
  370. def on_rectangular_cutout(self):
  371. log.debug("Cutout.on_rectangular_cutout() was launched ...")
  372. # def subtract_rectangle(obj_, x0, y0, x1, y1):
  373. # pts = [(x0, y0), (x1, y0), (x1, y1), (x0, y1)]
  374. # obj_.subtract_polygon(pts)
  375. name = self.ui.obj_combo.currentText()
  376. # Get source object.
  377. try:
  378. cutout_obj = self.app.collection.get_by_name(str(name))
  379. except Exception as e:
  380. log.debug("CutOut.on_rectangular_cutout() --> %s" % str(e))
  381. self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve object"), name))
  382. return "Could not retrieve object: %s" % name
  383. if cutout_obj is None:
  384. self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Object not found"), str(name)))
  385. dia = float(self.ui.dia.get_value())
  386. if 0 in {dia}:
  387. self.app.inform.emit('[ERROR_NOTCL] %s' %
  388. _("Tool Diameter is zero value. Change it to a positive real number."))
  389. return "Tool Diameter is zero value. Change it to a positive real number."
  390. try:
  391. kind = self.ui.obj_kind_combo.get_value()
  392. except ValueError:
  393. return
  394. margin = float(self.ui.margin.get_value())
  395. gapsize = float(self.ui.gapsize.get_value())
  396. try:
  397. gaps = self.ui.gaps.get_value()
  398. except TypeError:
  399. self.app.inform.emit('[WARNING_NOTCL] %s' %
  400. _("Number of gaps value is missing. Add it and retry."))
  401. return
  402. if gaps not in ['None', 'LR', 'TB', '2LR', '2TB', '4', '8']:
  403. self.app.inform.emit('[WARNING_NOTCL] %s' % _("Gaps value can be only one of: "
  404. "'None', 'lr', 'tb', '2lr', '2tb', 4 or 8. "
  405. "Fill in a correct value and retry. "))
  406. return
  407. if cutout_obj.multigeo is True:
  408. self.app.inform.emit('[ERROR] %s' % _("Cutout operation cannot be done on a multi-geo Geometry.\n"
  409. "Optionally, this Multi-geo Geometry can be converted to "
  410. "Single-geo Geometry,\n"
  411. "and after that perform Cutout."))
  412. return
  413. # Get min and max data for each object as we just cut rectangles across X or Y
  414. gapsize = gapsize / 2 + (dia / 2)
  415. def geo_init(geo_obj, app_obj):
  416. solid_geo = []
  417. object_geo = cutout_obj.solid_geometry
  418. def cutout_rect_handler(geom):
  419. proc_geometry = []
  420. px = 0.5 * (xmin + xmax) + margin
  421. py = 0.5 * (ymin + ymax) + margin
  422. lenx = (xmax - xmin) + (margin * 2)
  423. leny = (ymax - ymin) + (margin * 2)
  424. if gaps == 'None':
  425. pass
  426. else:
  427. if gaps == '8' or gaps == '2LR':
  428. geom = self.subtract_poly_from_geo(geom,
  429. xmin - gapsize, # botleft_x
  430. py - gapsize + leny / 4, # botleft_y
  431. xmax + gapsize, # topright_x
  432. py + gapsize + leny / 4) # topright_y
  433. geom = self.subtract_poly_from_geo(geom,
  434. xmin - gapsize,
  435. py - gapsize - leny / 4,
  436. xmax + gapsize,
  437. py + gapsize - leny / 4)
  438. if gaps == '8' or gaps == '2TB':
  439. geom = self.subtract_poly_from_geo(geom,
  440. px - gapsize + lenx / 4,
  441. ymin - gapsize,
  442. px + gapsize + lenx / 4,
  443. ymax + gapsize)
  444. geom = self.subtract_poly_from_geo(geom,
  445. px - gapsize - lenx / 4,
  446. ymin - gapsize,
  447. px + gapsize - lenx / 4,
  448. ymax + gapsize)
  449. if gaps == '4' or gaps == 'LR':
  450. geom = self.subtract_poly_from_geo(geom,
  451. xmin - gapsize,
  452. py - gapsize,
  453. xmax + gapsize,
  454. py + gapsize)
  455. if gaps == '4' or gaps == 'TB':
  456. geom = self.subtract_poly_from_geo(geom,
  457. px - gapsize,
  458. ymin - gapsize,
  459. px + gapsize,
  460. ymax + gapsize)
  461. try:
  462. for g in geom:
  463. proc_geometry.append(g)
  464. except TypeError:
  465. proc_geometry.append(geom)
  466. return proc_geometry
  467. if kind == 'single':
  468. # fuse the lines
  469. object_geo = unary_union(object_geo)
  470. xmin, ymin, xmax, ymax = object_geo.bounds
  471. geo = box(xmin, ymin, xmax, ymax)
  472. # if Gerber create a buffer at a distance
  473. # if Geometry then cut through the geometry
  474. if cutout_obj.kind == 'gerber':
  475. if margin >= 0:
  476. geo = geo.buffer(margin + abs(dia / 2))
  477. else:
  478. geo = geo.buffer(margin - abs(dia / 2))
  479. solid_geo = cutout_rect_handler(geom=geo)
  480. else:
  481. if cutout_obj.kind == 'geometry':
  482. try:
  483. __ = iter(object_geo)
  484. except TypeError:
  485. object_geo = [object_geo]
  486. for geom_struct in object_geo:
  487. geom_struct = unary_union(geom_struct)
  488. xmin, ymin, xmax, ymax = geom_struct.bounds
  489. geom_struct = box(xmin, ymin, xmax, ymax)
  490. solid_geo += cutout_rect_handler(geom=geom_struct)
  491. elif cutout_obj.kind == 'gerber' and margin >= 0:
  492. try:
  493. __ = iter(object_geo)
  494. except TypeError:
  495. object_geo = [object_geo]
  496. for geom_struct in object_geo:
  497. geom_struct = unary_union(geom_struct)
  498. xmin, ymin, xmax, ymax = geom_struct.bounds
  499. geom_struct = box(xmin, ymin, xmax, ymax)
  500. geom_struct = geom_struct.buffer(margin + abs(dia / 2))
  501. solid_geo += cutout_rect_handler(geom=geom_struct)
  502. elif cutout_obj.kind == 'gerber' and margin < 0:
  503. app_obj.inform.emit(
  504. '[WARNING_NOTCL] %s' % _("Rectangular cutout with negative margin is not possible."))
  505. return "fail"
  506. geo_obj.options['cnctooldia'] = str(dia)
  507. geo_obj.options['cutz'] = self.ui.cutz_entry.get_value()
  508. geo_obj.options['multidepth'] = self.ui.mpass_cb.get_value()
  509. geo_obj.options['depthperpass'] = self.ui.maxdepth_entry.get_value()
  510. if not solid_geo:
  511. app_obj.inform.emit('[ERROR_NOTCL] %s' % _("Failed."))
  512. return "fail"
  513. solid_geo = linemerge(solid_geo)
  514. geo_obj.solid_geometry = deepcopy(solid_geo)
  515. geo_obj.tools.update({
  516. 1: {
  517. 'tooldia': str(dia),
  518. 'offset': 'Path',
  519. 'offset_value': 0.0,
  520. 'type': _('Rough'),
  521. 'tool_type': 'C1',
  522. 'data': self.default_data,
  523. 'solid_geometry': geo_obj.solid_geometry
  524. }
  525. })
  526. geo_obj.multigeo = True
  527. geo_obj.tools[1]['data']['name'] = outname
  528. geo_obj.tools[1]['data']['cutz'] = self.ui.cutz_entry.get_value()
  529. geo_obj.tools[1]['data']['multidepth'] = self.ui.mpass_cb.get_value()
  530. geo_obj.tools[1]['data']['depthperpass'] = self.ui.maxdepth_entry.get_value()
  531. outname = cutout_obj.options["name"] + "_cutout"
  532. ret = self.app.app_obj.new_object('geometry', outname, geo_init)
  533. if ret != 'fail':
  534. # cutout_obj.plot()
  535. self.app.inform.emit('[success] %s' % _("Any form CutOut operation finished."))
  536. else:
  537. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Failed."))
  538. return
  539. # self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
  540. self.app.should_we_save = True
  541. def on_manual_gap_click(self):
  542. name = self.ui.man_object_combo.currentText()
  543. # Get source object.
  544. try:
  545. self.man_cutout_obj = self.app.collection.get_by_name(str(name))
  546. except Exception as e:
  547. log.debug("CutOut.on_manual_cutout() --> %s" % str(e))
  548. self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve Geometry object"), name))
  549. return
  550. if self.man_cutout_obj is None:
  551. self.app.inform.emit('[ERROR_NOTCL] %s: %s' %
  552. (_("Geometry object for manual cutout not found"), self.man_cutout_obj))
  553. return
  554. self.app.inform.emit(_("Click on the selected geometry object perimeter to create a bridge gap ..."))
  555. self.app.geo_editor.tool_shape.enabled = True
  556. self.cutting_dia = float(self.ui.dia.get_value())
  557. if 0 in {self.cutting_dia}:
  558. self.app.inform.emit('[ERROR_NOTCL] %s' %
  559. _("Tool Diameter is zero value. Change it to a positive real number."))
  560. return
  561. self.cutting_gapsize = float(self.ui.gapsize.get_value())
  562. name = self.ui.man_object_combo.currentText()
  563. # Get Geometry source object to be used as target for Manual adding Gaps
  564. try:
  565. self.man_cutout_obj = self.app.collection.get_by_name(str(name))
  566. except Exception as e:
  567. log.debug("CutOut.on_manual_cutout() --> %s" % str(e))
  568. self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve Geometry object"), name))
  569. return
  570. if self.app.is_legacy is False:
  571. self.app.plotcanvas.graph_event_disconnect('key_press', self.app.ui.keyPressEvent)
  572. self.app.plotcanvas.graph_event_disconnect('mouse_press', self.app.on_mouse_click_over_plot)
  573. self.app.plotcanvas.graph_event_disconnect('mouse_release', self.app.on_mouse_click_release_over_plot)
  574. self.app.plotcanvas.graph_event_disconnect('mouse_move', self.app.on_mouse_move_over_plot)
  575. else:
  576. self.app.plotcanvas.graph_event_disconnect(self.app.kp)
  577. self.app.plotcanvas.graph_event_disconnect(self.app.mp)
  578. self.app.plotcanvas.graph_event_disconnect(self.app.mr)
  579. self.app.plotcanvas.graph_event_disconnect(self.app.mm)
  580. self.kp = self.app.plotcanvas.graph_event_connect('key_press', self.on_key_press)
  581. self.mm = self.app.plotcanvas.graph_event_connect('mouse_move', self.on_mouse_move)
  582. self.mr = self.app.plotcanvas.graph_event_connect('mouse_release', self.on_mouse_click_release)
  583. self.mouse_events_connected = True
  584. if self.ui.big_cursor_cb.get_value():
  585. self.old_cursor_type = self.app.defaults["global_cursor_type"]
  586. self.app.on_cursor_type(val="big")
  587. self.app.defaults['global_selection_shape'] = False
  588. def on_manual_cutout(self, click_pos):
  589. if self.man_cutout_obj is None:
  590. self.app.inform.emit('[ERROR_NOTCL] %s: %s' %
  591. (_("Geometry object for manual cutout not found"), self.man_cutout_obj))
  592. return
  593. # use the snapped position as reference
  594. snapped_pos = self.app.geo_editor.snap(click_pos[0], click_pos[1])
  595. cut_poly = self.cutting_geo(pos=(snapped_pos[0], snapped_pos[1]))
  596. # first subtract geometry for the total solid_geometry
  597. new_solid_geometry = CutOut.subtract_polygon(self.man_cutout_obj.solid_geometry, cut_poly)
  598. new_solid_geometry = linemerge(new_solid_geometry)
  599. self.man_cutout_obj.solid_geometry = new_solid_geometry
  600. # then do it or each tool in the manual cutout Geometry object
  601. for tool in self.man_cutout_obj.tools:
  602. self.man_cutout_obj.tools[tool]['solid_geometry'] = new_solid_geometry
  603. self.man_cutout_obj.plot()
  604. self.app.inform.emit('[success] %s' % _("Added manual Bridge Gap."))
  605. self.app.should_we_save = True
  606. def on_manual_geo(self):
  607. name = self.ui.obj_combo.currentText()
  608. # Get source object.
  609. try:
  610. cutout_obj = self.app.collection.get_by_name(str(name))
  611. except Exception as e:
  612. log.debug("CutOut.on_manual_geo() --> %s" % str(e))
  613. self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve Gerber object"), name))
  614. return "Could not retrieve object: %s" % name
  615. if cutout_obj is None:
  616. self.app.inform.emit('[ERROR_NOTCL] %s' %
  617. _("There is no Gerber object selected for Cutout.\n"
  618. "Select one and try again."))
  619. return
  620. if cutout_obj.kind != 'gerber':
  621. self.app.inform.emit('[ERROR_NOTCL] %s' %
  622. _("The selected object has to be of Gerber type.\n"
  623. "Select a Gerber file and try again."))
  624. return
  625. dia = float(self.ui.dia.get_value())
  626. if 0 in {dia}:
  627. self.app.inform.emit('[ERROR_NOTCL] %s' %
  628. _("Tool Diameter is zero value. Change it to a positive real number."))
  629. return
  630. try:
  631. kind = self.ui.obj_kind_combo.get_value()
  632. except ValueError:
  633. return
  634. margin = float(self.ui.margin.get_value())
  635. convex_box = self.ui.convex_box.get_value()
  636. def geo_init(geo_obj, app_obj):
  637. geo_union = unary_union(cutout_obj.solid_geometry)
  638. if convex_box:
  639. geo = geo_union.convex_hull
  640. geo_obj.solid_geometry = geo.buffer(margin + abs(dia / 2))
  641. elif kind == 'single':
  642. if isinstance(geo_union, Polygon) or \
  643. (isinstance(geo_union, list) and len(geo_union) == 1) or \
  644. (isinstance(geo_union, MultiPolygon) and len(geo_union) == 1):
  645. geo_obj.solid_geometry = geo_union.buffer(margin + abs(dia / 2)).exterior
  646. elif isinstance(geo_union, MultiPolygon):
  647. x0, y0, x1, y1 = geo_union.bounds
  648. geo = box(x0, y0, x1, y1)
  649. geo_obj.solid_geometry = geo.buffer(margin + abs(dia / 2))
  650. else:
  651. app_obj.inform.emit('[ERROR_NOTCL] %s: %s' % (
  652. _("Geometry not supported for cutout"), type(geo_union)))
  653. return 'fail'
  654. else:
  655. geo = geo_union
  656. geo = geo.buffer(margin + abs(dia / 2))
  657. if isinstance(geo, Polygon):
  658. geo_obj.solid_geometry = geo.exterior
  659. elif isinstance(geo, MultiPolygon):
  660. solid_geo = []
  661. for poly in geo:
  662. solid_geo.append(poly.exterior)
  663. geo_obj.solid_geometry = deepcopy(solid_geo)
  664. geo_obj.options['cnctooldia'] = str(dia)
  665. geo_obj.options['cutz'] = self.ui.cutz_entry.get_value()
  666. geo_obj.options['multidepth'] = self.ui.mpass_cb.get_value()
  667. geo_obj.options['depthperpass'] = self.ui.maxdepth_entry.get_value()
  668. geo_obj.tools.update({
  669. 1: {
  670. 'tooldia': str(dia),
  671. 'offset': 'Path',
  672. 'offset_value': 0.0,
  673. 'type': _('Rough'),
  674. 'tool_type': 'C1',
  675. 'data': self.default_data,
  676. 'solid_geometry': geo_obj.solid_geometry
  677. }
  678. })
  679. geo_obj.multigeo = True
  680. geo_obj.tools[1]['data']['name'] = outname
  681. geo_obj.tools[1]['data']['cutz'] = self.ui.cutz_entry.get_value()
  682. geo_obj.tools[1]['data']['multidepth'] = self.ui.mpass_cb.get_value()
  683. geo_obj.tools[1]['data']['depthperpass'] = self.ui.maxdepth_entry.get_value()
  684. outname = cutout_obj.options["name"] + "_cutout"
  685. self.app.app_obj.new_object('geometry', outname, geo_init)
  686. def cutting_geo(self, pos):
  687. self.cutting_dia = float(self.ui.dia.get_value())
  688. self.cutting_gapsize = float(self.ui.gapsize.get_value())
  689. offset = self.cutting_dia / 2 + self.cutting_gapsize / 2
  690. # cutting area definition
  691. orig_x = pos[0]
  692. orig_y = pos[1]
  693. xmin = orig_x - offset
  694. ymin = orig_y - offset
  695. xmax = orig_x + offset
  696. ymax = orig_y + offset
  697. cut_poly = box(xmin, ymin, xmax, ymax)
  698. return cut_poly
  699. # To be called after clicking on the plot.
  700. def on_mouse_click_release(self, event):
  701. if self.app.is_legacy is False:
  702. event_pos = event.pos
  703. # event_is_dragging = event.is_dragging
  704. right_button = 2
  705. else:
  706. event_pos = (event.xdata, event.ydata)
  707. # event_is_dragging = self.app.plotcanvas.is_dragging
  708. right_button = 3
  709. try:
  710. x = float(event_pos[0])
  711. y = float(event_pos[1])
  712. except TypeError:
  713. return
  714. event_pos = (x, y)
  715. # do paint single only for left mouse clicks
  716. if event.button == 1:
  717. self.app.inform.emit(_("Making manual bridge gap..."))
  718. pos = self.app.plotcanvas.translate_coords(event_pos)
  719. self.on_manual_cutout(click_pos=pos)
  720. # if RMB then we exit
  721. elif event.button == right_button and self.mouse_is_dragging is False:
  722. if self.app.is_legacy is False:
  723. self.app.plotcanvas.graph_event_disconnect('key_press', self.on_key_press)
  724. self.app.plotcanvas.graph_event_disconnect('mouse_move', self.on_mouse_move)
  725. self.app.plotcanvas.graph_event_disconnect('mouse_release', self.on_mouse_click_release)
  726. else:
  727. self.app.plotcanvas.graph_event_disconnect(self.kp)
  728. self.app.plotcanvas.graph_event_disconnect(self.mm)
  729. self.app.plotcanvas.graph_event_disconnect(self.mr)
  730. self.app.kp = self.app.plotcanvas.graph_event_connect('key_press', self.app.ui.keyPressEvent)
  731. self.app.mp = self.app.plotcanvas.graph_event_connect('mouse_press', self.app.on_mouse_click_over_plot)
  732. self.app.mr = self.app.plotcanvas.graph_event_connect('mouse_release',
  733. self.app.on_mouse_click_release_over_plot)
  734. self.app.mm = self.app.plotcanvas.graph_event_connect('mouse_move', self.app.on_mouse_move_over_plot)
  735. # Remove any previous utility shape
  736. self.app.geo_editor.tool_shape.clear(update=True)
  737. self.app.geo_editor.tool_shape.enabled = False
  738. # signal that the mouse events are disconnected from local methods
  739. self.mouse_events_connected = False
  740. if self.ui.big_cursor_cb.get_value():
  741. # restore cursor
  742. self.app.on_cursor_type(val=self.old_cursor_type)
  743. # restore selection
  744. self.app.defaults['global_selection_shape'] = self.old_selection_state
  745. def on_mouse_move(self, event):
  746. self.app.on_mouse_move_over_plot(event=event)
  747. if self.app.is_legacy is False:
  748. event_pos = event.pos
  749. event_is_dragging = event.is_dragging
  750. # right_button = 2
  751. else:
  752. event_pos = (event.xdata, event.ydata)
  753. event_is_dragging = self.app.plotcanvas.is_dragging
  754. # right_button = 3
  755. try:
  756. x = float(event_pos[0])
  757. y = float(event_pos[1])
  758. except TypeError:
  759. return
  760. event_pos = (x, y)
  761. pos = self.canvas.translate_coords(event_pos)
  762. event.xdata, event.ydata = pos[0], pos[1]
  763. if event_is_dragging is True:
  764. self.mouse_is_dragging = True
  765. else:
  766. self.mouse_is_dragging = False
  767. try:
  768. x = float(event.xdata)
  769. y = float(event.ydata)
  770. except TypeError:
  771. return
  772. if self.app.grid_status():
  773. snap_x, snap_y = self.app.geo_editor.snap(x, y)
  774. else:
  775. snap_x, snap_y = x, y
  776. self.x_pos, self.y_pos = snap_x, snap_y
  777. # #################################################
  778. # ### This section makes the cutting geo to #######
  779. # ### rotate if it intersects the target geo ######
  780. # #################################################
  781. cut_geo = self.cutting_geo(pos=(snap_x, snap_y))
  782. man_geo = self.man_cutout_obj.solid_geometry
  783. def get_angle(geo):
  784. line = cut_geo.intersection(geo)
  785. try:
  786. pt1_x = line.coords[0][0]
  787. pt1_y = line.coords[0][1]
  788. pt2_x = line.coords[1][0]
  789. pt2_y = line.coords[1][1]
  790. dx = pt1_x - pt2_x
  791. dy = pt1_y - pt2_y
  792. if dx == 0 or dy == 0:
  793. angle = 0
  794. else:
  795. radian = math.atan(dx / dy)
  796. angle = radian * 180 / math.pi
  797. except Exception:
  798. angle = 0
  799. return angle
  800. try:
  801. rot_angle = 0
  802. for geo_el in man_geo:
  803. if isinstance(geo_el, Polygon):
  804. work_geo = geo_el.exterior
  805. if cut_geo.intersects(work_geo):
  806. rot_angle = get_angle(geo=work_geo)
  807. else:
  808. rot_angle = 0
  809. else:
  810. rot_angle = 0
  811. if cut_geo.intersects(geo_el):
  812. rot_angle = get_angle(geo=geo_el)
  813. if rot_angle != 0:
  814. break
  815. except TypeError:
  816. if isinstance(man_geo, Polygon):
  817. work_geo = man_geo.exterior
  818. if cut_geo.intersects(work_geo):
  819. rot_angle = get_angle(geo=work_geo)
  820. else:
  821. rot_angle = 0
  822. else:
  823. rot_angle = 0
  824. if cut_geo.intersects(man_geo):
  825. rot_angle = get_angle(geo=man_geo)
  826. # rotate only if there is an angle to rotate to
  827. if rot_angle != 0:
  828. cut_geo = affinity.rotate(cut_geo, -rot_angle)
  829. # Remove any previous utility shape
  830. self.app.geo_editor.tool_shape.clear(update=True)
  831. self.draw_utility_geometry(geo=cut_geo)
  832. def draw_utility_geometry(self, geo):
  833. self.app.geo_editor.tool_shape.add(
  834. shape=geo,
  835. color=(self.app.defaults["global_draw_color"] + '80'),
  836. update=False,
  837. layer=0,
  838. tolerance=None)
  839. self.app.geo_editor.tool_shape.redraw()
  840. def on_key_press(self, event):
  841. # events out of the self.app.collection view (it's about Project Tab) are of type int
  842. if type(event) is int:
  843. key = event
  844. # events from the GUI are of type QKeyEvent
  845. elif type(event) == QtGui.QKeyEvent:
  846. key = event.key()
  847. elif isinstance(event, mpl_key_event): # MatPlotLib key events are trickier to interpret than the rest
  848. key = event.key
  849. key = QtGui.QKeySequence(key)
  850. # check for modifiers
  851. key_string = key.toString().lower()
  852. if '+' in key_string:
  853. mod, __, key_text = key_string.rpartition('+')
  854. if mod.lower() == 'ctrl':
  855. # modifiers = QtCore.Qt.ControlModifier
  856. pass
  857. elif mod.lower() == 'alt':
  858. # modifiers = QtCore.Qt.AltModifier
  859. pass
  860. elif mod.lower() == 'shift':
  861. # modifiers = QtCore.Qt.ShiftModifier
  862. pass
  863. else:
  864. # modifiers = QtCore.Qt.NoModifier
  865. pass
  866. key = QtGui.QKeySequence(key_text)
  867. # events from Vispy are of type KeyEvent
  868. else:
  869. key = event.key
  870. # Escape = Deselect All
  871. if key == QtCore.Qt.Key_Escape or key == 'Escape':
  872. if self.mouse_events_connected is True:
  873. self.mouse_events_connected = False
  874. if self.app.is_legacy is False:
  875. self.app.plotcanvas.graph_event_disconnect('key_press', self.on_key_press)
  876. self.app.plotcanvas.graph_event_disconnect('mouse_move', self.on_mouse_move)
  877. self.app.plotcanvas.graph_event_disconnect('mouse_release', self.on_mouse_click_release)
  878. else:
  879. self.app.plotcanvas.graph_event_disconnect(self.kp)
  880. self.app.plotcanvas.graph_event_disconnect(self.mm)
  881. self.app.plotcanvas.graph_event_disconnect(self.mr)
  882. self.app.kp = self.app.plotcanvas.graph_event_connect('key_press', self.app.ui.keyPressEvent)
  883. self.app.mp = self.app.plotcanvas.graph_event_connect('mouse_press', self.app.on_mouse_click_over_plot)
  884. self.app.mr = self.app.plotcanvas.graph_event_connect('mouse_release',
  885. self.app.on_mouse_click_release_over_plot)
  886. self.app.mm = self.app.plotcanvas.graph_event_connect('mouse_move', self.app.on_mouse_move_over_plot)
  887. if self.ui.big_cursor_cb.get_value():
  888. # restore cursor
  889. self.app.on_cursor_type(val=self.old_cursor_type)
  890. # restore selection
  891. self.app.defaults['global_selection_shape'] = self.old_selection_state
  892. # Remove any previous utility shape
  893. self.app.geo_editor.tool_shape.clear(update=True)
  894. self.app.geo_editor.tool_shape.enabled = False
  895. # Grid toggle
  896. if key == QtCore.Qt.Key_G or key == 'G':
  897. self.app.ui.grid_snap_btn.trigger()
  898. # Jump to coords
  899. if key == QtCore.Qt.Key_J or key == 'J':
  900. l_x, l_y = self.app.on_jump_to()
  901. self.app.geo_editor.tool_shape.clear(update=True)
  902. geo = self.cutting_geo(pos=(l_x, l_y))
  903. self.draw_utility_geometry(geo=geo)
  904. @staticmethod
  905. def subtract_poly_from_geo(solid_geo, x0, y0, x1, y1):
  906. """
  907. Subtract polygon made from points from the given object.
  908. This only operates on the paths in the original geometry,
  909. i.e. it converts polygons into paths.
  910. :param x0: x coord for lower left vertex of the polygon.
  911. :param y0: y coord for lower left vertex of the polygon.
  912. :param x1: x coord for upper right vertex of the polygon.
  913. :param y1: y coord for upper right vertex of the polygon.
  914. :param solid_geo: Geometry from which to substract. If none, use the solid_geomety property of the object
  915. :return: none
  916. """
  917. points = [(x0, y0), (x1, y0), (x1, y1), (x0, y1)]
  918. # pathonly should be allways True, otherwise polygons are not subtracted
  919. flat_geometry = CutOut.flatten(geometry=solid_geo)
  920. log.debug("%d paths" % len(flat_geometry))
  921. polygon = Polygon(points)
  922. toolgeo = cascaded_union(polygon)
  923. diffs = []
  924. for target in flat_geometry:
  925. if type(target) == LineString or type(target) == LinearRing:
  926. diffs.append(target.difference(toolgeo))
  927. else:
  928. log.warning("Not implemented.")
  929. return unary_union(diffs)
  930. @staticmethod
  931. def flatten(geometry):
  932. """
  933. Creates a list of non-iterable linear geometry objects.
  934. Polygons are expanded into its exterior and interiors.
  935. Results are placed in self.flat_geometry
  936. :param geometry: Shapely type or list or list of list of such.
  937. """
  938. flat_geo = []
  939. try:
  940. for geo in geometry:
  941. if geo and not geo.is_empty:
  942. flat_geo += CutOut.flatten(geometry=geo)
  943. except TypeError:
  944. if isinstance(geometry, Polygon):
  945. flat_geo.append(geometry.exterior)
  946. CutOut.flatten(geometry=geometry.interiors)
  947. else:
  948. flat_geo.append(geometry)
  949. return flat_geo
  950. @staticmethod
  951. def recursive_bounds(geometry):
  952. """
  953. Return the bounds of the biggest bounding box in geometry, one that include all.
  954. :param geometry: a iterable object that holds geometry
  955. :return: Returns coordinates of rectangular bounds of geometry: (xmin, ymin, xmax, ymax).
  956. """
  957. # now it can get bounds for nested lists of objects
  958. def bounds_rec(obj):
  959. try:
  960. minx = Inf
  961. miny = Inf
  962. maxx = -Inf
  963. maxy = -Inf
  964. for k in obj:
  965. minx_, miny_, maxx_, maxy_ = bounds_rec(k)
  966. minx = min(minx, minx_)
  967. miny = min(miny, miny_)
  968. maxx = max(maxx, maxx_)
  969. maxy = max(maxy, maxy_)
  970. return minx, miny, maxx, maxy
  971. except TypeError:
  972. # it's a Shapely object, return it's bounds
  973. if obj:
  974. return obj.bounds
  975. return bounds_rec(geometry)
  976. @staticmethod
  977. def subtract_polygon(target_geo, subtractor):
  978. """
  979. Subtract subtractor polygon from the target_geo. This only operates on the paths in the target_geo,
  980. i.e. it converts polygons into paths.
  981. :param target_geo: geometry from which to subtract
  982. :param subtractor: a list of Points, a LinearRing or a Polygon that will be subtracted from target_geo
  983. :return: a cascaded union of the resulting geometry
  984. """
  985. if target_geo is None:
  986. target_geo = []
  987. # flatten() takes care of possible empty geometry making sure that is filtered
  988. flat_geometry = CutOut.flatten(target_geo)
  989. log.debug("%d paths" % len(flat_geometry))
  990. toolgeo = unary_union(subtractor)
  991. diffs = []
  992. for target in flat_geometry:
  993. if isinstance(target, LineString) or isinstance(target, LinearRing) or isinstance(target, MultiLineString):
  994. diffs.append(target.difference(toolgeo))
  995. else:
  996. log.warning("Not implemented.")
  997. return unary_union(diffs)
  998. def reset_fields(self):
  999. self.ui.obj_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  1000. class CutoutUI:
  1001. toolName = _("Cutout PCB")
  1002. def __init__(self, layout, app):
  1003. self.app = app
  1004. self.decimals = self.app.decimals
  1005. self.layout = layout
  1006. # Title
  1007. title_label = QtWidgets.QLabel("%s" % self.toolName)
  1008. title_label.setStyleSheet("""
  1009. QLabel
  1010. {
  1011. font-size: 16px;
  1012. font-weight: bold;
  1013. }
  1014. """)
  1015. self.layout.addWidget(title_label)
  1016. self.layout.addWidget(QtWidgets.QLabel(''))
  1017. # Form Layout
  1018. grid0 = QtWidgets.QGridLayout()
  1019. grid0.setColumnStretch(0, 0)
  1020. grid0.setColumnStretch(1, 1)
  1021. self.layout.addLayout(grid0)
  1022. self.object_label = QtWidgets.QLabel('<b>%s:</b>' % _("Source Object"))
  1023. self.object_label.setToolTip('%s.' % _("Object to be cutout"))
  1024. grid0.addWidget(self.object_label, 0, 0, 1, 2)
  1025. # Object kind
  1026. self.kindlabel = QtWidgets.QLabel('%s:' % _('Kind'))
  1027. self.kindlabel.setToolTip(
  1028. _("Choice of what kind the object we want to cutout is.<BR>"
  1029. "- <B>Single</B>: contain a single PCB Gerber outline object.<BR>"
  1030. "- <B>Panel</B>: a panel PCB Gerber object, which is made\n"
  1031. "out of many individual PCB outlines.")
  1032. )
  1033. self.obj_kind_combo = RadioSet([
  1034. {"label": _("Single"), "value": "single"},
  1035. {"label": _("Panel"), "value": "panel"},
  1036. ])
  1037. grid0.addWidget(self.kindlabel, 1, 0)
  1038. grid0.addWidget(self.obj_kind_combo, 1, 1)
  1039. # Type of object to be cutout
  1040. self.type_obj_radio = RadioSet([
  1041. {"label": _("Gerber"), "value": "grb"},
  1042. {"label": _("Geometry"), "value": "geo"},
  1043. ])
  1044. self.type_obj_combo_label = QtWidgets.QLabel('%s:' % _("Type"))
  1045. self.type_obj_combo_label.setToolTip(
  1046. _("Specify the type of object to be cutout.\n"
  1047. "It can be of type: Gerber or Geometry.\n"
  1048. "What is selected here will dictate the kind\n"
  1049. "of objects that will populate the 'Object' combobox.")
  1050. )
  1051. grid0.addWidget(self.type_obj_combo_label, 2, 0)
  1052. grid0.addWidget(self.type_obj_radio, 2, 1)
  1053. # Object to be cutout
  1054. self.obj_combo = FCComboBox()
  1055. self.obj_combo.setModel(self.app.collection)
  1056. self.obj_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  1057. self.obj_combo.is_last = True
  1058. grid0.addWidget(self.obj_combo, 3, 0, 1, 2)
  1059. separator_line = QtWidgets.QFrame()
  1060. separator_line.setFrameShape(QtWidgets.QFrame.HLine)
  1061. separator_line.setFrameShadow(QtWidgets.QFrame.Sunken)
  1062. grid0.addWidget(separator_line, 4, 0, 1, 2)
  1063. grid0.addWidget(QtWidgets.QLabel(''), 5, 0, 1, 2)
  1064. self.param_label = QtWidgets.QLabel('<b>%s:</b>' % _("Tool Parameters"))
  1065. grid0.addWidget(self.param_label, 6, 0, 1, 2)
  1066. # Tool Diameter
  1067. self.dia = FCDoubleSpinner(callback=self.confirmation_message)
  1068. self.dia.set_precision(self.decimals)
  1069. self.dia.set_range(0.0000, 9999.9999)
  1070. self.dia_label = QtWidgets.QLabel('%s:' % _("Tool Diameter"))
  1071. self.dia_label.setToolTip(
  1072. _("Diameter of the tool used to cutout\n"
  1073. "the PCB shape out of the surrounding material.")
  1074. )
  1075. grid0.addWidget(self.dia_label, 8, 0)
  1076. grid0.addWidget(self.dia, 8, 1)
  1077. # Cut Z
  1078. cutzlabel = QtWidgets.QLabel('%s:' % _('Cut Z'))
  1079. cutzlabel.setToolTip(
  1080. _(
  1081. "Cutting depth (negative)\n"
  1082. "below the copper surface."
  1083. )
  1084. )
  1085. self.cutz_entry = FCDoubleSpinner(callback=self.confirmation_message)
  1086. self.cutz_entry.set_precision(self.decimals)
  1087. if machinist_setting == 0:
  1088. self.cutz_entry.setRange(-9999.9999, -0.00001)
  1089. else:
  1090. self.cutz_entry.setRange(-9999.9999, 9999.9999)
  1091. self.cutz_entry.setSingleStep(0.1)
  1092. grid0.addWidget(cutzlabel, 9, 0)
  1093. grid0.addWidget(self.cutz_entry, 9, 1)
  1094. # Multi-pass
  1095. self.mpass_cb = FCCheckBox('%s:' % _("Multi-Depth"))
  1096. self.mpass_cb.setToolTip(
  1097. _(
  1098. "Use multiple passes to limit\n"
  1099. "the cut depth in each pass. Will\n"
  1100. "cut multiple times until Cut Z is\n"
  1101. "reached."
  1102. )
  1103. )
  1104. self.maxdepth_entry = FCDoubleSpinner(callback=self.confirmation_message)
  1105. self.maxdepth_entry.set_precision(self.decimals)
  1106. self.maxdepth_entry.setRange(0, 9999.9999)
  1107. self.maxdepth_entry.setSingleStep(0.1)
  1108. self.maxdepth_entry.setToolTip(
  1109. _(
  1110. "Depth of each pass (positive)."
  1111. )
  1112. )
  1113. grid0.addWidget(self.mpass_cb, 10, 0)
  1114. grid0.addWidget(self.maxdepth_entry, 10, 1)
  1115. # Suppress gaps
  1116. self.suppress_cb = FCCheckBox('%s:' % _("Suppress gaps"))
  1117. self.suppress_cb.setToolTip(
  1118. _("Active only when multi depth is active.\n"
  1119. "If used, the gaps will not be used until the specified depth."))
  1120. self.suppress_depth_entry = FCDoubleSpinner(callback=self.confirmation_message)
  1121. self.suppress_depth_entry.set_precision(self.decimals)
  1122. self.suppress_depth_entry.setRange(0, 9999.9999)
  1123. self.suppress_depth_entry.setSingleStep(0.1)
  1124. self.suppress_depth_entry.setToolTip(
  1125. _("Active only when multi depth is active.\n"
  1126. "If used, the gaps will not be used until the specified depth."))
  1127. grid0.addWidget(self.suppress_cb, 12, 0)
  1128. grid0.addWidget(self.suppress_depth_entry, 12, 1)
  1129. self.ois_mpass_geo = OptionalInputSection(self.mpass_cb,
  1130. [
  1131. self.maxdepth_entry,
  1132. self.suppress_cb,
  1133. self.suppress_depth_entry
  1134. ])
  1135. # Margin
  1136. self.margin = FCDoubleSpinner(callback=self.confirmation_message)
  1137. self.margin.set_range(-9999.9999, 9999.9999)
  1138. self.margin.setSingleStep(0.1)
  1139. self.margin.set_precision(self.decimals)
  1140. self.margin_label = QtWidgets.QLabel('%s:' % _("Margin"))
  1141. self.margin_label.setToolTip(
  1142. _("Margin over bounds. A positive value here\n"
  1143. "will make the cutout of the PCB further from\n"
  1144. "the actual PCB border")
  1145. )
  1146. grid0.addWidget(self.margin_label, 14, 0)
  1147. grid0.addWidget(self.margin, 14, 1)
  1148. # Gapsize
  1149. self.gapsize = FCDoubleSpinner(callback=self.confirmation_message)
  1150. self.gapsize.set_precision(self.decimals)
  1151. self.gapsize_label = QtWidgets.QLabel('%s:' % _("Gap size"))
  1152. self.gapsize_label.setToolTip(
  1153. _("The size of the bridge gaps in the cutout\n"
  1154. "used to keep the board connected to\n"
  1155. "the surrounding material (the one \n"
  1156. "from which the PCB is cutout).")
  1157. )
  1158. grid0.addWidget(self.gapsize_label, 16, 0)
  1159. grid0.addWidget(self.gapsize, 16, 1)
  1160. # How gaps wil be rendered:
  1161. # lr - left + right
  1162. # tb - top + bottom
  1163. # 4 - left + right +top + bottom
  1164. # 2lr - 2*left + 2*right
  1165. # 2tb - 2*top + 2*bottom
  1166. # 8 - 2*left + 2*right +2*top + 2*bottom
  1167. # Surrounding convex box shape
  1168. self.convex_box = FCCheckBox('%s' % _("Convex Shape"))
  1169. # self.convex_box_label = QtWidgets.QLabel('%s' % _("Convex Sh."))
  1170. self.convex_box.setToolTip(
  1171. _("Create a convex shape surrounding the entire PCB.\n"
  1172. "Used only if the source object type is Gerber.")
  1173. )
  1174. grid0.addWidget(self.convex_box, 18, 0, 1, 2)
  1175. separator_line = QtWidgets.QFrame()
  1176. separator_line.setFrameShape(QtWidgets.QFrame.HLine)
  1177. separator_line.setFrameShadow(QtWidgets.QFrame.Sunken)
  1178. grid0.addWidget(separator_line, 20, 0, 1, 2)
  1179. grid0.addWidget(QtWidgets.QLabel(''), 22, 0, 1, 2)
  1180. # Title2
  1181. title_param_label = QtWidgets.QLabel("<b>%s %s</b>:" % (_('Automatic'), _("Bridge Gaps")))
  1182. title_param_label.setToolTip(
  1183. _("This section handle creation of automatic bridge gaps.")
  1184. )
  1185. grid0.addWidget(title_param_label, 24, 0, 1, 2)
  1186. # Gaps
  1187. gaps_label = QtWidgets.QLabel('%s:' % _('Gaps'))
  1188. gaps_label.setToolTip(
  1189. _("Number of gaps used for the Automatic cutout.\n"
  1190. "There can be maximum 8 bridges/gaps.\n"
  1191. "The choices are:\n"
  1192. "- None - no gaps\n"
  1193. "- lr - left + right\n"
  1194. "- tb - top + bottom\n"
  1195. "- 4 - left + right +top + bottom\n"
  1196. "- 2lr - 2*left + 2*right\n"
  1197. "- 2tb - 2*top + 2*bottom\n"
  1198. "- 8 - 2*left + 2*right +2*top + 2*bottom")
  1199. )
  1200. # gaps_label.setMinimumWidth(60)
  1201. self.gaps = FCComboBox()
  1202. gaps_items = ['None', 'LR', 'TB', '4', '2LR', '2TB', '8']
  1203. for it in gaps_items:
  1204. self.gaps.addItem(it)
  1205. # self.gaps.setStyleSheet('background-color: rgb(255,255,255)')
  1206. grid0.addWidget(gaps_label, 26, 0)
  1207. grid0.addWidget(self.gaps, 26, 1)
  1208. # Buttons
  1209. self.ff_cutout_object_btn = FCButton(_("Generate Freeform Geometry"))
  1210. self.ff_cutout_object_btn.setToolTip(
  1211. _("Cutout the selected object.\n"
  1212. "The cutout shape can be of any shape.\n"
  1213. "Useful when the PCB has a non-rectangular shape.")
  1214. )
  1215. self.ff_cutout_object_btn.setStyleSheet("""
  1216. QPushButton
  1217. {
  1218. font-weight: bold;
  1219. }
  1220. """)
  1221. grid0.addWidget(self.ff_cutout_object_btn, 28, 0, 1, 2)
  1222. self.rect_cutout_object_btn = FCButton(_("Generate Rectangular Geometry"))
  1223. self.rect_cutout_object_btn.setToolTip(
  1224. _("Cutout the selected object.\n"
  1225. "The resulting cutout shape is\n"
  1226. "always a rectangle shape and it will be\n"
  1227. "the bounding box of the Object.")
  1228. )
  1229. self.rect_cutout_object_btn.setStyleSheet("""
  1230. QPushButton
  1231. {
  1232. font-weight: bold;
  1233. }
  1234. """)
  1235. grid0.addWidget(self.rect_cutout_object_btn, 30, 0, 1, 2)
  1236. separator_line = QtWidgets.QFrame()
  1237. separator_line.setFrameShape(QtWidgets.QFrame.HLine)
  1238. separator_line.setFrameShadow(QtWidgets.QFrame.Sunken)
  1239. grid0.addWidget(separator_line, 32, 0, 1, 2)
  1240. grid0.addWidget(QtWidgets.QLabel(''), 34, 0, 1, 2)
  1241. # MANUAL BRIDGE GAPS
  1242. title_manual_label = QtWidgets.QLabel("<b>%s %s</b>:" % (_('Manual'), _("Bridge Gaps")))
  1243. title_manual_label.setToolTip(
  1244. _("This section handle creation of manual bridge gaps.\n"
  1245. "This is done by mouse clicking on the perimeter of the\n"
  1246. "Geometry object that is used as a cutout object. ")
  1247. )
  1248. grid0.addWidget(title_manual_label, 36, 0, 1, 2)
  1249. # Big Cursor
  1250. big_cursor_label = QtWidgets.QLabel('%s:' % _("Big cursor"))
  1251. big_cursor_label.setToolTip(
  1252. _("Use a big cursor when adding manual gaps."))
  1253. self.big_cursor_cb = FCCheckBox()
  1254. grid0.addWidget(big_cursor_label, 38, 0)
  1255. grid0.addWidget(self.big_cursor_cb, 38, 1)
  1256. # Generate a surrounding Geometry object
  1257. self.man_geo_creation_btn = FCButton(_("Generate Manual Geometry"))
  1258. self.man_geo_creation_btn.setToolTip(
  1259. _("If the object to be cutout is a Gerber\n"
  1260. "first create a Geometry that surrounds it,\n"
  1261. "to be used as the cutout, if one doesn't exist yet.\n"
  1262. "Select the source Gerber file in the top object combobox.")
  1263. )
  1264. # self.man_geo_creation_btn.setStyleSheet("""
  1265. # QPushButton
  1266. # {
  1267. # font-weight: bold;
  1268. # }
  1269. # """)
  1270. grid0.addWidget(self.man_geo_creation_btn, 40, 0, 1, 2)
  1271. # Manual Geo Object
  1272. self.man_object_combo = FCComboBox()
  1273. self.man_object_combo.setModel(self.app.collection)
  1274. self.man_object_combo.setRootModelIndex(self.app.collection.index(2, 0, QtCore.QModelIndex()))
  1275. self.man_object_combo.is_last = True
  1276. self.man_object_combo.obj_type = "Geometry"
  1277. self.man_object_label = QtWidgets.QLabel('%s:' % _("Manual cutout Geometry"))
  1278. self.man_object_label.setToolTip(
  1279. _("Geometry object used to create the manual cutout.")
  1280. )
  1281. # self.man_object_label.setMinimumWidth(60)
  1282. grid0.addWidget(self.man_object_label, 42, 0, 1, 2)
  1283. grid0.addWidget(self.man_object_combo, 44, 0, 1, 2)
  1284. self.man_gaps_creation_btn = FCButton(_("Manual Add Bridge Gaps"))
  1285. self.man_gaps_creation_btn.setToolTip(
  1286. _("Use the left mouse button (LMB) click\n"
  1287. "to create a bridge gap to separate the PCB from\n"
  1288. "the surrounding material.\n"
  1289. "The LMB click has to be done on the perimeter of\n"
  1290. "the Geometry object used as a cutout geometry.")
  1291. )
  1292. self.man_gaps_creation_btn.setStyleSheet("""
  1293. QPushButton
  1294. {
  1295. font-weight: bold;
  1296. }
  1297. """)
  1298. grid0.addWidget(self.man_gaps_creation_btn, 46, 0, 1, 2)
  1299. self.layout.addStretch()
  1300. # ## Reset Tool
  1301. self.reset_button = FCButton(_("Reset Tool"))
  1302. self.reset_button.setIcon(QtGui.QIcon(self.app.resource_location + '/reset32.png'))
  1303. self.reset_button.setToolTip(
  1304. _("Will reset the tool parameters.")
  1305. )
  1306. self.reset_button.setStyleSheet("""
  1307. QPushButton
  1308. {
  1309. font-weight: bold;
  1310. }
  1311. """)
  1312. self.layout.addWidget(self.reset_button)
  1313. # ############################ FINSIHED GUI ###################################
  1314. # #############################################################################
  1315. def confirmation_message(self, accepted, minval, maxval):
  1316. if accepted is False:
  1317. self.app.inform[str, bool].emit('[WARNING_NOTCL] %s: [%.*f, %.*f]' % (_("Edited value is out of range"),
  1318. self.decimals,
  1319. minval,
  1320. self.decimals,
  1321. maxval), False)
  1322. else:
  1323. self.app.inform[str, bool].emit('[success] %s' % _("Edited value is within limits."), False)
  1324. def confirmation_message_int(self, accepted, minval, maxval):
  1325. if accepted is False:
  1326. self.app.inform[str, bool].emit('[WARNING_NOTCL] %s: [%d, %d]' %
  1327. (_("Edited value is out of range"), minval, maxval), False)
  1328. else:
  1329. self.app.inform[str, bool].emit('[success] %s' % _("Edited value is within limits."), False)