FlatCAMGerber.py 72 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514
  1. # ##########################################################
  2. # FlatCAM: 2D Post-processing for Manufacturing #
  3. # http://flatcam.org #
  4. # Author: Juan Pablo Caram (c) #
  5. # Date: 2/5/2014 #
  6. # MIT Licence #
  7. # ##########################################################
  8. # ##########################################################
  9. # File modified by: Marius Stanciu #
  10. # ##########################################################
  11. from shapely.geometry import Point, Polygon, MultiPolygon, MultiLineString, LineString, LinearRing
  12. from shapely.ops import cascaded_union
  13. from appParsers.ParseGerber import Gerber
  14. from appObjects.FlatCAMObj import *
  15. import math
  16. import numpy as np
  17. from copy import deepcopy
  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. class GerberObject(FlatCAMObj, Gerber):
  25. """
  26. Represents Gerber code.
  27. """
  28. optionChanged = QtCore.pyqtSignal(str)
  29. replotApertures = QtCore.pyqtSignal()
  30. do_buffer_signal = QtCore.pyqtSignal()
  31. ui_type = GerberObjectUI
  32. @staticmethod
  33. def merge(grb_list, grb_final):
  34. """
  35. Merges the geometry of objects in geo_list into
  36. the geometry of geo_final.
  37. :param grb_list: List of GerberObject Objects to join.
  38. :param grb_final: Destination GeometryObject object.
  39. :return: None
  40. """
  41. if grb_final.solid_geometry is None:
  42. grb_final.solid_geometry = []
  43. grb_final.follow_geometry = []
  44. if not grb_final.apertures:
  45. grb_final.apertures = {}
  46. if type(grb_final.solid_geometry) is not list:
  47. grb_final.solid_geometry = [grb_final.solid_geometry]
  48. grb_final.follow_geometry = [grb_final.follow_geometry]
  49. for grb in grb_list:
  50. # Expand lists
  51. if type(grb) is list:
  52. GerberObject.merge(grb_list=grb, grb_final=grb_final)
  53. else: # If not list, just append
  54. for option in grb.options:
  55. if option != 'name':
  56. try:
  57. grb_final.options[option] = grb.options[option]
  58. except KeyError:
  59. log.warning("Failed to copy option.", option)
  60. try:
  61. for geos in grb.solid_geometry:
  62. grb_final.solid_geometry.append(geos)
  63. grb_final.follow_geometry.append(geos)
  64. except TypeError:
  65. grb_final.solid_geometry.append(grb.solid_geometry)
  66. grb_final.follow_geometry.append(grb.solid_geometry)
  67. for ap in grb.apertures:
  68. if ap not in grb_final.apertures:
  69. grb_final.apertures[ap] = grb.apertures[ap]
  70. else:
  71. # create a list of integers out of the grb.apertures keys and find the max of that value
  72. # then, the aperture duplicate is assigned an id value incremented with 1,
  73. # and finally made string because the apertures dict keys are strings
  74. max_ap = str(max([int(k) for k in grb_final.apertures.keys()]) + 1)
  75. grb_final.apertures[max_ap] = {}
  76. grb_final.apertures[max_ap]['geometry'] = []
  77. for k, v in grb.apertures[ap].items():
  78. grb_final.apertures[max_ap][k] = deepcopy(v)
  79. grb_final.solid_geometry = MultiPolygon(grb_final.solid_geometry)
  80. grb_final.follow_geometry = MultiPolygon(grb_final.follow_geometry)
  81. def __init__(self, name):
  82. self.decimals = self.app.decimals
  83. self.circle_steps = int(self.app.defaults["gerber_circle_steps"])
  84. Gerber.__init__(self, steps_per_circle=self.circle_steps)
  85. FlatCAMObj.__init__(self, name)
  86. self.kind = "gerber"
  87. # The 'name' is already in self.options from FlatCAMObj
  88. # Automatically updates the UI
  89. self.options.update({
  90. "plot": True,
  91. "multicolored": False,
  92. "solid": False,
  93. "noncoppermargin": 0.0,
  94. "noncopperrounded": False,
  95. "bboxmargin": 0.0,
  96. "bboxrounded": False,
  97. "aperture_display": False,
  98. "follow": False,
  99. "milling_type": 'cl',
  100. })
  101. # type of isolation: 0 = exteriors, 1 = interiors, 2 = complete isolation (both interiors and exteriors)
  102. self.iso_type = 2
  103. self.multigeo = False
  104. self.follow = False
  105. self.apertures_row = 0
  106. # store the source file here
  107. self.source_file = ""
  108. # list of rows with apertures plotted
  109. self.marked_rows = []
  110. # Mouse events
  111. self.mr = None
  112. self.mm = None
  113. self.mp = None
  114. # dict to store the polygons selected for isolation; key is the shape added to be plotted and value is the poly
  115. self.poly_dict = {}
  116. # store the status of grid snapping
  117. self.grid_status_memory = None
  118. self.units_found = self.app.defaults['units']
  119. self.fill_color = self.app.defaults['gerber_plot_fill']
  120. self.outline_color = self.app.defaults['gerber_plot_line']
  121. self.alpha_level = 'bf'
  122. # keep track if the UI is built so we don't have to build it every time
  123. self.ui_build = False
  124. # build only once the aperture storage (takes time)
  125. self.build_aperture_storage = False
  126. # Attributes to be included in serialization
  127. # Always append to it because it carries contents
  128. # from predecessors.
  129. self.ser_attrs += ['options', 'kind', 'fill_color', 'outline_color', 'alpha_level']
  130. def set_ui(self, ui):
  131. """
  132. Maps options with GUI inputs.
  133. Connects GUI events to methods.
  134. :param ui: GUI object.
  135. :type ui: GerberObjectUI
  136. :return: None
  137. """
  138. FlatCAMObj.set_ui(self, ui)
  139. log.debug("GerberObject.set_ui()")
  140. self.units = self.app.defaults['units'].upper()
  141. self.replotApertures.connect(self.on_mark_cb_click_table)
  142. self.form_fields.update({
  143. "plot": self.ui.plot_cb,
  144. "multicolored": self.ui.multicolored_cb,
  145. "solid": self.ui.solid_cb,
  146. "noncoppermargin": self.ui.noncopper_margin_entry,
  147. "noncopperrounded": self.ui.noncopper_rounded_cb,
  148. "bboxmargin": self.ui.bbmargin_entry,
  149. "bboxrounded": self.ui.bbrounded_cb,
  150. "aperture_display": self.ui.aperture_table_visibility_cb,
  151. "follow": self.ui.follow_cb
  152. })
  153. # Fill form fields only on object create
  154. self.to_form()
  155. assert isinstance(self.ui, GerberObjectUI)
  156. self.ui.plot_cb.stateChanged.connect(self.on_plot_cb_click)
  157. self.ui.solid_cb.stateChanged.connect(self.on_solid_cb_click)
  158. self.ui.multicolored_cb.stateChanged.connect(self.on_multicolored_cb_click)
  159. # Tools
  160. self.ui.iso_button.clicked.connect(self.app.isolation_tool.run)
  161. self.ui.generate_ncc_button.clicked.connect(self.app.ncclear_tool.run)
  162. self.ui.generate_cutout_button.clicked.connect(self.app.cutout_tool.run)
  163. self.ui.generate_bb_button.clicked.connect(self.on_generatebb_button_click)
  164. self.ui.generate_noncopper_button.clicked.connect(self.on_generatenoncopper_button_click)
  165. self.ui.aperture_table_visibility_cb.stateChanged.connect(self.on_aperture_table_visibility_change)
  166. self.ui.follow_cb.stateChanged.connect(self.on_follow_cb_click)
  167. self.do_buffer_signal.connect(self.on_generate_buffer)
  168. # Show/Hide Advanced Options
  169. if self.app.defaults["global_app_level"] == 'b':
  170. self.ui.level.setText('<span style="color:green;"><b>%s</b></span>' % _('Basic'))
  171. self.ui.apertures_table_label.hide()
  172. self.ui.aperture_table_visibility_cb.hide()
  173. self.ui.follow_cb.hide()
  174. else:
  175. self.ui.level.setText('<span style="color:red;"><b>%s</b></span>' % _('Advanced'))
  176. if self.app.defaults["gerber_buffering"] == 'no':
  177. self.ui.create_buffer_button.show()
  178. try:
  179. self.ui.create_buffer_button.clicked.disconnect(self.on_generate_buffer)
  180. except TypeError:
  181. pass
  182. self.ui.create_buffer_button.clicked.connect(self.on_generate_buffer)
  183. else:
  184. self.ui.create_buffer_button.hide()
  185. # set initial state of the aperture table and associated widgets
  186. self.on_aperture_table_visibility_change()
  187. self.build_ui()
  188. self.units_found = self.app.defaults['units']
  189. def build_ui(self):
  190. FlatCAMObj.build_ui(self)
  191. if self.ui.aperture_table_visibility_cb.get_value() and self.ui_build is False:
  192. self.ui_build = True
  193. try:
  194. # if connected, disconnect the signal from the slot on item_changed as it creates issues
  195. self.ui.apertures_table.itemChanged.disconnect()
  196. except (TypeError, AttributeError):
  197. pass
  198. self.apertures_row = 0
  199. aper_no = self.apertures_row + 1
  200. sort = []
  201. for k, v in list(self.apertures.items()):
  202. sort.append(int(k))
  203. sorted_apertures = sorted(sort)
  204. n = len(sorted_apertures)
  205. self.ui.apertures_table.setRowCount(n)
  206. for ap_code in sorted_apertures:
  207. ap_code = str(ap_code)
  208. ap_id_item = QtWidgets.QTableWidgetItem('%d' % int(self.apertures_row + 1))
  209. ap_id_item.setFlags(QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
  210. self.ui.apertures_table.setItem(self.apertures_row, 0, ap_id_item) # Tool name/id
  211. ap_code_item = QtWidgets.QTableWidgetItem(ap_code)
  212. ap_code_item.setFlags(QtCore.Qt.ItemIsEnabled)
  213. ap_type_item = QtWidgets.QTableWidgetItem(str(self.apertures[ap_code]['type']))
  214. ap_type_item.setFlags(QtCore.Qt.ItemIsEnabled)
  215. if str(self.apertures[ap_code]['type']) == 'R' or str(self.apertures[ap_code]['type']) == 'O':
  216. ap_dim_item = QtWidgets.QTableWidgetItem(
  217. '%.*f, %.*f' % (self.decimals, self.apertures[ap_code]['width'],
  218. self.decimals, self.apertures[ap_code]['height']
  219. )
  220. )
  221. ap_dim_item.setFlags(QtCore.Qt.ItemIsEnabled)
  222. elif str(self.apertures[ap_code]['type']) == 'P':
  223. ap_dim_item = QtWidgets.QTableWidgetItem(
  224. '%.*f, %.*f' % (self.decimals, self.apertures[ap_code]['diam'],
  225. self.decimals, self.apertures[ap_code]['nVertices'])
  226. )
  227. ap_dim_item.setFlags(QtCore.Qt.ItemIsEnabled)
  228. else:
  229. ap_dim_item = QtWidgets.QTableWidgetItem('')
  230. ap_dim_item.setFlags(QtCore.Qt.ItemIsEnabled)
  231. try:
  232. if self.apertures[ap_code]['size'] is not None:
  233. ap_size_item = QtWidgets.QTableWidgetItem(
  234. '%.*f' % (self.decimals, float(self.apertures[ap_code]['size'])))
  235. else:
  236. ap_size_item = QtWidgets.QTableWidgetItem('')
  237. except KeyError:
  238. ap_size_item = QtWidgets.QTableWidgetItem('')
  239. ap_size_item.setFlags(QtCore.Qt.ItemIsEnabled)
  240. mark_item = FCCheckBox()
  241. mark_item.setLayoutDirection(QtCore.Qt.RightToLeft)
  242. # if self.ui.aperture_table_visibility_cb.isChecked():
  243. # mark_item.setChecked(True)
  244. self.ui.apertures_table.setItem(self.apertures_row, 1, ap_code_item) # Aperture Code
  245. self.ui.apertures_table.setItem(self.apertures_row, 2, ap_type_item) # Aperture Type
  246. self.ui.apertures_table.setItem(self.apertures_row, 3, ap_size_item) # Aperture Dimensions
  247. self.ui.apertures_table.setItem(self.apertures_row, 4, ap_dim_item) # Aperture Dimensions
  248. empty_plot_item = QtWidgets.QTableWidgetItem('')
  249. empty_plot_item.setFlags(~QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
  250. self.ui.apertures_table.setItem(self.apertures_row, 5, empty_plot_item)
  251. self.ui.apertures_table.setCellWidget(self.apertures_row, 5, mark_item)
  252. self.apertures_row += 1
  253. self.ui.apertures_table.selectColumn(0)
  254. self.ui.apertures_table.resizeColumnsToContents()
  255. self.ui.apertures_table.resizeRowsToContents()
  256. vertical_header = self.ui.apertures_table.verticalHeader()
  257. # vertical_header.setSectionResizeMode(QtWidgets.QHeaderView.ResizeToContents)
  258. vertical_header.hide()
  259. self.ui.apertures_table.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
  260. horizontal_header = self.ui.apertures_table.horizontalHeader()
  261. horizontal_header.setMinimumSectionSize(10)
  262. horizontal_header.setDefaultSectionSize(70)
  263. horizontal_header.setSectionResizeMode(0, QtWidgets.QHeaderView.Fixed)
  264. horizontal_header.resizeSection(0, 27)
  265. horizontal_header.setSectionResizeMode(1, QtWidgets.QHeaderView.ResizeToContents)
  266. horizontal_header.setSectionResizeMode(2, QtWidgets.QHeaderView.ResizeToContents)
  267. horizontal_header.setSectionResizeMode(3, QtWidgets.QHeaderView.ResizeToContents)
  268. horizontal_header.setSectionResizeMode(4, QtWidgets.QHeaderView.Stretch)
  269. horizontal_header.setSectionResizeMode(5, QtWidgets.QHeaderView.Fixed)
  270. horizontal_header.resizeSection(5, 17)
  271. self.ui.apertures_table.setColumnWidth(5, 17)
  272. self.ui.apertures_table.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
  273. self.ui.apertures_table.setSortingEnabled(False)
  274. self.ui.apertures_table.setMinimumHeight(self.ui.apertures_table.getHeight())
  275. self.ui.apertures_table.setMaximumHeight(self.ui.apertures_table.getHeight())
  276. # update the 'mark' checkboxes state according with what is stored in the self.marked_rows list
  277. if self.marked_rows:
  278. for row in range(self.ui.apertures_table.rowCount()):
  279. try:
  280. self.ui.apertures_table.cellWidget(row, 5).set_value(self.marked_rows[row])
  281. except IndexError:
  282. pass
  283. self.ui_connect()
  284. def ui_connect(self):
  285. for row in range(self.ui.apertures_table.rowCount()):
  286. try:
  287. self.ui.apertures_table.cellWidget(row, 5).clicked.disconnect(self.on_mark_cb_click_table)
  288. except (TypeError, AttributeError):
  289. pass
  290. self.ui.apertures_table.cellWidget(row, 5).clicked.connect(self.on_mark_cb_click_table)
  291. try:
  292. self.ui.mark_all_cb.clicked.disconnect(self.on_mark_all_click)
  293. except (TypeError, AttributeError):
  294. pass
  295. self.ui.mark_all_cb.clicked.connect(self.on_mark_all_click)
  296. def ui_disconnect(self):
  297. for row in range(self.ui.apertures_table.rowCount()):
  298. try:
  299. self.ui.apertures_table.cellWidget(row, 5).clicked.disconnect()
  300. except (TypeError, AttributeError):
  301. pass
  302. try:
  303. self.ui.mark_all_cb.clicked.disconnect(self.on_mark_all_click)
  304. except (TypeError, AttributeError):
  305. pass
  306. @staticmethod
  307. def buffer_handler(geo):
  308. new_geo = geo
  309. if isinstance(new_geo, list):
  310. new_geo = MultiPolygon(new_geo)
  311. new_geo = new_geo.buffer(0.0000001)
  312. new_geo = new_geo.buffer(-0.0000001)
  313. return new_geo
  314. def on_generate_buffer(self):
  315. self.app.inform.emit('[WARNING_NOTCL] %s...' % _("Buffering solid geometry"))
  316. def buffer_task():
  317. with self.app.proc_container.new('%s...' % _("Buffering")):
  318. output = self.app.pool.apply_async(self.buffer_handler, args=([self.solid_geometry]))
  319. self.solid_geometry = output.get()
  320. self.app.inform.emit('[success] %s.' % _("Done"))
  321. self.plot_single_object.emit()
  322. self.app.worker_task.emit({'fcn': buffer_task, 'params': []})
  323. def on_generatenoncopper_button_click(self, *args):
  324. self.app.defaults.report_usage("gerber_on_generatenoncopper_button")
  325. self.read_form()
  326. name = self.options["name"] + "_noncopper"
  327. def geo_init(geo_obj, app_obj):
  328. assert geo_obj.kind == 'geometry', "Expected a Geometry object got %s" % type(geo_obj)
  329. if isinstance(self.solid_geometry, list):
  330. try:
  331. self.solid_geometry = MultiPolygon(self.solid_geometry)
  332. except Exception:
  333. self.solid_geometry = cascaded_union(self.solid_geometry)
  334. bounding_box = self.solid_geometry.envelope.buffer(float(self.options["noncoppermargin"]))
  335. if not self.options["noncopperrounded"]:
  336. bounding_box = bounding_box.envelope
  337. non_copper = bounding_box.difference(self.solid_geometry)
  338. if non_copper is None or non_copper.is_empty:
  339. self.app.inform.emit("[ERROR_NOTCL] %s" % _("Operation could not be done."))
  340. return "fail"
  341. geo_obj.solid_geometry = non_copper
  342. self.app.app_obj.new_object("geometry", name, geo_init)
  343. def on_generatebb_button_click(self, *args):
  344. self.app.defaults.report_usage("gerber_on_generatebb_button")
  345. self.read_form()
  346. name = self.options["name"] + "_bbox"
  347. def geo_init(geo_obj, app_obj):
  348. assert geo_obj.kind == 'geometry', "Expected a Geometry object got %s" % type(geo_obj)
  349. if isinstance(self.solid_geometry, list):
  350. try:
  351. self.solid_geometry = MultiPolygon(self.solid_geometry)
  352. except Exception:
  353. self.solid_geometry = cascaded_union(self.solid_geometry)
  354. # Bounding box with rounded corners
  355. bounding_box = self.solid_geometry.envelope.buffer(float(self.options["bboxmargin"]))
  356. if not self.options["bboxrounded"]: # Remove rounded corners
  357. bounding_box = bounding_box.envelope
  358. if bounding_box is None or bounding_box.is_empty:
  359. self.app.inform.emit("[ERROR_NOTCL] %s" % _("Operation could not be done."))
  360. return "fail"
  361. geo_obj.solid_geometry = bounding_box
  362. self.app.app_obj.new_object("geometry", name, geo_init)
  363. def isolate(self, iso_type=None, geometry=None, dia=None, passes=None, overlap=None, outname=None, combine=None,
  364. milling_type=None, follow=None, plot=True):
  365. """
  366. Creates an isolation routing geometry object in the project.
  367. :param iso_type: type of isolation to be done: 0 = exteriors, 1 = interiors and 2 = both
  368. :param geometry: specific geometry to isolate
  369. :param dia: Tool diameter
  370. :param passes: Number of tool widths to cut
  371. :param overlap: Overlap between passes in fraction of tool diameter
  372. :param outname: Base name of the output object
  373. :param combine: Boolean: if to combine passes in one resulting object in case of multiple passes
  374. :param milling_type: type of milling: conventional or climbing
  375. :param follow: Boolean: if to generate a 'follow' geometry
  376. :param plot: Boolean: if to plot the resulting geometry object
  377. :return: None
  378. """
  379. if geometry is None:
  380. work_geo = self.follow_geometry if follow is True else self.solid_geometry
  381. else:
  382. work_geo = geometry
  383. if dia is None:
  384. dia = float(self.app.defaults["tools_iso_tooldia"])
  385. if passes is None:
  386. passes = int(self.app.defaults["tools_iso_passes"])
  387. if overlap is None:
  388. overlap = float(self.app.defaults["tools_iso_overlap"])
  389. overlap /= 100.0
  390. combine = self.app.defaults["tools_iso_combine_passes"] if combine is None else bool(combine)
  391. if milling_type is None:
  392. milling_type = self.app.defaults["tools_iso_milling_type"]
  393. if iso_type is None:
  394. iso_t = 2
  395. else:
  396. iso_t = iso_type
  397. base_name = self.options["name"]
  398. if combine:
  399. if outname is None:
  400. if self.iso_type == 0:
  401. iso_name = base_name + "_ext_iso"
  402. elif self.iso_type == 1:
  403. iso_name = base_name + "_int_iso"
  404. else:
  405. iso_name = base_name + "_iso"
  406. else:
  407. iso_name = outname
  408. def iso_init(geo_obj, app_obj):
  409. # Propagate options
  410. geo_obj.options["cnctooldia"] = str(dia)
  411. geo_obj.tool_type = self.app.defaults["tools_iso_tool_type"]
  412. geo_obj.solid_geometry = []
  413. # transfer the Cut Z and Vtip and VAngle values in case that we use the V-Shape tool in Gerber UI
  414. if geo_obj.tool_type.lower() == 'v':
  415. new_cutz = self.app.defaults["tools_iso_tool_cutz"]
  416. new_vtipdia = self.app.defaults["tools_iso_tool_vtipdia"]
  417. new_vtipangle = self.app.defaults["tools_iso_tool_vtipangle"]
  418. tool_type = 'V'
  419. else:
  420. new_cutz = self.app.defaults['geometry_cutz']
  421. new_vtipdia = self.app.defaults['geometry_vtipdia']
  422. new_vtipangle = self.app.defaults['geometry_vtipangle']
  423. tool_type = 'C1'
  424. # store here the default data for Geometry Data
  425. default_data = {}
  426. default_data.update({
  427. "name": iso_name,
  428. "plot": self.app.defaults['geometry_plot'],
  429. "cutz": new_cutz,
  430. "vtipdia": new_vtipdia,
  431. "vtipangle": new_vtipangle,
  432. "travelz": self.app.defaults['geometry_travelz'],
  433. "feedrate": self.app.defaults['geometry_feedrate'],
  434. "feedrate_z": self.app.defaults['geometry_feedrate_z'],
  435. "feedrate_rapid": self.app.defaults['geometry_feedrate_rapid'],
  436. "dwell": self.app.defaults['geometry_dwell'],
  437. "dwelltime": self.app.defaults['geometry_dwelltime'],
  438. "multidepth": self.app.defaults['geometry_multidepth'],
  439. "ppname_g": self.app.defaults['geometry_ppname_g'],
  440. "depthperpass": self.app.defaults['geometry_depthperpass'],
  441. "extracut": self.app.defaults['geometry_extracut'],
  442. "extracut_length": self.app.defaults['geometry_extracut_length'],
  443. "toolchange": self.app.defaults['geometry_toolchange'],
  444. "toolchangez": self.app.defaults['geometry_toolchangez'],
  445. "endz": self.app.defaults['geometry_endz'],
  446. "spindlespeed": self.app.defaults['geometry_spindlespeed'],
  447. "toolchangexy": self.app.defaults['geometry_toolchangexy'],
  448. "startz": self.app.defaults['geometry_startz']
  449. })
  450. geo_obj.tools = {}
  451. geo_obj.tools['1'] = {}
  452. geo_obj.tools.update({
  453. '1': {
  454. 'tooldia': dia,
  455. 'offset': 'Path',
  456. 'offset_value': 0.0,
  457. 'type': _('Rough'),
  458. 'tool_type': tool_type,
  459. 'data': default_data,
  460. 'solid_geometry': geo_obj.solid_geometry
  461. }
  462. })
  463. for nr_pass in range(passes):
  464. iso_offset = dia * ((2 * nr_pass + 1) / 2.0) - (nr_pass * overlap * dia)
  465. # if milling type is climb then the move is counter-clockwise around features
  466. mill_dir = 1 if milling_type == 'cl' else 0
  467. geom = self.generate_envelope(iso_offset, mill_dir, geometry=work_geo, env_iso_type=iso_t,
  468. follow=follow, nr_passes=nr_pass)
  469. if geom == 'fail':
  470. app_obj.inform.emit('[ERROR_NOTCL] %s' % _("Isolation geometry could not be generated."))
  471. return 'fail'
  472. geo_obj.solid_geometry.append(geom)
  473. # update the geometry in the tools
  474. geo_obj.tools['1']['solid_geometry'] = geo_obj.solid_geometry
  475. # detect if solid_geometry is empty and this require list flattening which is "heavy"
  476. # or just looking in the lists (they are one level depth) and if any is not empty
  477. # proceed with object creation, if there are empty and the number of them is the length
  478. # of the list then we have an empty solid_geometry which should raise a Custom Exception
  479. empty_cnt = 0
  480. if not isinstance(geo_obj.solid_geometry, list) and \
  481. not isinstance(geo_obj.solid_geometry, MultiPolygon):
  482. geo_obj.solid_geometry = [geo_obj.solid_geometry]
  483. for g in geo_obj.solid_geometry:
  484. if g:
  485. break
  486. else:
  487. empty_cnt += 1
  488. if empty_cnt == len(geo_obj.solid_geometry):
  489. raise ValidationError("Empty Geometry", None)
  490. else:
  491. app_obj.inform.emit('[success] %s" %s' % (_("Isolation geometry created"), geo_obj.options["name"]))
  492. # even if combine is checked, one pass is still single-geo
  493. geo_obj.multigeo = True if passes > 1 else False
  494. # ############################################################
  495. # ########## AREA SUBTRACTION ################################
  496. # ############################################################
  497. # if self.app.defaults["tools_iso_except"]:
  498. # self.app.proc_container.update_view_text(' %s' % _("Subtracting Geo"))
  499. # geo_obj.solid_geometry = self.area_subtraction(geo_obj.solid_geometry)
  500. self.app.app_obj.new_object("geometry", iso_name, iso_init, plot=plot)
  501. else:
  502. for i in range(passes):
  503. offset = dia * ((2 * i + 1) / 2.0) - (i * overlap * dia)
  504. if passes > 1:
  505. if outname is None:
  506. if self.iso_type == 0:
  507. iso_name = base_name + "_ext_iso" + str(i + 1)
  508. elif self.iso_type == 1:
  509. iso_name = base_name + "_int_iso" + str(i + 1)
  510. else:
  511. iso_name = base_name + "_iso" + str(i + 1)
  512. else:
  513. iso_name = outname
  514. else:
  515. if outname is None:
  516. if self.iso_type == 0:
  517. iso_name = base_name + "_ext_iso"
  518. elif self.iso_type == 1:
  519. iso_name = base_name + "_int_iso"
  520. else:
  521. iso_name = base_name + "_iso"
  522. else:
  523. iso_name = outname
  524. def iso_init(geo_obj, app_obj):
  525. # Propagate options
  526. geo_obj.options["cnctooldia"] = str(dia)
  527. geo_obj.tool_type = self.app.defaults["tools_iso_tool_type"]
  528. # if milling type is climb then the move is counter-clockwise around features
  529. mill_dir = 1 if milling_type == 'cl' else 0
  530. geom = self.generate_envelope(offset, mill_dir, geometry=work_geo, env_iso_type=iso_t,
  531. follow=follow, nr_passes=i)
  532. if geom == 'fail':
  533. app_obj.inform.emit('[ERROR_NOTCL] %s' % _("Isolation geometry could not be generated."))
  534. return 'fail'
  535. geo_obj.solid_geometry = geom
  536. # transfer the Cut Z and Vtip and VAngle values in case that we use the V-Shape tool in Gerber UI
  537. # even if the resulting geometry is not multigeo we add the tools dict which will hold the data
  538. # required to be transfered to the Geometry object
  539. if self.app.defaults["tools_iso_tool_type"].lower() == 'v':
  540. new_cutz = self.app.defaults["tools_iso_tool_cutz"]
  541. new_vtipdia = self.app.defaults["tools_iso_tool_vtipdia"]
  542. new_vtipangle = self.app.defaults["tools_iso_tool_vtipangle"]
  543. tool_type = 'V'
  544. else:
  545. new_cutz = self.app.defaults['geometry_cutz']
  546. new_vtipdia = self.app.defaults['geometry_vtipdia']
  547. new_vtipangle = self.app.defaults['geometry_vtipangle']
  548. tool_type = 'C1'
  549. # store here the default data for Geometry Data
  550. default_data = {}
  551. default_data.update({
  552. "name": iso_name,
  553. "plot": self.app.defaults['geometry_plot'],
  554. "cutz": new_cutz,
  555. "vtipdia": new_vtipdia,
  556. "vtipangle": new_vtipangle,
  557. "travelz": self.app.defaults['geometry_travelz'],
  558. "feedrate": self.app.defaults['geometry_feedrate'],
  559. "feedrate_z": self.app.defaults['geometry_feedrate_z'],
  560. "feedrate_rapid": self.app.defaults['geometry_feedrate_rapid'],
  561. "dwell": self.app.defaults['geometry_dwell'],
  562. "dwelltime": self.app.defaults['geometry_dwelltime'],
  563. "multidepth": self.app.defaults['geometry_multidepth'],
  564. "ppname_g": self.app.defaults['geometry_ppname_g'],
  565. "depthperpass": self.app.defaults['geometry_depthperpass'],
  566. "extracut": self.app.defaults['geometry_extracut'],
  567. "extracut_length": self.app.defaults['geometry_extracut_length'],
  568. "toolchange": self.app.defaults['geometry_toolchange'],
  569. "toolchangez": self.app.defaults['geometry_toolchangez'],
  570. "endz": self.app.defaults['geometry_endz'],
  571. "spindlespeed": self.app.defaults['geometry_spindlespeed'],
  572. "toolchangexy": self.app.defaults['geometry_toolchangexy'],
  573. "startz": self.app.defaults['geometry_startz']
  574. })
  575. geo_obj.tools = {}
  576. geo_obj.tools['1'] = {}
  577. geo_obj.tools.update({
  578. '1': {
  579. 'tooldia': dia,
  580. 'offset': 'Path',
  581. 'offset_value': 0.0,
  582. 'type': _('Rough'),
  583. 'tool_type': tool_type,
  584. 'data': default_data,
  585. 'solid_geometry': geo_obj.solid_geometry
  586. }
  587. })
  588. # detect if solid_geometry is empty and this require list flattening which is "heavy"
  589. # or just looking in the lists (they are one level depth) and if any is not empty
  590. # proceed with object creation, if there are empty and the number of them is the length
  591. # of the list then we have an empty solid_geometry which should raise a Custom Exception
  592. empty_cnt = 0
  593. if not isinstance(geo_obj.solid_geometry, list):
  594. geo_obj.solid_geometry = [geo_obj.solid_geometry]
  595. for g in geo_obj.solid_geometry:
  596. if g:
  597. break
  598. else:
  599. empty_cnt += 1
  600. if empty_cnt == len(geo_obj.solid_geometry):
  601. raise ValidationError("Empty Geometry", None)
  602. else:
  603. app_obj.inform.emit('[success] %s: %s' %
  604. (_("Isolation geometry created"), geo_obj.options["name"]))
  605. geo_obj.multigeo = False
  606. # ############################################################
  607. # ########## AREA SUBTRACTION ################################
  608. # ############################################################
  609. # if self.app.defaults["tools_iso_except"]:
  610. # self.app.proc_container.update_view_text(' %s' % _("Subtracting Geo"))
  611. # geo_obj.solid_geometry = self.area_subtraction(geo_obj.solid_geometry)
  612. self.app.app_obj.new_object("geometry", iso_name, iso_init, plot=plot)
  613. def generate_envelope(self, offset, invert, geometry=None, env_iso_type=2, follow=None, nr_passes=0):
  614. # isolation_geometry produces an envelope that is going on the left of the geometry
  615. # (the copper features). To leave the least amount of burrs on the features
  616. # the tool needs to travel on the right side of the features (this is called conventional milling)
  617. # the first pass is the one cutting all of the features, so it needs to be reversed
  618. # the other passes overlap preceding ones and cut the left over copper. It is better for them
  619. # to cut on the right side of the left over copper i.e on the left side of the features.
  620. if follow:
  621. geom = self.isolation_geometry(offset, geometry=geometry, follow=follow)
  622. else:
  623. try:
  624. geom = self.isolation_geometry(offset, geometry=geometry, iso_type=env_iso_type, passes=nr_passes)
  625. except Exception as e:
  626. log.debug('GerberObject.isolate().generate_envelope() --> %s' % str(e))
  627. return 'fail'
  628. if invert:
  629. try:
  630. pl = []
  631. for p in geom:
  632. if p is not None:
  633. if isinstance(p, Polygon):
  634. pl.append(Polygon(p.exterior.coords[::-1], p.interiors))
  635. elif isinstance(p, LinearRing):
  636. pl.append(Polygon(p.coords[::-1]))
  637. geom = MultiPolygon(pl)
  638. except TypeError:
  639. if isinstance(geom, Polygon) and geom is not None:
  640. geom = Polygon(geom.exterior.coords[::-1], geom.interiors)
  641. elif isinstance(geom, LinearRing) and geom is not None:
  642. geom = Polygon(geom.coords[::-1])
  643. else:
  644. log.debug("GerberObject.isolate().generate_envelope() Error --> Unexpected Geometry %s" %
  645. type(geom))
  646. except Exception as e:
  647. log.debug("GerberObject.isolate().generate_envelope() Error --> %s" % str(e))
  648. return 'fail'
  649. return geom
  650. def follow_geo(self, outname=None):
  651. """
  652. Creates a geometry object "following" the gerber paths.
  653. :return: None
  654. """
  655. if outname is None:
  656. follow_name = self.options["name"] + "_follow"
  657. else:
  658. follow_name = outname
  659. def follow_init(follow_obj, app):
  660. # Propagate options
  661. follow_obj.options["cnctooldia"] = str(self.app.defaults["tools_iso_tooldia"])
  662. follow_obj.solid_geometry = self.follow_geometry
  663. # TODO: Do something if this is None. Offer changing name?
  664. try:
  665. self.app.app_obj.new_object("geometry", follow_name, follow_init)
  666. except Exception as e:
  667. return "Operation failed: %s" % str(e)
  668. def on_plot_cb_click(self, *args):
  669. if self.muted_ui:
  670. return
  671. self.read_form_item('plot')
  672. self.plot()
  673. def on_solid_cb_click(self, *args):
  674. if self.muted_ui:
  675. return
  676. self.read_form_item('solid')
  677. self.plot()
  678. def on_multicolored_cb_click(self, *args):
  679. if self.muted_ui:
  680. return
  681. self.read_form_item('multicolored')
  682. self.plot()
  683. def on_follow_cb_click(self):
  684. if self.muted_ui:
  685. return
  686. self.plot()
  687. def on_aperture_table_visibility_change(self):
  688. if self.ui.aperture_table_visibility_cb.isChecked():
  689. # add the shapes storage for marking apertures
  690. if self.build_aperture_storage is False:
  691. self.build_aperture_storage = True
  692. if self.app.is_legacy is False:
  693. for ap_code in self.apertures:
  694. self.mark_shapes[ap_code] = self.app.plotcanvas.new_shape_collection(layers=1)
  695. else:
  696. for ap_code in self.apertures:
  697. self.mark_shapes[ap_code] = ShapeCollectionLegacy(obj=self, app=self.app,
  698. name=self.options['name'] + str(ap_code))
  699. self.ui.apertures_table.setVisible(True)
  700. for ap in self.mark_shapes:
  701. self.mark_shapes[ap].enabled = True
  702. self.ui.mark_all_cb.setVisible(True)
  703. self.ui.mark_all_cb.setChecked(False)
  704. self.build_ui()
  705. else:
  706. self.ui.apertures_table.setVisible(False)
  707. self.ui.mark_all_cb.setVisible(False)
  708. # on hide disable all mark plots
  709. try:
  710. for row in range(self.ui.apertures_table.rowCount()):
  711. self.ui.apertures_table.cellWidget(row, 5).set_value(False)
  712. self.clear_plot_apertures()
  713. # for ap in list(self.mark_shapes.keys()):
  714. # # self.mark_shapes[ap].enabled = False
  715. # del self.mark_shapes[ap]
  716. except Exception as e:
  717. log.debug(" GerberObject.on_aperture_visibility_changed() --> %s" % str(e))
  718. def convert_units(self, units):
  719. """
  720. Converts the units of the object by scaling dimensions in all geometry
  721. and options.
  722. :param units: Units to which to convert the object: "IN" or "MM".
  723. :type units: str
  724. :return: None
  725. :rtype: None
  726. """
  727. # units conversion to get a conversion should be done only once even if we found multiple
  728. # units declaration inside a Gerber file (it can happen to find also the obsolete declaration)
  729. if self.conversion_done is True:
  730. log.debug("Gerber units conversion cancelled. Already done.")
  731. return
  732. log.debug("FlatCAMObj.GerberObject.convert_units()")
  733. factor = Gerber.convert_units(self, units)
  734. # self.options['isotooldia'] = float(self.options['isotooldia']) * factor
  735. # self.options['bboxmargin'] = float(self.options['bboxmargin']) * factor
  736. def plot(self, kind=None, **kwargs):
  737. """
  738. :param kind: Not used, for compatibility with the plot method for other objects
  739. :param kwargs: Color and face_color, visible
  740. :return:
  741. """
  742. log.debug(str(inspect.stack()[1][3]) + " --> GerberObject.plot()")
  743. # Does all the required setup and returns False
  744. # if the 'ptint' option is set to False.
  745. if not FlatCAMObj.plot(self):
  746. return
  747. if 'color' in kwargs:
  748. color = kwargs['color']
  749. else:
  750. color = self.outline_color
  751. if 'face_color' in kwargs:
  752. face_color = kwargs['face_color']
  753. else:
  754. face_color = self.fill_color
  755. if 'visible' not in kwargs:
  756. visible = self.options['plot']
  757. else:
  758. visible = kwargs['visible']
  759. # if the Follow Geometry checkbox is checked then plot only the follow geometry
  760. if self.ui.follow_cb.get_value():
  761. geometry = self.follow_geometry
  762. else:
  763. geometry = self.solid_geometry
  764. # Make sure geometry is iterable.
  765. try:
  766. __ = iter(geometry)
  767. except TypeError:
  768. geometry = [geometry]
  769. if self.app.is_legacy is False:
  770. def random_color():
  771. r_color = np.random.rand(4)
  772. r_color[3] = 1
  773. return r_color
  774. else:
  775. def random_color():
  776. while True:
  777. r_color = np.random.rand(4)
  778. r_color[3] = 1
  779. new_color = '#'
  780. for idx in range(len(r_color)):
  781. new_color += '%x' % int(r_color[idx] * 255)
  782. # do it until a valid color is generated
  783. # a valid color has the # symbol, another 6 chars for the color and the last 2 chars for alpha
  784. # for a total of 9 chars
  785. if len(new_color) == 9:
  786. break
  787. return new_color
  788. try:
  789. if self.options["solid"]:
  790. for g in geometry:
  791. if type(g) == Polygon or type(g) == LineString:
  792. self.add_shape(shape=g, color=color,
  793. face_color=random_color() if self.options['multicolored']
  794. else face_color, visible=visible)
  795. elif type(g) == Point:
  796. pass
  797. else:
  798. try:
  799. for el in g:
  800. self.add_shape(shape=el, color=color,
  801. face_color=random_color() if self.options['multicolored']
  802. else face_color, visible=visible)
  803. except TypeError:
  804. self.add_shape(shape=g, color=color,
  805. face_color=random_color() if self.options['multicolored']
  806. else face_color, visible=visible)
  807. else:
  808. for g in geometry:
  809. if type(g) == Polygon or type(g) == LineString:
  810. self.add_shape(shape=g, color=random_color() if self.options['multicolored'] else 'black',
  811. visible=visible)
  812. elif type(g) == Point:
  813. pass
  814. else:
  815. for el in g:
  816. self.add_shape(shape=el, color=random_color() if self.options['multicolored'] else 'black',
  817. visible=visible)
  818. self.shapes.redraw(
  819. # update_colors=(self.fill_color, self.outline_color),
  820. # indexes=self.app.plotcanvas.shape_collection.data.keys()
  821. )
  822. except (ObjectDeleted, AttributeError):
  823. self.shapes.clear(update=True)
  824. except Exception as e:
  825. log.debug("GerberObject.plot() --> %s" % str(e))
  826. # experimental plot() when the solid_geometry is stored in the self.apertures
  827. def plot_aperture(self, run_thread=True, **kwargs):
  828. """
  829. :param run_thread: if True run the aperture plot as a thread in a worker
  830. :param kwargs: color and face_color
  831. :return:
  832. """
  833. log.debug(str(inspect.stack()[1][3]) + " --> GerberObject.plot_aperture()")
  834. # Does all the required setup and returns False
  835. # if the 'ptint' option is set to False.
  836. # if not FlatCAMObj.plot(self):
  837. # return
  838. # for marking apertures, line color and fill color are the same
  839. if 'color' in kwargs:
  840. color = kwargs['color']
  841. else:
  842. color = self.app.defaults['gerber_plot_fill']
  843. if 'marked_aperture' not in kwargs:
  844. return
  845. else:
  846. aperture_to_plot_mark = kwargs['marked_aperture']
  847. if aperture_to_plot_mark is None:
  848. return
  849. if 'visible' not in kwargs:
  850. visibility = True
  851. else:
  852. visibility = kwargs['visible']
  853. with self.app.proc_container.new(_("Plotting Apertures")):
  854. def job_thread(app_obj):
  855. try:
  856. if aperture_to_plot_mark in self.apertures:
  857. for elem in self.apertures[aperture_to_plot_mark]['geometry']:
  858. if 'solid' in elem:
  859. geo = elem['solid']
  860. if type(geo) == Polygon or type(geo) == LineString:
  861. self.add_mark_shape(apid=aperture_to_plot_mark, shape=geo, color=color,
  862. face_color=color, visible=visibility)
  863. else:
  864. for el in geo:
  865. self.add_mark_shape(apid=aperture_to_plot_mark, shape=el, color=color,
  866. face_color=color, visible=visibility)
  867. self.mark_shapes[aperture_to_plot_mark].redraw()
  868. except (ObjectDeleted, AttributeError):
  869. self.clear_plot_apertures()
  870. except Exception as e:
  871. log.debug("GerberObject.plot_aperture() --> %s" % str(e))
  872. if run_thread:
  873. self.app.worker_task.emit({'fcn': job_thread, 'params': [self]})
  874. else:
  875. job_thread(self)
  876. def clear_plot_apertures(self, aperture='all'):
  877. """
  878. :param aperture: string; aperture for which to clear the mark shapes
  879. :return:
  880. """
  881. if self.mark_shapes:
  882. if aperture == 'all':
  883. for apid in list(self.apertures.keys()):
  884. try:
  885. if self.app.is_legacy is True:
  886. self.mark_shapes[apid].clear(update=False)
  887. else:
  888. self.mark_shapes[apid].clear(update=True)
  889. except Exception as e:
  890. log.debug("GerberObject.clear_plot_apertures() 'all' --> %s" % str(e))
  891. else:
  892. try:
  893. if self.app.is_legacy is True:
  894. self.mark_shapes[aperture].clear(update=False)
  895. else:
  896. self.mark_shapes[aperture].clear(update=True)
  897. except Exception as e:
  898. log.debug("GerberObject.clear_plot_apertures() 'aperture' --> %s" % str(e))
  899. def clear_mark_all(self):
  900. self.ui.mark_all_cb.set_value(False)
  901. self.marked_rows[:] = []
  902. def on_mark_cb_click_table(self):
  903. """
  904. Will mark aperture geometries on canvas or delete the markings depending on the checkbox state
  905. :return:
  906. """
  907. self.ui_disconnect()
  908. try:
  909. cw = self.sender()
  910. cw_index = self.ui.apertures_table.indexAt(cw.pos())
  911. cw_row = cw_index.row()
  912. except AttributeError:
  913. cw_row = 0
  914. except TypeError:
  915. return
  916. self.marked_rows[:] = []
  917. try:
  918. aperture = self.ui.apertures_table.item(cw_row, 1).text()
  919. except AttributeError:
  920. return
  921. if self.ui.apertures_table.cellWidget(cw_row, 5).isChecked():
  922. self.marked_rows.append(True)
  923. # self.plot_aperture(color='#2d4606bf', marked_aperture=aperture, visible=True)
  924. self.plot_aperture(color=self.app.defaults['global_sel_draw_color'] + 'AF',
  925. marked_aperture=aperture, visible=True, run_thread=True)
  926. # self.mark_shapes[aperture].redraw()
  927. else:
  928. self.marked_rows.append(False)
  929. self.clear_plot_apertures(aperture=aperture)
  930. # make sure that the Mark All is disabled if one of the row mark's are disabled and
  931. # if all the row mark's are enabled also enable the Mark All checkbox
  932. cb_cnt = 0
  933. total_row = self.ui.apertures_table.rowCount()
  934. for row in range(total_row):
  935. if self.ui.apertures_table.cellWidget(row, 5).isChecked():
  936. cb_cnt += 1
  937. else:
  938. cb_cnt -= 1
  939. if cb_cnt < total_row:
  940. self.ui.mark_all_cb.setChecked(False)
  941. else:
  942. self.ui.mark_all_cb.setChecked(True)
  943. self.ui_connect()
  944. def on_mark_all_click(self):
  945. self.ui_disconnect()
  946. mark_all = self.ui.mark_all_cb.isChecked()
  947. for row in range(self.ui.apertures_table.rowCount()):
  948. # update the mark_rows list
  949. if mark_all:
  950. self.marked_rows.append(True)
  951. else:
  952. self.marked_rows[:] = []
  953. mark_cb = self.ui.apertures_table.cellWidget(row, 5)
  954. mark_cb.setChecked(mark_all)
  955. if mark_all:
  956. for aperture in self.apertures:
  957. # self.plot_aperture(color='#2d4606bf', marked_aperture=aperture, visible=True)
  958. self.plot_aperture(color=self.app.defaults['global_sel_draw_color'] + 'AF',
  959. marked_aperture=aperture, visible=True)
  960. # HACK: enable/disable the grid for a better look
  961. self.app.ui.grid_snap_btn.trigger()
  962. self.app.ui.grid_snap_btn.trigger()
  963. else:
  964. self.clear_plot_apertures()
  965. self.marked_rows[:] = []
  966. self.ui_connect()
  967. def export_gerber(self, whole, fract, g_zeros='L', factor=1):
  968. """
  969. Creates a Gerber file content to be exported to a file.
  970. :param whole: how many digits in the whole part of coordinates
  971. :param fract: how many decimals in coordinates
  972. :param g_zeros: type of the zero suppression used: LZ or TZ; string
  973. :param factor: factor to be applied onto the Gerber coordinates
  974. :return: Gerber_code
  975. """
  976. log.debug("GerberObject.export_gerber() --> Generating the Gerber code from the selected Gerber file")
  977. def tz_format(x, y, fac):
  978. x_c = x * fac
  979. y_c = y * fac
  980. x_form = "{:.{dec}f}".format(x_c, dec=fract)
  981. y_form = "{:.{dec}f}".format(y_c, dec=fract)
  982. # extract whole part and decimal part
  983. x_form = x_form.partition('.')
  984. y_form = y_form.partition('.')
  985. # left padd the 'whole' part with zeros
  986. x_whole = x_form[0].rjust(whole, '0')
  987. y_whole = y_form[0].rjust(whole, '0')
  988. # restore the coordinate padded in the left with 0 and added the decimal part
  989. # without the decinal dot
  990. x_form = x_whole + x_form[2]
  991. y_form = y_whole + y_form[2]
  992. return x_form, y_form
  993. def lz_format(x, y, fac):
  994. x_c = x * fac
  995. y_c = y * fac
  996. x_form = "{:.{dec}f}".format(x_c, dec=fract).replace('.', '')
  997. y_form = "{:.{dec}f}".format(y_c, dec=fract).replace('.', '')
  998. # pad with rear zeros
  999. x_form.ljust(length, '0')
  1000. y_form.ljust(length, '0')
  1001. return x_form, y_form
  1002. # Gerber code is stored here
  1003. gerber_code = ''
  1004. # apertures processing
  1005. try:
  1006. length = whole + fract
  1007. if '0' in self.apertures:
  1008. if 'geometry' in self.apertures['0']:
  1009. for geo_elem in self.apertures['0']['geometry']:
  1010. if 'solid' in geo_elem:
  1011. geo = geo_elem['solid']
  1012. if not geo.is_empty:
  1013. gerber_code += 'G36*\n'
  1014. geo_coords = list(geo.exterior.coords)
  1015. # first command is a move with pen-up D02 at the beginning of the geo
  1016. if g_zeros == 'T':
  1017. x_formatted, y_formatted = tz_format(geo_coords[0][0], geo_coords[0][1], factor)
  1018. gerber_code += "X{xform}Y{yform}D02*\n".format(xform=x_formatted,
  1019. yform=y_formatted)
  1020. else:
  1021. x_formatted, y_formatted = lz_format(geo_coords[0][0], geo_coords[0][1], factor)
  1022. gerber_code += "X{xform}Y{yform}D02*\n".format(xform=x_formatted,
  1023. yform=y_formatted)
  1024. for coord in geo_coords[1:]:
  1025. if g_zeros == 'T':
  1026. x_formatted, y_formatted = tz_format(coord[0], coord[1], factor)
  1027. gerber_code += "X{xform}Y{yform}D01*\n".format(xform=x_formatted,
  1028. yform=y_formatted)
  1029. else:
  1030. x_formatted, y_formatted = lz_format(coord[0], coord[1], factor)
  1031. gerber_code += "X{xform}Y{yform}D01*\n".format(xform=x_formatted,
  1032. yform=y_formatted)
  1033. gerber_code += 'D02*\n'
  1034. gerber_code += 'G37*\n'
  1035. clear_list = list(geo.interiors)
  1036. if clear_list:
  1037. gerber_code += '%LPC*%\n'
  1038. for clear_geo in clear_list:
  1039. gerber_code += 'G36*\n'
  1040. geo_coords = list(clear_geo.coords)
  1041. # first command is a move with pen-up D02 at the beginning of the geo
  1042. if g_zeros == 'T':
  1043. x_formatted, y_formatted = tz_format(
  1044. geo_coords[0][0], geo_coords[0][1], factor)
  1045. gerber_code += "X{xform}Y{yform}D02*\n".format(xform=x_formatted,
  1046. yform=y_formatted)
  1047. else:
  1048. x_formatted, y_formatted = lz_format(
  1049. geo_coords[0][0], geo_coords[0][1], factor)
  1050. gerber_code += "X{xform}Y{yform}D02*\n".format(xform=x_formatted,
  1051. yform=y_formatted)
  1052. prev_coord = geo_coords[0]
  1053. for coord in geo_coords[1:]:
  1054. if coord != prev_coord:
  1055. if g_zeros == 'T':
  1056. x_formatted, y_formatted = tz_format(coord[0], coord[1], factor)
  1057. gerber_code += "X{xform}Y{yform}D01*\n".format(xform=x_formatted,
  1058. yform=y_formatted)
  1059. else:
  1060. x_formatted, y_formatted = lz_format(coord[0], coord[1], factor)
  1061. gerber_code += "X{xform}Y{yform}D01*\n".format(xform=x_formatted,
  1062. yform=y_formatted)
  1063. prev_coord = coord
  1064. gerber_code += 'D02*\n'
  1065. gerber_code += 'G37*\n'
  1066. gerber_code += '%LPD*%\n'
  1067. if 'clear' in geo_elem:
  1068. geo = geo_elem['clear']
  1069. if not geo.is_empty:
  1070. gerber_code += '%LPC*%\n'
  1071. gerber_code += 'G36*\n'
  1072. geo_coords = list(geo.exterior.coords)
  1073. # first command is a move with pen-up D02 at the beginning of the geo
  1074. if g_zeros == 'T':
  1075. x_formatted, y_formatted = tz_format(geo_coords[0][0], geo_coords[0][1], factor)
  1076. gerber_code += "X{xform}Y{yform}D02*\n".format(xform=x_formatted,
  1077. yform=y_formatted)
  1078. else:
  1079. x_formatted, y_formatted = lz_format(geo_coords[0][0], geo_coords[0][1], factor)
  1080. gerber_code += "X{xform}Y{yform}D02*\n".format(xform=x_formatted,
  1081. yform=y_formatted)
  1082. prev_coord = geo_coords[0]
  1083. for coord in geo_coords[1:]:
  1084. if coord != prev_coord:
  1085. if g_zeros == 'T':
  1086. x_formatted, y_formatted = tz_format(coord[0], coord[1], factor)
  1087. gerber_code += "X{xform}Y{yform}D01*\n".format(xform=x_formatted,
  1088. yform=y_formatted)
  1089. else:
  1090. x_formatted, y_formatted = lz_format(coord[0], coord[1], factor)
  1091. gerber_code += "X{xform}Y{yform}D01*\n".format(xform=x_formatted,
  1092. yform=y_formatted)
  1093. prev_coord = coord
  1094. gerber_code += 'D02*\n'
  1095. gerber_code += 'G37*\n'
  1096. gerber_code += '%LPD*%\n'
  1097. except Exception as e:
  1098. log.debug("FlatCAMObj.GerberObject.export_gerber() '0' aperture --> %s" % str(e))
  1099. for apid in self.apertures:
  1100. if apid == '0':
  1101. continue
  1102. else:
  1103. gerber_code += 'D%s*\n' % str(apid)
  1104. if 'geometry' in self.apertures[apid]:
  1105. for geo_elem in self.apertures[apid]['geometry']:
  1106. try:
  1107. if 'follow' in geo_elem:
  1108. geo = geo_elem['follow']
  1109. if not geo.is_empty:
  1110. if isinstance(geo, Point):
  1111. if g_zeros == 'T':
  1112. x_formatted, y_formatted = tz_format(geo.x, geo.y, factor)
  1113. gerber_code += "X{xform}Y{yform}D03*\n".format(xform=x_formatted,
  1114. yform=y_formatted)
  1115. else:
  1116. x_formatted, y_formatted = lz_format(geo.x, geo.y, factor)
  1117. gerber_code += "X{xform}Y{yform}D03*\n".format(xform=x_formatted,
  1118. yform=y_formatted)
  1119. else:
  1120. geo_coords = list(geo.coords)
  1121. # first command is a move with pen-up D02 at the beginning of the geo
  1122. if g_zeros == 'T':
  1123. x_formatted, y_formatted = tz_format(
  1124. geo_coords[0][0], geo_coords[0][1], factor)
  1125. gerber_code += "X{xform}Y{yform}D02*\n".format(xform=x_formatted,
  1126. yform=y_formatted)
  1127. else:
  1128. x_formatted, y_formatted = lz_format(
  1129. geo_coords[0][0], geo_coords[0][1], factor)
  1130. gerber_code += "X{xform}Y{yform}D02*\n".format(xform=x_formatted,
  1131. yform=y_formatted)
  1132. prev_coord = geo_coords[0]
  1133. for coord in geo_coords[1:]:
  1134. if coord != prev_coord:
  1135. if g_zeros == 'T':
  1136. x_formatted, y_formatted = tz_format(coord[0], coord[1], factor)
  1137. gerber_code += "X{xform}Y{yform}D01*\n".format(xform=x_formatted,
  1138. yform=y_formatted)
  1139. else:
  1140. x_formatted, y_formatted = lz_format(coord[0], coord[1], factor)
  1141. gerber_code += "X{xform}Y{yform}D01*\n".format(xform=x_formatted,
  1142. yform=y_formatted)
  1143. prev_coord = coord
  1144. # gerber_code += "D02*\n"
  1145. except Exception as e:
  1146. log.debug("FlatCAMObj.GerberObject.export_gerber() 'follow' --> %s" % str(e))
  1147. try:
  1148. if 'clear' in geo_elem:
  1149. gerber_code += '%LPC*%\n'
  1150. geo = geo_elem['clear']
  1151. if not geo.is_empty:
  1152. if isinstance(geo, Point):
  1153. if g_zeros == 'T':
  1154. x_formatted, y_formatted = tz_format(geo.x, geo.y, factor)
  1155. gerber_code += "X{xform}Y{yform}D03*\n".format(xform=x_formatted,
  1156. yform=y_formatted)
  1157. else:
  1158. x_formatted, y_formatted = lz_format(geo.x, geo.y, factor)
  1159. gerber_code += "X{xform}Y{yform}D03*\n".format(xform=x_formatted,
  1160. yform=y_formatted)
  1161. elif isinstance(geo, Polygon):
  1162. geo_coords = list(geo.exterior.coords)
  1163. # first command is a move with pen-up D02 at the beginning of the geo
  1164. if g_zeros == 'T':
  1165. x_formatted, y_formatted = tz_format(
  1166. geo_coords[0][0], geo_coords[0][1], factor)
  1167. gerber_code += "X{xform}Y{yform}D02*\n".format(xform=x_formatted,
  1168. yform=y_formatted)
  1169. else:
  1170. x_formatted, y_formatted = lz_format(
  1171. geo_coords[0][0], geo_coords[0][1], factor)
  1172. gerber_code += "X{xform}Y{yform}D02*\n".format(xform=x_formatted,
  1173. yform=y_formatted)
  1174. prev_coord = geo_coords[0]
  1175. for coord in geo_coords[1:]:
  1176. if coord != prev_coord:
  1177. if g_zeros == 'T':
  1178. x_formatted, y_formatted = tz_format(coord[0], coord[1], factor)
  1179. gerber_code += "X{xform}Y{yform}D01*\n".format(xform=x_formatted,
  1180. yform=y_formatted)
  1181. else:
  1182. x_formatted, y_formatted = lz_format(coord[0], coord[1], factor)
  1183. gerber_code += "X{xform}Y{yform}D01*\n".format(xform=x_formatted,
  1184. yform=y_formatted)
  1185. prev_coord = coord
  1186. for geo_int in geo.interiors:
  1187. geo_coords = list(geo_int.coords)
  1188. # first command is a move with pen-up D02 at the beginning of the geo
  1189. if g_zeros == 'T':
  1190. x_formatted, y_formatted = tz_format(
  1191. geo_coords[0][0], geo_coords[0][1], factor)
  1192. gerber_code += "X{xform}Y{yform}D02*\n".format(xform=x_formatted,
  1193. yform=y_formatted)
  1194. else:
  1195. x_formatted, y_formatted = lz_format(
  1196. geo_coords[0][0], geo_coords[0][1], factor)
  1197. gerber_code += "X{xform}Y{yform}D02*\n".format(xform=x_formatted,
  1198. yform=y_formatted)
  1199. prev_coord = geo_coords[0]
  1200. for coord in geo_coords[1:]:
  1201. if coord != prev_coord:
  1202. if g_zeros == 'T':
  1203. x_formatted, y_formatted = tz_format(coord[0], coord[1], factor)
  1204. gerber_code += "X{xform}Y{yform}D01*\n".format(
  1205. xform=x_formatted,
  1206. yform=y_formatted)
  1207. else:
  1208. x_formatted, y_formatted = lz_format(coord[0], coord[1], factor)
  1209. gerber_code += "X{xform}Y{yform}D01*\n".format(
  1210. xform=x_formatted,
  1211. yform=y_formatted)
  1212. prev_coord = coord
  1213. else:
  1214. geo_coords = list(geo.coords)
  1215. # first command is a move with pen-up D02 at the beginning of the geo
  1216. if g_zeros == 'T':
  1217. x_formatted, y_formatted = tz_format(
  1218. geo_coords[0][0], geo_coords[0][1], factor)
  1219. gerber_code += "X{xform}Y{yform}D02*\n".format(xform=x_formatted,
  1220. yform=y_formatted)
  1221. else:
  1222. x_formatted, y_formatted = lz_format(
  1223. geo_coords[0][0], geo_coords[0][1], factor)
  1224. gerber_code += "X{xform}Y{yform}D02*\n".format(xform=x_formatted,
  1225. yform=y_formatted)
  1226. prev_coord = geo_coords[0]
  1227. for coord in geo_coords[1:]:
  1228. if coord != prev_coord:
  1229. if g_zeros == 'T':
  1230. x_formatted, y_formatted = tz_format(coord[0], coord[1], factor)
  1231. gerber_code += "X{xform}Y{yform}D01*\n".format(xform=x_formatted,
  1232. yform=y_formatted)
  1233. else:
  1234. x_formatted, y_formatted = lz_format(coord[0], coord[1], factor)
  1235. gerber_code += "X{xform}Y{yform}D01*\n".format(xform=x_formatted,
  1236. yform=y_formatted)
  1237. prev_coord = coord
  1238. # gerber_code += "D02*\n"
  1239. gerber_code += '%LPD*%\n'
  1240. except Exception as e:
  1241. log.debug("FlatCAMObj.GerberObject.export_gerber() 'clear' --> %s" % str(e))
  1242. if not self.apertures:
  1243. log.debug("FlatCAMObj.GerberObject.export_gerber() --> Gerber Object is empty: no apertures.")
  1244. return 'fail'
  1245. return gerber_code
  1246. def mirror(self, axis, point):
  1247. Gerber.mirror(self, axis=axis, point=point)
  1248. self.replotApertures.emit()
  1249. def offset(self, vect):
  1250. Gerber.offset(self, vect=vect)
  1251. self.replotApertures.emit()
  1252. def rotate(self, angle, point):
  1253. Gerber.rotate(self, angle=angle, point=point)
  1254. self.replotApertures.emit()
  1255. def scale(self, xfactor, yfactor=None, point=None):
  1256. Gerber.scale(self, xfactor=xfactor, yfactor=yfactor, point=point)
  1257. self.replotApertures.emit()
  1258. def skew(self, angle_x, angle_y, point):
  1259. Gerber.skew(self, angle_x=angle_x, angle_y=angle_y, point=point)
  1260. self.replotApertures.emit()
  1261. def buffer(self, distance, join=2, factor=None):
  1262. Gerber.buffer(self, distance=distance, join=join, factor=factor)
  1263. self.replotApertures.emit()
  1264. def serialize(self):
  1265. return {
  1266. "options": self.options,
  1267. "kind": self.kind
  1268. }