FlatCAMGerber.py 75 KB

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