FlatCAMCNCJob.py 78 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841
  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 copy import deepcopy
  12. from io import StringIO
  13. from datetime import datetime
  14. from appEditors.AppTextEditor import AppTextEditor
  15. from appObjects.FlatCAMObj import *
  16. from matplotlib.backend_bases import KeyEvent as mpl_key_event
  17. from camlib import CNCjob
  18. from shapely.ops import unary_union
  19. from shapely.geometry import Point
  20. try:
  21. from shapely.ops import voronoi_diagram
  22. except Exception:
  23. pass
  24. import os
  25. import sys
  26. import serial
  27. import glob
  28. import math
  29. import gettext
  30. import appTranslation as fcTranslate
  31. import builtins
  32. fcTranslate.apply_language('strings')
  33. if '_' not in builtins.__dict__:
  34. _ = gettext.gettext
  35. class CNCJobObject(FlatCAMObj, CNCjob):
  36. """
  37. Represents G-Code.
  38. """
  39. optionChanged = QtCore.pyqtSignal(str)
  40. ui_type = CNCObjectUI
  41. def __init__(self, name, units="in", kind="generic", z_move=0.1,
  42. feedrate=3.0, feedrate_rapid=3.0, z_cut=-0.002, tooldia=0.0,
  43. spindlespeed=None):
  44. log.debug("Creating CNCJob object...")
  45. self.decimals = self.app.decimals
  46. CNCjob.__init__(self, units=units, kind=kind, z_move=z_move,
  47. feedrate=feedrate, feedrate_rapid=feedrate_rapid, z_cut=z_cut, tooldia=tooldia,
  48. spindlespeed=spindlespeed, steps_per_circle=int(self.app.defaults["cncjob_steps_per_circle"]))
  49. FlatCAMObj.__init__(self, name)
  50. self.kind = "cncjob"
  51. self.options.update({
  52. "plot": True,
  53. "tooldia": 0.03937, # 0.4mm in inches
  54. "append": "",
  55. "prepend": "",
  56. "dwell": False,
  57. "dwelltime": 1,
  58. "type": 'Geometry',
  59. # "toolchange_macro": '',
  60. # "toolchange_macro_enable": False
  61. })
  62. '''
  63. This is a dict of dictionaries. Each dict is associated with a tool present in the file. The key is the
  64. diameter of the tools and the value is another dict that will hold the data under the following form:
  65. {tooldia: {
  66. 'tooluid': 1,
  67. 'offset': 'Path',
  68. 'type_item': 'Rough',
  69. 'tool_type': 'C1',
  70. 'data': {} # a dict to hold the parameters
  71. 'gcode': "" # a string with the actual GCODE
  72. 'gcode_parsed': {} # dictionary holding the CNCJob geometry and type of geometry
  73. (cut or move)
  74. 'solid_geometry': []
  75. },
  76. ...
  77. }
  78. It is populated in the GeometryObject.mtool_gen_cncjob()
  79. BEWARE: I rely on the ordered nature of the Python 3.7 dictionary. Things might change ...
  80. '''
  81. self.cnc_tools = {}
  82. '''
  83. This is a dict of dictionaries. Each dict is associated with a tool present in the file. The key is the
  84. diameter of the tools and the value is another dict that will hold the data under the following form:
  85. {tooldia: {
  86. 'tool': int,
  87. 'nr_drills': int,
  88. 'nr_slots': int,
  89. 'offset': float,
  90. 'data': {}, a dict to hold the parameters
  91. 'gcode': "", a string with the actual GCODE
  92. 'gcode_parsed': [], list of dicts holding the CNCJob geometry and
  93. type of geometry (cut or move)
  94. 'solid_geometry': [],
  95. },
  96. ...
  97. }
  98. It is populated in the ExcellonObject.on_create_cncjob_click() but actually
  99. it's done in camlib.CNCJob.generate_from_excellon_by_tool()
  100. BEWARE: I rely on the ordered nature of the Python 3.7 dictionary. Things might change ...
  101. '''
  102. self.exc_cnc_tools = {}
  103. # flag to store if the CNCJob is part of a special group of CNCJob objects that can't be processed by the
  104. # default engine of FlatCAM. They generated by some of tools and are special cases of CNCJob objects.
  105. self.special_group = None
  106. # for now it show if the plot will be done for multi-tool CNCJob (True) or for single tool
  107. # (like the one in the TCL Command), False
  108. self.multitool = False
  109. # determine if the GCode was generated out of a Excellon object or a Geometry object
  110. self.origin_kind = None
  111. self.coords_decimals = 4
  112. self.fr_decimals = 2
  113. # used for parsing the GCode lines to adjust the GCode when the GCode is offseted or scaled
  114. gcodex_re_string = r'(?=.*(X[-\+]?\d*\.\d*))'
  115. self.g_x_re = re.compile(gcodex_re_string)
  116. gcodey_re_string = r'(?=.*(Y[-\+]?\d*\.\d*))'
  117. self.g_y_re = re.compile(gcodey_re_string)
  118. gcodez_re_string = r'(?=.*(Z[-\+]?\d*\.\d*))'
  119. self.g_z_re = re.compile(gcodez_re_string)
  120. gcodef_re_string = r'(?=.*(F[-\+]?\d*\.\d*))'
  121. self.g_f_re = re.compile(gcodef_re_string)
  122. gcodet_re_string = r'(?=.*(\=\s*[-\+]?\d*\.\d*))'
  123. self.g_t_re = re.compile(gcodet_re_string)
  124. gcodenr_re_string = r'([+-]?\d*\.\d+)'
  125. self.g_nr_re = re.compile(gcodenr_re_string)
  126. if self.app.is_legacy is False:
  127. self.text_col = self.app.plotcanvas.new_text_collection()
  128. self.text_col.enabled = True
  129. self.annotation = self.app.plotcanvas.new_text_group(collection=self.text_col)
  130. self.gcode_editor_tab = None
  131. self.gcode_viewer_tab = None
  132. self.source_file = ''
  133. self.units_found = self.app.defaults['units']
  134. self.probing_gcode_text = ''
  135. # store the current selection shape status to be restored after manual adding test points
  136. self.old_selection_state = self.app.defaults['global_selection_shape']
  137. # if mouse is dragging set the object True
  138. self.mouse_is_dragging = False
  139. # if mouse events are bound to local methods
  140. self.mouse_events_connected = False
  141. # event handlers references
  142. self.kp = None
  143. self.mm = None
  144. self.mr = None
  145. self.append_snippet = ''
  146. self.prepend_snippet = ''
  147. self.gc_header = self.gcode_header()
  148. self.gc_start = ''
  149. self.gc_end = ''
  150. '''
  151. dictionary of dictionaries to store the informations for the autolevelling
  152. format:
  153. {
  154. id: {
  155. 'point': Shapely Point
  156. 'geo': Shapely Polygon from Voronoi diagram,
  157. 'height': float
  158. }
  159. }
  160. '''
  161. self.al_geometry_dict = {}
  162. self.grbl_ser_port = None
  163. # Attributes to be included in serialization
  164. # Always append to it because it carries contents
  165. # from predecessors.
  166. self.ser_attrs += [
  167. 'options', 'kind', 'origin_kind', 'cnc_tools', 'exc_cnc_tools', 'multitool', 'append_snippet',
  168. 'prepend_snippet', 'gc_header'
  169. ]
  170. def build_ui(self):
  171. self.ui_disconnect()
  172. FlatCAMObj.build_ui(self)
  173. self.units = self.app.defaults['units'].upper()
  174. # if the FlatCAM object is Excellon don't build the CNC Tools Table but hide it
  175. self.ui.cnc_tools_table.hide()
  176. if self.cnc_tools:
  177. self.ui.cnc_tools_table.show()
  178. self.build_cnc_tools_table()
  179. self.ui.exc_cnc_tools_table.hide()
  180. if self.exc_cnc_tools:
  181. self.ui.exc_cnc_tools_table.show()
  182. self.build_excellon_cnc_tools()
  183. if self.ui.sal_cb.get_value():
  184. self.build_al_table()
  185. self.ui_connect()
  186. def build_cnc_tools_table(self):
  187. tool_idx = 0
  188. n = len(self.cnc_tools)
  189. self.ui.cnc_tools_table.setRowCount(n)
  190. for dia_key, dia_value in self.cnc_tools.items():
  191. tool_idx += 1
  192. row_no = tool_idx - 1
  193. t_id = QtWidgets.QTableWidgetItem('%d' % int(tool_idx))
  194. # id.setFlags(QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
  195. self.ui.cnc_tools_table.setItem(row_no, 0, t_id) # Tool name/id
  196. # Make sure that the tool diameter when in MM is with no more than 2 decimals.
  197. # There are no tool bits in MM with more than 2 decimals diameter.
  198. # For INCH the decimals should be no more than 4. There are no tools under 10mils.
  199. dia_item = QtWidgets.QTableWidgetItem('%.*f' % (self.decimals, float(dia_value['tooldia'])))
  200. offset_txt = list(str(dia_value['offset']))
  201. offset_txt[0] = offset_txt[0].upper()
  202. offset_item = QtWidgets.QTableWidgetItem(''.join(offset_txt))
  203. type_item = QtWidgets.QTableWidgetItem(str(dia_value['type']))
  204. tool_type_item = QtWidgets.QTableWidgetItem(str(dia_value['tool_type']))
  205. t_id.setFlags(QtCore.Qt.ItemIsEnabled)
  206. dia_item.setFlags(QtCore.Qt.ItemIsEnabled)
  207. offset_item.setFlags(QtCore.Qt.ItemIsEnabled)
  208. type_item.setFlags(QtCore.Qt.ItemIsEnabled)
  209. tool_type_item.setFlags(QtCore.Qt.ItemIsEnabled)
  210. # hack so the checkbox stay centered in the table cell
  211. # used this:
  212. # https://stackoverflow.com/questions/32458111/pyqt-allign-checkbox-and-put-it-in-every-row
  213. # plot_item = QtWidgets.QWidget()
  214. # checkbox = FCCheckBox()
  215. # checkbox.setCheckState(QtCore.Qt.Checked)
  216. # qhboxlayout = QtWidgets.QHBoxLayout(plot_item)
  217. # qhboxlayout.addWidget(checkbox)
  218. # qhboxlayout.setAlignment(QtCore.Qt.AlignCenter)
  219. # qhboxlayout.setContentsMargins(0, 0, 0, 0)
  220. plot_item = FCCheckBox()
  221. plot_item.setLayoutDirection(QtCore.Qt.RightToLeft)
  222. tool_uid_item = QtWidgets.QTableWidgetItem(str(dia_key))
  223. if self.ui.plot_cb.isChecked():
  224. plot_item.setChecked(True)
  225. self.ui.cnc_tools_table.setItem(row_no, 1, dia_item) # Diameter
  226. self.ui.cnc_tools_table.setItem(row_no, 2, offset_item) # Offset
  227. self.ui.cnc_tools_table.setItem(row_no, 3, type_item) # Toolpath Type
  228. self.ui.cnc_tools_table.setItem(row_no, 4, tool_type_item) # Tool Type
  229. # ## REMEMBER: THIS COLUMN IS HIDDEN IN OBJECTUI.PY # ##
  230. self.ui.cnc_tools_table.setItem(row_no, 5, tool_uid_item) # Tool unique ID)
  231. self.ui.cnc_tools_table.setCellWidget(row_no, 6, plot_item)
  232. # make the diameter column editable
  233. # for row in range(tool_idx):
  234. # self.ui.cnc_tools_table.item(row, 1).setFlags(QtCore.Qt.ItemIsSelectable |
  235. # QtCore.Qt.ItemIsEnabled)
  236. for row in range(tool_idx):
  237. self.ui.cnc_tools_table.item(row, 0).setFlags(
  238. self.ui.cnc_tools_table.item(row, 0).flags() ^ QtCore.Qt.ItemIsSelectable)
  239. self.ui.cnc_tools_table.resizeColumnsToContents()
  240. self.ui.cnc_tools_table.resizeRowsToContents()
  241. vertical_header = self.ui.cnc_tools_table.verticalHeader()
  242. # vertical_header.setSectionResizeMode(QtWidgets.QHeaderView.ResizeToContents)
  243. vertical_header.hide()
  244. self.ui.cnc_tools_table.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
  245. horizontal_header = self.ui.cnc_tools_table.horizontalHeader()
  246. horizontal_header.setMinimumSectionSize(10)
  247. horizontal_header.setDefaultSectionSize(70)
  248. horizontal_header.setSectionResizeMode(0, QtWidgets.QHeaderView.Fixed)
  249. horizontal_header.resizeSection(0, 20)
  250. horizontal_header.setSectionResizeMode(1, QtWidgets.QHeaderView.Stretch)
  251. horizontal_header.setSectionResizeMode(3, QtWidgets.QHeaderView.ResizeToContents)
  252. horizontal_header.setSectionResizeMode(4, QtWidgets.QHeaderView.Fixed)
  253. horizontal_header.resizeSection(4, 40)
  254. horizontal_header.setSectionResizeMode(6, QtWidgets.QHeaderView.Fixed)
  255. horizontal_header.resizeSection(4, 17)
  256. # horizontal_header.setStretchLastSection(True)
  257. self.ui.cnc_tools_table.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
  258. self.ui.cnc_tools_table.setColumnWidth(0, 20)
  259. self.ui.cnc_tools_table.setColumnWidth(4, 40)
  260. self.ui.cnc_tools_table.setColumnWidth(6, 17)
  261. # self.ui.geo_tools_table.setSortingEnabled(True)
  262. self.ui.cnc_tools_table.setMinimumHeight(self.ui.cnc_tools_table.getHeight())
  263. self.ui.cnc_tools_table.setMaximumHeight(self.ui.cnc_tools_table.getHeight())
  264. def build_excellon_cnc_tools(self):
  265. tool_idx = 0
  266. n = len(self.exc_cnc_tools)
  267. self.ui.exc_cnc_tools_table.setRowCount(n)
  268. for tooldia_key, dia_value in self.exc_cnc_tools.items():
  269. tool_idx += 1
  270. row_no = tool_idx - 1
  271. t_id = QtWidgets.QTableWidgetItem('%d' % int(tool_idx))
  272. dia_item = QtWidgets.QTableWidgetItem('%.*f' % (self.decimals, float(tooldia_key)))
  273. nr_drills_item = QtWidgets.QTableWidgetItem('%d' % int(dia_value['nr_drills']))
  274. nr_slots_item = QtWidgets.QTableWidgetItem('%d' % int(dia_value['nr_slots']))
  275. cutz_item = QtWidgets.QTableWidgetItem('%.*f' % (self.decimals, float(dia_value['offset']) + self.z_cut))
  276. t_id.setFlags(QtCore.Qt.ItemIsEnabled)
  277. dia_item.setFlags(QtCore.Qt.ItemIsEnabled)
  278. nr_drills_item.setFlags(QtCore.Qt.ItemIsEnabled)
  279. nr_slots_item.setFlags(QtCore.Qt.ItemIsEnabled)
  280. cutz_item.setFlags(QtCore.Qt.ItemIsEnabled)
  281. # hack so the checkbox stay centered in the table cell
  282. # used this:
  283. # https://stackoverflow.com/questions/32458111/pyqt-allign-checkbox-and-put-it-in-every-row
  284. # plot_item = QtWidgets.QWidget()
  285. # checkbox = FCCheckBox()
  286. # checkbox.setCheckState(QtCore.Qt.Checked)
  287. # qhboxlayout = QtWidgets.QHBoxLayout(plot_item)
  288. # qhboxlayout.addWidget(checkbox)
  289. # qhboxlayout.setAlignment(QtCore.Qt.AlignCenter)
  290. # qhboxlayout.setContentsMargins(0, 0, 0, 0)
  291. plot_item = FCCheckBox()
  292. plot_item.setLayoutDirection(QtCore.Qt.RightToLeft)
  293. tool_uid_item = QtWidgets.QTableWidgetItem(str(dia_value['tool']))
  294. if self.ui.plot_cb.isChecked():
  295. plot_item.setChecked(True)
  296. self.ui.exc_cnc_tools_table.setItem(row_no, 0, t_id) # Tool name/id
  297. self.ui.exc_cnc_tools_table.setItem(row_no, 1, dia_item) # Diameter
  298. self.ui.exc_cnc_tools_table.setItem(row_no, 2, nr_drills_item) # Nr of drills
  299. self.ui.exc_cnc_tools_table.setItem(row_no, 3, nr_slots_item) # Nr of slots
  300. # ## REMEMBER: THIS COLUMN IS HIDDEN IN OBJECTUI.PY # ##
  301. self.ui.exc_cnc_tools_table.setItem(row_no, 4, tool_uid_item) # Tool unique ID)
  302. self.ui.exc_cnc_tools_table.setItem(row_no, 5, cutz_item)
  303. self.ui.exc_cnc_tools_table.setCellWidget(row_no, 6, plot_item)
  304. for row in range(tool_idx):
  305. self.ui.exc_cnc_tools_table.item(row, 0).setFlags(
  306. self.ui.exc_cnc_tools_table.item(row, 0).flags() ^ QtCore.Qt.ItemIsSelectable)
  307. self.ui.exc_cnc_tools_table.resizeColumnsToContents()
  308. self.ui.exc_cnc_tools_table.resizeRowsToContents()
  309. vertical_header = self.ui.exc_cnc_tools_table.verticalHeader()
  310. vertical_header.hide()
  311. self.ui.exc_cnc_tools_table.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
  312. horizontal_header = self.ui.exc_cnc_tools_table.horizontalHeader()
  313. horizontal_header.setMinimumSectionSize(10)
  314. horizontal_header.setDefaultSectionSize(70)
  315. horizontal_header.setSectionResizeMode(0, QtWidgets.QHeaderView.Fixed)
  316. horizontal_header.resizeSection(0, 20)
  317. horizontal_header.setSectionResizeMode(1, QtWidgets.QHeaderView.Stretch)
  318. horizontal_header.setSectionResizeMode(2, QtWidgets.QHeaderView.ResizeToContents)
  319. horizontal_header.setSectionResizeMode(3, QtWidgets.QHeaderView.ResizeToContents)
  320. horizontal_header.setSectionResizeMode(5, QtWidgets.QHeaderView.ResizeToContents)
  321. horizontal_header.setSectionResizeMode(6, QtWidgets.QHeaderView.Fixed)
  322. # horizontal_header.setStretchLastSection(True)
  323. self.ui.exc_cnc_tools_table.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
  324. self.ui.exc_cnc_tools_table.setColumnWidth(0, 20)
  325. self.ui.exc_cnc_tools_table.setColumnWidth(6, 17)
  326. self.ui.exc_cnc_tools_table.setMinimumHeight(self.ui.exc_cnc_tools_table.getHeight())
  327. self.ui.exc_cnc_tools_table.setMaximumHeight(self.ui.exc_cnc_tools_table.getHeight())
  328. def build_al_table(self):
  329. tool_idx = 0
  330. n = len(self.al_geometry_dict)
  331. self.ui.al_probe_points_table.setRowCount(n)
  332. for id_key, value in self.al_geometry_dict.items():
  333. tool_idx += 1
  334. row_no = tool_idx - 1
  335. t_id = QtWidgets.QTableWidgetItem('%d' % int(tool_idx))
  336. x = value['point'].x
  337. y = value['point'].y
  338. xy_coords = self.app.dec_format(x, dec=self.app.decimals), self.app.dec_format(y, dec=self.app.decimals)
  339. coords_item = QtWidgets.QTableWidgetItem(str(xy_coords))
  340. height = self.app.dec_format(value['height'], dec=self.app.decimals)
  341. height_item = QtWidgets.QTableWidgetItem(str(height))
  342. t_id.setFlags(QtCore.Qt.ItemIsEnabled)
  343. coords_item.setFlags(QtCore.Qt.ItemIsEnabled)
  344. height_item.setFlags(QtCore.Qt.ItemIsEnabled)
  345. self.ui.al_probe_points_table.setItem(row_no, 0, t_id) # Tool name/id
  346. self.ui.al_probe_points_table.setItem(row_no, 1, coords_item) # X-Y coords
  347. self.ui.al_probe_points_table.setItem(row_no, 2, height_item) # Determined Height
  348. self.ui.al_probe_points_table.resizeColumnsToContents()
  349. self.ui.al_probe_points_table.resizeRowsToContents()
  350. h_header = self.ui.al_probe_points_table.horizontalHeader()
  351. h_header.setMinimumSectionSize(10)
  352. h_header.setDefaultSectionSize(70)
  353. h_header.setSectionResizeMode(0, QtWidgets.QHeaderView.Fixed)
  354. h_header.resizeSection(0, 20)
  355. h_header.setSectionResizeMode(1, QtWidgets.QHeaderView.Stretch)
  356. h_header.setSectionResizeMode(2, QtWidgets.QHeaderView.ResizeToContents)
  357. self.ui.al_probe_points_table.setMinimumHeight(self.ui.al_probe_points_table.getHeight())
  358. self.ui.al_probe_points_table.setMaximumHeight(self.ui.al_probe_points_table.getHeight())
  359. if self.ui.al_probe_points_table.model().rowCount():
  360. self.ui.voronoi_cb.setDisabled(False)
  361. self.ui.grbl_get_heightmap_button.setDisabled(False)
  362. self.ui.h_gcode_button.setDisabled(False)
  363. self.ui.view_h_gcode_button.setDisabled(False)
  364. else:
  365. self.ui.voronoi_cb.setDisabled(True)
  366. self.ui.grbl_get_heightmap_button.setDisabled(True)
  367. self.ui.h_gcode_button.setDisabled(True)
  368. self.ui.view_h_gcode_button.setDisabled(True)
  369. def set_ui(self, ui):
  370. FlatCAMObj.set_ui(self, ui)
  371. log.debug("FlatCAMCNCJob.set_ui()")
  372. assert isinstance(self.ui, CNCObjectUI), \
  373. "Expected a CNCObjectUI, got %s" % type(self.ui)
  374. self.units = self.app.defaults['units'].upper()
  375. self.units_found = self.app.defaults['units']
  376. # this signal has to be connected to it's slot before the defaults are populated
  377. # the decision done in the slot has to override the default value set below
  378. # self.ui.toolchange_cb.toggled.connect(self.on_toolchange_custom_clicked)
  379. self.form_fields.update({
  380. "plot": self.ui.plot_cb,
  381. "tooldia": self.ui.tooldia_entry,
  382. # "append": self.ui.append_text,
  383. # "prepend": self.ui.prepend_text,
  384. # "toolchange_macro": self.ui.toolchange_text,
  385. # "toolchange_macro_enable": self.ui.toolchange_cb,
  386. "al_travelz": self.ui.ptravelz_entry,
  387. "al_probe_depth": self.ui.pdepth_entry,
  388. "al_probe_fr": self.ui.feedrate_probe_entry,
  389. "al_controller": self.ui.al_controller_combo,
  390. "al_mode": self.ui.al_mode_radio,
  391. "al_rows": self.ui.al_rows_entry,
  392. "al_columns": self.ui.al_columns_entry,
  393. })
  394. self.append_snippet = self.app.defaults['cncjob_append']
  395. self.prepend_snippet = self.app.defaults['cncjob_prepend']
  396. if self.append_snippet != '' or self.prepend_snippet:
  397. self.ui.snippets_cb.set_value(True)
  398. # Fill form fields only on object create
  399. self.to_form()
  400. # this means that the object that created this CNCJob was an Excellon or Geometry
  401. try:
  402. if self.travel_distance:
  403. self.ui.t_distance_label.show()
  404. self.ui.t_distance_entry.setVisible(True)
  405. self.ui.t_distance_entry.setDisabled(True)
  406. self.ui.t_distance_entry.set_value('%.*f' % (self.decimals, float(self.travel_distance)))
  407. self.ui.units_label.setText(str(self.units).lower())
  408. self.ui.units_label.setDisabled(True)
  409. self.ui.t_time_label.show()
  410. self.ui.t_time_entry.setVisible(True)
  411. self.ui.t_time_entry.setDisabled(True)
  412. # if time is more than 1 then we have minutes, else we have seconds
  413. if self.routing_time > 1:
  414. self.ui.t_time_entry.set_value('%.*f' % (self.decimals, math.ceil(float(self.routing_time))))
  415. self.ui.units_time_label.setText('min')
  416. else:
  417. time_r = self.routing_time * 60
  418. self.ui.t_time_entry.set_value('%.*f' % (self.decimals, math.ceil(float(time_r))))
  419. self.ui.units_time_label.setText('sec')
  420. self.ui.units_time_label.setDisabled(True)
  421. except AttributeError:
  422. pass
  423. if self.multitool is False:
  424. self.ui.tooldia_entry.show()
  425. self.ui.updateplot_button.show()
  426. else:
  427. self.ui.tooldia_entry.hide()
  428. self.ui.updateplot_button.hide()
  429. # set the kind of geometries are plotted by default with plot2() from camlib.CNCJob
  430. self.ui.cncplot_method_combo.set_value(self.app.defaults["cncjob_plot_kind"])
  431. try:
  432. self.ui.annotation_cb.stateChanged.disconnect(self.on_annotation_change)
  433. except (TypeError, AttributeError):
  434. pass
  435. self.ui.annotation_cb.stateChanged.connect(self.on_annotation_change)
  436. # set if to display text annotations
  437. self.ui.annotation_cb.set_value(self.app.defaults["cncjob_annotation"])
  438. # Show/Hide Advanced Options
  439. if self.app.defaults["global_app_level"] == 'b':
  440. self.ui.level.setText(_(
  441. '<span style="color:green;"><b>Basic</b></span>'
  442. ))
  443. self.ui.sal_cb.hide()
  444. else:
  445. self.ui.level.setText(_(
  446. '<span style="color:red;"><b>Advanced</b></span>'
  447. ))
  448. self.ui.sal_cb.show()
  449. self.ui.updateplot_button.clicked.connect(self.on_updateplot_button_click)
  450. self.ui.export_gcode_button.clicked.connect(self.on_exportgcode_button_click)
  451. self.ui.review_gcode_button.clicked.connect(self.on_edit_code_click)
  452. self.ui.editor_button.clicked.connect(lambda: self.app.object2editor())
  453. # autolevelling signals
  454. self.ui.al_mode_radio.activated_custom.connect(self.on_mode_radio)
  455. self.ui.al_controller_combo.currentIndexChanged.connect(self.on_controller_change)
  456. self.ui.com_search_button.clicked.connect(self.on_search_ports)
  457. self.ui.add_bd_button.clicked.connect(self.on_add_baudrate_grbl)
  458. self.ui.del_bd_button.clicked.connect(self.on_delete_baudrate_grbl)
  459. self.ui.com_connect_button.clicked.connect(self.on_connect_grbl)
  460. self.ui.view_h_gcode_button.clicked.connect(self.on_view_probing_gcode)
  461. self.ui.h_gcode_button.clicked.connect(self.on_generate_probing_gcode)
  462. # self.ui.tc_variable_combo.currentIndexChanged[str].connect(self.on_cnc_custom_parameters)
  463. self.ui.cncplot_method_combo.activated_custom.connect(self.on_plot_kind_change)
  464. preamble = self.append_snippet
  465. postamble = self.prepend_snippet
  466. gc = self.export_gcode(preamble=preamble, postamble=postamble, to_file=True)
  467. self.source_file = gc.getvalue()
  468. self.ui.al_mode_radio.set_value(self.options['al_mode'])
  469. self.on_controller_change()
  470. # def on_cnc_custom_parameters(self, signal_text):
  471. # if signal_text == 'Parameters':
  472. # return
  473. # else:
  474. # self.ui.toolchange_text.insertPlainText('%%%s%%' % signal_text)
  475. def ui_connect(self):
  476. for row in range(self.ui.cnc_tools_table.rowCount()):
  477. self.ui.cnc_tools_table.cellWidget(row, 6).clicked.connect(self.on_plot_cb_click_table)
  478. for row in range(self.ui.exc_cnc_tools_table.rowCount()):
  479. self.ui.exc_cnc_tools_table.cellWidget(row, 6).clicked.connect(self.on_plot_cb_click_table)
  480. self.ui.plot_cb.stateChanged.connect(self.on_plot_cb_click)
  481. self.ui.al_add_button.clicked.connect(self.on_add_al_probepoints)
  482. self.ui.show_al_table.stateChanged.connect(self.on_show_al_table)
  483. def ui_disconnect(self):
  484. for row in range(self.ui.cnc_tools_table.rowCount()):
  485. try:
  486. self.ui.cnc_tools_table.cellWidget(row, 6).clicked.disconnect(self.on_plot_cb_click_table)
  487. except (TypeError, AttributeError):
  488. pass
  489. for row in range(self.ui.exc_cnc_tools_table.rowCount()):
  490. try:
  491. self.ui.exc_cnc_tools_table.cellWidget(row, 6).clicked.disconnect(self.on_plot_cb_click_table)
  492. except (TypeError, AttributeError):
  493. pass
  494. try:
  495. self.ui.plot_cb.stateChanged.disconnect(self.on_plot_cb_click)
  496. except (TypeError, AttributeError):
  497. pass
  498. try:
  499. self.ui.al_add_button.clicked.disconnect()
  500. except (TypeError, AttributeError):
  501. pass
  502. try:
  503. self.ui.show_al_table.stateChanged.disconnect()
  504. except (TypeError, AttributeError):
  505. pass
  506. def on_add_al_probepoints(self):
  507. # create the solid_geo
  508. solid_geo = [geo['geom'] for geo in self.gcode_parsed if geo['kind'][0] == 'C']
  509. solid_geo = unary_union(solid_geo)
  510. # reset al table
  511. self.ui.al_probe_points_table.setRowCount(0)
  512. # reset the al dict
  513. self.al_geometry_dict.clear()
  514. xmin, ymin, xmax, ymax = solid_geo.bounds
  515. if self.ui.al_mode_radio.get_value() == 'grid':
  516. width = abs(xmax - xmin)
  517. height = abs(ymax - ymin)
  518. cols = self.ui.al_columns_entry.get_value()
  519. rows = self.ui.al_rows_entry.get_value()
  520. dx = width / (cols + 1)
  521. dy = height / (rows + 1)
  522. points = []
  523. new_y = ymin
  524. for x in range(rows):
  525. new_y += dy
  526. new_x = xmin
  527. for x in range(cols):
  528. new_x += dx
  529. points.append((new_x, new_y))
  530. pt_id = 0
  531. for point in points:
  532. pt_id += 1
  533. new_dict = {
  534. 'point': Point(point),
  535. 'geo': None,
  536. 'height': 0.0
  537. }
  538. self.al_geometry_dict[pt_id] = deepcopy(new_dict)
  539. else:
  540. self.app.inform.emit(_("Click on canvas to add a Test Point..."))
  541. if self.app.is_legacy is False:
  542. self.app.plotcanvas.graph_event_disconnect('key_press', self.app.ui.keyPressEvent)
  543. self.app.plotcanvas.graph_event_disconnect('mouse_press', self.app.on_mouse_click_over_plot)
  544. self.app.plotcanvas.graph_event_disconnect('mouse_release', self.app.on_mouse_click_release_over_plot)
  545. else:
  546. self.app.plotcanvas.graph_event_disconnect(self.app.kp)
  547. self.app.plotcanvas.graph_event_disconnect(self.app.mp)
  548. self.app.plotcanvas.graph_event_disconnect(self.app.mr)
  549. self.kp = self.app.plotcanvas.graph_event_connect('key_press', self.on_key_press)
  550. self.mr = self.app.plotcanvas.graph_event_connect('mouse_release', self.on_mouse_click_release)
  551. self.mouse_events_connected = True
  552. # self.calculate_voronoi_diagram()
  553. self.build_al_table()
  554. # def calculate_voronoi_diagram(self):
  555. # return voronoi_diagram()
  556. # To be called after clicking on the plot.
  557. def on_mouse_click_release(self, event):
  558. if self.app.is_legacy is False:
  559. event_pos = event.pos
  560. # event_is_dragging = event.is_dragging
  561. right_button = 2
  562. else:
  563. event_pos = (event.xdata, event.ydata)
  564. # event_is_dragging = self.app.plotcanvas.is_dragging
  565. right_button = 3
  566. try:
  567. x = float(event_pos[0])
  568. y = float(event_pos[1])
  569. except TypeError:
  570. return
  571. event_pos = (x, y)
  572. # do paint single only for left mouse clicks
  573. if event.button == 1:
  574. pos = self.app.plotcanvas.translate_coords(event_pos)
  575. # use the snapped position as reference
  576. snapped_pos = self.app.geo_editor.snap(pos[0], pos[1])
  577. if not self.al_geometry_dict:
  578. new_dict = {
  579. 'point': Point(snapped_pos),
  580. 'geo': None,
  581. 'height': 0.0
  582. }
  583. self.al_geometry_dict[1] = deepcopy(new_dict)
  584. else:
  585. int_keys = [int(k) for k in self.al_geometry_dict.keys()]
  586. new_id = max(int_keys) + 1
  587. new_dict = {
  588. 'point': Point(snapped_pos),
  589. 'geo': None,
  590. 'height': 0.0
  591. }
  592. self.al_geometry_dict[new_id] = deepcopy(new_dict)
  593. # rebuild the al table
  594. self.build_al_table()
  595. self.app.inform.emit(_("Added Test Point... Click again to add another or right click to finish ..."))
  596. # if RMB then we exit
  597. elif event.button == right_button and self.mouse_is_dragging is False:
  598. if self.app.is_legacy is False:
  599. self.app.plotcanvas.graph_event_disconnect('key_press', self.on_key_press)
  600. self.app.plotcanvas.graph_event_disconnect('mouse_release', self.on_mouse_click_release)
  601. else:
  602. self.app.plotcanvas.graph_event_disconnect(self.kp)
  603. self.app.plotcanvas.graph_event_disconnect(self.mr)
  604. self.app.kp = self.app.plotcanvas.graph_event_connect('key_press', self.app.ui.keyPressEvent)
  605. self.app.mp = self.app.plotcanvas.graph_event_connect('mouse_press', self.app.on_mouse_click_over_plot)
  606. self.app.mr = self.app.plotcanvas.graph_event_connect('mouse_release',
  607. self.app.on_mouse_click_release_over_plot)
  608. # signal that the mouse events are disconnected from local methods
  609. self.mouse_events_connected = False
  610. # restore selection
  611. self.app.defaults['global_selection_shape'] = self.old_selection_state
  612. self.app.inform.emit(_("Finished manual adding of Test Point..."))
  613. # rebuild the al table
  614. self.build_al_table()
  615. def on_key_press(self, event):
  616. # events out of the self.app.collection view (it's about Project Tab) are of type int
  617. if type(event) is int:
  618. key = event
  619. # events from the GUI are of type QKeyEvent
  620. elif type(event) == QtGui.QKeyEvent:
  621. key = event.key()
  622. elif isinstance(event, mpl_key_event): # MatPlotLib key events are trickier to interpret than the rest
  623. key = event.key
  624. key = QtGui.QKeySequence(key)
  625. # check for modifiers
  626. key_string = key.toString().lower()
  627. if '+' in key_string:
  628. mod, __, key_text = key_string.rpartition('+')
  629. if mod.lower() == 'ctrl':
  630. # modifiers = QtCore.Qt.ControlModifier
  631. pass
  632. elif mod.lower() == 'alt':
  633. # modifiers = QtCore.Qt.AltModifier
  634. pass
  635. elif mod.lower() == 'shift':
  636. # modifiers = QtCore.Qt.ShiftModifier
  637. pass
  638. else:
  639. # modifiers = QtCore.Qt.NoModifier
  640. pass
  641. key = QtGui.QKeySequence(key_text)
  642. # events from Vispy are of type KeyEvent
  643. else:
  644. key = event.key
  645. # Escape = Deselect All
  646. if key == QtCore.Qt.Key_Escape or key == 'Escape':
  647. if self.mouse_events_connected is True:
  648. self.mouse_events_connected = False
  649. if self.app.is_legacy is False:
  650. self.app.plotcanvas.graph_event_disconnect('key_press', self.on_key_press)
  651. self.app.plotcanvas.graph_event_disconnect('mouse_release', self.on_mouse_click_release)
  652. else:
  653. self.app.plotcanvas.graph_event_disconnect(self.kp)
  654. self.app.plotcanvas.graph_event_disconnect(self.mr)
  655. self.app.kp = self.app.plotcanvas.graph_event_connect('key_press', self.app.ui.keyPressEvent)
  656. self.app.mp = self.app.plotcanvas.graph_event_connect('mouse_press', self.app.on_mouse_click_over_plot)
  657. self.app.mr = self.app.plotcanvas.graph_event_connect('mouse_release',
  658. self.app.on_mouse_click_release_over_plot)
  659. if self.ui.big_cursor_cb.get_value():
  660. # restore cursor
  661. self.app.on_cursor_type(val=self.old_cursor_type)
  662. # restore selection
  663. self.app.defaults['global_selection_shape'] = self.old_selection_state
  664. # Grid toggle
  665. if key == QtCore.Qt.Key_G or key == 'G':
  666. self.app.ui.grid_snap_btn.trigger()
  667. # Jump to coords
  668. if key == QtCore.Qt.Key_J or key == 'J':
  669. l_x, l_y = self.app.on_jump_to()
  670. def on_show_al_table(self, state):
  671. self.ui.al_probe_points_table.show() if state else self.ui.al_probe_points_table.hide()
  672. def on_mode_radio(self, val):
  673. # reset al table
  674. self.ui.al_probe_points_table.setRowCount(0)
  675. # reset the al dict
  676. self.al_geometry_dict.clear()
  677. # build AL table
  678. self.build_al_table()
  679. if val == "manual":
  680. self.ui.al_rows_entry.setDisabled(True)
  681. self.ui.al_rows_label.setDisabled(True)
  682. self.ui.al_columns_entry.setDisabled(True)
  683. self.ui.al_columns_label.setDisabled(True)
  684. else:
  685. self.ui.al_rows_entry.setDisabled(False)
  686. self.ui.al_rows_label.setDisabled(False)
  687. self.ui.al_columns_entry.setDisabled(False)
  688. self.ui.al_columns_label.setDisabled(False)
  689. def on_controller_change(self):
  690. if self.ui.al_controller_combo.get_value() == 'GRBL':
  691. self.ui.h_gcode_button.hide()
  692. self.ui.view_h_gcode_button.hide()
  693. self.ui.import_heights_button.hide()
  694. self.ui.grbl_frame.show()
  695. self.on_search_ports(muted=True)
  696. else:
  697. self.ui.h_gcode_button.show()
  698. self.ui.view_h_gcode_button.show()
  699. self.ui.import_heights_button.show()
  700. self.ui.grbl_frame.hide()
  701. def list_serial_ports(self):
  702. """
  703. Lists serial port names.
  704. From here: https://stackoverflow.com/questions/12090503/listing-available-com-ports-with-python
  705. :raises EnvironmentError: On unsupported or unknown platforms
  706. :returns: A list of the serial ports available on the system
  707. """
  708. if sys.platform.startswith('win'):
  709. ports = ['COM%s' % (i + 1) for i in range(256)]
  710. elif sys.platform.startswith('linux') or sys.platform.startswith('cygwin'):
  711. # this excludes your current terminal "/dev/tty"
  712. ports = glob.glob('/dev/tty[A-Za-z]*')
  713. elif sys.platform.startswith('darwin'):
  714. ports = glob.glob('/dev/tty.*')
  715. else:
  716. raise EnvironmentError('Unsupported platform')
  717. result = []
  718. s = serial.Serial()
  719. for port in ports:
  720. s.port = port
  721. try:
  722. s.open()
  723. s.close()
  724. result.append(port)
  725. except (OSError, serial.SerialException):
  726. # result.append(port + " (in use)")
  727. pass
  728. return result
  729. def on_search_ports(self, muted=None):
  730. port_list = self.list_serial_ports()
  731. self.ui.com_list_combo.clear()
  732. self.ui.com_list_combo.addItems(port_list)
  733. if muted is not True:
  734. self.app.inform.emit('[WARNING_NOTCL] %s' % _("COM list updated ..."))
  735. def on_connect_grbl(self):
  736. port_name = self.ui.com_list_combo.currentText()
  737. if " (" in port_name:
  738. port_name = port_name.rpartition(" (")[0]
  739. baudrate = int(self.ui.baudrates_list_combo.currentText())
  740. try:
  741. self.grbl_ser_port = serial.serial_for_url(port_name, baudrate,
  742. bytesize=serial.EIGHTBITS,
  743. parity=serial.PARITY_NONE,
  744. stopbits=serial.STOPBITS_ONE,
  745. timeout=0.1,
  746. xonxoff=False,
  747. rtscts=False)
  748. self.app.inform.emit("%s: %s" % (_("Port connected"), port_name))
  749. self.ui.com_connect_button.setStyleSheet("QPushButton {color: seagreen;}")
  750. # Toggle DTR to reset the controller loaded with GRBL (Arduino, ESP32, etc)
  751. try:
  752. self.grbl_ser_port.dtr = False
  753. except IOError:
  754. pass
  755. self.grbl_ser_port.reset_input_buffer()
  756. try:
  757. self.grbl_ser_port.dtr = True
  758. except IOError:
  759. pass
  760. except serial.SerialException:
  761. self.grbl_ser_port = serial.Serial()
  762. self.grbl_ser_port.port = port_name
  763. self.grbl_ser_port.close()
  764. self.ui.com_connect_button.setStyleSheet("")
  765. self.app.inform.emit("%s: %s" % (_("Port is connected. Disconnecting"), port_name))
  766. except Exception:
  767. self.app.inform.emit("[ERROR_NOTCL] %s: %s" % (_("Could not connect to port"), port_name))
  768. def on_add_baudrate_grbl(self):
  769. new_bd = str(self.ui.new_baudrate_entry.get_value())
  770. if int(new_bd) >= 40 and new_bd not in self.ui.baudrates_list_combo.model().stringList():
  771. self.ui.baudrates_list_combo.addItem(new_bd)
  772. self.ui.baudrates_list_combo.setCurrentText(new_bd)
  773. def on_delete_baudrate_grbl(self):
  774. current_idx = self.ui.baudrates_list_combo.currentIndex()
  775. self.ui.baudrates_list_combo.removeItem(current_idx)
  776. def probing_gcode(self, coords, pr_travel, probe_fr, pr_depth, controller):
  777. """
  778. :param coords: a list of (x, y) tuples of probe points coordinates
  779. :type coords: list
  780. :param pr_travel: the height (z) where the probe travel between probe points
  781. :type pr_travel: float
  782. :param probe_fr: feedrate when probing
  783. :type probe_fr: float
  784. :param pr_depth: how much to lower the probe searching for contact
  785. :type pr_depth: float
  786. :param controller: a string with the name of the GCode sender for which to create the probing GCode.
  787. Can be: 'MACH3', 'MACH4', 'LinuxCNC', 'GRBL'
  788. :type controller: str
  789. :return: Probing GCode
  790. :rtype: str
  791. """
  792. p_gcode = ''
  793. header = ''
  794. time_str = "{:%A, %d %B %Y at %H:%M}".format(datetime.now())
  795. header += '(G-CODE GENERATED BY FLATCAM v%s - www.flatcam.org - Version Date: %s)\n' % \
  796. (str(self.app.version), str(self.app.version_date)) + '\n'
  797. header += '(This is a autolevelling probing GCode.)\n' \
  798. '(Make sure that before you start the job you first do a zero for all axis.)\n\n'
  799. header += '(Name: ' + str(self.options['name']) + ')\n'
  800. header += '(Type: ' + "Autolevelling Probing GCode " + ')\n'
  801. header += '(Units: ' + self.units.upper() + ')\n'
  802. header += '(Created on ' + time_str + ')\n\n'
  803. # commands
  804. if controller == 'MACH3':
  805. probing_command = 'G31'
  806. probing_var = '#2002'
  807. openfile_command = 'M40'
  808. closefile_command = 'M41'
  809. elif controller == 'MACH4':
  810. probing_command = 'G31'
  811. probing_var = '#5063'
  812. openfile_command = 'M40'
  813. closefile_command = 'M41'
  814. elif controller == 'LinuxCNC':
  815. probing_command = 'G38.2'
  816. probing_var = '#5422'
  817. openfile_command = '(PROBEOPEN a_probing_points_file.txt)'
  818. closefile_command = '(PROBECLOSE)'
  819. else:
  820. log.debug("CNCJobObject.probing_gcode() -> controller not supported")
  821. return
  822. # #############################################################################################################
  823. # ########################### GCODE construction ##############################################################
  824. # #############################################################################################################
  825. # header
  826. p_gcode += header + '\n\n'
  827. # supplementary message for LinuxCNC
  828. if controller == 'LinuxCNC':
  829. probing_var += "The file with the stored probing points can be found\n" \
  830. "in the configuration folder for LinuxCNC.\n" \
  831. "The name of the file is: a_probing_points_file.txt.\n"
  832. # units
  833. p_gcode += 'G21\n' if self.units == 'MM' else 'G20\n'
  834. # reference mode = absolute
  835. p_gcode += 'G90\n'
  836. # open a new file
  837. p_gcode += openfile_command + '\n'
  838. # move to safe height (probe travel Z)
  839. p_gcode += 'G0 Z%s\n' % str(self.app.dec_format(pr_travel, self.coords_decimals))
  840. # probing points
  841. for idx, xy_tuple in enumerate(coords, 1): # index starts from 1
  842. x = xy_tuple[0]
  843. y = xy_tuple[1]
  844. # move to probing point
  845. p_gcode += "G0 X%sY%s\n" % (
  846. str(self.app.dec_format(x, self.coords_decimals)),
  847. str(self.app.dec_format(y, self.coords_decimals))
  848. )
  849. # do the probing
  850. p_gcode += "%s Z%s F%s\n" % (
  851. probing_command,
  852. str(self.app.dec_format(pr_depth, self.coords_decimals)),
  853. str(self.app.dec_format(probe_fr, self.fr_decimals)),
  854. )
  855. # store in a global numeric variable the value of the detected probe Z
  856. # I offset the global numeric variable by 500 so it does not conflict with something else
  857. temp_var = int(idx + 500)
  858. p_gcode += "#%d = %s\n" % (temp_var, probing_var)
  859. # move to safe height (probe travel Z)
  860. p_gcode += 'G0 Z%s\n' % str(self.app.dec_format(pr_travel, self.coords_decimals))
  861. # close the file
  862. p_gcode += closefile_command + '\n'
  863. # finish the GCode
  864. p_gcode += 'M2'
  865. return p_gcode
  866. def on_generate_probing_gcode(self):
  867. coords = []
  868. for id_key, value in self.al_geometry_dict.items():
  869. x = value['point'].x
  870. y = value['point'].y
  871. coords.append(
  872. (
  873. self.app.dec_format(x, dec=self.app.decimals),
  874. self.app.dec_format(y, dec=self.app.decimals)
  875. )
  876. )
  877. pr_travel = self.ui.ptravelz_entry.get_value()
  878. probe_fr = self.ui.feedrate_probe_entry.get_value()
  879. pr_depth = self.ui.pdepth_entry.get_value()
  880. controller = self.ui.al_controller_combo.get_value()
  881. self.probing_gcode_text = self.probing_gcode(coords, pr_travel, probe_fr, pr_depth, controller)
  882. def on_view_probing_gcode(self):
  883. self.app.proc_container.view.set_busy(_("Loading..."))
  884. gco = self.probing_gcode_text
  885. if gco is None or gco == '':
  886. self.app.inform.emit('[WARNING_NOTCL] %s...' % _('There is nothing to view'))
  887. return
  888. self.gcode_viewer_tab = AppTextEditor(app=self.app, plain_text=True)
  889. # add the tab if it was closed
  890. self.app.ui.plot_tab_area.addTab(self.gcode_viewer_tab, '%s' % _("Code Viewer"))
  891. self.gcode_viewer_tab.setObjectName('code_viewer_tab')
  892. # delete the absolute and relative position and messages in the infobar
  893. self.app.ui.position_label.setText("")
  894. self.app.ui.rel_position_label.setText("")
  895. self.gcode_viewer_tab.code_editor.completer_enable = False
  896. self.gcode_viewer_tab.buttonRun.hide()
  897. # Switch plot_area to CNCJob tab
  898. self.app.ui.plot_tab_area.setCurrentWidget(self.gcode_viewer_tab)
  899. self.gcode_viewer_tab.t_frame.hide()
  900. # then append the text from GCode to the text editor
  901. try:
  902. self.gcode_viewer_tab.load_text(gco, move_to_start=True, clear_text=True)
  903. except Exception as e:
  904. log.debug('FlatCAMCNCJob.on_edit_code_click() -->%s' % str(e))
  905. return
  906. self.gcode_viewer_tab.t_frame.show()
  907. self.app.proc_container.view.set_idle()
  908. self.gcode_viewer_tab.buttonSave.hide()
  909. self.gcode_viewer_tab.buttonOpen.hide()
  910. self.gcode_viewer_tab.buttonPrint.hide()
  911. self.gcode_viewer_tab.buttonPreview.hide()
  912. self.gcode_viewer_tab.buttonReplace.hide()
  913. self.gcode_viewer_tab.sel_all_cb.hide()
  914. self.gcode_viewer_tab.entryReplace.hide()
  915. self.gcode_viewer_tab.code_editor.setReadOnly(True)
  916. self.app.inform.emit('[success] %s...' % _('Loaded Machine Code into Code Viewer'))
  917. def on_updateplot_button_click(self, *args):
  918. """
  919. Callback for the "Updata Plot" button. Reads the form for updates
  920. and plots the object.
  921. """
  922. self.read_form()
  923. self.on_plot_kind_change()
  924. def on_plot_kind_change(self):
  925. kind = self.ui.cncplot_method_combo.get_value()
  926. def worker_task():
  927. with self.app.proc_container.new(_("Plotting...")):
  928. self.plot(kind=kind)
  929. self.app.worker_task.emit({'fcn': worker_task, 'params': []})
  930. def on_exportgcode_button_click(self):
  931. """
  932. Handler activated by a button clicked when exporting GCode.
  933. :param args:
  934. :return:
  935. """
  936. self.app.defaults.report_usage("cncjob_on_exportgcode_button")
  937. self.read_form()
  938. name = self.app.collection.get_active().options['name']
  939. save_gcode = False
  940. if 'Roland' in self.pp_excellon_name or 'Roland' in self.pp_geometry_name:
  941. _filter_ = "RML1 Files .rol (*.rol);;All Files (*.*)"
  942. elif 'hpgl' in self.pp_geometry_name:
  943. _filter_ = "HPGL Files .plt (*.plt);;All Files (*.*)"
  944. else:
  945. save_gcode = True
  946. _filter_ = self.app.defaults['cncjob_save_filters']
  947. try:
  948. dir_file_to_save = self.app.get_last_save_folder() + '/' + str(name)
  949. filename, _f = FCFileSaveDialog.get_saved_filename(
  950. caption=_("Export Code ..."),
  951. directory=dir_file_to_save,
  952. ext_filter=_filter_
  953. )
  954. except TypeError:
  955. filename, _f = FCFileSaveDialog.get_saved_filename(caption=_("Export Code ..."), ext_filter=_filter_)
  956. self.export_gcode_handler(filename, is_gcode=save_gcode)
  957. def export_gcode_handler(self, filename, is_gcode=True):
  958. preamble = ''
  959. postamble = ''
  960. filename = str(filename)
  961. if filename == '':
  962. self.app.inform.emit('[WARNING_NOTCL] %s' % _("Export cancelled ..."))
  963. return
  964. else:
  965. if is_gcode is True:
  966. used_extension = filename.rpartition('.')[2]
  967. self.update_filters(last_ext=used_extension, filter_string='cncjob_save_filters')
  968. new_name = os.path.split(str(filename))[1].rpartition('.')[0]
  969. self.ui.name_entry.set_value(new_name)
  970. self.on_name_activate(silent=True)
  971. try:
  972. if self.ui.snippets_cb.get_value():
  973. preamble = self.append_snippet
  974. postamble = self.prepend_snippet
  975. gc = self.export_gcode(filename, preamble=preamble, postamble=postamble)
  976. except Exception as err:
  977. log.debug("CNCJobObject.export_gcode_handler() --> %s" % str(err))
  978. gc = self.export_gcode(filename)
  979. if gc == 'fail':
  980. return
  981. if self.app.defaults["global_open_style"] is False:
  982. self.app.file_opened.emit("gcode", filename)
  983. self.app.file_saved.emit("gcode", filename)
  984. self.app.inform.emit('[success] %s: %s' % (_("File saved to"), filename))
  985. def on_edit_code_click(self, *args):
  986. """
  987. Handler activated by a button clicked when reviewing GCode.
  988. :param args:
  989. :return:
  990. """
  991. self.app.proc_container.view.set_busy(_("Loading..."))
  992. preamble = self.append_snippet
  993. postamble = self.prepend_snippet
  994. gco = self.export_gcode(preamble=preamble, postamble=postamble, to_file=True)
  995. if gco == 'fail':
  996. return
  997. else:
  998. self.app.gcode_edited = gco
  999. self.gcode_editor_tab = AppTextEditor(app=self.app, plain_text=True)
  1000. # add the tab if it was closed
  1001. self.app.ui.plot_tab_area.addTab(self.gcode_editor_tab, '%s' % _("Code Review"))
  1002. self.gcode_editor_tab.setObjectName('code_editor_tab')
  1003. # delete the absolute and relative position and messages in the infobar
  1004. self.app.ui.position_label.setText("")
  1005. self.app.ui.rel_position_label.setText("")
  1006. self.gcode_editor_tab.code_editor.completer_enable = False
  1007. self.gcode_editor_tab.buttonRun.hide()
  1008. # Switch plot_area to CNCJob tab
  1009. self.app.ui.plot_tab_area.setCurrentWidget(self.gcode_editor_tab)
  1010. self.gcode_editor_tab.t_frame.hide()
  1011. # then append the text from GCode to the text editor
  1012. try:
  1013. self.gcode_editor_tab.load_text(self.app.gcode_edited.getvalue(), move_to_start=True, clear_text=True)
  1014. except Exception as e:
  1015. log.debug('FlatCAMCNCJob.on_edit_code_click() -->%s' % str(e))
  1016. return
  1017. self.gcode_editor_tab.t_frame.show()
  1018. self.app.proc_container.view.set_idle()
  1019. self.gcode_editor_tab.buttonSave.hide()
  1020. self.gcode_editor_tab.buttonOpen.hide()
  1021. self.gcode_editor_tab.buttonPrint.hide()
  1022. self.gcode_editor_tab.buttonPreview.hide()
  1023. self.gcode_editor_tab.buttonReplace.hide()
  1024. self.gcode_editor_tab.sel_all_cb.hide()
  1025. self.gcode_editor_tab.entryReplace.hide()
  1026. self.gcode_editor_tab.code_editor.setReadOnly(True)
  1027. self.app.inform.emit('[success] %s...' % _('Loaded Machine Code into Code Editor'))
  1028. def on_update_source_file(self):
  1029. self.source_file = self.gcode_editor_tab.code_editor.toPlainText()
  1030. def gcode_header(self, comment_start_symbol=None, comment_stop_symbol=None):
  1031. """
  1032. Will create a header to be added to all GCode files generated by FlatCAM
  1033. :param comment_start_symbol: A symbol to be used as the first symbol in a comment
  1034. :param comment_stop_symbol: A symbol to be used as the last symbol in a comment
  1035. :return: A string with a GCode header
  1036. """
  1037. log.debug("FlatCAMCNCJob.gcode_header()")
  1038. time_str = "{:%A, %d %B %Y at %H:%M}".format(datetime.now())
  1039. marlin = False
  1040. hpgl = False
  1041. probe_pp = False
  1042. gcode = ''
  1043. start_comment = comment_start_symbol if comment_start_symbol is not None else '('
  1044. stop_comment = comment_stop_symbol if comment_stop_symbol is not None else ')'
  1045. try:
  1046. for key in self.cnc_tools:
  1047. ppg = self.cnc_tools[key]['data']['ppname_g']
  1048. if 'marlin' in ppg.lower() or 'repetier' in ppg.lower():
  1049. marlin = True
  1050. break
  1051. if ppg == 'hpgl':
  1052. hpgl = True
  1053. break
  1054. if "toolchange_probe" in ppg.lower():
  1055. probe_pp = True
  1056. break
  1057. except KeyError:
  1058. # log.debug("FlatCAMCNCJob.gcode_header() error: --> %s" % str(e))
  1059. pass
  1060. try:
  1061. if 'marlin' in self.options['ppname_e'].lower() or 'repetier' in self.options['ppname_e'].lower():
  1062. marlin = True
  1063. except KeyError:
  1064. # log.debug("FlatCAMCNCJob.gcode_header(): --> There is no such self.option: %s" % str(e))
  1065. pass
  1066. try:
  1067. if "toolchange_probe" in self.options['ppname_e'].lower():
  1068. probe_pp = True
  1069. except KeyError:
  1070. # log.debug("FlatCAMCNCJob.gcode_header(): --> There is no such self.option: %s" % str(e))
  1071. pass
  1072. if marlin is True:
  1073. gcode += ';Marlin(Repetier) G-CODE GENERATED BY FLATCAM v%s - www.flatcam.org - Version Date: %s\n' % \
  1074. (str(self.app.version), str(self.app.version_date)) + '\n'
  1075. gcode += ';Name: ' + str(self.options['name']) + '\n'
  1076. gcode += ';Type: ' + "G-code from " + str(self.options['type']) + '\n'
  1077. # if str(p['options']['type']) == 'Excellon' or str(p['options']['type']) == 'Excellon Geometry':
  1078. # gcode += '(Tools in use: ' + str(p['options']['Tools_in_use']) + ')\n'
  1079. gcode += ';Units: ' + self.units.upper() + '\n' + "\n"
  1080. gcode += ';Created on ' + time_str + '\n' + '\n'
  1081. elif hpgl is True:
  1082. gcode += 'CO "HPGL CODE GENERATED BY FLATCAM v%s - www.flatcam.org - Version Date: %s' % \
  1083. (str(self.app.version), str(self.app.version_date)) + '";\n'
  1084. gcode += 'CO "Name: ' + str(self.options['name']) + '";\n'
  1085. gcode += 'CO "Type: ' + "HPGL code from " + str(self.options['type']) + '";\n'
  1086. # if str(p['options']['type']) == 'Excellon' or str(p['options']['type']) == 'Excellon Geometry':
  1087. # gcode += '(Tools in use: ' + str(p['options']['Tools_in_use']) + ')\n'
  1088. gcode += 'CO "Units: ' + self.units.upper() + '";\n'
  1089. gcode += 'CO "Created on ' + time_str + '";\n'
  1090. elif probe_pp is True:
  1091. gcode += '(G-CODE GENERATED BY FLATCAM v%s - www.flatcam.org - Version Date: %s)\n' % \
  1092. (str(self.app.version), str(self.app.version_date)) + '\n'
  1093. gcode += '(This GCode tool change is done by using a Probe.)\n' \
  1094. '(Make sure that before you start the job you first do a rough zero for Z axis.)\n' \
  1095. '(This means that you need to zero the CNC axis and then jog to the toolchange X, Y location,)\n' \
  1096. '(mount the probe and adjust the Z so more or less the probe tip touch the plate. ' \
  1097. 'Then zero the Z axis.)\n' + '\n'
  1098. gcode += '(Name: ' + str(self.options['name']) + ')\n'
  1099. gcode += '(Type: ' + "G-code from " + str(self.options['type']) + ')\n'
  1100. # if str(p['options']['type']) == 'Excellon' or str(p['options']['type']) == 'Excellon Geometry':
  1101. # gcode += '(Tools in use: ' + str(p['options']['Tools_in_use']) + ')\n'
  1102. gcode += '(Units: ' + self.units.upper() + ')\n' + "\n"
  1103. gcode += '(Created on ' + time_str + ')\n' + '\n'
  1104. else:
  1105. gcode += '%sG-CODE GENERATED BY FLATCAM v%s - www.flatcam.org - Version Date: %s%s\n' % \
  1106. (start_comment, str(self.app.version), str(self.app.version_date), stop_comment) + '\n'
  1107. gcode += '%sName: ' % start_comment + str(self.options['name']) + '%s\n' % stop_comment
  1108. gcode += '%sType: ' % start_comment + "G-code from " + str(self.options['type']) + '%s\n' % stop_comment
  1109. # if str(p['options']['type']) == 'Excellon' or str(p['options']['type']) == 'Excellon Geometry':
  1110. # gcode += '(Tools in use: ' + str(p['options']['Tools_in_use']) + ')\n'
  1111. gcode += '%sUnits: ' % start_comment + self.units.upper() + '%s\n' % stop_comment + "\n"
  1112. gcode += '%sCreated on ' % start_comment + time_str + '%s\n' % stop_comment + '\n'
  1113. return gcode
  1114. @staticmethod
  1115. def gcode_footer(end_command=None):
  1116. """
  1117. Will add the M02 to the end of GCode, if requested.
  1118. :param end_command: 'M02' or 'M30' - String
  1119. :return:
  1120. """
  1121. if end_command:
  1122. return end_command
  1123. else:
  1124. return 'M02'
  1125. def export_gcode(self, filename=None, preamble='', postamble='', to_file=False, from_tcl=False):
  1126. """
  1127. This will save the GCode from the Gcode object to a file on the OS filesystem
  1128. :param filename: filename for the GCode file
  1129. :param preamble: a custom Gcode block to be added at the beginning of the Gcode file
  1130. :param postamble: a custom Gcode block to be added at the end of the Gcode file
  1131. :param to_file: if False then no actual file is saved but the app will know that a file was created
  1132. :param from_tcl: True if run from Tcl Shell
  1133. :return: None
  1134. """
  1135. # gcode = ''
  1136. # roland = False
  1137. # hpgl = False
  1138. # isel_icp = False
  1139. include_header = True
  1140. if preamble == '':
  1141. preamble = self.app.defaults["cncjob_prepend"]
  1142. if postamble == '':
  1143. preamble = self.app.defaults["cncjob_append"]
  1144. try:
  1145. if self.special_group:
  1146. self.app.inform.emit('[WARNING_NOTCL] %s %s %s.' %
  1147. (_("This CNCJob object can't be processed because it is a"),
  1148. str(self.special_group),
  1149. _("CNCJob object")))
  1150. return 'fail'
  1151. except AttributeError:
  1152. pass
  1153. # if this dict is not empty then the object is a Geometry object
  1154. if self.cnc_tools:
  1155. first_key = next(iter(self.cnc_tools))
  1156. include_header = self.app.preprocessors[self.cnc_tools[first_key]['data']['ppname_g']].include_header
  1157. # if this dict is not empty then the object is an Excellon object
  1158. if self.exc_cnc_tools:
  1159. first_key = next(iter(self.exc_cnc_tools))
  1160. include_header = self.app.preprocessors[
  1161. self.exc_cnc_tools[first_key]['data']['tools_drill_ppname_e']
  1162. ].include_header
  1163. gcode = ''
  1164. if include_header is False:
  1165. g = preamble
  1166. # detect if using multi-tool and make the Gcode summation correctly for each case
  1167. if self.multitool is True:
  1168. for tooluid_key in self.cnc_tools:
  1169. for key, value in self.cnc_tools[tooluid_key].items():
  1170. if key == 'gcode':
  1171. gcode += value
  1172. break
  1173. else:
  1174. gcode += self.gcode
  1175. g = g + gcode + postamble
  1176. else:
  1177. # search for the GCode beginning which is usually a G20 or G21
  1178. # fix so the preamble gets inserted in between the comments header and the actual start of GCODE
  1179. # g_idx = gcode.rfind('G20')
  1180. #
  1181. # # if it did not find 'G20' then search for 'G21'
  1182. # if g_idx == -1:
  1183. # g_idx = gcode.rfind('G21')
  1184. #
  1185. # # if it did not find 'G20' and it did not find 'G21' then there is an error and return
  1186. # if g_idx == -1:
  1187. # self.app.inform.emit('[ERROR_NOTCL] %s' % _("G-code does not have a units code: either G20 or G21"))
  1188. # return
  1189. # detect if using multi-tool and make the Gcode summation correctly for each case
  1190. if self.multitool is True:
  1191. if self.origin_kind == 'excellon':
  1192. for tooluid_key in self.exc_cnc_tools:
  1193. for key, value in self.exc_cnc_tools[tooluid_key].items():
  1194. if key == 'gcode' and value:
  1195. gcode += value
  1196. break
  1197. else:
  1198. for tooluid_key in self.cnc_tools:
  1199. for key, value in self.cnc_tools[tooluid_key].items():
  1200. if key == 'gcode' and value:
  1201. gcode += value
  1202. break
  1203. else:
  1204. gcode += self.gcode
  1205. end_gcode = self.gcode_footer() if self.app.defaults['cncjob_footer'] is True else ''
  1206. # detect if using a HPGL preprocessor
  1207. hpgl = False
  1208. if self.cnc_tools:
  1209. for key in self.cnc_tools:
  1210. if 'ppname_g' in self.cnc_tools[key]['data']:
  1211. if 'hpgl' in self.cnc_tools[key]['data']['ppname_g']:
  1212. hpgl = True
  1213. break
  1214. elif self.exc_cnc_tools:
  1215. for key in self.cnc_tools:
  1216. if 'ppname_e' in self.cnc_tools[key]['data']:
  1217. if 'hpgl' in self.cnc_tools[key]['data']['ppname_e']:
  1218. hpgl = True
  1219. break
  1220. if hpgl:
  1221. processed_gcode = ''
  1222. pa_re = re.compile(r"^PA\s*(-?\d+\.\d*),?\s*(-?\d+\.\d*)*;?$")
  1223. for gline in gcode.splitlines():
  1224. match = pa_re.search(gline)
  1225. if match:
  1226. x_int = int(float(match.group(1)))
  1227. y_int = int(float(match.group(2)))
  1228. new_line = 'PA%d,%d;\n' % (x_int, y_int)
  1229. processed_gcode += new_line
  1230. else:
  1231. processed_gcode += gline + '\n'
  1232. gcode = processed_gcode
  1233. g = self.gc_header + '\n' + preamble + '\n' + gcode + postamble + end_gcode
  1234. else:
  1235. try:
  1236. g_idx = gcode.index('G94')
  1237. if preamble != '' and postamble != '':
  1238. g = self.gc_header + gcode[:g_idx + 3] + '\n' + preamble + '\n' + \
  1239. gcode[(g_idx + 3):] + postamble + end_gcode
  1240. elif preamble == '':
  1241. g = self.gc_header + gcode[:g_idx + 3] + '\n' + \
  1242. gcode[(g_idx + 3):] + postamble + end_gcode
  1243. elif postamble == '':
  1244. g = self.gc_header + gcode[:g_idx + 3] + '\n' + preamble + '\n' + \
  1245. gcode[(g_idx + 3):] + end_gcode
  1246. else:
  1247. g = self.gc_header + gcode[:g_idx + 3] + gcode[(g_idx + 3):] + end_gcode
  1248. except ValueError:
  1249. self.app.inform.emit('[ERROR_NOTCL] %s' %
  1250. _("G-code does not have a G94 code.\n"
  1251. "Append Code snippet will not be used.."))
  1252. g = self.gc_header + '\n' + gcode + postamble + end_gcode
  1253. # if toolchange custom is used, replace M6 code with the code from the Toolchange Custom Text box
  1254. # if self.ui.toolchange_cb.get_value() is True:
  1255. # # match = self.re_toolchange.search(g)
  1256. # if 'M6' in g:
  1257. # m6_code = self.parse_custom_toolchange_code(self.ui.toolchange_text.get_value())
  1258. # if m6_code is None or m6_code == '':
  1259. # self.app.inform.emit(
  1260. # '[ERROR_NOTCL] %s' % _("Cancelled. The Toolchange Custom code is enabled but it's empty.")
  1261. # )
  1262. # return 'fail'
  1263. #
  1264. # g = g.replace('M6', m6_code)
  1265. # self.app.inform.emit('[success] %s' % _("Toolchange G-code was replaced by a custom code."))
  1266. lines = StringIO(g)
  1267. # Write
  1268. if filename is not None:
  1269. try:
  1270. force_windows_line_endings = self.app.defaults['cncjob_line_ending']
  1271. if force_windows_line_endings and sys.platform != 'win32':
  1272. with open(filename, 'w', newline='\r\n') as f:
  1273. for line in lines:
  1274. f.write(line)
  1275. else:
  1276. with open(filename, 'w') as f:
  1277. for line in lines:
  1278. f.write(line)
  1279. except FileNotFoundError:
  1280. self.app.inform.emit('[WARNING_NOTCL] %s' % _("No such file or directory"))
  1281. return
  1282. except PermissionError:
  1283. self.app.inform.emit(
  1284. '[WARNING] %s' % _("Permission denied, saving not possible.\n"
  1285. "Most likely another app is holding the file open and not accessible.")
  1286. )
  1287. return 'fail'
  1288. elif to_file is False:
  1289. # Just for adding it to the recent files list.
  1290. if self.app.defaults["global_open_style"] is False:
  1291. self.app.file_opened.emit("cncjob", filename)
  1292. self.app.file_saved.emit("cncjob", filename)
  1293. self.app.inform.emit('[success] %s: %s' % (_("Saved to"), filename))
  1294. else:
  1295. return lines
  1296. # def on_toolchange_custom_clicked(self, signal):
  1297. # """
  1298. # Handler for clicking toolchange custom.
  1299. #
  1300. # :param signal:
  1301. # :return:
  1302. # """
  1303. #
  1304. # try:
  1305. # if 'toolchange_custom' not in str(self.options['ppname_e']).lower():
  1306. # if self.ui.toolchange_cb.get_value():
  1307. # self.ui.toolchange_cb.set_value(False)
  1308. # self.app.inform.emit('[WARNING_NOTCL] %s' %
  1309. # _("The used preprocessor file has to have in it's name: 'toolchange_custom'"))
  1310. # except KeyError:
  1311. # try:
  1312. # for key in self.cnc_tools:
  1313. # ppg = self.cnc_tools[key]['data']['ppname_g']
  1314. # if 'toolchange_custom' not in str(ppg).lower():
  1315. # if self.ui.toolchange_cb.get_value():
  1316. # self.ui.toolchange_cb.set_value(False)
  1317. # self.app.inform.emit('[WARNING_NOTCL] %s' %
  1318. # _("The used preprocessor file has to have in it's name: "
  1319. # "'toolchange_custom'"))
  1320. # except KeyError:
  1321. # self.app.inform.emit('[ERROR] %s' % _("There is no preprocessor file."))
  1322. def get_gcode(self, preamble='', postamble=''):
  1323. """
  1324. We need this to be able to get_gcode separately for shell command export_gcode
  1325. :param preamble: Extra GCode added to the beginning of the GCode
  1326. :param postamble: Extra GCode added at the end of the GCode
  1327. :return: The modified GCode
  1328. """
  1329. return preamble + '\n' + self.gcode + "\n" + postamble
  1330. def get_svg(self):
  1331. # we need this to be able get_svg separately for shell command export_svg
  1332. pass
  1333. def on_plot_cb_click(self, *args):
  1334. """
  1335. Handler for clicking on the Plot checkbox.
  1336. :param args:
  1337. :return:
  1338. """
  1339. if self.muted_ui:
  1340. return
  1341. kind = self.ui.cncplot_method_combo.get_value()
  1342. self.plot(kind=kind)
  1343. self.read_form_item('plot')
  1344. self.ui_disconnect()
  1345. cb_flag = self.ui.plot_cb.isChecked()
  1346. for row in range(self.ui.cnc_tools_table.rowCount()):
  1347. table_cb = self.ui.cnc_tools_table.cellWidget(row, 6)
  1348. if cb_flag:
  1349. table_cb.setChecked(True)
  1350. else:
  1351. table_cb.setChecked(False)
  1352. self.ui_connect()
  1353. def on_plot_cb_click_table(self):
  1354. """
  1355. Handler for clicking the plot checkboxes added into a Table on each row. Purpose: toggle visibility for the
  1356. tool/aperture found on that row.
  1357. :return:
  1358. """
  1359. # self.ui.cnc_tools_table.cellWidget(row, 2).widget().setCheckState(QtCore.Qt.Unchecked)
  1360. self.ui_disconnect()
  1361. # cw = self.sender()
  1362. # cw_index = self.ui.cnc_tools_table.indexAt(cw.pos())
  1363. # cw_row = cw_index.row()
  1364. kind = self.ui.cncplot_method_combo.get_value()
  1365. self.shapes.clear(update=True)
  1366. if self.origin_kind == "excellon":
  1367. for r in range(self.ui.exc_cnc_tools_table.rowCount()):
  1368. row_dia = float('%.*f' % (self.decimals, float(self.ui.exc_cnc_tools_table.item(r, 1).text())))
  1369. for tooluid_key in self.exc_cnc_tools:
  1370. tooldia = float('%.*f' % (self.decimals, float(tooluid_key)))
  1371. if row_dia == tooldia:
  1372. gcode_parsed = self.exc_cnc_tools[tooluid_key]['gcode_parsed']
  1373. if self.ui.exc_cnc_tools_table.cellWidget(r, 6).isChecked():
  1374. self.plot2(tooldia=tooldia, obj=self, visible=True, gcode_parsed=gcode_parsed, kind=kind)
  1375. else:
  1376. for tooluid_key in self.cnc_tools:
  1377. tooldia = float('%.*f' % (self.decimals, float(self.cnc_tools[tooluid_key]['tooldia'])))
  1378. gcode_parsed = self.cnc_tools[tooluid_key]['gcode_parsed']
  1379. # tool_uid = int(self.ui.cnc_tools_table.item(cw_row, 3).text())
  1380. for r in range(self.ui.cnc_tools_table.rowCount()):
  1381. if int(self.ui.cnc_tools_table.item(r, 5).text()) == int(tooluid_key):
  1382. if self.ui.cnc_tools_table.cellWidget(r, 6).isChecked():
  1383. self.plot2(tooldia=tooldia, obj=self, visible=True, gcode_parsed=gcode_parsed, kind=kind)
  1384. self.shapes.redraw()
  1385. # make sure that the general plot is disabled if one of the row plot's are disabled and
  1386. # if all the row plot's are enabled also enable the general plot checkbox
  1387. cb_cnt = 0
  1388. total_row = self.ui.cnc_tools_table.rowCount()
  1389. for row in range(total_row):
  1390. if self.ui.cnc_tools_table.cellWidget(row, 6).isChecked():
  1391. cb_cnt += 1
  1392. else:
  1393. cb_cnt -= 1
  1394. if cb_cnt < total_row:
  1395. self.ui.plot_cb.setChecked(False)
  1396. else:
  1397. self.ui.plot_cb.setChecked(True)
  1398. self.ui_connect()
  1399. def plot(self, visible=None, kind='all'):
  1400. """
  1401. # Does all the required setup and returns False
  1402. # if the 'ptint' option is set to False.
  1403. :param visible: Boolean to decide if the object will be plotted as visible or disabled on canvas
  1404. :param kind: String. Can be "all" or "travel" or "cut". For CNCJob plotting
  1405. :return: None
  1406. """
  1407. if not FlatCAMObj.plot(self):
  1408. return
  1409. visible = visible if visible else self.options['plot']
  1410. if self.app.is_legacy is False:
  1411. if self.ui.annotation_cb.get_value() and self.ui.plot_cb.get_value():
  1412. self.text_col.enabled = True
  1413. else:
  1414. self.text_col.enabled = False
  1415. self.annotation.redraw()
  1416. try:
  1417. if self.multitool is False: # single tool usage
  1418. try:
  1419. dia_plot = float(self.options["tooldia"])
  1420. except ValueError:
  1421. # we may have a tuple with only one element and a comma
  1422. dia_plot = [float(el) for el in self.options["tooldia"].split(',') if el != ''][0]
  1423. self.plot2(tooldia=dia_plot, obj=self, visible=visible, kind=kind)
  1424. else:
  1425. # I do this so the travel lines thickness will reflect the tool diameter
  1426. # may work only for objects created within the app and not Gcode imported from elsewhere for which we
  1427. # don't know the origin
  1428. if self.origin_kind == "excellon":
  1429. if self.exc_cnc_tools:
  1430. for tooldia_key in self.exc_cnc_tools:
  1431. tooldia = float('%.*f' % (self.decimals, float(tooldia_key)))
  1432. gcode_parsed = self.exc_cnc_tools[tooldia_key]['gcode_parsed']
  1433. if not gcode_parsed:
  1434. continue
  1435. # gcode_parsed = self.gcode_parsed
  1436. self.plot2(tooldia=tooldia, obj=self, visible=visible, gcode_parsed=gcode_parsed, kind=kind)
  1437. else:
  1438. # multiple tools usage
  1439. if self.cnc_tools:
  1440. for tooluid_key in self.cnc_tools:
  1441. tooldia = float('%.*f' % (self.decimals, float(self.cnc_tools[tooluid_key]['tooldia'])))
  1442. gcode_parsed = self.cnc_tools[tooluid_key]['gcode_parsed']
  1443. self.plot2(tooldia=tooldia, obj=self, visible=visible, gcode_parsed=gcode_parsed, kind=kind)
  1444. self.shapes.redraw()
  1445. except (ObjectDeleted, AttributeError):
  1446. self.shapes.clear(update=True)
  1447. if self.app.is_legacy is False:
  1448. self.annotation.clear(update=True)
  1449. def on_annotation_change(self):
  1450. """
  1451. Handler for toggling the annotation display by clicking a checkbox.
  1452. :return:
  1453. """
  1454. if self.app.is_legacy is False:
  1455. if self.ui.annotation_cb.get_value():
  1456. self.text_col.enabled = True
  1457. else:
  1458. self.text_col.enabled = False
  1459. # kind = self.ui.cncplot_method_combo.get_value()
  1460. # self.plot(kind=kind)
  1461. self.annotation.redraw()
  1462. else:
  1463. kind = self.ui.cncplot_method_combo.get_value()
  1464. self.plot(kind=kind)
  1465. def convert_units(self, units):
  1466. """
  1467. Units conversion used by the CNCJob objects.
  1468. :param units: Can be "MM" or "IN"
  1469. :return:
  1470. """
  1471. log.debug("FlatCAMObj.FlatCAMECNCjob.convert_units()")
  1472. factor = CNCjob.convert_units(self, units)
  1473. self.options["tooldia"] = float(self.options["tooldia"]) * factor
  1474. param_list = ['cutz', 'depthperpass', 'travelz', 'feedrate', 'feedrate_z', 'feedrate_rapid',
  1475. 'endz', 'toolchangez']
  1476. temp_tools_dict = {}
  1477. tool_dia_copy = {}
  1478. data_copy = {}
  1479. for tooluid_key, tooluid_value in self.cnc_tools.items():
  1480. for dia_key, dia_value in tooluid_value.items():
  1481. if dia_key == 'tooldia':
  1482. dia_value *= factor
  1483. dia_value = float('%.*f' % (self.decimals, dia_value))
  1484. tool_dia_copy[dia_key] = dia_value
  1485. if dia_key == 'offset':
  1486. tool_dia_copy[dia_key] = dia_value
  1487. if dia_key == 'offset_value':
  1488. dia_value *= factor
  1489. tool_dia_copy[dia_key] = dia_value
  1490. if dia_key == 'type':
  1491. tool_dia_copy[dia_key] = dia_value
  1492. if dia_key == 'tool_type':
  1493. tool_dia_copy[dia_key] = dia_value
  1494. if dia_key == 'data':
  1495. for data_key, data_value in dia_value.items():
  1496. # convert the form fields that are convertible
  1497. for param in param_list:
  1498. if data_key == param and data_value is not None:
  1499. data_copy[data_key] = data_value * factor
  1500. # copy the other dict entries that are not convertible
  1501. if data_key not in param_list:
  1502. data_copy[data_key] = data_value
  1503. tool_dia_copy[dia_key] = deepcopy(data_copy)
  1504. data_copy.clear()
  1505. if dia_key == 'gcode':
  1506. tool_dia_copy[dia_key] = dia_value
  1507. if dia_key == 'gcode_parsed':
  1508. tool_dia_copy[dia_key] = dia_value
  1509. if dia_key == 'solid_geometry':
  1510. tool_dia_copy[dia_key] = dia_value
  1511. # if dia_key == 'solid_geometry':
  1512. # tool_dia_copy[dia_key] = affinity.scale(dia_value, xfact=factor, origin=(0, 0))
  1513. # if dia_key == 'gcode_parsed':
  1514. # for g in dia_value:
  1515. # g['geom'] = affinity.scale(g['geom'], factor, factor, origin=(0, 0))
  1516. #
  1517. # tool_dia_copy['gcode_parsed'] = deepcopy(dia_value)
  1518. # tool_dia_copy['solid_geometry'] = cascaded_union([geo['geom'] for geo in dia_value])
  1519. temp_tools_dict.update({
  1520. tooluid_key: deepcopy(tool_dia_copy)
  1521. })
  1522. tool_dia_copy.clear()
  1523. self.cnc_tools.clear()
  1524. self.cnc_tools = deepcopy(temp_tools_dict)