ToolCutOut.py 66 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566
  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
  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. geo_obj.solid_geometry = deepcopy(solid_geo)
  335. xmin, ymin, xmax, ymax = CutOut.recursive_bounds(geo_obj.solid_geometry)
  336. geo_obj.options['xmin'] = xmin
  337. geo_obj.options['ymin'] = ymin
  338. geo_obj.options['xmax'] = xmax
  339. geo_obj.options['ymax'] = ymax
  340. geo_obj.options['cnctooldia'] = str(dia)
  341. geo_obj.options['cutz'] = self.ui.cutz_entry.get_value()
  342. geo_obj.options['multidepth'] = self.ui.mpass_cb.get_value()
  343. geo_obj.options['depthperpass'] = self.ui.maxdepth_entry.get_value()
  344. geo_obj.tools.update({
  345. 1: {
  346. 'tooldia': str(dia),
  347. 'offset': 'Path',
  348. 'offset_value': 0.0,
  349. 'type': _('Rough'),
  350. 'tool_type': 'C1',
  351. 'data': self.default_data,
  352. 'solid_geometry': geo_obj.solid_geometry
  353. }
  354. })
  355. geo_obj.multigeo = True
  356. geo_obj.tools[1]['data']['name'] = outname
  357. geo_obj.tools[1]['data']['cutz'] = self.ui.cutz_entry.get_value()
  358. geo_obj.tools[1]['data']['multidepth'] = self.ui.mpass_cb.get_value()
  359. geo_obj.tools[1]['data']['depthperpass'] = self.ui.maxdepth_entry.get_value()
  360. outname = cutout_obj.options["name"] + "_cutout"
  361. ret = self.app.app_obj.new_object('geometry', outname, geo_init)
  362. if ret == 'fail':
  363. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Failed."))
  364. return
  365. cutout_obj.plot()
  366. self.app.inform.emit('[success] %s' % _("Any form CutOut operation finished."))
  367. # self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
  368. self.app.should_we_save = True
  369. def on_rectangular_cutout(self):
  370. log.debug("Cutout.on_rectangular_cutout() was launched ...")
  371. # def subtract_rectangle(obj_, x0, y0, x1, y1):
  372. # pts = [(x0, y0), (x1, y0), (x1, y1), (x0, y1)]
  373. # obj_.subtract_polygon(pts)
  374. name = self.ui.obj_combo.currentText()
  375. # Get source object.
  376. try:
  377. cutout_obj = self.app.collection.get_by_name(str(name))
  378. except Exception as e:
  379. log.debug("CutOut.on_rectangular_cutout() --> %s" % str(e))
  380. self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve object"), name))
  381. return "Could not retrieve object: %s" % name
  382. if cutout_obj is None:
  383. self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Object not found"), str(name)))
  384. dia = float(self.ui.dia.get_value())
  385. if 0 in {dia}:
  386. self.app.inform.emit('[ERROR_NOTCL] %s' %
  387. _("Tool Diameter is zero value. Change it to a positive real number."))
  388. return "Tool Diameter is zero value. Change it to a positive real number."
  389. try:
  390. kind = self.ui.obj_kind_combo.get_value()
  391. except ValueError:
  392. return
  393. margin = float(self.ui.margin.get_value())
  394. gapsize = float(self.ui.gapsize.get_value())
  395. try:
  396. gaps = self.ui.gaps.get_value()
  397. except TypeError:
  398. self.app.inform.emit('[WARNING_NOTCL] %s' %
  399. _("Number of gaps value is missing. Add it and retry."))
  400. return
  401. if gaps not in ['None', 'LR', 'TB', '2LR', '2TB', '4', '8']:
  402. self.app.inform.emit('[WARNING_NOTCL] %s' % _("Gaps value can be only one of: "
  403. "'None', 'lr', 'tb', '2lr', '2tb', 4 or 8. "
  404. "Fill in a correct value and retry. "))
  405. return
  406. if cutout_obj.multigeo is True:
  407. self.app.inform.emit('[ERROR] %s' % _("Cutout operation cannot be done on a multi-geo Geometry.\n"
  408. "Optionally, this Multi-geo Geometry can be converted to "
  409. "Single-geo Geometry,\n"
  410. "and after that perform Cutout."))
  411. return
  412. # Get min and max data for each object as we just cut rectangles across X or Y
  413. gapsize = gapsize / 2 + (dia / 2)
  414. def geo_init(geo_obj, app_obj):
  415. solid_geo = []
  416. object_geo = cutout_obj.solid_geometry
  417. def cutout_rect_handler(geom):
  418. proc_geometry = []
  419. px = 0.5 * (xmin + xmax) + margin
  420. py = 0.5 * (ymin + ymax) + margin
  421. lenx = (xmax - xmin) + (margin * 2)
  422. leny = (ymax - ymin) + (margin * 2)
  423. if gaps == 'None':
  424. pass
  425. else:
  426. if gaps == '8' or gaps == '2LR':
  427. geom = self.subtract_poly_from_geo(geom,
  428. xmin - gapsize, # botleft_x
  429. py - gapsize + leny / 4, # botleft_y
  430. xmax + gapsize, # topright_x
  431. py + gapsize + leny / 4) # topright_y
  432. geom = self.subtract_poly_from_geo(geom,
  433. xmin - gapsize,
  434. py - gapsize - leny / 4,
  435. xmax + gapsize,
  436. py + gapsize - leny / 4)
  437. if gaps == '8' or gaps == '2TB':
  438. geom = self.subtract_poly_from_geo(geom,
  439. px - gapsize + lenx / 4,
  440. ymin - gapsize,
  441. px + gapsize + lenx / 4,
  442. ymax + gapsize)
  443. geom = self.subtract_poly_from_geo(geom,
  444. px - gapsize - lenx / 4,
  445. ymin - gapsize,
  446. px + gapsize - lenx / 4,
  447. ymax + gapsize)
  448. if gaps == '4' or gaps == 'LR':
  449. geom = self.subtract_poly_from_geo(geom,
  450. xmin - gapsize,
  451. py - gapsize,
  452. xmax + gapsize,
  453. py + gapsize)
  454. if gaps == '4' or gaps == 'TB':
  455. geom = self.subtract_poly_from_geo(geom,
  456. px - gapsize,
  457. ymin - gapsize,
  458. px + gapsize,
  459. ymax + gapsize)
  460. try:
  461. for g in geom:
  462. proc_geometry.append(g)
  463. except TypeError:
  464. proc_geometry.append(geom)
  465. return proc_geometry
  466. if kind == 'single':
  467. # fuse the lines
  468. object_geo = unary_union(object_geo)
  469. xmin, ymin, xmax, ymax = object_geo.bounds
  470. geo = box(xmin, ymin, xmax, ymax)
  471. # if Gerber create a buffer at a distance
  472. # if Geometry then cut through the geometry
  473. if cutout_obj.kind == 'gerber':
  474. if margin >= 0:
  475. geo = geo.buffer(margin + abs(dia / 2))
  476. else:
  477. geo = geo.buffer(margin - abs(dia / 2))
  478. solid_geo = cutout_rect_handler(geom=geo)
  479. else:
  480. if cutout_obj.kind == 'geometry':
  481. try:
  482. __ = iter(object_geo)
  483. except TypeError:
  484. object_geo = [object_geo]
  485. for geom_struct in object_geo:
  486. geom_struct = unary_union(geom_struct)
  487. xmin, ymin, xmax, ymax = geom_struct.bounds
  488. geom_struct = box(xmin, ymin, xmax, ymax)
  489. solid_geo += cutout_rect_handler(geom=geom_struct)
  490. elif cutout_obj.kind == 'gerber' and margin >= 0:
  491. try:
  492. __ = iter(object_geo)
  493. except TypeError:
  494. object_geo = [object_geo]
  495. for geom_struct in object_geo:
  496. geom_struct = unary_union(geom_struct)
  497. xmin, ymin, xmax, ymax = geom_struct.bounds
  498. geom_struct = box(xmin, ymin, xmax, ymax)
  499. geom_struct = geom_struct.buffer(margin + abs(dia / 2))
  500. solid_geo += cutout_rect_handler(geom=geom_struct)
  501. elif cutout_obj.kind == 'gerber' and margin < 0:
  502. app_obj.inform.emit(
  503. '[WARNING_NOTCL] %s' % _("Rectangular cutout with negative margin is not possible."))
  504. return "fail"
  505. geo_obj.options['cnctooldia'] = str(dia)
  506. geo_obj.options['cutz'] = self.ui.cutz_entry.get_value()
  507. geo_obj.options['multidepth'] = self.ui.mpass_cb.get_value()
  508. geo_obj.options['depthperpass'] = self.ui.maxdepth_entry.get_value()
  509. geo_obj.solid_geometry = deepcopy(solid_geo)
  510. geo_obj.tools.update({
  511. 1: {
  512. 'tooldia': str(dia),
  513. 'offset': 'Path',
  514. 'offset_value': 0.0,
  515. 'type': _('Rough'),
  516. 'tool_type': 'C1',
  517. 'data': self.default_data,
  518. 'solid_geometry': geo_obj.solid_geometry
  519. }
  520. })
  521. geo_obj.multigeo = True
  522. geo_obj.tools[1]['data']['name'] = outname
  523. geo_obj.tools[1]['data']['cutz'] = self.ui.cutz_entry.get_value()
  524. geo_obj.tools[1]['data']['multidepth'] = self.ui.mpass_cb.get_value()
  525. geo_obj.tools[1]['data']['depthperpass'] = self.ui.maxdepth_entry.get_value()
  526. outname = cutout_obj.options["name"] + "_cutout"
  527. ret = self.app.app_obj.new_object('geometry', outname, geo_init)
  528. if ret != 'fail':
  529. # cutout_obj.plot()
  530. self.app.inform.emit('[success] %s' % _("Any form CutOut operation finished."))
  531. else:
  532. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Failed."))
  533. return
  534. # self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
  535. self.app.should_we_save = True
  536. def on_manual_gap_click(self):
  537. name = self.ui.man_object_combo.currentText()
  538. # Get source object.
  539. try:
  540. self.man_cutout_obj = self.app.collection.get_by_name(str(name))
  541. except Exception as e:
  542. log.debug("CutOut.on_manual_cutout() --> %s" % str(e))
  543. self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve Geometry object"), name))
  544. return
  545. if self.man_cutout_obj is None:
  546. self.app.inform.emit('[ERROR_NOTCL] %s: %s' %
  547. (_("Geometry object for manual cutout not found"), self.man_cutout_obj))
  548. return
  549. self.app.inform.emit(_("Click on the selected geometry object perimeter to create a bridge gap ..."))
  550. self.app.geo_editor.tool_shape.enabled = True
  551. self.cutting_dia = float(self.ui.dia.get_value())
  552. if 0 in {self.cutting_dia}:
  553. self.app.inform.emit('[ERROR_NOTCL] %s' %
  554. _("Tool Diameter is zero value. Change it to a positive real number."))
  555. return
  556. self.cutting_gapsize = float(self.ui.gapsize.get_value())
  557. name = self.ui.man_object_combo.currentText()
  558. # Get Geometry source object to be used as target for Manual adding Gaps
  559. try:
  560. self.man_cutout_obj = self.app.collection.get_by_name(str(name))
  561. except Exception as e:
  562. log.debug("CutOut.on_manual_cutout() --> %s" % str(e))
  563. self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve Geometry object"), name))
  564. return
  565. if self.app.is_legacy is False:
  566. self.app.plotcanvas.graph_event_disconnect('key_press', self.app.ui.keyPressEvent)
  567. self.app.plotcanvas.graph_event_disconnect('mouse_press', self.app.on_mouse_click_over_plot)
  568. self.app.plotcanvas.graph_event_disconnect('mouse_release', self.app.on_mouse_click_release_over_plot)
  569. self.app.plotcanvas.graph_event_disconnect('mouse_move', self.app.on_mouse_move_over_plot)
  570. else:
  571. self.app.plotcanvas.graph_event_disconnect(self.app.kp)
  572. self.app.plotcanvas.graph_event_disconnect(self.app.mp)
  573. self.app.plotcanvas.graph_event_disconnect(self.app.mr)
  574. self.app.plotcanvas.graph_event_disconnect(self.app.mm)
  575. self.kp = self.app.plotcanvas.graph_event_connect('key_press', self.on_key_press)
  576. self.mm = self.app.plotcanvas.graph_event_connect('mouse_move', self.on_mouse_move)
  577. self.mr = self.app.plotcanvas.graph_event_connect('mouse_release', self.on_mouse_click_release)
  578. self.mouse_events_connected = True
  579. if self.ui.big_cursor_cb.get_value():
  580. self.old_cursor_type = self.app.defaults["global_cursor_type"]
  581. self.app.on_cursor_type(val="big")
  582. self.app.defaults['global_selection_shape'] = False
  583. def on_manual_cutout(self, click_pos):
  584. if self.man_cutout_obj is None:
  585. self.app.inform.emit('[ERROR_NOTCL] %s: %s' %
  586. (_("Geometry object for manual cutout not found"), self.man_cutout_obj))
  587. return
  588. # use the snapped position as reference
  589. snapped_pos = self.app.geo_editor.snap(click_pos[0], click_pos[1])
  590. cut_poly = self.cutting_geo(pos=(snapped_pos[0], snapped_pos[1]))
  591. # first subtract geometry for the total solid_geometry
  592. new_solid_geometry = CutOut.subtract_polygon(self.man_cutout_obj.solid_geometry, cut_poly)
  593. self.man_cutout_obj.solid_geometry = new_solid_geometry
  594. # then do it or each tool in the manual cutout Geometry object
  595. for tool in self.man_cutout_obj.tools:
  596. self.man_cutout_obj.tools[tool]['solid_geometry'] = new_solid_geometry
  597. self.man_cutout_obj.plot()
  598. self.app.inform.emit('[success] %s' % _("Added manual Bridge Gap."))
  599. self.app.should_we_save = True
  600. def on_manual_geo(self):
  601. name = self.ui.obj_combo.currentText()
  602. # Get source object.
  603. try:
  604. cutout_obj = self.app.collection.get_by_name(str(name))
  605. except Exception as e:
  606. log.debug("CutOut.on_manual_geo() --> %s" % str(e))
  607. self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve Gerber object"), name))
  608. return "Could not retrieve object: %s" % name
  609. if cutout_obj is None:
  610. self.app.inform.emit('[ERROR_NOTCL] %s' %
  611. _("There is no Gerber object selected for Cutout.\n"
  612. "Select one and try again."))
  613. return
  614. if cutout_obj.kind != 'gerber':
  615. self.app.inform.emit('[ERROR_NOTCL] %s' %
  616. _("The selected object has to be of Gerber type.\n"
  617. "Select a Gerber file and try again."))
  618. return
  619. dia = float(self.ui.dia.get_value())
  620. if 0 in {dia}:
  621. self.app.inform.emit('[ERROR_NOTCL] %s' %
  622. _("Tool Diameter is zero value. Change it to a positive real number."))
  623. return
  624. try:
  625. kind = self.ui.obj_kind_combo.get_value()
  626. except ValueError:
  627. return
  628. margin = float(self.ui.margin.get_value())
  629. convex_box = self.ui.convex_box.get_value()
  630. def geo_init(geo_obj, app_obj):
  631. geo_union = unary_union(cutout_obj.solid_geometry)
  632. if convex_box:
  633. geo = geo_union.convex_hull
  634. geo_obj.solid_geometry = geo.buffer(margin + abs(dia / 2))
  635. elif kind == 'single':
  636. if isinstance(geo_union, Polygon) or \
  637. (isinstance(geo_union, list) and len(geo_union) == 1) or \
  638. (isinstance(geo_union, MultiPolygon) and len(geo_union) == 1):
  639. geo_obj.solid_geometry = geo_union.buffer(margin + abs(dia / 2)).exterior
  640. elif isinstance(geo_union, MultiPolygon):
  641. x0, y0, x1, y1 = geo_union.bounds
  642. geo = box(x0, y0, x1, y1)
  643. geo_obj.solid_geometry = geo.buffer(margin + abs(dia / 2))
  644. else:
  645. app_obj.inform.emit('[ERROR_NOTCL] %s: %s' % (
  646. _("Geometry not supported for cutout"), type(geo_union)))
  647. return 'fail'
  648. else:
  649. geo = geo_union
  650. geo = geo.buffer(margin + abs(dia / 2))
  651. if isinstance(geo, Polygon):
  652. geo_obj.solid_geometry = geo.exterior
  653. elif isinstance(geo, MultiPolygon):
  654. solid_geo = []
  655. for poly in geo:
  656. solid_geo.append(poly.exterior)
  657. geo_obj.solid_geometry = deepcopy(solid_geo)
  658. geo_obj.options['cnctooldia'] = str(dia)
  659. geo_obj.options['cutz'] = self.ui.cutz_entry.get_value()
  660. geo_obj.options['multidepth'] = self.ui.mpass_cb.get_value()
  661. geo_obj.options['depthperpass'] = self.ui.maxdepth_entry.get_value()
  662. geo_obj.tools.update({
  663. 1: {
  664. 'tooldia': str(dia),
  665. 'offset': 'Path',
  666. 'offset_value': 0.0,
  667. 'type': _('Rough'),
  668. 'tool_type': 'C1',
  669. 'data': self.default_data,
  670. 'solid_geometry': geo_obj.solid_geometry
  671. }
  672. })
  673. geo_obj.multigeo = True
  674. geo_obj.tools[1]['data']['name'] = outname
  675. geo_obj.tools[1]['data']['cutz'] = self.ui.cutz_entry.get_value()
  676. geo_obj.tools[1]['data']['multidepth'] = self.ui.mpass_cb.get_value()
  677. geo_obj.tools[1]['data']['depthperpass'] = self.ui.maxdepth_entry.get_value()
  678. outname = cutout_obj.options["name"] + "_cutout"
  679. self.app.app_obj.new_object('geometry', outname, geo_init)
  680. def cutting_geo(self, pos):
  681. self.cutting_dia = float(self.ui.dia.get_value())
  682. self.cutting_gapsize = float(self.ui.gapsize.get_value())
  683. offset = self.cutting_dia / 2 + self.cutting_gapsize / 2
  684. # cutting area definition
  685. orig_x = pos[0]
  686. orig_y = pos[1]
  687. xmin = orig_x - offset
  688. ymin = orig_y - offset
  689. xmax = orig_x + offset
  690. ymax = orig_y + offset
  691. cut_poly = box(xmin, ymin, xmax, ymax)
  692. return cut_poly
  693. # To be called after clicking on the plot.
  694. def on_mouse_click_release(self, event):
  695. if self.app.is_legacy is False:
  696. event_pos = event.pos
  697. # event_is_dragging = event.is_dragging
  698. right_button = 2
  699. else:
  700. event_pos = (event.xdata, event.ydata)
  701. # event_is_dragging = self.app.plotcanvas.is_dragging
  702. right_button = 3
  703. try:
  704. x = float(event_pos[0])
  705. y = float(event_pos[1])
  706. except TypeError:
  707. return
  708. event_pos = (x, y)
  709. # do paint single only for left mouse clicks
  710. if event.button == 1:
  711. self.app.inform.emit(_("Making manual bridge gap..."))
  712. pos = self.app.plotcanvas.translate_coords(event_pos)
  713. self.on_manual_cutout(click_pos=pos)
  714. # if RMB then we exit
  715. elif event.button == right_button and self.mouse_is_dragging is False:
  716. if self.app.is_legacy is False:
  717. self.app.plotcanvas.graph_event_disconnect('key_press', self.on_key_press)
  718. self.app.plotcanvas.graph_event_disconnect('mouse_move', self.on_mouse_move)
  719. self.app.plotcanvas.graph_event_disconnect('mouse_release', self.on_mouse_click_release)
  720. else:
  721. self.app.plotcanvas.graph_event_disconnect(self.kp)
  722. self.app.plotcanvas.graph_event_disconnect(self.mm)
  723. self.app.plotcanvas.graph_event_disconnect(self.mr)
  724. self.app.kp = self.app.plotcanvas.graph_event_connect('key_press', self.app.ui.keyPressEvent)
  725. self.app.mp = self.app.plotcanvas.graph_event_connect('mouse_press', self.app.on_mouse_click_over_plot)
  726. self.app.mr = self.app.plotcanvas.graph_event_connect('mouse_release',
  727. self.app.on_mouse_click_release_over_plot)
  728. self.app.mm = self.app.plotcanvas.graph_event_connect('mouse_move', self.app.on_mouse_move_over_plot)
  729. # Remove any previous utility shape
  730. self.app.geo_editor.tool_shape.clear(update=True)
  731. self.app.geo_editor.tool_shape.enabled = False
  732. # signal that the mouse events are disconnected from local methods
  733. self.mouse_events_connected = False
  734. if self.ui.big_cursor_cb.get_value():
  735. # restore cursor
  736. self.app.on_cursor_type(val=self.old_cursor_type)
  737. # restore selection
  738. self.app.defaults['global_selection_shape'] = self.old_selection_state
  739. def on_mouse_move(self, event):
  740. self.app.on_mouse_move_over_plot(event=event)
  741. if self.app.is_legacy is False:
  742. event_pos = event.pos
  743. event_is_dragging = event.is_dragging
  744. # right_button = 2
  745. else:
  746. event_pos = (event.xdata, event.ydata)
  747. event_is_dragging = self.app.plotcanvas.is_dragging
  748. # right_button = 3
  749. try:
  750. x = float(event_pos[0])
  751. y = float(event_pos[1])
  752. except TypeError:
  753. return
  754. event_pos = (x, y)
  755. pos = self.canvas.translate_coords(event_pos)
  756. event.xdata, event.ydata = pos[0], pos[1]
  757. if event_is_dragging is True:
  758. self.mouse_is_dragging = True
  759. else:
  760. self.mouse_is_dragging = False
  761. try:
  762. x = float(event.xdata)
  763. y = float(event.ydata)
  764. except TypeError:
  765. return
  766. if self.app.grid_status():
  767. snap_x, snap_y = self.app.geo_editor.snap(x, y)
  768. else:
  769. snap_x, snap_y = x, y
  770. self.x_pos, self.y_pos = snap_x, snap_y
  771. # #################################################
  772. # ### This section makes the cutting geo to #######
  773. # ### rotate if it intersects the target geo ######
  774. # #################################################
  775. cut_geo = self.cutting_geo(pos=(snap_x, snap_y))
  776. man_geo = self.man_cutout_obj.solid_geometry
  777. def get_angle(geo):
  778. line = cut_geo.intersection(geo)
  779. try:
  780. pt1_x = line.coords[0][0]
  781. pt1_y = line.coords[0][1]
  782. pt2_x = line.coords[1][0]
  783. pt2_y = line.coords[1][1]
  784. dx = pt1_x - pt2_x
  785. dy = pt1_y - pt2_y
  786. if dx == 0 or dy == 0:
  787. angle = 0
  788. else:
  789. radian = math.atan(dx / dy)
  790. angle = radian * 180 / math.pi
  791. except Exception:
  792. angle = 0
  793. return angle
  794. try:
  795. rot_angle = 0
  796. for geo_el in man_geo:
  797. if isinstance(geo_el, Polygon):
  798. work_geo = geo_el.exterior
  799. if cut_geo.intersects(work_geo):
  800. rot_angle = get_angle(geo=work_geo)
  801. else:
  802. rot_angle = 0
  803. else:
  804. rot_angle = 0
  805. if cut_geo.intersects(geo_el):
  806. rot_angle = get_angle(geo=geo_el)
  807. if rot_angle != 0:
  808. break
  809. except TypeError:
  810. if isinstance(man_geo, Polygon):
  811. work_geo = man_geo.exterior
  812. if cut_geo.intersects(work_geo):
  813. rot_angle = get_angle(geo=work_geo)
  814. else:
  815. rot_angle = 0
  816. else:
  817. rot_angle = 0
  818. if cut_geo.intersects(man_geo):
  819. rot_angle = get_angle(geo=man_geo)
  820. # rotate only if there is an angle to rotate to
  821. if rot_angle != 0:
  822. cut_geo = affinity.rotate(cut_geo, -rot_angle)
  823. # Remove any previous utility shape
  824. self.app.geo_editor.tool_shape.clear(update=True)
  825. self.draw_utility_geometry(geo=cut_geo)
  826. def draw_utility_geometry(self, geo):
  827. self.app.geo_editor.tool_shape.add(
  828. shape=geo,
  829. color=(self.app.defaults["global_draw_color"] + '80'),
  830. update=False,
  831. layer=0,
  832. tolerance=None)
  833. self.app.geo_editor.tool_shape.redraw()
  834. def on_key_press(self, event):
  835. # events out of the self.app.collection view (it's about Project Tab) are of type int
  836. if type(event) is int:
  837. key = event
  838. # events from the GUI are of type QKeyEvent
  839. elif type(event) == QtGui.QKeyEvent:
  840. key = event.key()
  841. elif isinstance(event, mpl_key_event): # MatPlotLib key events are trickier to interpret than the rest
  842. key = event.key
  843. key = QtGui.QKeySequence(key)
  844. # check for modifiers
  845. key_string = key.toString().lower()
  846. if '+' in key_string:
  847. mod, __, key_text = key_string.rpartition('+')
  848. if mod.lower() == 'ctrl':
  849. # modifiers = QtCore.Qt.ControlModifier
  850. pass
  851. elif mod.lower() == 'alt':
  852. # modifiers = QtCore.Qt.AltModifier
  853. pass
  854. elif mod.lower() == 'shift':
  855. # modifiers = QtCore.Qt.ShiftModifier
  856. pass
  857. else:
  858. # modifiers = QtCore.Qt.NoModifier
  859. pass
  860. key = QtGui.QKeySequence(key_text)
  861. # events from Vispy are of type KeyEvent
  862. else:
  863. key = event.key
  864. # Escape = Deselect All
  865. if key == QtCore.Qt.Key_Escape or key == 'Escape':
  866. if self.mouse_events_connected is True:
  867. self.mouse_events_connected = False
  868. if self.app.is_legacy is False:
  869. self.app.plotcanvas.graph_event_disconnect('key_press', self.on_key_press)
  870. self.app.plotcanvas.graph_event_disconnect('mouse_move', self.on_mouse_move)
  871. self.app.plotcanvas.graph_event_disconnect('mouse_release', self.on_mouse_click_release)
  872. else:
  873. self.app.plotcanvas.graph_event_disconnect(self.kp)
  874. self.app.plotcanvas.graph_event_disconnect(self.mm)
  875. self.app.plotcanvas.graph_event_disconnect(self.mr)
  876. self.app.kp = self.app.plotcanvas.graph_event_connect('key_press', self.app.ui.keyPressEvent)
  877. self.app.mp = self.app.plotcanvas.graph_event_connect('mouse_press', self.app.on_mouse_click_over_plot)
  878. self.app.mr = self.app.plotcanvas.graph_event_connect('mouse_release',
  879. self.app.on_mouse_click_release_over_plot)
  880. self.app.mm = self.app.plotcanvas.graph_event_connect('mouse_move', self.app.on_mouse_move_over_plot)
  881. if self.ui.big_cursor_cb.get_value():
  882. # restore cursor
  883. self.app.on_cursor_type(val=self.old_cursor_type)
  884. # restore selection
  885. self.app.defaults['global_selection_shape'] = self.old_selection_state
  886. # Remove any previous utility shape
  887. self.app.geo_editor.tool_shape.clear(update=True)
  888. self.app.geo_editor.tool_shape.enabled = False
  889. # Grid toggle
  890. if key == QtCore.Qt.Key_G or key == 'G':
  891. self.app.ui.grid_snap_btn.trigger()
  892. # Jump to coords
  893. if key == QtCore.Qt.Key_J or key == 'J':
  894. l_x, l_y = self.app.on_jump_to()
  895. self.app.geo_editor.tool_shape.clear(update=True)
  896. geo = self.cutting_geo(pos=(l_x, l_y))
  897. self.draw_utility_geometry(geo=geo)
  898. @staticmethod
  899. def subtract_poly_from_geo(solid_geo, x0, y0, x1, y1):
  900. """
  901. Subtract polygon made from points from the given object.
  902. This only operates on the paths in the original geometry,
  903. i.e. it converts polygons into paths.
  904. :param x0: x coord for lower left vertex of the polygon.
  905. :param y0: y coord for lower left vertex of the polygon.
  906. :param x1: x coord for upper right vertex of the polygon.
  907. :param y1: y coord for upper right vertex of the polygon.
  908. :param solid_geo: Geometry from which to substract. If none, use the solid_geomety property of the object
  909. :return: none
  910. """
  911. points = [(x0, y0), (x1, y0), (x1, y1), (x0, y1)]
  912. # pathonly should be allways True, otherwise polygons are not subtracted
  913. flat_geometry = CutOut.flatten(geometry=solid_geo)
  914. log.debug("%d paths" % len(flat_geometry))
  915. polygon = Polygon(points)
  916. toolgeo = cascaded_union(polygon)
  917. diffs = []
  918. for target in flat_geometry:
  919. if type(target) == LineString or type(target) == LinearRing:
  920. diffs.append(target.difference(toolgeo))
  921. else:
  922. log.warning("Not implemented.")
  923. return unary_union(diffs)
  924. @staticmethod
  925. def flatten(geometry):
  926. """
  927. Creates a list of non-iterable linear geometry objects.
  928. Polygons are expanded into its exterior and interiors.
  929. Results are placed in self.flat_geometry
  930. :param geometry: Shapely type or list or list of list of such.
  931. """
  932. flat_geo = []
  933. try:
  934. for geo in geometry:
  935. if geo and not geo.is_empty:
  936. flat_geo += CutOut.flatten(geometry=geo)
  937. except TypeError:
  938. if isinstance(geometry, Polygon):
  939. flat_geo.append(geometry.exterior)
  940. CutOut.flatten(geometry=geometry.interiors)
  941. else:
  942. flat_geo.append(geometry)
  943. return flat_geo
  944. @staticmethod
  945. def recursive_bounds(geometry):
  946. """
  947. Return the bounds of the biggest bounding box in geometry, one that include all.
  948. :param geometry: a iterable object that holds geometry
  949. :return: Returns coordinates of rectangular bounds of geometry: (xmin, ymin, xmax, ymax).
  950. """
  951. # now it can get bounds for nested lists of objects
  952. def bounds_rec(obj):
  953. try:
  954. minx = Inf
  955. miny = Inf
  956. maxx = -Inf
  957. maxy = -Inf
  958. for k in obj:
  959. minx_, miny_, maxx_, maxy_ = bounds_rec(k)
  960. minx = min(minx, minx_)
  961. miny = min(miny, miny_)
  962. maxx = max(maxx, maxx_)
  963. maxy = max(maxy, maxy_)
  964. return minx, miny, maxx, maxy
  965. except TypeError:
  966. # it's a Shapely object, return it's bounds
  967. if obj:
  968. return obj.bounds
  969. return bounds_rec(geometry)
  970. @staticmethod
  971. def subtract_polygon(target_geo, subtractor):
  972. """
  973. Subtract subtractor polygon from the target_geo. This only operates on the paths in the target_geo,
  974. i.e. it converts polygons into paths.
  975. :param target_geo: geometry from which to subtract
  976. :param subtractor: a list of Points, a LinearRing or a Polygon that will be subtracted from target_geo
  977. :return: a cascaded union of the resulting geometry
  978. """
  979. if target_geo is None:
  980. target_geo = []
  981. # flatten() takes care of possible empty geometry making sure that is filtered
  982. flat_geometry = CutOut.flatten(target_geo)
  983. log.debug("%d paths" % len(flat_geometry))
  984. toolgeo = unary_union(subtractor)
  985. diffs = []
  986. for target in flat_geometry:
  987. if isinstance(target, LineString) or isinstance(target, LinearRing) or isinstance(target, MultiLineString):
  988. diffs.append(target.difference(toolgeo))
  989. else:
  990. log.warning("Not implemented.")
  991. return unary_union(diffs)
  992. def reset_fields(self):
  993. self.ui.obj_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  994. class CutoutUI:
  995. toolName = _("Cutout PCB")
  996. def __init__(self, layout, app):
  997. self.app = app
  998. self.decimals = self.app.decimals
  999. self.layout = layout
  1000. # Title
  1001. title_label = QtWidgets.QLabel("%s" % self.toolName)
  1002. title_label.setStyleSheet("""
  1003. QLabel
  1004. {
  1005. font-size: 16px;
  1006. font-weight: bold;
  1007. }
  1008. """)
  1009. self.layout.addWidget(title_label)
  1010. self.layout.addWidget(QtWidgets.QLabel(''))
  1011. # Form Layout
  1012. grid0 = QtWidgets.QGridLayout()
  1013. grid0.setColumnStretch(0, 0)
  1014. grid0.setColumnStretch(1, 1)
  1015. self.layout.addLayout(grid0)
  1016. self.object_label = QtWidgets.QLabel('<b>%s:</b>' % _("Source Object"))
  1017. self.object_label.setToolTip('%s.' % _("Object to be cutout"))
  1018. grid0.addWidget(self.object_label, 0, 0, 1, 2)
  1019. # Object kind
  1020. self.kindlabel = QtWidgets.QLabel('%s:' % _('Kind'))
  1021. self.kindlabel.setToolTip(
  1022. _("Choice of what kind the object we want to cutout is.<BR>"
  1023. "- <B>Single</B>: contain a single PCB Gerber outline object.<BR>"
  1024. "- <B>Panel</B>: a panel PCB Gerber object, which is made\n"
  1025. "out of many individual PCB outlines.")
  1026. )
  1027. self.obj_kind_combo = RadioSet([
  1028. {"label": _("Single"), "value": "single"},
  1029. {"label": _("Panel"), "value": "panel"},
  1030. ])
  1031. grid0.addWidget(self.kindlabel, 1, 0)
  1032. grid0.addWidget(self.obj_kind_combo, 1, 1)
  1033. # Type of object to be cutout
  1034. self.type_obj_radio = RadioSet([
  1035. {"label": _("Gerber"), "value": "grb"},
  1036. {"label": _("Geometry"), "value": "geo"},
  1037. ])
  1038. self.type_obj_combo_label = QtWidgets.QLabel('%s:' % _("Type"))
  1039. self.type_obj_combo_label.setToolTip(
  1040. _("Specify the type of object to be cutout.\n"
  1041. "It can be of type: Gerber or Geometry.\n"
  1042. "What is selected here will dictate the kind\n"
  1043. "of objects that will populate the 'Object' combobox.")
  1044. )
  1045. grid0.addWidget(self.type_obj_combo_label, 2, 0)
  1046. grid0.addWidget(self.type_obj_radio, 2, 1)
  1047. # Object to be cutout
  1048. self.obj_combo = FCComboBox()
  1049. self.obj_combo.setModel(self.app.collection)
  1050. self.obj_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  1051. self.obj_combo.is_last = True
  1052. grid0.addWidget(self.obj_combo, 3, 0, 1, 2)
  1053. separator_line = QtWidgets.QFrame()
  1054. separator_line.setFrameShape(QtWidgets.QFrame.HLine)
  1055. separator_line.setFrameShadow(QtWidgets.QFrame.Sunken)
  1056. grid0.addWidget(separator_line, 4, 0, 1, 2)
  1057. grid0.addWidget(QtWidgets.QLabel(''), 5, 0, 1, 2)
  1058. self.param_label = QtWidgets.QLabel('<b>%s:</b>' % _("Tool Parameters"))
  1059. grid0.addWidget(self.param_label, 6, 0, 1, 2)
  1060. # Tool Diameter
  1061. self.dia = FCDoubleSpinner(callback=self.confirmation_message)
  1062. self.dia.set_precision(self.decimals)
  1063. self.dia.set_range(0.0000, 9999.9999)
  1064. self.dia_label = QtWidgets.QLabel('%s:' % _("Tool Diameter"))
  1065. self.dia_label.setToolTip(
  1066. _("Diameter of the tool used to cutout\n"
  1067. "the PCB shape out of the surrounding material.")
  1068. )
  1069. grid0.addWidget(self.dia_label, 8, 0)
  1070. grid0.addWidget(self.dia, 8, 1)
  1071. # Cut Z
  1072. cutzlabel = QtWidgets.QLabel('%s:' % _('Cut Z'))
  1073. cutzlabel.setToolTip(
  1074. _(
  1075. "Cutting depth (negative)\n"
  1076. "below the copper surface."
  1077. )
  1078. )
  1079. self.cutz_entry = FCDoubleSpinner(callback=self.confirmation_message)
  1080. self.cutz_entry.set_precision(self.decimals)
  1081. if machinist_setting == 0:
  1082. self.cutz_entry.setRange(-9999.9999, -0.00001)
  1083. else:
  1084. self.cutz_entry.setRange(-9999.9999, 9999.9999)
  1085. self.cutz_entry.setSingleStep(0.1)
  1086. grid0.addWidget(cutzlabel, 9, 0)
  1087. grid0.addWidget(self.cutz_entry, 9, 1)
  1088. # Multi-pass
  1089. self.mpass_cb = FCCheckBox('%s:' % _("Multi-Depth"))
  1090. self.mpass_cb.setToolTip(
  1091. _(
  1092. "Use multiple passes to limit\n"
  1093. "the cut depth in each pass. Will\n"
  1094. "cut multiple times until Cut Z is\n"
  1095. "reached."
  1096. )
  1097. )
  1098. self.maxdepth_entry = FCDoubleSpinner(callback=self.confirmation_message)
  1099. self.maxdepth_entry.set_precision(self.decimals)
  1100. self.maxdepth_entry.setRange(0, 9999.9999)
  1101. self.maxdepth_entry.setSingleStep(0.1)
  1102. self.maxdepth_entry.setToolTip(
  1103. _(
  1104. "Depth of each pass (positive)."
  1105. )
  1106. )
  1107. self.ois_mpass_geo = OptionalInputSection(self.mpass_cb, [self.maxdepth_entry])
  1108. grid0.addWidget(self.mpass_cb, 10, 0)
  1109. grid0.addWidget(self.maxdepth_entry, 10, 1)
  1110. # Margin
  1111. self.margin = FCDoubleSpinner(callback=self.confirmation_message)
  1112. self.margin.set_range(-9999.9999, 9999.9999)
  1113. self.margin.setSingleStep(0.1)
  1114. self.margin.set_precision(self.decimals)
  1115. self.margin_label = QtWidgets.QLabel('%s:' % _("Margin"))
  1116. self.margin_label.setToolTip(
  1117. _("Margin over bounds. A positive value here\n"
  1118. "will make the cutout of the PCB further from\n"
  1119. "the actual PCB border")
  1120. )
  1121. grid0.addWidget(self.margin_label, 11, 0)
  1122. grid0.addWidget(self.margin, 11, 1)
  1123. # Gapsize
  1124. self.gapsize = FCDoubleSpinner(callback=self.confirmation_message)
  1125. self.gapsize.set_precision(self.decimals)
  1126. self.gapsize_label = QtWidgets.QLabel('%s:' % _("Gap size"))
  1127. self.gapsize_label.setToolTip(
  1128. _("The size of the bridge gaps in the cutout\n"
  1129. "used to keep the board connected to\n"
  1130. "the surrounding material (the one \n"
  1131. "from which the PCB is cutout).")
  1132. )
  1133. grid0.addWidget(self.gapsize_label, 13, 0)
  1134. grid0.addWidget(self.gapsize, 13, 1)
  1135. # How gaps wil be rendered:
  1136. # lr - left + right
  1137. # tb - top + bottom
  1138. # 4 - left + right +top + bottom
  1139. # 2lr - 2*left + 2*right
  1140. # 2tb - 2*top + 2*bottom
  1141. # 8 - 2*left + 2*right +2*top + 2*bottom
  1142. # Surrounding convex box shape
  1143. self.convex_box = FCCheckBox('%s' % _("Convex Shape"))
  1144. # self.convex_box_label = QtWidgets.QLabel('%s' % _("Convex Sh."))
  1145. self.convex_box.setToolTip(
  1146. _("Create a convex shape surrounding the entire PCB.\n"
  1147. "Used only if the source object type is Gerber.")
  1148. )
  1149. grid0.addWidget(self.convex_box, 15, 0, 1, 2)
  1150. separator_line = QtWidgets.QFrame()
  1151. separator_line.setFrameShape(QtWidgets.QFrame.HLine)
  1152. separator_line.setFrameShadow(QtWidgets.QFrame.Sunken)
  1153. grid0.addWidget(separator_line, 16, 0, 1, 2)
  1154. grid0.addWidget(QtWidgets.QLabel(''), 17, 0, 1, 2)
  1155. # Title2
  1156. title_param_label = QtWidgets.QLabel("<b>%s %s</b>:" % (_('Automatic'), _("Bridge Gaps")))
  1157. title_param_label.setToolTip(
  1158. _("This section handle creation of automatic bridge gaps.")
  1159. )
  1160. grid0.addWidget(title_param_label, 18, 0, 1, 2)
  1161. # Gaps
  1162. gaps_label = QtWidgets.QLabel('%s:' % _('Gaps'))
  1163. gaps_label.setToolTip(
  1164. _("Number of gaps used for the Automatic cutout.\n"
  1165. "There can be maximum 8 bridges/gaps.\n"
  1166. "The choices are:\n"
  1167. "- None - no gaps\n"
  1168. "- lr - left + right\n"
  1169. "- tb - top + bottom\n"
  1170. "- 4 - left + right +top + bottom\n"
  1171. "- 2lr - 2*left + 2*right\n"
  1172. "- 2tb - 2*top + 2*bottom\n"
  1173. "- 8 - 2*left + 2*right +2*top + 2*bottom")
  1174. )
  1175. # gaps_label.setMinimumWidth(60)
  1176. self.gaps = FCComboBox()
  1177. gaps_items = ['None', 'LR', 'TB', '4', '2LR', '2TB', '8']
  1178. for it in gaps_items:
  1179. self.gaps.addItem(it)
  1180. # self.gaps.setStyleSheet('background-color: rgb(255,255,255)')
  1181. grid0.addWidget(gaps_label, 19, 0)
  1182. grid0.addWidget(self.gaps, 19, 1)
  1183. # Buttons
  1184. self.ff_cutout_object_btn = FCButton(_("Generate Freeform Geometry"))
  1185. self.ff_cutout_object_btn.setToolTip(
  1186. _("Cutout the selected object.\n"
  1187. "The cutout shape can be of any shape.\n"
  1188. "Useful when the PCB has a non-rectangular shape.")
  1189. )
  1190. self.ff_cutout_object_btn.setStyleSheet("""
  1191. QPushButton
  1192. {
  1193. font-weight: bold;
  1194. }
  1195. """)
  1196. grid0.addWidget(self.ff_cutout_object_btn, 20, 0, 1, 2)
  1197. self.rect_cutout_object_btn = FCButton(_("Generate Rectangular Geometry"))
  1198. self.rect_cutout_object_btn.setToolTip(
  1199. _("Cutout the selected object.\n"
  1200. "The resulting cutout shape is\n"
  1201. "always a rectangle shape and it will be\n"
  1202. "the bounding box of the Object.")
  1203. )
  1204. self.rect_cutout_object_btn.setStyleSheet("""
  1205. QPushButton
  1206. {
  1207. font-weight: bold;
  1208. }
  1209. """)
  1210. grid0.addWidget(self.rect_cutout_object_btn, 21, 0, 1, 2)
  1211. separator_line = QtWidgets.QFrame()
  1212. separator_line.setFrameShape(QtWidgets.QFrame.HLine)
  1213. separator_line.setFrameShadow(QtWidgets.QFrame.Sunken)
  1214. grid0.addWidget(separator_line, 22, 0, 1, 2)
  1215. grid0.addWidget(QtWidgets.QLabel(''), 24, 0, 1, 2)
  1216. # MANUAL BRIDGE GAPS
  1217. title_manual_label = QtWidgets.QLabel("<b>%s %s</b>:" % (_('Manual'), _("Bridge Gaps")))
  1218. title_manual_label.setToolTip(
  1219. _("This section handle creation of manual bridge gaps.\n"
  1220. "This is done by mouse clicking on the perimeter of the\n"
  1221. "Geometry object that is used as a cutout object. ")
  1222. )
  1223. grid0.addWidget(title_manual_label, 25, 0, 1, 2)
  1224. # Big Cursor
  1225. big_cursor_label = QtWidgets.QLabel('%s:' % _("Big cursor"))
  1226. big_cursor_label.setToolTip(
  1227. _("Use a big cursor when adding manual gaps."))
  1228. self.big_cursor_cb = FCCheckBox()
  1229. grid0.addWidget(big_cursor_label, 27, 0)
  1230. grid0.addWidget(self.big_cursor_cb, 27, 1)
  1231. # Generate a surrounding Geometry object
  1232. self.man_geo_creation_btn = FCButton(_("Generate Manual Geometry"))
  1233. self.man_geo_creation_btn.setToolTip(
  1234. _("If the object to be cutout is a Gerber\n"
  1235. "first create a Geometry that surrounds it,\n"
  1236. "to be used as the cutout, if one doesn't exist yet.\n"
  1237. "Select the source Gerber file in the top object combobox.")
  1238. )
  1239. # self.man_geo_creation_btn.setStyleSheet("""
  1240. # QPushButton
  1241. # {
  1242. # font-weight: bold;
  1243. # }
  1244. # """)
  1245. grid0.addWidget(self.man_geo_creation_btn, 28, 0, 1, 2)
  1246. # Manual Geo Object
  1247. self.man_object_combo = FCComboBox()
  1248. self.man_object_combo.setModel(self.app.collection)
  1249. self.man_object_combo.setRootModelIndex(self.app.collection.index(2, 0, QtCore.QModelIndex()))
  1250. self.man_object_combo.is_last = True
  1251. self.man_object_combo.obj_type = "Geometry"
  1252. self.man_object_label = QtWidgets.QLabel('%s:' % _("Manual cutout Geometry"))
  1253. self.man_object_label.setToolTip(
  1254. _("Geometry object used to create the manual cutout.")
  1255. )
  1256. # self.man_object_label.setMinimumWidth(60)
  1257. grid0.addWidget(self.man_object_label, 30, 0, 1, 2)
  1258. grid0.addWidget(self.man_object_combo, 31, 0, 1, 2)
  1259. self.man_gaps_creation_btn = FCButton(_("Manual Add Bridge Gaps"))
  1260. self.man_gaps_creation_btn.setToolTip(
  1261. _("Use the left mouse button (LMB) click\n"
  1262. "to create a bridge gap to separate the PCB from\n"
  1263. "the surrounding material.\n"
  1264. "The LMB click has to be done on the perimeter of\n"
  1265. "the Geometry object used as a cutout geometry.")
  1266. )
  1267. self.man_gaps_creation_btn.setStyleSheet("""
  1268. QPushButton
  1269. {
  1270. font-weight: bold;
  1271. }
  1272. """)
  1273. grid0.addWidget(self.man_gaps_creation_btn, 35, 0, 1, 2)
  1274. self.layout.addStretch()
  1275. # ## Reset Tool
  1276. self.reset_button = FCButton(_("Reset Tool"))
  1277. self.reset_button.setToolTip(
  1278. _("Will reset the tool parameters.")
  1279. )
  1280. self.reset_button.setStyleSheet("""
  1281. QPushButton
  1282. {
  1283. font-weight: bold;
  1284. }
  1285. """)
  1286. self.layout.addWidget(self.reset_button)
  1287. # ############################ FINSIHED GUI ###################################
  1288. # #############################################################################
  1289. def confirmation_message(self, accepted, minval, maxval):
  1290. if accepted is False:
  1291. self.app.inform[str, bool].emit('[WARNING_NOTCL] %s: [%.*f, %.*f]' % (_("Edited value is out of range"),
  1292. self.decimals,
  1293. minval,
  1294. self.decimals,
  1295. maxval), False)
  1296. else:
  1297. self.app.inform[str, bool].emit('[success] %s' % _("Edited value is within limits."), False)
  1298. def confirmation_message_int(self, accepted, minval, maxval):
  1299. if accepted is False:
  1300. self.app.inform[str, bool].emit('[WARNING_NOTCL] %s: [%d, %d]' %
  1301. (_("Edited value is out of range"), minval, maxval), False)
  1302. else:
  1303. self.app.inform[str, bool].emit('[success] %s' % _("Edited value is within limits."), False)