ToolCopperThieving.py 68 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579
  1. # ##########################################################
  2. # FlatCAM: 2D Post-processing for Manufacturing #
  3. # File Author: Marius Adrian Stanciu (c) #
  4. # Date: 10/25/2019 #
  5. # MIT Licence #
  6. # ##########################################################
  7. from PyQt5 import QtWidgets, QtCore
  8. from camlib import grace
  9. from AppTool import AppTool
  10. from AppGUI.GUIElements import FCDoubleSpinner, RadioSet, FCEntry, FCComboBox
  11. import shapely.geometry.base as base
  12. from shapely.ops import cascaded_union, unary_union
  13. from shapely.geometry import Polygon, MultiPolygon, Point, LineString
  14. from shapely.geometry import box as box
  15. import shapely.affinity as affinity
  16. import logging
  17. from copy import deepcopy
  18. import numpy as np
  19. from collections import Iterable
  20. import gettext
  21. import AppTranslation as fcTranslate
  22. import builtins
  23. fcTranslate.apply_language('strings')
  24. if '_' not in builtins.__dict__:
  25. _ = gettext.gettext
  26. log = logging.getLogger('base')
  27. class ToolCopperThieving(AppTool):
  28. work_finished = QtCore.pyqtSignal()
  29. toolName = _("Copper Thieving Tool")
  30. def __init__(self, app):
  31. AppTool.__init__(self, app)
  32. self.app = app
  33. self.canvas = self.app.plotcanvas
  34. self.decimals = self.app.decimals
  35. self.units = self.app.defaults['units']
  36. # ## Title
  37. title_label = QtWidgets.QLabel("%s" % self.toolName)
  38. title_label.setStyleSheet("""
  39. QLabel
  40. {
  41. font-size: 16px;
  42. font-weight: bold;
  43. }
  44. """)
  45. self.layout.addWidget(title_label)
  46. self.layout.addWidget(QtWidgets.QLabel(''))
  47. # ## Grid Layout
  48. i_grid_lay = QtWidgets.QGridLayout()
  49. self.layout.addLayout(i_grid_lay)
  50. i_grid_lay.setColumnStretch(0, 0)
  51. i_grid_lay.setColumnStretch(1, 1)
  52. self.grb_object_combo = FCComboBox()
  53. self.grb_object_combo.setModel(self.app.collection)
  54. self.grb_object_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  55. self.grb_object_combo.is_last = True
  56. self.grb_object_combo.obj_type = 'Gerber'
  57. self.grbobj_label = QtWidgets.QLabel("<b>%s:</b>" % _("GERBER"))
  58. self.grbobj_label.setToolTip(
  59. _("Gerber Object to which will be added a copper thieving.")
  60. )
  61. i_grid_lay.addWidget(self.grbobj_label, 0, 0)
  62. i_grid_lay.addWidget(self.grb_object_combo, 1, 0, 1, 2)
  63. separator_line = QtWidgets.QFrame()
  64. separator_line.setFrameShape(QtWidgets.QFrame.HLine)
  65. separator_line.setFrameShadow(QtWidgets.QFrame.Sunken)
  66. i_grid_lay.addWidget(separator_line, 2, 0, 1, 2)
  67. # ## Grid Layout
  68. grid_lay = QtWidgets.QGridLayout()
  69. self.layout.addLayout(grid_lay)
  70. grid_lay.setColumnStretch(0, 0)
  71. grid_lay.setColumnStretch(1, 1)
  72. self.copper_fill_label = QtWidgets.QLabel('<b>%s</b>' % _('Parameters'))
  73. self.copper_fill_label.setToolTip(
  74. _("Parameters used for this tool.")
  75. )
  76. grid_lay.addWidget(self.copper_fill_label, 0, 0, 1, 2)
  77. # CLEARANCE #
  78. self.clearance_label = QtWidgets.QLabel('%s:' % _("Clearance"))
  79. self.clearance_label.setToolTip(
  80. _("This set the distance between the copper thieving components\n"
  81. "(the polygon fill may be split in multiple polygons)\n"
  82. "and the copper traces in the Gerber file.")
  83. )
  84. self.clearance_entry = FCDoubleSpinner(callback=self.confirmation_message)
  85. self.clearance_entry.set_range(0.00001, 9999.9999)
  86. self.clearance_entry.set_precision(self.decimals)
  87. self.clearance_entry.setSingleStep(0.1)
  88. grid_lay.addWidget(self.clearance_label, 1, 0)
  89. grid_lay.addWidget(self.clearance_entry, 1, 1)
  90. # MARGIN #
  91. self.margin_label = QtWidgets.QLabel('%s:' % _("Margin"))
  92. self.margin_label.setToolTip(
  93. _("Bounding box margin.")
  94. )
  95. self.margin_entry = FCDoubleSpinner(callback=self.confirmation_message)
  96. self.margin_entry.set_range(0.0, 9999.9999)
  97. self.margin_entry.set_precision(self.decimals)
  98. self.margin_entry.setSingleStep(0.1)
  99. grid_lay.addWidget(self.margin_label, 2, 0)
  100. grid_lay.addWidget(self.margin_entry, 2, 1)
  101. # Reference #
  102. self.reference_radio = RadioSet([
  103. {'label': _('Itself'), 'value': 'itself'},
  104. {"label": _("Area Selection"), "value": "area"},
  105. {'label': _("Reference Object"), 'value': 'box'}
  106. ], orientation='vertical', stretch=False)
  107. self.reference_label = QtWidgets.QLabel(_("Reference:"))
  108. self.reference_label.setToolTip(
  109. _("- 'Itself' - the copper thieving extent is based on the object extent.\n"
  110. "- 'Area Selection' - left mouse click to start selection of the area to be filled.\n"
  111. "- 'Reference Object' - will do copper thieving within the area specified by another object.")
  112. )
  113. grid_lay.addWidget(self.reference_label, 3, 0)
  114. grid_lay.addWidget(self.reference_radio, 3, 1)
  115. self.ref_combo_type_label = QtWidgets.QLabel('%s:' % _("Ref. Type"))
  116. self.ref_combo_type_label.setToolTip(
  117. _("The type of FlatCAM object to be used as copper thieving reference.\n"
  118. "It can be Gerber, Excellon or Geometry.")
  119. )
  120. self.ref_combo_type = FCComboBox()
  121. self.ref_combo_type.addItems([_("Gerber"), _("Excellon"), _("Geometry")])
  122. grid_lay.addWidget(self.ref_combo_type_label, 4, 0)
  123. grid_lay.addWidget(self.ref_combo_type, 4, 1)
  124. self.ref_combo_label = QtWidgets.QLabel('%s:' % _("Ref. Object"))
  125. self.ref_combo_label.setToolTip(
  126. _("The FlatCAM object to be used as non copper clearing reference.")
  127. )
  128. self.ref_combo = FCComboBox()
  129. self.ref_combo.setModel(self.app.collection)
  130. self.ref_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  131. self.ref_combo.is_last = True
  132. self.ref_combo.obj_type = {
  133. _("Gerber"): "Gerber", _("Excellon"): "Excellon", _("Geometry"): "Geometry"
  134. }[self.ref_combo_type.get_value()]
  135. grid_lay.addWidget(self.ref_combo_label, 5, 0)
  136. grid_lay.addWidget(self.ref_combo, 5, 1)
  137. self.ref_combo.hide()
  138. self.ref_combo_label.hide()
  139. self.ref_combo_type.hide()
  140. self.ref_combo_type_label.hide()
  141. # Bounding Box Type #
  142. self.bbox_type_radio = RadioSet([
  143. {'label': _('Rectangular'), 'value': 'rect'},
  144. {"label": _("Minimal"), "value": "min"}
  145. ], stretch=False)
  146. self.bbox_type_label = QtWidgets.QLabel(_("Box Type:"))
  147. self.bbox_type_label.setToolTip(
  148. _("- 'Rectangular' - the bounding box will be of rectangular shape.\n"
  149. "- 'Minimal' - the bounding box will be the convex hull shape.")
  150. )
  151. grid_lay.addWidget(self.bbox_type_label, 6, 0)
  152. grid_lay.addWidget(self.bbox_type_radio, 6, 1)
  153. self.bbox_type_label.hide()
  154. self.bbox_type_radio.hide()
  155. separator_line = QtWidgets.QFrame()
  156. separator_line.setFrameShape(QtWidgets.QFrame.HLine)
  157. separator_line.setFrameShadow(QtWidgets.QFrame.Sunken)
  158. grid_lay.addWidget(separator_line, 7, 0, 1, 2)
  159. # Fill Type
  160. self.fill_type_radio = RadioSet([
  161. {'label': _('Solid'), 'value': 'solid'},
  162. {"label": _("Dots Grid"), "value": "dot"},
  163. {"label": _("Squares Grid"), "value": "square"},
  164. {"label": _("Lines Grid"), "value": "line"}
  165. ], orientation='vertical', stretch=False)
  166. self.fill_type_label = QtWidgets.QLabel(_("Fill Type:"))
  167. self.fill_type_label.setToolTip(
  168. _("- 'Solid' - copper thieving will be a solid polygon.\n"
  169. "- 'Dots Grid' - the empty area will be filled with a pattern of dots.\n"
  170. "- 'Squares Grid' - the empty area will be filled with a pattern of squares.\n"
  171. "- 'Lines Grid' - the empty area will be filled with a pattern of lines.")
  172. )
  173. grid_lay.addWidget(self.fill_type_label, 8, 0)
  174. grid_lay.addWidget(self.fill_type_radio, 8, 1)
  175. # DOTS FRAME
  176. self.dots_frame = QtWidgets.QFrame()
  177. self.dots_frame.setContentsMargins(0, 0, 0, 0)
  178. self.layout.addWidget(self.dots_frame)
  179. dots_grid = QtWidgets.QGridLayout()
  180. dots_grid.setColumnStretch(0, 0)
  181. dots_grid.setColumnStretch(1, 1)
  182. dots_grid.setContentsMargins(0, 0, 0, 0)
  183. self.dots_frame.setLayout(dots_grid)
  184. self.dots_frame.hide()
  185. self.dots_label = QtWidgets.QLabel('<b>%s</b>:' % _("Dots Grid Parameters"))
  186. dots_grid.addWidget(self.dots_label, 0, 0, 1, 2)
  187. # Dot diameter #
  188. self.dotdia_label = QtWidgets.QLabel('%s:' % _("Dia"))
  189. self.dotdia_label.setToolTip(
  190. _("Dot diameter in Dots Grid.")
  191. )
  192. self.dot_dia_entry = FCDoubleSpinner(callback=self.confirmation_message)
  193. self.dot_dia_entry.set_range(0.0, 9999.9999)
  194. self.dot_dia_entry.set_precision(self.decimals)
  195. self.dot_dia_entry.setSingleStep(0.1)
  196. dots_grid.addWidget(self.dotdia_label, 1, 0)
  197. dots_grid.addWidget(self.dot_dia_entry, 1, 1)
  198. # Dot spacing #
  199. self.dotspacing_label = QtWidgets.QLabel('%s:' % _("Spacing"))
  200. self.dotspacing_label.setToolTip(
  201. _("Distance between each two dots in Dots Grid.")
  202. )
  203. self.dot_spacing_entry = FCDoubleSpinner(callback=self.confirmation_message)
  204. self.dot_spacing_entry.set_range(0.0, 9999.9999)
  205. self.dot_spacing_entry.set_precision(self.decimals)
  206. self.dot_spacing_entry.setSingleStep(0.1)
  207. dots_grid.addWidget(self.dotspacing_label, 2, 0)
  208. dots_grid.addWidget(self.dot_spacing_entry, 2, 1)
  209. # SQUARES FRAME
  210. self.squares_frame = QtWidgets.QFrame()
  211. self.squares_frame.setContentsMargins(0, 0, 0, 0)
  212. self.layout.addWidget(self.squares_frame)
  213. squares_grid = QtWidgets.QGridLayout()
  214. squares_grid.setColumnStretch(0, 0)
  215. squares_grid.setColumnStretch(1, 1)
  216. squares_grid.setContentsMargins(0, 0, 0, 0)
  217. self.squares_frame.setLayout(squares_grid)
  218. self.squares_frame.hide()
  219. self.squares_label = QtWidgets.QLabel('<b>%s</b>:' % _("Squares Grid Parameters"))
  220. squares_grid.addWidget(self.squares_label, 0, 0, 1, 2)
  221. # Square Size #
  222. self.square_size_label = QtWidgets.QLabel('%s:' % _("Size"))
  223. self.square_size_label.setToolTip(
  224. _("Square side size in Squares Grid.")
  225. )
  226. self.square_size_entry = FCDoubleSpinner(callback=self.confirmation_message)
  227. self.square_size_entry.set_range(0.0, 9999.9999)
  228. self.square_size_entry.set_precision(self.decimals)
  229. self.square_size_entry.setSingleStep(0.1)
  230. squares_grid.addWidget(self.square_size_label, 1, 0)
  231. squares_grid.addWidget(self.square_size_entry, 1, 1)
  232. # Squares spacing #
  233. self.squares_spacing_label = QtWidgets.QLabel('%s:' % _("Spacing"))
  234. self.squares_spacing_label.setToolTip(
  235. _("Distance between each two squares in Squares Grid.")
  236. )
  237. self.squares_spacing_entry = FCDoubleSpinner(callback=self.confirmation_message)
  238. self.squares_spacing_entry.set_range(0.0, 9999.9999)
  239. self.squares_spacing_entry.set_precision(self.decimals)
  240. self.squares_spacing_entry.setSingleStep(0.1)
  241. squares_grid.addWidget(self.squares_spacing_label, 2, 0)
  242. squares_grid.addWidget(self.squares_spacing_entry, 2, 1)
  243. # LINES FRAME
  244. self.lines_frame = QtWidgets.QFrame()
  245. self.lines_frame.setContentsMargins(0, 0, 0, 0)
  246. self.layout.addWidget(self.lines_frame)
  247. lines_grid = QtWidgets.QGridLayout()
  248. lines_grid.setColumnStretch(0, 0)
  249. lines_grid.setColumnStretch(1, 1)
  250. lines_grid.setContentsMargins(0, 0, 0, 0)
  251. self.lines_frame.setLayout(lines_grid)
  252. self.lines_frame.hide()
  253. self.lines_label = QtWidgets.QLabel('<b>%s</b>:' % _("Lines Grid Parameters"))
  254. lines_grid.addWidget(self.lines_label, 0, 0, 1, 2)
  255. # Square Size #
  256. self.line_size_label = QtWidgets.QLabel('%s:' % _("Size"))
  257. self.line_size_label.setToolTip(
  258. _("Line thickness size in Lines Grid.")
  259. )
  260. self.line_size_entry = FCDoubleSpinner(callback=self.confirmation_message)
  261. self.line_size_entry.set_range(0.0, 9999.9999)
  262. self.line_size_entry.set_precision(self.decimals)
  263. self.line_size_entry.setSingleStep(0.1)
  264. lines_grid.addWidget(self.line_size_label, 1, 0)
  265. lines_grid.addWidget(self.line_size_entry, 1, 1)
  266. # Lines spacing #
  267. self.lines_spacing_label = QtWidgets.QLabel('%s:' % _("Spacing"))
  268. self.lines_spacing_label.setToolTip(
  269. _("Distance between each two lines in Lines Grid.")
  270. )
  271. self.lines_spacing_entry = FCDoubleSpinner(callback=self.confirmation_message)
  272. self.lines_spacing_entry.set_range(0.0, 9999.9999)
  273. self.lines_spacing_entry.set_precision(self.decimals)
  274. self.lines_spacing_entry.setSingleStep(0.1)
  275. lines_grid.addWidget(self.lines_spacing_label, 2, 0)
  276. lines_grid.addWidget(self.lines_spacing_entry, 2, 1)
  277. # ## Insert Copper Thieving
  278. self.fill_button = QtWidgets.QPushButton(_("Insert Copper thieving"))
  279. self.fill_button.setToolTip(
  280. _("Will add a polygon (may be split in multiple parts)\n"
  281. "that will surround the actual Gerber traces at a certain distance.")
  282. )
  283. self.fill_button.setStyleSheet("""
  284. QPushButton
  285. {
  286. font-weight: bold;
  287. }
  288. """)
  289. self.layout.addWidget(self.fill_button)
  290. # ## Grid Layout
  291. grid_lay_1 = QtWidgets.QGridLayout()
  292. self.layout.addLayout(grid_lay_1)
  293. grid_lay_1.setColumnStretch(0, 0)
  294. grid_lay_1.setColumnStretch(1, 1)
  295. grid_lay_1.setColumnStretch(2, 0)
  296. separator_line_1 = QtWidgets.QFrame()
  297. separator_line_1.setFrameShape(QtWidgets.QFrame.HLine)
  298. separator_line_1.setFrameShadow(QtWidgets.QFrame.Sunken)
  299. grid_lay_1.addWidget(separator_line_1, 0, 0, 1, 3)
  300. grid_lay_1.addWidget(QtWidgets.QLabel(''))
  301. self.robber_bar_label = QtWidgets.QLabel('<b>%s</b>' % _('Robber Bar Parameters'))
  302. self.robber_bar_label.setToolTip(
  303. _("Parameters used for the robber bar.\n"
  304. "Robber bar = copper border to help in pattern hole plating.")
  305. )
  306. grid_lay_1.addWidget(self.robber_bar_label, 1, 0, 1, 3)
  307. # ROBBER BAR MARGIN #
  308. self.rb_margin_label = QtWidgets.QLabel('%s:' % _("Margin"))
  309. self.rb_margin_label.setToolTip(
  310. _("Bounding box margin for robber bar.")
  311. )
  312. self.rb_margin_entry = FCDoubleSpinner(callback=self.confirmation_message)
  313. self.rb_margin_entry.set_range(-9999.9999, 9999.9999)
  314. self.rb_margin_entry.set_precision(self.decimals)
  315. self.rb_margin_entry.setSingleStep(0.1)
  316. grid_lay_1.addWidget(self.rb_margin_label, 2, 0)
  317. grid_lay_1.addWidget(self.rb_margin_entry, 2, 1, 1, 2)
  318. # THICKNESS #
  319. self.rb_thickness_label = QtWidgets.QLabel('%s:' % _("Thickness"))
  320. self.rb_thickness_label.setToolTip(
  321. _("The robber bar thickness.")
  322. )
  323. self.rb_thickness_entry = FCDoubleSpinner(callback=self.confirmation_message)
  324. self.rb_thickness_entry.set_range(0.0000, 9999.9999)
  325. self.rb_thickness_entry.set_precision(self.decimals)
  326. self.rb_thickness_entry.setSingleStep(0.1)
  327. grid_lay_1.addWidget(self.rb_thickness_label, 3, 0)
  328. grid_lay_1.addWidget(self.rb_thickness_entry, 3, 1, 1, 2)
  329. # ## Insert Robber Bar
  330. self.rb_button = QtWidgets.QPushButton(_("Insert Robber Bar"))
  331. self.rb_button.setToolTip(
  332. _("Will add a polygon with a defined thickness\n"
  333. "that will surround the actual Gerber object\n"
  334. "at a certain distance.\n"
  335. "Required when doing holes pattern plating.")
  336. )
  337. self.rb_button.setStyleSheet("""
  338. QPushButton
  339. {
  340. font-weight: bold;
  341. }
  342. """)
  343. grid_lay_1.addWidget(self.rb_button, 4, 0, 1, 3)
  344. separator_line_2 = QtWidgets.QFrame()
  345. separator_line_2.setFrameShape(QtWidgets.QFrame.HLine)
  346. separator_line_2.setFrameShadow(QtWidgets.QFrame.Sunken)
  347. grid_lay_1.addWidget(separator_line_2, 5, 0, 1, 3)
  348. self.patern_mask_label = QtWidgets.QLabel('<b>%s</b>' % _('Pattern Plating Mask'))
  349. self.patern_mask_label.setToolTip(
  350. _("Generate a mask for pattern plating.")
  351. )
  352. grid_lay_1.addWidget(self.patern_mask_label, 6, 0, 1, 3)
  353. self.sm_obj_label = QtWidgets.QLabel("%s:" % _("Select Soldermask object"))
  354. self.sm_obj_label.setToolTip(
  355. _("Gerber Object with the soldermask.\n"
  356. "It will be used as a base for\n"
  357. "the pattern plating mask.")
  358. )
  359. self.sm_object_combo = FCComboBox()
  360. self.sm_object_combo.setModel(self.app.collection)
  361. self.sm_object_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  362. self.sm_object_combo.is_last = True
  363. self.sm_object_combo.obj_type = 'Gerber'
  364. grid_lay_1.addWidget(self.sm_obj_label, 7, 0, 1, 3)
  365. grid_lay_1.addWidget(self.sm_object_combo, 8, 0, 1, 3)
  366. # Openings CLEARANCE #
  367. self.clearance_ppm_label = QtWidgets.QLabel('%s:' % _("Clearance"))
  368. self.clearance_ppm_label.setToolTip(
  369. _("The distance between the possible copper thieving elements\n"
  370. "and/or robber bar and the actual openings in the mask.")
  371. )
  372. self.clearance_ppm_entry = FCDoubleSpinner(callback=self.confirmation_message)
  373. self.clearance_ppm_entry.set_range(-9999.9999, 9999.9999)
  374. self.clearance_ppm_entry.set_precision(self.decimals)
  375. self.clearance_ppm_entry.setSingleStep(0.1)
  376. grid_lay_1.addWidget(self.clearance_ppm_label, 9, 0)
  377. grid_lay_1.addWidget(self.clearance_ppm_entry, 9, 1, 1, 2)
  378. # Plated area
  379. self.plated_area_label = QtWidgets.QLabel('%s:' % _("Plated area"))
  380. self.plated_area_label.setToolTip(
  381. _("The area to be plated by pattern plating.\n"
  382. "Basically is made from the openings in the plating mask.\n\n"
  383. "<<WARNING>> - the calculated area is actually a bit larger\n"
  384. "due of the fact that the soldermask openings are by design\n"
  385. "a bit larger than the copper pads, and this area is\n"
  386. "calculated from the soldermask openings.")
  387. )
  388. self.plated_area_entry = FCEntry()
  389. self.plated_area_entry.setDisabled(True)
  390. if self.units.upper() == 'MM':
  391. self.units_area_label = QtWidgets.QLabel('%s<sup>2</sup>' % _("mm"))
  392. else:
  393. self.units_area_label = QtWidgets.QLabel('%s<sup>2</sup>' % _("in"))
  394. grid_lay_1.addWidget(self.plated_area_label, 10, 0)
  395. grid_lay_1.addWidget(self.plated_area_entry, 10, 1)
  396. grid_lay_1.addWidget(self.units_area_label, 10, 2)
  397. # ## Pattern Plating Mask
  398. self.ppm_button = QtWidgets.QPushButton(_("Generate pattern plating mask"))
  399. self.ppm_button.setToolTip(
  400. _("Will add to the soldermask gerber geometry\n"
  401. "the geometries of the copper thieving and/or\n"
  402. "the robber bar if those were generated.")
  403. )
  404. self.ppm_button.setStyleSheet("""
  405. QPushButton
  406. {
  407. font-weight: bold;
  408. }
  409. """)
  410. grid_lay_1.addWidget(self.ppm_button, 11, 0, 1, 3)
  411. self.layout.addStretch()
  412. # ## Reset Tool
  413. self.reset_button = QtWidgets.QPushButton(_("Reset Tool"))
  414. self.reset_button.setToolTip(
  415. _("Will reset the tool parameters.")
  416. )
  417. self.reset_button.setStyleSheet("""
  418. QPushButton
  419. {
  420. font-weight: bold;
  421. }
  422. """)
  423. self.layout.addWidget(self.reset_button)
  424. # Objects involved in Copper thieving
  425. self.grb_object = None
  426. self.ref_obj = None
  427. self.sel_rect = []
  428. self.sm_object = None
  429. # store the flattened geometry here:
  430. self.flat_geometry = []
  431. # Events ID
  432. self.mr = None
  433. self.mm = None
  434. # Mouse cursor positions
  435. self.mouse_is_dragging = False
  436. self.cursor_pos = (0, 0)
  437. self.first_click = False
  438. self.area_method = False
  439. # Tool properties
  440. self.clearance_val = None
  441. self.margin_val = None
  442. self.geo_steps_per_circle = 128
  443. # Thieving geometry storage
  444. self.new_solid_geometry = []
  445. # Robber bar geometry storage
  446. self.robber_geo = None
  447. self.robber_line = None
  448. self.rb_thickness = None
  449. # SIGNALS
  450. self.ref_combo_type.currentIndexChanged.connect(self.on_ref_combo_type_change)
  451. self.reference_radio.group_toggle_fn = self.on_toggle_reference
  452. self.fill_type_radio.activated_custom.connect(self.on_thieving_type)
  453. self.fill_button.clicked.connect(self.execute)
  454. self.rb_button.clicked.connect(self.add_robber_bar)
  455. self.ppm_button.clicked.connect(self.on_add_ppm)
  456. self.reset_button.clicked.connect(self.set_tool_ui)
  457. self.work_finished.connect(self.on_new_pattern_plating_object)
  458. def run(self, toggle=True):
  459. self.app.defaults.report_usage("ToolCopperThieving()")
  460. if toggle:
  461. # if the splitter is hidden, display it, else hide it but only if the current widget is the same
  462. if self.app.ui.splitter.sizes()[0] == 0:
  463. self.app.ui.splitter.setSizes([1, 1])
  464. else:
  465. try:
  466. if self.app.ui.tool_scroll_area.widget().objectName() == self.toolName:
  467. # if tab is populated with the tool but it does not have the focus, focus on it
  468. if not self.app.ui.notebook.currentWidget() is self.app.ui.tool_tab:
  469. # focus on Tool Tab
  470. self.app.ui.notebook.setCurrentWidget(self.app.ui.tool_tab)
  471. else:
  472. self.app.ui.splitter.setSizes([0, 1])
  473. except AttributeError:
  474. pass
  475. else:
  476. if self.app.ui.splitter.sizes()[0] == 0:
  477. self.app.ui.splitter.setSizes([1, 1])
  478. AppTool.run(self)
  479. self.set_tool_ui()
  480. self.app.ui.notebook.setTabText(2, _("Copper Thieving Tool"))
  481. def install(self, icon=None, separator=None, **kwargs):
  482. AppTool.install(self, icon, separator, shortcut='Alt+J', **kwargs)
  483. def set_tool_ui(self):
  484. self.units = self.app.defaults['units']
  485. self.clearance_entry.set_value(float(self.app.defaults["tools_copper_thieving_clearance"]))
  486. self.margin_entry.set_value(float(self.app.defaults["tools_copper_thieving_margin"]))
  487. self.reference_radio.set_value(self.app.defaults["tools_copper_thieving_reference"])
  488. self.bbox_type_radio.set_value(self.app.defaults["tools_copper_thieving_box_type"])
  489. self.fill_type_radio.set_value(self.app.defaults["tools_copper_thieving_fill_type"])
  490. self.geo_steps_per_circle = int(self.app.defaults["tools_copper_thieving_circle_steps"])
  491. self.dot_dia_entry.set_value(self.app.defaults["tools_copper_thieving_dots_dia"])
  492. self.dot_spacing_entry.set_value(self.app.defaults["tools_copper_thieving_dots_spacing"])
  493. self.square_size_entry.set_value(self.app.defaults["tools_copper_thieving_squares_size"])
  494. self.squares_spacing_entry.set_value(self.app.defaults["tools_copper_thieving_squares_spacing"])
  495. self.line_size_entry.set_value(self.app.defaults["tools_copper_thieving_lines_size"])
  496. self.lines_spacing_entry.set_value(self.app.defaults["tools_copper_thieving_lines_spacing"])
  497. self.rb_margin_entry.set_value(self.app.defaults["tools_copper_thieving_rb_margin"])
  498. self.rb_thickness_entry.set_value(self.app.defaults["tools_copper_thieving_rb_thickness"])
  499. self.clearance_ppm_entry.set_value(self.app.defaults["tools_copper_thieving_mask_clearance"])
  500. # INIT SECTION
  501. self.area_method = False
  502. self.robber_geo = None
  503. self.robber_line = None
  504. self.new_solid_geometry = None
  505. def on_ref_combo_type_change(self):
  506. obj_type = self.ref_combo_type.currentIndex()
  507. self.ref_combo.setRootModelIndex(self.app.collection.index(obj_type, 0, QtCore.QModelIndex()))
  508. self.ref_combo.setCurrentIndex(0)
  509. self.ref_combo.obj_type = {
  510. _("Gerber"): "Gerber", _("Excellon"): "Excellon", _("Geometry"): "Geometry"
  511. }[self.ref_combo_type.get_value()]
  512. def on_toggle_reference(self):
  513. if self.reference_radio.get_value() == "itself" or self.reference_radio.get_value() == "area":
  514. self.ref_combo.hide()
  515. self.ref_combo_label.hide()
  516. self.ref_combo_type.hide()
  517. self.ref_combo_type_label.hide()
  518. else:
  519. self.ref_combo.show()
  520. self.ref_combo_label.show()
  521. self.ref_combo_type.show()
  522. self.ref_combo_type_label.show()
  523. if self.reference_radio.get_value() == "itself":
  524. self.bbox_type_label.show()
  525. self.bbox_type_radio.show()
  526. else:
  527. if self.fill_type_radio.get_value() == 'line':
  528. self.reference_radio.set_value('itself')
  529. self.app.inform.emit('[WARNING_NOTCL] %s' % _("Lines Grid works only for 'itself' reference ..."))
  530. return
  531. self.bbox_type_label.hide()
  532. self.bbox_type_radio.hide()
  533. def on_thieving_type(self, choice):
  534. if choice == 'solid':
  535. self.dots_frame.hide()
  536. self.squares_frame.hide()
  537. self.lines_frame.hide()
  538. self.app.inform.emit(_("Solid fill selected."))
  539. elif choice == 'dot':
  540. self.dots_frame.show()
  541. self.squares_frame.hide()
  542. self.lines_frame.hide()
  543. self.app.inform.emit(_("Dots grid fill selected."))
  544. elif choice == 'square':
  545. self.dots_frame.hide()
  546. self.squares_frame.show()
  547. self.lines_frame.hide()
  548. self.app.inform.emit(_("Squares grid fill selected."))
  549. else:
  550. if self.reference_radio.get_value() != 'itself':
  551. self.reference_radio.set_value('itself')
  552. self.app.inform.emit('[WARNING_NOTCL] %s' % _("Lines Grid works only for 'itself' reference ..."))
  553. self.dots_frame.hide()
  554. self.squares_frame.hide()
  555. self.lines_frame.show()
  556. def add_robber_bar(self):
  557. rb_margin = self.rb_margin_entry.get_value()
  558. self.rb_thickness = self.rb_thickness_entry.get_value()
  559. # get the Gerber object on which the Robber bar will be inserted
  560. selection_index = self.grb_object_combo.currentIndex()
  561. model_index = self.app.collection.index(selection_index, 0, self.grb_object_combo.rootModelIndex())
  562. try:
  563. self.grb_object = model_index.internalPointer().obj
  564. except Exception as e:
  565. log.debug("ToolCopperThieving.add_robber_bar() --> %s" % str(e))
  566. self.app.inform.emit('[WARNING_NOTCL] %s' % _("There is no Gerber object loaded ..."))
  567. return 'fail'
  568. try:
  569. outline_pol = self.grb_object.solid_geometry.envelope
  570. except TypeError:
  571. outline_pol = MultiPolygon(self.grb_object.solid_geometry).envelope
  572. rb_distance = rb_margin + (self.rb_thickness / 2.0)
  573. self.robber_line = outline_pol.buffer(rb_distance).exterior
  574. self.robber_geo = self.robber_line.buffer(self.rb_thickness / 2.0)
  575. self.app.proc_container.update_view_text(' %s' % _("Append geometry"))
  576. aperture_found = None
  577. for ap_id, ap_val in self.grb_object.apertures.items():
  578. if ap_val['type'] == 'C' and ap_val['size'] == self.rb_thickness:
  579. aperture_found = ap_id
  580. break
  581. if aperture_found:
  582. geo_elem = {}
  583. geo_elem['solid'] = self.robber_geo
  584. geo_elem['follow'] = self.robber_line
  585. self.grb_object.apertures[aperture_found]['geometry'].append(deepcopy(geo_elem))
  586. else:
  587. ap_keys = list(self.grb_object.apertures.keys())
  588. if ap_keys:
  589. new_apid = str(int(max(ap_keys)) + 1)
  590. else:
  591. new_apid = '10'
  592. self.grb_object.apertures[new_apid] = {}
  593. self.grb_object.apertures[new_apid]['type'] = 'C'
  594. self.grb_object.apertures[new_apid]['size'] = self.rb_thickness
  595. self.grb_object.apertures[new_apid]['geometry'] = []
  596. geo_elem = {}
  597. geo_elem['solid'] = self.robber_geo
  598. geo_elem['follow'] = self.robber_line
  599. self.grb_object.apertures[new_apid]['geometry'].append(deepcopy(geo_elem))
  600. geo_obj = self.grb_object.solid_geometry
  601. if isinstance(geo_obj, MultiPolygon):
  602. s_list = []
  603. for pol in geo_obj.geoms:
  604. s_list.append(pol)
  605. s_list.append(self.robber_geo)
  606. geo_obj = MultiPolygon(s_list)
  607. elif isinstance(geo_obj, list):
  608. geo_obj.append(self.robber_geo)
  609. elif isinstance(geo_obj, Polygon):
  610. geo_obj = MultiPolygon([geo_obj, self.robber_geo])
  611. self.grb_object.solid_geometry = geo_obj
  612. self.app.proc_container.update_view_text(' %s' % _("Append source file"))
  613. # update the source file with the new geometry:
  614. self.grb_object.source_file = self.app.export_gerber(obj_name=self.grb_object.options['name'],
  615. filename=None,
  616. local_use=self.grb_object,
  617. use_thread=False)
  618. self.app.proc_container.update_view_text(' %s' % '')
  619. self.on_exit()
  620. self.app.inform.emit('[success] %s' % _("Copper Thieving Tool done."))
  621. def execute(self):
  622. self.app.call_source = "copper_thieving_tool"
  623. self.clearance_val = self.clearance_entry.get_value()
  624. self.margin_val = self.margin_entry.get_value()
  625. reference_method = self.reference_radio.get_value()
  626. # get the Gerber object on which the Copper thieving will be inserted
  627. selection_index = self.grb_object_combo.currentIndex()
  628. model_index = self.app.collection.index(selection_index, 0, self.grb_object_combo.rootModelIndex())
  629. try:
  630. self.grb_object = model_index.internalPointer().obj
  631. except Exception as e:
  632. log.debug("ToolCopperThieving.execute() --> %s" % str(e))
  633. self.app.inform.emit('[WARNING_NOTCL] %s' % _("There is no Gerber object loaded ..."))
  634. return 'fail'
  635. if reference_method == 'itself':
  636. bound_obj_name = self.grb_object_combo.currentText()
  637. # Get reference object.
  638. try:
  639. self.ref_obj = self.app.collection.get_by_name(bound_obj_name)
  640. except Exception as e:
  641. self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve object"), str(e)))
  642. return "Could not retrieve object: %s" % self.obj_name
  643. self.on_copper_thieving(
  644. thieving_obj=self.grb_object,
  645. c_val=self.clearance_val,
  646. margin=self.margin_val
  647. )
  648. elif reference_method == 'area':
  649. self.app.inform.emit('[WARNING_NOTCL] %s' % _("Click the start point of the area."))
  650. self.area_method = True
  651. if self.app.is_legacy is False:
  652. self.app.plotcanvas.graph_event_disconnect('mouse_press', self.app.on_mouse_click_over_plot)
  653. self.app.plotcanvas.graph_event_disconnect('mouse_move', self.app.on_mouse_move_over_plot)
  654. self.app.plotcanvas.graph_event_disconnect('mouse_release', self.app.on_mouse_click_release_over_plot)
  655. else:
  656. self.app.plotcanvas.graph_event_disconnect(self.app.mp)
  657. self.app.plotcanvas.graph_event_disconnect(self.app.mm)
  658. self.app.plotcanvas.graph_event_disconnect(self.app.mr)
  659. self.mr = self.app.plotcanvas.graph_event_connect('mouse_release', self.on_mouse_release)
  660. self.mm = self.app.plotcanvas.graph_event_connect('mouse_move', self.on_mouse_move)
  661. elif reference_method == 'box':
  662. bound_obj_name = self.ref_combo.currentText()
  663. # Get reference object.
  664. try:
  665. self.ref_obj = self.app.collection.get_by_name(bound_obj_name)
  666. except Exception as e:
  667. self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve object"), bound_obj_name))
  668. return "Could not retrieve object: %s. Error: %s" % (bound_obj_name, str(e))
  669. self.on_copper_thieving(
  670. thieving_obj=self.grb_object,
  671. ref_obj=self.ref_obj,
  672. c_val=self.clearance_val,
  673. margin=self.margin_val
  674. )
  675. # To be called after clicking on the plot.
  676. def on_mouse_release(self, event):
  677. if self.app.is_legacy is False:
  678. event_pos = event.pos
  679. # event_is_dragging = event.is_dragging
  680. right_button = 2
  681. else:
  682. event_pos = (event.xdata, event.ydata)
  683. # event_is_dragging = self.app.plotcanvas.is_dragging
  684. right_button = 3
  685. event_pos = self.app.plotcanvas.translate_coords(event_pos)
  686. # do clear area only for left mouse clicks
  687. if event.button == 1:
  688. if self.first_click is False:
  689. self.first_click = True
  690. self.app.inform.emit('[WARNING_NOTCL] %s' % _("Click the end point of the filling area."))
  691. self.cursor_pos = self.app.plotcanvas.translate_coords(event_pos)
  692. if self.app.grid_status() is True:
  693. self.cursor_pos = self.app.geo_editor.snap(event_pos[0], event_pos[1])
  694. else:
  695. self.app.inform.emit(_("Zone added. Click to start adding next zone or right click to finish."))
  696. self.app.delete_selection_shape()
  697. if self.app.grid_status() is True:
  698. curr_pos = self.app.geo_editor.snap(event_pos[0], event_pos[1])
  699. else:
  700. curr_pos = (event_pos[0], event_pos[1])
  701. x0, y0 = self.cursor_pos[0], self.cursor_pos[1]
  702. x1, y1 = curr_pos[0], curr_pos[1]
  703. pt1 = (x0, y0)
  704. pt2 = (x1, y0)
  705. pt3 = (x1, y1)
  706. pt4 = (x0, y1)
  707. new_rectangle = Polygon([pt1, pt2, pt3, pt4])
  708. self.sel_rect.append(new_rectangle)
  709. # add a temporary shape on canvas
  710. self.draw_tool_selection_shape(old_coords=(x0, y0), coords=(x1, y1))
  711. self.first_click = False
  712. return
  713. elif event.button == right_button and self.mouse_is_dragging is False:
  714. self.area_method = False
  715. self.first_click = False
  716. self.delete_tool_selection_shape()
  717. if self.app.is_legacy is False:
  718. self.app.plotcanvas.graph_event_disconnect('mouse_release', self.on_mouse_release)
  719. self.app.plotcanvas.graph_event_disconnect('mouse_move', self.on_mouse_move)
  720. else:
  721. self.app.plotcanvas.graph_event_disconnect(self.mr)
  722. self.app.plotcanvas.graph_event_disconnect(self.mm)
  723. self.app.mp = self.app.plotcanvas.graph_event_connect('mouse_press',
  724. self.app.on_mouse_click_over_plot)
  725. self.app.mm = self.app.plotcanvas.graph_event_connect('mouse_move',
  726. self.app.on_mouse_move_over_plot)
  727. self.app.mr = self.app.plotcanvas.graph_event_connect('mouse_release',
  728. self.app.on_mouse_click_release_over_plot)
  729. if len(self.sel_rect) == 0:
  730. return
  731. self.sel_rect = cascaded_union(self.sel_rect)
  732. if not isinstance(self.sel_rect, Iterable):
  733. self.sel_rect = [self.sel_rect]
  734. self.on_copper_thieving(
  735. thieving_obj=self.grb_object,
  736. ref_obj=self.sel_rect,
  737. c_val=self.clearance_val,
  738. margin=self.margin_val
  739. )
  740. # called on mouse move
  741. def on_mouse_move(self, event):
  742. if self.app.is_legacy is False:
  743. event_pos = event.pos
  744. event_is_dragging = event.is_dragging
  745. # right_button = 2
  746. else:
  747. event_pos = (event.xdata, event.ydata)
  748. event_is_dragging = self.app.plotcanvas.is_dragging
  749. # right_button = 3
  750. curr_pos = self.app.plotcanvas.translate_coords(event_pos)
  751. # detect mouse dragging motion
  752. if event_is_dragging is True:
  753. self.mouse_is_dragging = True
  754. else:
  755. self.mouse_is_dragging = False
  756. # update the cursor position
  757. if self.app.grid_status() is True:
  758. # Update cursor
  759. curr_pos = self.app.geo_editor.snap(curr_pos[0], curr_pos[1])
  760. self.app.app_cursor.set_data(np.asarray([(curr_pos[0], curr_pos[1])]),
  761. symbol='++', edge_color=self.app.cursor_color_3D,
  762. edge_width=self.app.defaults["global_cursor_width"],
  763. size=self.app.defaults["global_cursor_size"])
  764. if self.cursor_pos is None:
  765. self.cursor_pos = (0, 0)
  766. self.app.dx = curr_pos[0] - float(self.cursor_pos[0])
  767. self.app.dy = curr_pos[1] - float(self.cursor_pos[1])
  768. # # update the positions on status bar
  769. self.app.ui.position_label.setText("&nbsp;<b>X</b>: %.4f&nbsp;&nbsp; "
  770. "<b>Y</b>: %.4f&nbsp;" % (curr_pos[0], curr_pos[1]))
  771. # self.app.ui.rel_position_label.setText("<b>Dx</b>: %.4f&nbsp;&nbsp; <b>Dy</b>: "
  772. # "%.4f&nbsp;&nbsp;&nbsp;&nbsp;" % (self.app.dx, self.app.dy))
  773. units = self.app.defaults["units"].lower()
  774. self.app.plotcanvas.text_hud.text = \
  775. 'Dx:\t{:<.4f} [{:s}]\nDy:\t{:<.4f} [{:s}]\n\nX: \t{:<.4f} [{:s}]\nY: \t{:<.4f} [{:s}]'.format(
  776. self.app.dx, units, self.app.dy, units, curr_pos[0], units, curr_pos[1], units)
  777. # draw the utility geometry
  778. if self.first_click:
  779. self.app.delete_selection_shape()
  780. self.app.draw_moving_selection_shape(old_coords=(self.cursor_pos[0], self.cursor_pos[1]),
  781. coords=(curr_pos[0], curr_pos[1]))
  782. def on_copper_thieving(self, thieving_obj, ref_obj=None, c_val=None, margin=None, run_threaded=True):
  783. """
  784. :param thieving_obj:
  785. :param ref_obj:
  786. :param c_val:
  787. :param margin:
  788. :param run_threaded:
  789. :return:
  790. """
  791. if run_threaded:
  792. proc = self.app.proc_container.new('%s ...' % _("Thieving"))
  793. else:
  794. QtWidgets.QApplication.processEvents()
  795. self.app.proc_container.view.set_busy('%s ...' % _("Thieving"))
  796. # #####################################################################
  797. # ####### Read the parameters #########################################
  798. # #####################################################################
  799. log.debug("Copper Thieving Tool started. Reading parameters.")
  800. self.app.inform.emit(_("Copper Thieving Tool started. Reading parameters."))
  801. ref_selected = self.reference_radio.get_value()
  802. if c_val is None:
  803. c_val = float(self.app.defaults["tools_copperfill_clearance"])
  804. if margin is None:
  805. margin = float(self.app.defaults["tools_copperfill_margin"])
  806. fill_type = self.fill_type_radio.get_value()
  807. dot_dia = self.dot_dia_entry.get_value()
  808. dot_spacing = self.dot_spacing_entry.get_value()
  809. square_size = self.square_size_entry.get_value()
  810. square_spacing = self.squares_spacing_entry.get_value()
  811. line_size = self.line_size_entry.get_value()
  812. line_spacing = self.lines_spacing_entry.get_value()
  813. # make sure that the source object solid geometry is an Iterable
  814. if not isinstance(self.grb_object.solid_geometry, Iterable):
  815. self.grb_object.solid_geometry = [self.grb_object.solid_geometry]
  816. def job_thread_thieving(app_obj):
  817. # #########################################################################################
  818. # Prepare isolation polygon. This will create the clearance over the Gerber features ######
  819. # #########################################################################################
  820. log.debug("Copper Thieving Tool. Preparing isolation polygons.")
  821. app_obj.app.inform.emit(_("Copper Thieving Tool. Preparing isolation polygons."))
  822. # variables to display the percentage of work done
  823. geo_len = 0
  824. try:
  825. for pol in app_obj.grb_object.solid_geometry:
  826. geo_len += 1
  827. except TypeError:
  828. geo_len = 1
  829. old_disp_number = 0
  830. pol_nr = 0
  831. clearance_geometry = []
  832. try:
  833. for pol in app_obj.grb_object.solid_geometry:
  834. if app_obj.app.abort_flag:
  835. # graceful abort requested by the user
  836. raise grace
  837. clearance_geometry.append(
  838. pol.buffer(c_val, int(int(app_obj.geo_steps_per_circle) / 4))
  839. )
  840. pol_nr += 1
  841. disp_number = int(np.interp(pol_nr, [0, geo_len], [0, 100]))
  842. if old_disp_number < disp_number <= 100:
  843. app_obj.app.proc_container.update_view_text(' %s ... %d%%' %
  844. (_("Thieving"), int(disp_number)))
  845. old_disp_number = disp_number
  846. except TypeError:
  847. # taking care of the case when the self.solid_geometry is just a single Polygon, not a list or a
  848. # MultiPolygon (not an iterable)
  849. clearance_geometry.append(
  850. app_obj.grb_object.solid_geometry.buffer(c_val, int(int(app_obj.geo_steps_per_circle) / 4))
  851. )
  852. app_obj.app.proc_container.update_view_text(' %s ...' % _("Buffering"))
  853. clearance_geometry = unary_union(clearance_geometry)
  854. # #########################################################################################
  855. # Prepare the area to fill with copper. ###################################################
  856. # #########################################################################################
  857. log.debug("Copper Thieving Tool. Preparing areas to fill with copper.")
  858. app_obj.app.inform.emit(_("Copper Thieving Tool. Preparing areas to fill with copper."))
  859. try:
  860. if ref_obj is None or ref_obj == 'itself':
  861. working_obj = thieving_obj
  862. else:
  863. working_obj = ref_obj
  864. except Exception as e:
  865. log.debug("ToolCopperThieving.on_copper_thieving() --> %s" % str(e))
  866. return 'fail'
  867. app_obj.app.proc_container.update_view_text(' %s' % _("Working..."))
  868. if ref_selected == 'itself':
  869. geo_n = working_obj.solid_geometry
  870. try:
  871. if app_obj.bbox_type_radio.get_value() == 'min':
  872. if isinstance(geo_n, MultiPolygon):
  873. env_obj = geo_n.convex_hull
  874. elif (isinstance(geo_n, MultiPolygon) and len(geo_n) == 1) or \
  875. (isinstance(geo_n, list) and len(geo_n) == 1) and isinstance(geo_n[0], Polygon):
  876. env_obj = cascaded_union(geo_n)
  877. else:
  878. env_obj = cascaded_union(geo_n)
  879. env_obj = env_obj.convex_hull
  880. bounding_box = env_obj.buffer(distance=margin, join_style=base.JOIN_STYLE.mitre)
  881. else:
  882. if isinstance(geo_n, Polygon):
  883. bounding_box = geo_n.buffer(distance=margin, join_style=base.JOIN_STYLE.mitre).exterior
  884. elif isinstance(geo_n, list):
  885. geo_n = unary_union(geo_n)
  886. bounding_box = geo_n.buffer(distance=margin, join_style=base.JOIN_STYLE.mitre).exterior
  887. elif isinstance(geo_n, MultiPolygon):
  888. x0, y0, x1, y1 = geo_n.bounds
  889. geo = box(x0, y0, x1, y1)
  890. bounding_box = geo.buffer(distance=margin, join_style=base.JOIN_STYLE.mitre)
  891. else:
  892. app_obj.app.inform.emit(
  893. '[ERROR_NOTCL] %s: %s' % (_("Geometry not supported for bounding box"), type(geo_n))
  894. )
  895. return 'fail'
  896. except Exception as e:
  897. log.debug("ToolCopperFIll.on_copper_thieving() 'itself' --> %s" % str(e))
  898. app_obj.app.inform.emit('[ERROR_NOTCL] %s' % _("No object available."))
  899. return 'fail'
  900. elif ref_selected == 'area':
  901. geo_buff_list = []
  902. try:
  903. for poly in working_obj:
  904. if app_obj.app.abort_flag:
  905. # graceful abort requested by the user
  906. raise grace
  907. geo_buff_list.append(poly.buffer(distance=margin, join_style=base.JOIN_STYLE.mitre))
  908. except TypeError:
  909. geo_buff_list.append(working_obj.buffer(distance=margin, join_style=base.JOIN_STYLE.mitre))
  910. bounding_box = MultiPolygon(geo_buff_list)
  911. else: # ref_selected == 'box'
  912. geo_n = working_obj.solid_geometry
  913. if working_obj.kind == 'geometry':
  914. try:
  915. __ = iter(geo_n)
  916. except Exception as e:
  917. log.debug("ToolCopperFIll.on_copper_thieving() 'box' --> %s" % str(e))
  918. geo_n = [geo_n]
  919. geo_buff_list = []
  920. for poly in geo_n:
  921. if app_obj.app.abort_flag:
  922. # graceful abort requested by the user
  923. raise grace
  924. geo_buff_list.append(poly.buffer(distance=margin, join_style=base.JOIN_STYLE.mitre))
  925. bounding_box = cascaded_union(geo_buff_list)
  926. elif working_obj.kind == 'gerber':
  927. geo_n = cascaded_union(geo_n).convex_hull
  928. bounding_box = cascaded_union(thieving_obj.solid_geometry).convex_hull.intersection(geo_n)
  929. bounding_box = bounding_box.buffer(distance=margin, join_style=base.JOIN_STYLE.mitre)
  930. else:
  931. app_obj.app.inform.emit('[ERROR_NOTCL] %s' % _("The reference object type is not supported."))
  932. return 'fail'
  933. log.debug("Copper Thieving Tool. Finished creating areas to fill with copper.")
  934. app_obj.app.inform.emit(_("Copper Thieving Tool. Appending new geometry and buffering."))
  935. # #########################################################################################
  936. # ########## Generate filling geometry. ###################################################
  937. # #########################################################################################
  938. app_obj.new_solid_geometry = bounding_box.difference(clearance_geometry)
  939. # determine the bounding box polygon for the entire Gerber object to which we add copper thieving
  940. # if isinstance(geo_n, list):
  941. # env_obj = unary_union(geo_n).buffer(distance=margin, join_style=base.JOIN_STYLE.mitre)
  942. # else:
  943. # env_obj = geo_n.buffer(distance=margin, join_style=base.JOIN_STYLE.mitre)
  944. #
  945. # x0, y0, x1, y1 = env_obj.bounds
  946. # bounding_box = box(x0, y0, x1, y1)
  947. app_obj.app.proc_container.update_view_text(' %s' % _("Create geometry"))
  948. bounding_box = thieving_obj.solid_geometry.envelope.buffer(
  949. distance=margin,
  950. join_style=base.JOIN_STYLE.mitre
  951. )
  952. x0, y0, x1, y1 = bounding_box.bounds
  953. if fill_type == 'dot' or fill_type == 'square':
  954. # build the MultiPolygon of dots/squares that will fill the entire bounding box
  955. thieving_list = []
  956. if fill_type == 'dot':
  957. radius = dot_dia / 2.0
  958. new_x = x0 + radius
  959. new_y = y0 + radius
  960. while new_x <= x1 - radius:
  961. while new_y <= y1 - radius:
  962. dot_geo = Point((new_x, new_y)).buffer(radius, resolution=64)
  963. thieving_list.append(dot_geo)
  964. new_y += dot_dia + dot_spacing
  965. new_x += dot_dia + dot_spacing
  966. new_y = y0 + radius
  967. else:
  968. h_size = square_size / 2.0
  969. new_x = x0 + h_size
  970. new_y = y0 + h_size
  971. while new_x <= x1 - h_size:
  972. while new_y <= y1 - h_size:
  973. a, b, c, d = (Point((new_x, new_y)).buffer(h_size)).bounds
  974. square_geo = box(a, b, c, d)
  975. thieving_list.append(square_geo)
  976. new_y += square_size + square_spacing
  977. new_x += square_size + square_spacing
  978. new_y = y0 + h_size
  979. thieving_box_geo = MultiPolygon(thieving_list)
  980. dx = bounding_box.centroid.x - thieving_box_geo.centroid.x
  981. dy = bounding_box.centroid.y - thieving_box_geo.centroid.y
  982. thieving_box_geo = affinity.translate(thieving_box_geo, xoff=dx, yoff=dy)
  983. try:
  984. _it = iter(app_obj.new_solid_geometry)
  985. except TypeError:
  986. app_obj.new_solid_geometry = [app_obj.new_solid_geometry]
  987. try:
  988. _it = iter(thieving_box_geo)
  989. except TypeError:
  990. thieving_box_geo = [thieving_box_geo]
  991. thieving_geo = []
  992. for dot_geo in thieving_box_geo:
  993. for geo_t in app_obj.new_solid_geometry:
  994. if dot_geo.within(geo_t):
  995. thieving_geo.append(dot_geo)
  996. app_obj.new_solid_geometry = thieving_geo
  997. if fill_type == 'line':
  998. half_thick_line = line_size / 2.0
  999. # create a thick polygon-line that surrounds the copper features
  1000. outline_geometry = []
  1001. try:
  1002. for pol in app_obj.grb_object.solid_geometry:
  1003. if app_obj.app.abort_flag:
  1004. # graceful abort requested by the user
  1005. raise grace
  1006. outline_geometry.append(
  1007. pol.buffer(c_val+half_thick_line, int(int(app_obj.geo_steps_per_circle) / 4))
  1008. )
  1009. pol_nr += 1
  1010. disp_number = int(np.interp(pol_nr, [0, geo_len], [0, 100]))
  1011. if old_disp_number < disp_number <= 100:
  1012. app_obj.app.proc_container.update_view_text(' %s ... %d%%' %
  1013. (_("Buffering"), int(disp_number)))
  1014. old_disp_number = disp_number
  1015. except TypeError:
  1016. # taking care of the case when the self.solid_geometry is just a single Polygon, not a list or a
  1017. # MultiPolygon (not an iterable)
  1018. outline_geometry.append(
  1019. app_obj.grb_object.solid_geometry.buffer(
  1020. c_val+half_thick_line,
  1021. int(int(app_obj.geo_steps_per_circle) / 4)
  1022. )
  1023. )
  1024. app_obj.app.proc_container.update_view_text(' %s' % _("Buffering"))
  1025. outline_geometry = unary_union(outline_geometry)
  1026. outline_line = []
  1027. try:
  1028. for geo_o in outline_geometry:
  1029. outline_line.append(
  1030. geo_o.exterior.buffer(
  1031. half_thick_line, resolution=int(int(app_obj.geo_steps_per_circle) / 4)
  1032. )
  1033. )
  1034. except TypeError:
  1035. outline_line.append(
  1036. outline_geometry.exterior.buffer(
  1037. half_thick_line, resolution=int(int(app_obj.geo_steps_per_circle) / 4)
  1038. )
  1039. )
  1040. outline_geometry = unary_union(outline_line)
  1041. # create a polygon-line that surrounds in the inside the bounding box polygon of the target Gerber
  1042. box_outline_geo = box(x0, y0, x1, y1).buffer(-half_thick_line)
  1043. box_outline_geo_exterior = box_outline_geo.exterior
  1044. box_outline_geometry = box_outline_geo_exterior.buffer(
  1045. half_thick_line,
  1046. resolution=int(int(app_obj.geo_steps_per_circle) / 4)
  1047. )
  1048. bx0, by0, bx1, by1 = box_outline_geo.bounds
  1049. thieving_lines_geo = []
  1050. new_x = bx0
  1051. new_y = by0
  1052. while new_x <= x1 - half_thick_line:
  1053. line_geo = LineString([(new_x, by0), (new_x, by1)]).buffer(
  1054. half_thick_line,
  1055. resolution=int(int(app_obj.geo_steps_per_circle) / 4)
  1056. )
  1057. thieving_lines_geo.append(line_geo)
  1058. new_x += line_size + line_spacing
  1059. while new_y <= y1 - half_thick_line:
  1060. line_geo = LineString([(bx0, new_y), (bx1, new_y)]).buffer(
  1061. half_thick_line,
  1062. resolution=int(int(app_obj.geo_steps_per_circle) / 4)
  1063. )
  1064. thieving_lines_geo.append(line_geo)
  1065. new_y += line_size + line_spacing
  1066. # merge everything together
  1067. diff_lines_geo = []
  1068. for line_poly in thieving_lines_geo:
  1069. rest_line = line_poly.difference(clearance_geometry)
  1070. diff_lines_geo.append(rest_line)
  1071. app_obj.flatten([outline_geometry, box_outline_geometry, diff_lines_geo])
  1072. app_obj.new_solid_geometry = app_obj.flat_geometry
  1073. app_obj.app.proc_container.update_view_text(' %s' % _("Append geometry"))
  1074. geo_list = app_obj.grb_object.solid_geometry
  1075. if isinstance(app_obj.grb_object.solid_geometry, MultiPolygon):
  1076. geo_list = list(app_obj.grb_object.solid_geometry.geoms)
  1077. if '0' not in app_obj.grb_object.apertures:
  1078. app_obj.grb_object.apertures['0'] = {}
  1079. app_obj.grb_object.apertures['0']['geometry'] = []
  1080. app_obj.grb_object.apertures['0']['type'] = 'REG'
  1081. app_obj.grb_object.apertures['0']['size'] = 0.0
  1082. try:
  1083. for poly in app_obj.new_solid_geometry:
  1084. # append to the new solid geometry
  1085. geo_list.append(poly)
  1086. # append into the '0' aperture
  1087. geo_elem = {}
  1088. geo_elem['solid'] = poly
  1089. geo_elem['follow'] = poly.exterior
  1090. app_obj.grb_object.apertures['0']['geometry'].append(deepcopy(geo_elem))
  1091. except TypeError:
  1092. # append to the new solid geometry
  1093. geo_list.append(app_obj.new_solid_geometry)
  1094. # append into the '0' aperture
  1095. geo_elem = {}
  1096. geo_elem['solid'] = app_obj.new_solid_geometry
  1097. geo_elem['follow'] = app_obj.new_solid_geometry.exterior
  1098. app_obj.grb_object.apertures['0']['geometry'].append(deepcopy(geo_elem))
  1099. app_obj.grb_object.solid_geometry = MultiPolygon(geo_list).buffer(0.0000001).buffer(-0.0000001)
  1100. app_obj.app.proc_container.update_view_text(' %s' % _("Append source file"))
  1101. # update the source file with the new geometry:
  1102. app_obj.grb_object.source_file = app_obj.app.export_gerber(obj_name=app_obj.grb_object.options['name'],
  1103. filename=None,
  1104. local_use=app_obj.grb_object,
  1105. use_thread=False)
  1106. app_obj.app.proc_container.update_view_text(' %s' % '')
  1107. app_obj.on_exit()
  1108. app_obj.app.inform.emit('[success] %s' % _("Copper Thieving Tool done."))
  1109. if run_threaded:
  1110. self.app.worker_task.emit({'fcn': job_thread_thieving, 'params': [self]})
  1111. else:
  1112. job_thread_thieving(self)
  1113. def on_add_ppm(self):
  1114. run_threaded = True
  1115. if run_threaded:
  1116. proc = self.app.proc_container.new('%s ...' % _("P-Plating Mask"))
  1117. else:
  1118. QtWidgets.QApplication.processEvents()
  1119. self.app.proc_container.view.set_busy('%s ...' % _("P-Plating Mask"))
  1120. if run_threaded:
  1121. self.app.worker_task.emit({'fcn': self.on_new_pattern_plating_object, 'params': []})
  1122. else:
  1123. self.on_new_pattern_plating_object()
  1124. def on_new_pattern_plating_object(self):
  1125. # get the Gerber object on which the Copper thieving will be inserted
  1126. selection_index = self.sm_object_combo.currentIndex()
  1127. model_index = self.app.collection.index(selection_index, 0, self.sm_object_combo.rootModelIndex())
  1128. try:
  1129. self.sm_object = model_index.internalPointer().obj
  1130. except Exception as e:
  1131. log.debug("ToolCopperThieving.on_add_ppm() --> %s" % str(e))
  1132. self.app.inform.emit('[WARNING_NOTCL] %s' % _("There is no Gerber object loaded ..."))
  1133. return 'fail'
  1134. ppm_clearance = self.clearance_ppm_entry.get_value()
  1135. rb_thickness = self.rb_thickness
  1136. self.app.proc_container.update_view_text(' %s' % _("Append PP-M geometry"))
  1137. geo_list = self.sm_object.solid_geometry
  1138. if isinstance(self.sm_object.solid_geometry, MultiPolygon):
  1139. geo_list = list(self.sm_object.solid_geometry.geoms)
  1140. # if the clearance is negative apply it to the original soldermask too
  1141. if ppm_clearance < 0:
  1142. temp_geo_list = []
  1143. for geo in geo_list:
  1144. temp_geo_list.append(geo.buffer(ppm_clearance))
  1145. geo_list = temp_geo_list
  1146. plated_area = 0.0
  1147. for geo in geo_list:
  1148. plated_area += geo.area
  1149. if self.new_solid_geometry:
  1150. for geo in self.new_solid_geometry:
  1151. plated_area += geo.area
  1152. if self.robber_geo:
  1153. plated_area += self.robber_geo.area
  1154. self.plated_area_entry.set_value(plated_area)
  1155. thieving_solid_geo = self.new_solid_geometry
  1156. robber_solid_geo = self.robber_geo
  1157. robber_line = self.robber_line
  1158. def obj_init(grb_obj, app_obj):
  1159. grb_obj.multitool = False
  1160. grb_obj.source_file = []
  1161. grb_obj.multigeo = False
  1162. grb_obj.follow = False
  1163. grb_obj.apertures = {}
  1164. grb_obj.solid_geometry = []
  1165. # try:
  1166. # grb_obj.options['xmin'] = 0
  1167. # grb_obj.options['ymin'] = 0
  1168. # grb_obj.options['xmax'] = 0
  1169. # grb_obj.options['ymax'] = 0
  1170. # except KeyError:
  1171. # pass
  1172. # if we have copper thieving geometry, add it
  1173. if thieving_solid_geo:
  1174. if '0' not in grb_obj.apertures:
  1175. grb_obj.apertures['0'] = {}
  1176. grb_obj.apertures['0']['geometry'] = []
  1177. grb_obj.apertures['0']['type'] = 'REG'
  1178. grb_obj.apertures['0']['size'] = 0.0
  1179. try:
  1180. for poly in thieving_solid_geo:
  1181. poly_b = poly.buffer(ppm_clearance)
  1182. # append to the new solid geometry
  1183. geo_list.append(poly_b)
  1184. # append into the '0' aperture
  1185. geo_elem = {}
  1186. geo_elem['solid'] = poly_b
  1187. geo_elem['follow'] = poly_b.exterior
  1188. grb_obj.apertures['0']['geometry'].append(deepcopy(geo_elem))
  1189. except TypeError:
  1190. # append to the new solid geometry
  1191. geo_list.append(thieving_solid_geo.buffer(ppm_clearance))
  1192. # append into the '0' aperture
  1193. geo_elem = {}
  1194. geo_elem['solid'] = thieving_solid_geo.buffer(ppm_clearance)
  1195. geo_elem['follow'] = thieving_solid_geo.buffer(ppm_clearance).exterior
  1196. grb_obj.apertures['0']['geometry'].append(deepcopy(geo_elem))
  1197. # if we have robber bar geometry, add it
  1198. if robber_solid_geo:
  1199. aperture_found = None
  1200. for ap_id, ap_val in grb_obj.apertures.items():
  1201. if ap_val['type'] == 'C' and ap_val['size'] == app_obj.rb_thickness + ppm_clearance:
  1202. aperture_found = ap_id
  1203. break
  1204. if aperture_found:
  1205. geo_elem = {}
  1206. geo_elem['solid'] = robber_solid_geo
  1207. geo_elem['follow'] = robber_line
  1208. grb_obj.apertures[aperture_found]['geometry'].append(deepcopy(geo_elem))
  1209. else:
  1210. ap_keys = list(grb_obj.apertures.keys())
  1211. max_apid = int(max(ap_keys))
  1212. if ap_keys and max_apid != 0:
  1213. new_apid = str(max_apid + 1)
  1214. else:
  1215. new_apid = '10'
  1216. grb_obj.apertures[new_apid] = {}
  1217. grb_obj.apertures[new_apid]['type'] = 'C'
  1218. grb_obj.apertures[new_apid]['size'] = rb_thickness + ppm_clearance
  1219. grb_obj.apertures[new_apid]['geometry'] = []
  1220. geo_elem = {}
  1221. geo_elem['solid'] = robber_solid_geo.buffer(ppm_clearance)
  1222. geo_elem['follow'] = Polygon(robber_line).buffer(ppm_clearance / 2.0).exterior
  1223. grb_obj.apertures[new_apid]['geometry'].append(deepcopy(geo_elem))
  1224. geo_list.append(robber_solid_geo.buffer(ppm_clearance))
  1225. grb_obj.solid_geometry = MultiPolygon(geo_list).buffer(0.0000001).buffer(-0.0000001)
  1226. app_obj.proc_container.update_view_text(' %s' % _("Append source file"))
  1227. # update the source file with the new geometry:
  1228. grb_obj.source_file = app_obj.export_gerber(obj_name=name,
  1229. filename=None,
  1230. local_use=grb_obj,
  1231. use_thread=False)
  1232. app_obj.proc_container.update_view_text(' %s' % '')
  1233. # Object name
  1234. obj_name, separatpr, obj_extension = self.sm_object.options['name'].rpartition('.')
  1235. name = '%s_%s.%s' % (obj_name, 'plating_mask', obj_extension)
  1236. self.app.app_obj.new_object('gerber', name, obj_init, autoselected=False)
  1237. # Register recent file
  1238. self.app.file_opened.emit("gerber", name)
  1239. self.on_exit()
  1240. self.app.inform.emit('[success] %s' % _("Generating Pattern Plating Mask done."))
  1241. def replot(self, obj):
  1242. def worker_task():
  1243. with self.app.proc_container.new('%s...' % _("Plotting")):
  1244. obj.plot()
  1245. self.app.worker_task.emit({'fcn': worker_task, 'params': []})
  1246. def on_exit(self):
  1247. # plot the objects
  1248. if self.grb_object:
  1249. self.replot(obj=self.grb_object)
  1250. if self.sm_object:
  1251. self.replot(obj=self.sm_object)
  1252. # update the bounding box values
  1253. try:
  1254. a, b, c, d = self.grb_object.bounds()
  1255. self.grb_object.options['xmin'] = a
  1256. self.grb_object.options['ymin'] = b
  1257. self.grb_object.options['xmax'] = c
  1258. self.grb_object.options['ymax'] = d
  1259. except Exception as e:
  1260. log.debug("ToolCopperThieving.on_exit() bounds -> copper thieving Gerber error --> %s" % str(e))
  1261. # update the bounding box values
  1262. try:
  1263. a, b, c, d = self.sm_object.bounds()
  1264. self.sm_object.options['xmin'] = a
  1265. self.sm_object.options['ymin'] = b
  1266. self.sm_object.options['xmax'] = c
  1267. self.sm_object.options['ymax'] = d
  1268. except Exception as e:
  1269. log.debug("ToolCopperThieving.on_exit() bounds -> pattern plating mask error --> %s" % str(e))
  1270. # reset the variables
  1271. self.grb_object = None
  1272. self.sm_object = None
  1273. self.ref_obj = None
  1274. self.sel_rect = []
  1275. # Events ID
  1276. self.mr = None
  1277. self.mm = None
  1278. # Mouse cursor positions
  1279. self.mouse_is_dragging = False
  1280. self.cursor_pos = (0, 0)
  1281. self.first_click = False
  1282. # if True it means we exited from tool in the middle of area adding therefore disconnect the events
  1283. if self.area_method is True:
  1284. self.app.delete_selection_shape()
  1285. self.area_method = False
  1286. if self.app.is_legacy is False:
  1287. self.app.plotcanvas.graph_event_disconnect('mouse_release', self.on_mouse_release)
  1288. self.app.plotcanvas.graph_event_disconnect('mouse_move', self.on_mouse_move)
  1289. else:
  1290. self.app.plotcanvas.graph_event_disconnect(self.mr)
  1291. self.app.plotcanvas.graph_event_disconnect(self.mm)
  1292. self.app.mp = self.app.plotcanvas.graph_event_connect('mouse_press',
  1293. self.app.on_mouse_click_over_plot)
  1294. self.app.mm = self.app.plotcanvas.graph_event_connect('mouse_move',
  1295. self.app.on_mouse_move_over_plot)
  1296. self.app.mr = self.app.plotcanvas.graph_event_connect('mouse_release',
  1297. self.app.on_mouse_click_release_over_plot)
  1298. self.app.call_source = "app"
  1299. self.app.inform.emit('[success] %s' % _("Copper Thieving Tool exit."))
  1300. def flatten(self, geometry):
  1301. """
  1302. Creates a list of non-iterable linear geometry objects.
  1303. :param geometry: Shapely type or list or list of list of such.
  1304. Results are placed in self.flat_geometry
  1305. """
  1306. # ## If iterable, expand recursively.
  1307. try:
  1308. for geo in geometry:
  1309. if geo is not None:
  1310. self.flatten(geometry=geo)
  1311. # ## Not iterable, do the actual indexing and add.
  1312. except TypeError:
  1313. self.flat_geometry.append(geometry)
  1314. return self.flat_geometry