FlatCAMCommon.py 101 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470
  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 (major mod): Marius Adrian Stanciu #
  10. # Date: 11/4/2019 #
  11. # ##########################################################
  12. from PyQt5 import QtGui, QtCore, QtWidgets
  13. from flatcamGUI.GUIElements import FCTable, FCEntry, FCButton, FCDoubleSpinner, FCComboBox, FCCheckBox, FCSpinner, \
  14. FCTree, RadioSet
  15. from camlib import to_dict
  16. import sys
  17. import webbrowser
  18. import json
  19. from copy import deepcopy
  20. from datetime import datetime
  21. import gettext
  22. import FlatCAMTranslation as fcTranslate
  23. import builtins
  24. fcTranslate.apply_language('strings')
  25. if '_' not in builtins.__dict__:
  26. _ = gettext.gettext
  27. class LoudDict(dict):
  28. """
  29. A Dictionary with a callback for
  30. item changes.
  31. """
  32. def __init__(self, *args, **kwargs):
  33. dict.__init__(self, *args, **kwargs)
  34. self.callback = lambda x: None
  35. def __setitem__(self, key, value):
  36. """
  37. Overridden __setitem__ method. Will emit 'changed(QString)'
  38. if the item was changed, with key as parameter.
  39. """
  40. if key in self and self.__getitem__(key) == value:
  41. return
  42. dict.__setitem__(self, key, value)
  43. self.callback(key)
  44. def update(self, *args, **kwargs):
  45. if len(args) > 1:
  46. raise TypeError("update expected at most 1 arguments, got %d" % len(args))
  47. other = dict(*args, **kwargs)
  48. for key in other:
  49. self[key] = other[key]
  50. def set_change_callback(self, callback):
  51. """
  52. Assigns a function as callback on item change. The callback
  53. will receive the key of the object that was changed.
  54. :param callback: Function to call on item change.
  55. :type callback: func
  56. :return: None
  57. """
  58. self.callback = callback
  59. class FCSignal:
  60. """
  61. Taken from here: https://blog.abstractfactory.io/dynamic-signals-in-pyqt/
  62. """
  63. def __init__(self):
  64. self.__subscribers = []
  65. def emit(self, *args, **kwargs):
  66. for subs in self.__subscribers:
  67. subs(*args, **kwargs)
  68. def connect(self, func):
  69. self.__subscribers.append(func)
  70. def disconnect(self, func):
  71. try:
  72. self.__subscribers.remove(func)
  73. except ValueError:
  74. print('Warning: function %s not removed '
  75. 'from signal %s' % (func, self))
  76. class BookmarkManager(QtWidgets.QWidget):
  77. mark_rows = QtCore.pyqtSignal()
  78. def __init__(self, app, storage, parent=None):
  79. super(BookmarkManager, self).__init__(parent)
  80. self.app = app
  81. assert isinstance(storage, dict), "Storage argument is not a dictionary"
  82. self.bm_dict = deepcopy(storage)
  83. # Icon and title
  84. # self.setWindowIcon(parent.app_icon)
  85. # self.setWindowTitle(_("Bookmark Manager"))
  86. # self.resize(600, 400)
  87. # title = QtWidgets.QLabel(
  88. # "<font size=8><B>FlatCAM</B></font><BR>"
  89. # )
  90. # title.setOpenExternalLinks(True)
  91. # layouts
  92. layout = QtWidgets.QVBoxLayout()
  93. self.setLayout(layout)
  94. table_hlay = QtWidgets.QHBoxLayout()
  95. layout.addLayout(table_hlay)
  96. self.table_widget = FCTable(drag_drop=True, protected_rows=[0, 1])
  97. self.table_widget.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
  98. table_hlay.addWidget(self.table_widget)
  99. self.table_widget.setColumnCount(3)
  100. self.table_widget.setColumnWidth(0, 20)
  101. self.table_widget.setHorizontalHeaderLabels(
  102. [
  103. '#',
  104. _('Title'),
  105. _('Web Link')
  106. ]
  107. )
  108. self.table_widget.horizontalHeaderItem(0).setToolTip(
  109. _("Index.\n"
  110. "The rows in gray color will populate the Bookmarks menu.\n"
  111. "The number of gray colored rows is set in Preferences."))
  112. self.table_widget.horizontalHeaderItem(1).setToolTip(
  113. _("Description of the link that is set as an menu action.\n"
  114. "Try to keep it short because it is installed as a menu item."))
  115. self.table_widget.horizontalHeaderItem(2).setToolTip(
  116. _("Web Link. E.g: https://your_website.org "))
  117. # pal = QtGui.QPalette()
  118. # pal.setColor(QtGui.QPalette.Background, Qt.white)
  119. # New Bookmark
  120. new_vlay = QtWidgets.QVBoxLayout()
  121. layout.addLayout(new_vlay)
  122. new_title_lbl = QtWidgets.QLabel('<b>%s</b>' % _("New Bookmark"))
  123. new_vlay.addWidget(new_title_lbl)
  124. form0 = QtWidgets.QFormLayout()
  125. new_vlay.addLayout(form0)
  126. title_lbl = QtWidgets.QLabel('%s:' % _("Title"))
  127. self.title_entry = FCEntry()
  128. form0.addRow(title_lbl, self.title_entry)
  129. link_lbl = QtWidgets.QLabel('%s:' % _("Web Link"))
  130. self.link_entry = FCEntry()
  131. self.link_entry.set_value('http://')
  132. form0.addRow(link_lbl, self.link_entry)
  133. # Buttons Layout
  134. button_hlay = QtWidgets.QHBoxLayout()
  135. layout.addLayout(button_hlay)
  136. add_entry_btn = FCButton(_("Add Entry"))
  137. remove_entry_btn = FCButton(_("Remove Entry"))
  138. export_list_btn = FCButton(_("Export List"))
  139. import_list_btn = FCButton(_("Import List"))
  140. # closebtn = QtWidgets.QPushButton(_("Close"))
  141. # button_hlay.addStretch()
  142. button_hlay.addWidget(add_entry_btn)
  143. button_hlay.addWidget(remove_entry_btn)
  144. button_hlay.addWidget(export_list_btn)
  145. button_hlay.addWidget(import_list_btn)
  146. # button_hlay.addWidget(closebtn)
  147. # ##############################################################################
  148. # ######################## SIGNALS #############################################
  149. # ##############################################################################
  150. add_entry_btn.clicked.connect(self.on_add_entry)
  151. remove_entry_btn.clicked.connect(self.on_remove_entry)
  152. export_list_btn.clicked.connect(self.on_export_bookmarks)
  153. import_list_btn.clicked.connect(self.on_import_bookmarks)
  154. self.title_entry.returnPressed.connect(self.on_add_entry)
  155. self.link_entry.returnPressed.connect(self.on_add_entry)
  156. # closebtn.clicked.connect(self.accept)
  157. self.table_widget.drag_drop_sig.connect(self.mark_table_rows_for_actions)
  158. self.build_bm_ui()
  159. def build_bm_ui(self):
  160. self.table_widget.setRowCount(len(self.bm_dict))
  161. nr_crt = 0
  162. sorted_bookmarks = sorted(list(self.bm_dict.items()), key=lambda x: int(x[0]))
  163. for entry, bookmark in sorted_bookmarks:
  164. row = nr_crt
  165. nr_crt += 1
  166. title = bookmark[0]
  167. weblink = bookmark[1]
  168. id_item = QtWidgets.QTableWidgetItem('%d' % int(nr_crt))
  169. # id.setFlags(QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
  170. self.table_widget.setItem(row, 0, id_item) # Tool name/id
  171. title_item = QtWidgets.QTableWidgetItem(title)
  172. self.table_widget.setItem(row, 1, title_item)
  173. weblink_txt = QtWidgets.QTextBrowser()
  174. weblink_txt.setOpenExternalLinks(True)
  175. weblink_txt.setFrameStyle(QtWidgets.QFrame.NoFrame)
  176. weblink_txt.document().setDefaultStyleSheet("a{ text-decoration: none; }")
  177. weblink_txt.setHtml('<a href=%s>%s</a>' % (weblink, weblink))
  178. self.table_widget.setCellWidget(row, 2, weblink_txt)
  179. vertical_header = self.table_widget.verticalHeader()
  180. vertical_header.hide()
  181. horizontal_header = self.table_widget.horizontalHeader()
  182. horizontal_header.setMinimumSectionSize(10)
  183. horizontal_header.setDefaultSectionSize(70)
  184. horizontal_header.setSectionResizeMode(0, QtWidgets.QHeaderView.Fixed)
  185. horizontal_header.resizeSection(0, 20)
  186. horizontal_header.setSectionResizeMode(1, QtWidgets.QHeaderView.ResizeToContents)
  187. horizontal_header.setSectionResizeMode(2, QtWidgets.QHeaderView.Stretch)
  188. self.mark_table_rows_for_actions()
  189. self.app.defaults["global_bookmarks"].clear()
  190. for key, val in self.bm_dict.items():
  191. self.app.defaults["global_bookmarks"][key] = deepcopy(val)
  192. def on_add_entry(self, **kwargs):
  193. """
  194. Add a entry in the Bookmark Table and in the menu actions
  195. :return: None
  196. """
  197. if 'title' in kwargs:
  198. title = kwargs['title']
  199. else:
  200. title = self.title_entry.get_value()
  201. if title == '':
  202. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Title entry is empty."))
  203. return 'fail'
  204. if 'link' is kwargs:
  205. link = kwargs['link']
  206. else:
  207. link = self.link_entry.get_value()
  208. if link == 'http://':
  209. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Web link entry is empty."))
  210. return 'fail'
  211. # if 'http' not in link or 'https' not in link:
  212. # link = 'http://' + link
  213. for bookmark in self.bm_dict.values():
  214. if title == bookmark[0] or link == bookmark[1]:
  215. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Either the Title or the Weblink already in the table."))
  216. return 'fail'
  217. # for some reason if the last char in the weblink is a slash it does not make the link clickable
  218. # so I remove it
  219. if link[-1] == '/':
  220. link = link[:-1]
  221. # add the new entry to storage
  222. new_entry = len(self.bm_dict) + 1
  223. self.bm_dict[str(new_entry)] = [title, link]
  224. # add the link to the menu but only if it is within the set limit
  225. bm_limit = int(self.app.defaults["global_bookmarks_limit"])
  226. if len(self.bm_dict) < bm_limit:
  227. act = QtWidgets.QAction(parent=self.app.ui.menuhelp_bookmarks)
  228. act.setText(title)
  229. act.setIcon(QtGui.QIcon(self.app.resource_location + '/link16.png'))
  230. act.triggered.connect(lambda: webbrowser.open(link))
  231. self.app.ui.menuhelp_bookmarks.insertAction(self.app.ui.menuhelp_bookmarks_manager, act)
  232. self.app.inform.emit('[success] %s' % _("Bookmark added."))
  233. # add the new entry to the bookmark manager table
  234. self.build_bm_ui()
  235. def on_remove_entry(self):
  236. """
  237. Remove an Entry in the Bookmark table and from the menu actions
  238. :return:
  239. """
  240. index_list = []
  241. for model_index in self.table_widget.selectionModel().selectedRows():
  242. index = QtCore.QPersistentModelIndex(model_index)
  243. index_list.append(index)
  244. title_to_remove = self.table_widget.item(model_index.row(), 1).text()
  245. if title_to_remove == 'FlatCAM' or title_to_remove == 'Backup Site':
  246. self.app.inform.emit('[WARNING_NOTCL] %s.' % _("This bookmark can not be removed"))
  247. self.build_bm_ui()
  248. return
  249. else:
  250. for k, bookmark in list(self.bm_dict.items()):
  251. if title_to_remove == bookmark[0]:
  252. # remove from the storage
  253. self.bm_dict.pop(k, None)
  254. for act in self.app.ui.menuhelp_bookmarks.actions():
  255. if act.text() == title_to_remove:
  256. # disconnect the signal
  257. try:
  258. act.triggered.disconnect()
  259. except TypeError:
  260. pass
  261. # remove the action from the menu
  262. self.app.ui.menuhelp_bookmarks.removeAction(act)
  263. # house keeping: it pays to have keys increased by one
  264. new_key = 0
  265. new_dict = {}
  266. for k, v in self.bm_dict.items():
  267. # we start with key 1 so we can use the len(self.bm_dict)
  268. # when adding bookmarks (keys in bm_dict)
  269. new_key += 1
  270. new_dict[str(new_key)] = v
  271. self.bm_dict = deepcopy(new_dict)
  272. new_dict.clear()
  273. self.app.inform.emit('[success] %s' % _("Bookmark removed."))
  274. # for index in index_list:
  275. # self.table_widget.model().removeRow(index.row())
  276. self.build_bm_ui()
  277. def on_export_bookmarks(self):
  278. self.app.report_usage("on_export_bookmarks")
  279. self.app.log.debug("on_export_bookmarks()")
  280. date = str(datetime.today()).rpartition('.')[0]
  281. date = ''.join(c for c in date if c not in ':-')
  282. date = date.replace(' ', '_')
  283. filter__ = "Text File (*.TXT);;All Files (*.*)"
  284. filename, _f = QtWidgets.QFileDialog.getSaveFileName(caption=_("Export FlatCAM Bookmarks"),
  285. directory='{l_save}/FlatCAM_{n}_{date}'.format(
  286. l_save=str(self.app.get_last_save_folder()),
  287. n=_("Bookmarks"),
  288. date=date),
  289. filter=filter__)
  290. filename = str(filename)
  291. if filename == "":
  292. self.app.inform.emit('[WARNING_NOTCL] %s' % _("FlatCAM bookmarks export cancelled."))
  293. return
  294. else:
  295. try:
  296. f = open(filename, 'w')
  297. f.close()
  298. except PermissionError:
  299. self.app.inform.emit('[WARNING] %s' %
  300. _("Permission denied, saving not possible.\n"
  301. "Most likely another app is holding the file open and not accessible."))
  302. return
  303. except IOError:
  304. self.app.log.debug('Creating a new bookmarks file ...')
  305. f = open(filename, 'w')
  306. f.close()
  307. except Exception:
  308. e = sys.exc_info()[0]
  309. self.app.log.error("Could not load defaults file.")
  310. self.app.log.error(str(e))
  311. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Could not load bookmarks file."))
  312. return
  313. # Save Bookmarks to a file
  314. try:
  315. with open(filename, "w") as f:
  316. for title, link in self.bm_dict.items():
  317. line2write = str(title) + ':' + str(link) + '\n'
  318. f.write(line2write)
  319. except Exception:
  320. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Failed to write bookmarks to file."))
  321. return
  322. self.app.inform.emit('[success] %s: %s' % (_("Exported bookmarks to"), filename))
  323. def on_import_bookmarks(self):
  324. self.app.log.debug("on_import_bookmarks()")
  325. filter_ = "Text File (*.txt);;All Files (*.*)"
  326. filename, _f = QtWidgets.QFileDialog.getOpenFileName(caption=_("Import FlatCAM Bookmarks"), filter=filter_)
  327. filename = str(filename)
  328. if filename == "":
  329. self.app.inform.emit('[WARNING_NOTCL] %s' % _("FlatCAM bookmarks import cancelled."))
  330. else:
  331. try:
  332. with open(filename) as f:
  333. bookmarks = f.readlines()
  334. except IOError:
  335. self.app.log.error("Could not load bookmarks file.")
  336. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Could not load bookmarks file."))
  337. return
  338. for line in bookmarks:
  339. proc_line = line.replace(' ', '').partition(':')
  340. self.on_add_entry(title=proc_line[0], link=proc_line[2])
  341. self.app.inform.emit('[success] %s: %s' % (_("Imported Bookmarks from"), filename))
  342. def mark_table_rows_for_actions(self):
  343. for row in range(self.table_widget.rowCount()):
  344. item_to_paint = self.table_widget.item(row, 0)
  345. if row < self.app.defaults["global_bookmarks_limit"]:
  346. item_to_paint.setBackground(QtGui.QColor('gray'))
  347. # item_to_paint.setForeground(QtGui.QColor('black'))
  348. else:
  349. item_to_paint.setBackground(QtGui.QColor('white'))
  350. # item_to_paint.setForeground(QtGui.QColor('black'))
  351. def rebuild_actions(self):
  352. # rebuild the storage to reflect the order of the lines
  353. self.bm_dict.clear()
  354. for row in range(self.table_widget.rowCount()):
  355. title = self.table_widget.item(row, 1).text()
  356. wlink = self.table_widget.cellWidget(row, 2).toPlainText()
  357. entry = int(row) + 1
  358. self.bm_dict.update(
  359. {
  360. str(entry): [title, wlink]
  361. }
  362. )
  363. self.app.install_bookmarks(book_dict=self.bm_dict)
  364. # def accept(self):
  365. # self.rebuild_actions()
  366. # super().accept()
  367. def closeEvent(self, QCloseEvent):
  368. self.rebuild_actions()
  369. super().closeEvent(QCloseEvent)
  370. class ToolsDB(QtWidgets.QWidget):
  371. mark_tools_rows = QtCore.pyqtSignal()
  372. def __init__(self, app, callback_on_edited, callback_on_tool_request, parent=None):
  373. super(ToolsDB, self).__init__(parent)
  374. self.app = app
  375. self.decimals = 4
  376. self.callback_app = callback_on_edited
  377. self.on_tool_request = callback_on_tool_request
  378. self.offset_item_options = ["Path", "In", "Out", "Custom"]
  379. self.type_item_options = ["Iso", "Rough", "Finish"]
  380. self.tool_type_item_options = ["C1", "C2", "C3", "C4", "B", "V"]
  381. '''
  382. dict to hold all the tools in the Tools DB
  383. format:
  384. {
  385. tool_id: {
  386. 'name': 'new_tool'
  387. 'tooldia': self.app.defaults["geometry_cnctooldia"]
  388. 'offset': 'Path'
  389. 'offset_value': 0.0
  390. 'type': _('Rough'),
  391. 'tool_type': 'C1'
  392. 'data': dict()
  393. }
  394. }
  395. '''
  396. self.db_tool_dict = {}
  397. # layouts
  398. layout = QtWidgets.QVBoxLayout()
  399. self.setLayout(layout)
  400. table_hlay = QtWidgets.QHBoxLayout()
  401. layout.addLayout(table_hlay)
  402. self.table_widget = FCTable(drag_drop=True)
  403. self.table_widget.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
  404. table_hlay.addWidget(self.table_widget)
  405. # set the number of columns and the headers tool tips
  406. self.configure_table()
  407. # pal = QtGui.QPalette()
  408. # pal.setColor(QtGui.QPalette.Background, Qt.white)
  409. # New Bookmark
  410. new_vlay = QtWidgets.QVBoxLayout()
  411. layout.addLayout(new_vlay)
  412. # new_tool_lbl = QtWidgets.QLabel('<b>%s</b>' % _("New Tool"))
  413. # new_vlay.addWidget(new_tool_lbl, alignment=QtCore.Qt.AlignBottom)
  414. self.buttons_frame = QtWidgets.QFrame()
  415. self.buttons_frame.setContentsMargins(0, 0, 0, 0)
  416. layout.addWidget(self.buttons_frame)
  417. self.buttons_box = QtWidgets.QHBoxLayout()
  418. self.buttons_box.setContentsMargins(0, 0, 0, 0)
  419. self.buttons_frame.setLayout(self.buttons_box)
  420. self.buttons_frame.show()
  421. add_entry_btn = FCButton(_("Add Geometry Tool in DB"))
  422. add_entry_btn.setToolTip(
  423. _("Add a new tool in the Tools Database.\n"
  424. "It will be used in the Geometry UI.\n"
  425. "You can edit it after it is added.")
  426. )
  427. self.buttons_box.addWidget(add_entry_btn)
  428. # add_fct_entry_btn = FCButton(_("Add Paint/NCC Tool in DB"))
  429. # add_fct_entry_btn.setToolTip(
  430. # _("Add a new tool in the Tools Database.\n"
  431. # "It will be used in the Paint/NCC Tools UI.\n"
  432. # "You can edit it after it is added.")
  433. # )
  434. # self.buttons_box.addWidget(add_fct_entry_btn)
  435. remove_entry_btn = FCButton(_("Delete Tool from DB"))
  436. remove_entry_btn.setToolTip(
  437. _("Remove a selection of tools in the Tools Database.")
  438. )
  439. self.buttons_box.addWidget(remove_entry_btn)
  440. export_db_btn = FCButton(_("Export DB"))
  441. export_db_btn.setToolTip(
  442. _("Save the Tools Database to a custom text file.")
  443. )
  444. self.buttons_box.addWidget(export_db_btn)
  445. import_db_btn = FCButton(_("Import DB"))
  446. import_db_btn.setToolTip(
  447. _("Load the Tools Database information's from a custom text file.")
  448. )
  449. self.buttons_box.addWidget(import_db_btn)
  450. self.add_tool_from_db = FCButton(_("Add Tool from Tools DB"))
  451. self.add_tool_from_db.setToolTip(
  452. _("Add a new tool in the Tools Table of the\n"
  453. "active Geometry object after selecting a tool\n"
  454. "in the Tools Database.")
  455. )
  456. self.add_tool_from_db.hide()
  457. self.cancel_tool_from_db = FCButton(_("Cancel"))
  458. self.cancel_tool_from_db.hide()
  459. hlay = QtWidgets.QHBoxLayout()
  460. layout.addLayout(hlay)
  461. hlay.addWidget(self.add_tool_from_db)
  462. hlay.addWidget(self.cancel_tool_from_db)
  463. hlay.addStretch()
  464. # ##############################################################################
  465. # ######################## SIGNALS #############################################
  466. # ##############################################################################
  467. add_entry_btn.clicked.connect(self.on_tool_add)
  468. remove_entry_btn.clicked.connect(self.on_tool_delete)
  469. export_db_btn.clicked.connect(self.on_export_tools_db_file)
  470. import_db_btn.clicked.connect(self.on_import_tools_db_file)
  471. # closebtn.clicked.connect(self.accept)
  472. self.add_tool_from_db.clicked.connect(self.on_tool_requested_from_app)
  473. self.cancel_tool_from_db.clicked.connect(self.on_cancel_tool)
  474. self.setup_db_ui()
  475. def configure_table(self):
  476. self.table_widget.setColumnCount(27)
  477. # self.table_widget.setColumnWidth(0, 20)
  478. self.table_widget.setHorizontalHeaderLabels(
  479. [
  480. '#',
  481. _("Tool Name"),
  482. _("Tool Dia"),
  483. _("Tool Offset"),
  484. _("Custom Offset"),
  485. _("Tool Type"),
  486. _("Tool Shape"),
  487. _("Cut Z"),
  488. _("MultiDepth"),
  489. _("DPP"),
  490. _("V-Dia"),
  491. _("V-Angle"),
  492. _("Travel Z"),
  493. _("FR"),
  494. _("FR Z"),
  495. _("FR Rapids"),
  496. _("Spindle Speed"),
  497. _("Dwell"),
  498. _("Dwelltime"),
  499. _("Preprocessor"),
  500. _("ExtraCut"),
  501. _("E-Cut Length"),
  502. _("Toolchange"),
  503. _("Toolchange XY"),
  504. _("Toolchange Z"),
  505. _("Start Z"),
  506. _("End Z"),
  507. ]
  508. )
  509. self.table_widget.horizontalHeaderItem(0).setToolTip(
  510. _("Tool Index."))
  511. self.table_widget.horizontalHeaderItem(1).setToolTip(
  512. _("Tool name.\n"
  513. "This is not used in the app, it's function\n"
  514. "is to serve as a note for the user."))
  515. self.table_widget.horizontalHeaderItem(2).setToolTip(
  516. _("Tool Diameter."))
  517. self.table_widget.horizontalHeaderItem(3).setToolTip(
  518. _("Tool Offset.\n"
  519. "Can be of a few types:\n"
  520. "Path = zero offset\n"
  521. "In = offset inside by half of tool diameter\n"
  522. "Out = offset outside by half of tool diameter\n"
  523. "Custom = custom offset using the Custom Offset value"))
  524. self.table_widget.horizontalHeaderItem(4).setToolTip(
  525. _("Custom Offset.\n"
  526. "A value to be used as offset from the current path."))
  527. self.table_widget.horizontalHeaderItem(5).setToolTip(
  528. _("Tool Type.\n"
  529. "Can be:\n"
  530. "Iso = isolation cut\n"
  531. "Rough = rough cut, low feedrate, multiple passes\n"
  532. "Finish = finishing cut, high feedrate"))
  533. self.table_widget.horizontalHeaderItem(6).setToolTip(
  534. _("Tool Shape. \n"
  535. "Can be:\n"
  536. "C1 ... C4 = circular tool with x flutes\n"
  537. "B = ball tip milling tool\n"
  538. "V = v-shape milling tool"))
  539. self.table_widget.horizontalHeaderItem(7).setToolTip(
  540. _("Cutting Depth.\n"
  541. "The depth at which to cut into material."))
  542. self.table_widget.horizontalHeaderItem(8).setToolTip(
  543. _("Multi Depth.\n"
  544. "Selecting this will allow cutting in multiple passes,\n"
  545. "each pass adding a DPP parameter depth."))
  546. self.table_widget.horizontalHeaderItem(9).setToolTip(
  547. _("DPP. Depth per Pass.\n"
  548. "The value used to cut into material on each pass."))
  549. self.table_widget.horizontalHeaderItem(10).setToolTip(
  550. _("V-Dia.\n"
  551. "Diameter of the tip for V-Shape Tools."))
  552. self.table_widget.horizontalHeaderItem(11).setToolTip(
  553. _("V-Agle.\n"
  554. "Angle at the tip for the V-Shape Tools."))
  555. self.table_widget.horizontalHeaderItem(12).setToolTip(
  556. _("Clearance Height.\n"
  557. "Height at which the milling bit will travel between cuts,\n"
  558. "above the surface of the material, avoiding all fixtures."))
  559. self.table_widget.horizontalHeaderItem(13).setToolTip(
  560. _("FR. Feedrate\n"
  561. "The speed on XY plane used while cutting into material."))
  562. self.table_widget.horizontalHeaderItem(14).setToolTip(
  563. _("FR Z. Feedrate Z\n"
  564. "The speed on Z plane."))
  565. self.table_widget.horizontalHeaderItem(15).setToolTip(
  566. _("FR Rapids. Feedrate Rapids\n"
  567. "Speed used while moving as fast as possible.\n"
  568. "This is used only by some devices that can't use\n"
  569. "the G0 g-code command. Mostly 3D printers."))
  570. self.table_widget.horizontalHeaderItem(16).setToolTip(
  571. _("Spindle Speed.\n"
  572. "If it's left empty it will not be used.\n"
  573. "The speed of the spindle in RPM."))
  574. self.table_widget.horizontalHeaderItem(17).setToolTip(
  575. _("Dwell.\n"
  576. "Check this if a delay is needed to allow\n"
  577. "the spindle motor to reach it's set speed."))
  578. self.table_widget.horizontalHeaderItem(18).setToolTip(
  579. _("Dwell Time.\n"
  580. "A delay used to allow the motor spindle reach it's set speed."))
  581. self.table_widget.horizontalHeaderItem(19).setToolTip(
  582. _("Preprocessor.\n"
  583. "A selection of files that will alter the generated G-code\n"
  584. "to fit for a number of use cases."))
  585. self.table_widget.horizontalHeaderItem(20).setToolTip(
  586. _("Extra Cut.\n"
  587. "If checked, after a isolation is finished an extra cut\n"
  588. "will be added where the start and end of isolation meet\n"
  589. "such as that this point is covered by this extra cut to\n"
  590. "ensure a complete isolation."))
  591. self.table_widget.horizontalHeaderItem(21).setToolTip(
  592. _("Extra Cut length.\n"
  593. "If checked, after a isolation is finished an extra cut\n"
  594. "will be added where the start and end of isolation meet\n"
  595. "such as that this point is covered by this extra cut to\n"
  596. "ensure a complete isolation. This is the length of\n"
  597. "the extra cut."))
  598. self.table_widget.horizontalHeaderItem(22).setToolTip(
  599. _("Toolchange.\n"
  600. "It will create a toolchange event.\n"
  601. "The kind of toolchange is determined by\n"
  602. "the preprocessor file."))
  603. self.table_widget.horizontalHeaderItem(23).setToolTip(
  604. _("Toolchange XY.\n"
  605. "A set of coordinates in the format (x, y).\n"
  606. "Will determine the cartesian position of the point\n"
  607. "where the tool change event take place."))
  608. self.table_widget.horizontalHeaderItem(24).setToolTip(
  609. _("Toolchange Z.\n"
  610. "The position on Z plane where the tool change event take place."))
  611. self.table_widget.horizontalHeaderItem(25).setToolTip(
  612. _("Start Z.\n"
  613. "If it's left empty it will not be used.\n"
  614. "A position on Z plane to move immediately after job start."))
  615. self.table_widget.horizontalHeaderItem(26).setToolTip(
  616. _("End Z.\n"
  617. "A position on Z plane to move immediately after job stop."))
  618. def setup_db_ui(self):
  619. filename = self.app.data_path + '/geo_tools_db.FlatDB'
  620. # load the database tools from the file
  621. try:
  622. with open(filename) as f:
  623. tools = f.read()
  624. except IOError:
  625. self.app.log.error("Could not load tools DB file.")
  626. self.app.inform.emit('[ERROR] %s' % _("Could not load Tools DB file."))
  627. return
  628. try:
  629. self.db_tool_dict = json.loads(tools)
  630. except Exception:
  631. e = sys.exc_info()[0]
  632. self.app.log.error(str(e))
  633. self.app.inform.emit('[ERROR] %s' % _("Failed to parse Tools DB file."))
  634. return
  635. self.app.inform.emit('[success] %s: %s' % (_("Loaded FlatCAM Tools DB from"), filename))
  636. self.build_db_ui()
  637. self.table_widget.setupContextMenu()
  638. self.table_widget.addContextMenu(
  639. _("Add to DB"), self.on_tool_add, icon=QtGui.QIcon(self.app.resource_location + "/plus16.png"))
  640. self.table_widget.addContextMenu(
  641. _("Copy from DB"), self.on_tool_copy, icon=QtGui.QIcon(self.app.resource_location + "/copy16.png"))
  642. self.table_widget.addContextMenu(
  643. _("Delete from DB"), self.on_tool_delete, icon=QtGui.QIcon(self.app.resource_location + "/delete32.png"))
  644. def build_db_ui(self):
  645. self.ui_disconnect()
  646. self.table_widget.setRowCount(len(self.db_tool_dict))
  647. nr_crt = 0
  648. for toolid, dict_val in self.db_tool_dict.items():
  649. row = nr_crt
  650. nr_crt += 1
  651. t_name = dict_val['name']
  652. try:
  653. self.add_tool_table_line(row, name=t_name, widget=self.table_widget, tooldict=dict_val)
  654. except Exception as e:
  655. self.app.log.debug("ToolDB.build_db_ui.add_tool_table_line() --> %s" % str(e))
  656. vertical_header = self.table_widget.verticalHeader()
  657. vertical_header.hide()
  658. horizontal_header = self.table_widget.horizontalHeader()
  659. horizontal_header.setMinimumSectionSize(10)
  660. horizontal_header.setDefaultSectionSize(70)
  661. self.table_widget.setSizeAdjustPolicy(QtWidgets.QAbstractScrollArea.AdjustToContents)
  662. for x in range(27):
  663. self.table_widget.resizeColumnToContents(x)
  664. horizontal_header.setSectionResizeMode(0, QtWidgets.QHeaderView.Fixed)
  665. # horizontal_header.setSectionResizeMode(1, QtWidgets.QHeaderView.Stretch)
  666. # horizontal_header.setSectionResizeMode(13, QtWidgets.QHeaderView.Fixed)
  667. horizontal_header.resizeSection(0, 20)
  668. # horizontal_header.setSectionResizeMode(1, QtWidgets.QHeaderView.ResizeToContents)
  669. # horizontal_header.setSectionResizeMode(2, QtWidgets.QHeaderView.Stretch)
  670. self.ui_connect()
  671. def add_tool_table_line(self, row, name, widget, tooldict):
  672. data = tooldict['data']
  673. nr_crt = row + 1
  674. id_item = QtWidgets.QTableWidgetItem('%d' % int(nr_crt))
  675. # id_item.setFlags(QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
  676. flags = id_item.flags() & ~QtCore.Qt.ItemIsEditable
  677. id_item.setFlags(flags)
  678. widget.setItem(row, 0, id_item) # Tool name/id
  679. tool_name_item = QtWidgets.QTableWidgetItem(name)
  680. widget.setItem(row, 1, tool_name_item)
  681. dia_item = FCDoubleSpinner()
  682. dia_item.set_precision(self.decimals)
  683. dia_item.setSingleStep(0.1)
  684. dia_item.set_range(0.0, 9999.9999)
  685. dia_item.set_value(float(tooldict['tooldia']))
  686. widget.setCellWidget(row, 2, dia_item)
  687. tool_offset_item = FCComboBox()
  688. for item in self.offset_item_options:
  689. tool_offset_item.addItem(item)
  690. tool_offset_item.set_value(tooldict['offset'])
  691. widget.setCellWidget(row, 3, tool_offset_item)
  692. c_offset_item = FCDoubleSpinner()
  693. c_offset_item.set_precision(self.decimals)
  694. c_offset_item.setSingleStep(0.1)
  695. c_offset_item.set_range(-9999.9999, 9999.9999)
  696. c_offset_item.set_value(float(tooldict['offset_value']))
  697. widget.setCellWidget(row, 4, c_offset_item)
  698. tt_item = FCComboBox()
  699. for item in self.type_item_options:
  700. tt_item.addItem(item)
  701. tt_item.set_value(tooldict['type'])
  702. widget.setCellWidget(row, 5, tt_item)
  703. tshape_item = FCComboBox()
  704. for item in self.tool_type_item_options:
  705. tshape_item.addItem(item)
  706. tshape_item.set_value(tooldict['tool_type'])
  707. widget.setCellWidget(row, 6, tshape_item)
  708. cutz_item = FCDoubleSpinner()
  709. cutz_item.set_precision(self.decimals)
  710. cutz_item.setSingleStep(0.1)
  711. if self.app.defaults['global_machinist_setting']:
  712. cutz_item.set_range(-9999.9999, 9999.9999)
  713. else:
  714. cutz_item.set_range(-9999.9999, -0.0000)
  715. cutz_item.set_value(float(data['cutz']))
  716. widget.setCellWidget(row, 7, cutz_item)
  717. multidepth_item = FCCheckBox()
  718. multidepth_item.set_value(data['multidepth'])
  719. widget.setCellWidget(row, 8, multidepth_item)
  720. # to make the checkbox centered but it can no longer have it's value accessed - needs a fix using findchild()
  721. # multidepth_item = QtWidgets.QWidget()
  722. # cb = FCCheckBox()
  723. # cb.set_value(data['multidepth'])
  724. # qhboxlayout = QtWidgets.QHBoxLayout(multidepth_item)
  725. # qhboxlayout.addWidget(cb)
  726. # qhboxlayout.setAlignment(QtCore.Qt.AlignCenter)
  727. # qhboxlayout.setContentsMargins(0, 0, 0, 0)
  728. # widget.setCellWidget(row, 8, multidepth_item)
  729. depth_per_pass_item = FCDoubleSpinner()
  730. depth_per_pass_item.set_precision(self.decimals)
  731. depth_per_pass_item.setSingleStep(0.1)
  732. depth_per_pass_item.set_range(0.0, 9999.9999)
  733. depth_per_pass_item.set_value(float(data['depthperpass']))
  734. widget.setCellWidget(row, 9, depth_per_pass_item)
  735. vtip_dia_item = FCDoubleSpinner()
  736. vtip_dia_item.set_precision(self.decimals)
  737. vtip_dia_item.setSingleStep(0.1)
  738. vtip_dia_item.set_range(0.0, 9999.9999)
  739. vtip_dia_item.set_value(float(data['vtipdia']))
  740. widget.setCellWidget(row, 10, vtip_dia_item)
  741. vtip_angle_item = FCDoubleSpinner()
  742. vtip_angle_item.set_precision(self.decimals)
  743. vtip_angle_item.setSingleStep(0.1)
  744. vtip_angle_item.set_range(-360.0, 360.0)
  745. vtip_angle_item.set_value(float(data['vtipangle']))
  746. widget.setCellWidget(row, 11, vtip_angle_item)
  747. travelz_item = FCDoubleSpinner()
  748. travelz_item.set_precision(self.decimals)
  749. travelz_item.setSingleStep(0.1)
  750. if self.app.defaults['global_machinist_setting']:
  751. travelz_item.set_range(-9999.9999, 9999.9999)
  752. else:
  753. travelz_item.set_range(0.0000, 9999.9999)
  754. travelz_item.set_value(float(data['travelz']))
  755. widget.setCellWidget(row, 12, travelz_item)
  756. fr_item = FCDoubleSpinner()
  757. fr_item.set_precision(self.decimals)
  758. fr_item.set_range(0.0, 9999.9999)
  759. fr_item.set_value(float(data['feedrate']))
  760. widget.setCellWidget(row, 13, fr_item)
  761. frz_item = FCDoubleSpinner()
  762. frz_item.set_precision(self.decimals)
  763. frz_item.set_range(0.0, 9999.9999)
  764. frz_item.set_value(float(data['feedrate_z']))
  765. widget.setCellWidget(row, 14, frz_item)
  766. frrapids_item = FCDoubleSpinner()
  767. frrapids_item.set_precision(self.decimals)
  768. frrapids_item.set_range(0.0, 9999.9999)
  769. frrapids_item.set_value(float(data['feedrate_rapid']))
  770. widget.setCellWidget(row, 15, frrapids_item)
  771. spindlespeed_item = FCSpinner()
  772. spindlespeed_item.set_range(0, 1000000)
  773. spindlespeed_item.set_value(int(data['spindlespeed']))
  774. spindlespeed_item.set_step(100)
  775. widget.setCellWidget(row, 16, spindlespeed_item)
  776. dwell_item = FCCheckBox()
  777. dwell_item.set_value(data['dwell'])
  778. widget.setCellWidget(row, 17, dwell_item)
  779. dwelltime_item = FCDoubleSpinner()
  780. dwelltime_item.set_precision(self.decimals)
  781. dwelltime_item.set_range(0.0000, 9999.9999)
  782. dwelltime_item.set_value(float(data['dwelltime']))
  783. widget.setCellWidget(row, 18, dwelltime_item)
  784. pp_item = FCComboBox()
  785. for item in self.app.preprocessors:
  786. pp_item.addItem(item)
  787. pp_item.set_value(data['ppname_g'])
  788. widget.setCellWidget(row, 19, pp_item)
  789. ecut_item = FCCheckBox()
  790. ecut_item.set_value(data['extracut'])
  791. widget.setCellWidget(row, 20, ecut_item)
  792. ecut_length_item = FCDoubleSpinner()
  793. ecut_length_item.set_precision(self.decimals)
  794. ecut_length_item.set_range(0.0000, 9999.9999)
  795. ecut_length_item.set_value(data['extracut_length'])
  796. widget.setCellWidget(row, 21, ecut_length_item)
  797. toolchange_item = FCCheckBox()
  798. toolchange_item.set_value(data['toolchange'])
  799. widget.setCellWidget(row, 22, toolchange_item)
  800. toolchangexy_item = QtWidgets.QTableWidgetItem(str(data['toolchangexy']) if data['toolchangexy'] else '')
  801. widget.setItem(row, 23, toolchangexy_item)
  802. toolchangez_item = FCDoubleSpinner()
  803. toolchangez_item.set_precision(self.decimals)
  804. toolchangez_item.setSingleStep(0.1)
  805. if self.app.defaults['global_machinist_setting']:
  806. toolchangez_item.set_range(-9999.9999, 9999.9999)
  807. else:
  808. toolchangez_item.set_range(0.0000, 9999.9999)
  809. toolchangez_item.set_value(float(data['toolchangez']))
  810. widget.setCellWidget(row, 24, toolchangez_item)
  811. startz_item = QtWidgets.QTableWidgetItem(str(data['startz']) if data['startz'] else '')
  812. widget.setItem(row, 25, startz_item)
  813. endz_item = FCDoubleSpinner()
  814. endz_item.set_precision(self.decimals)
  815. endz_item.setSingleStep(0.1)
  816. if self.app.defaults['global_machinist_setting']:
  817. endz_item.set_range(-9999.9999, 9999.9999)
  818. else:
  819. endz_item.set_range(0.0000, 9999.9999)
  820. endz_item.set_value(float(data['endz']))
  821. widget.setCellWidget(row, 26, endz_item)
  822. def on_tool_add(self):
  823. """
  824. Add a tool in the DB Tool Table
  825. :return: None
  826. """
  827. default_data = {}
  828. default_data.update({
  829. "cutz": float(self.app.defaults["geometry_cutz"]),
  830. "multidepth": self.app.defaults["geometry_multidepth"],
  831. "depthperpass": float(self.app.defaults["geometry_depthperpass"]),
  832. "vtipdia": float(self.app.defaults["geometry_vtipdia"]),
  833. "vtipangle": float(self.app.defaults["geometry_vtipangle"]),
  834. "travelz": float(self.app.defaults["geometry_travelz"]),
  835. "feedrate": float(self.app.defaults["geometry_feedrate"]),
  836. "feedrate_z": float(self.app.defaults["geometry_feedrate_z"]),
  837. "feedrate_rapid": float(self.app.defaults["geometry_feedrate_rapid"]),
  838. "spindlespeed": self.app.defaults["geometry_spindlespeed"],
  839. "dwell": self.app.defaults["geometry_dwell"],
  840. "dwelltime": float(self.app.defaults["geometry_dwelltime"]),
  841. "ppname_g": self.app.defaults["geometry_ppname_g"],
  842. "extracut": self.app.defaults["geometry_extracut"],
  843. "extracut_length": float(self.app.defaults["geometry_extracut_length"]),
  844. "toolchange": self.app.defaults["geometry_toolchange"],
  845. "toolchangexy": self.app.defaults["geometry_toolchangexy"],
  846. "toolchangez": float(self.app.defaults["geometry_toolchangez"]),
  847. "startz": self.app.defaults["geometry_startz"],
  848. "endz": float(self.app.defaults["geometry_endz"])
  849. })
  850. dict_elem = {}
  851. dict_elem['name'] = 'new_tool'
  852. if type(self.app.defaults["geometry_cnctooldia"]) == float:
  853. dict_elem['tooldia'] = self.app.defaults["geometry_cnctooldia"]
  854. else:
  855. try:
  856. tools_string = self.app.defaults["geometry_cnctooldia"].split(",")
  857. tools_diameters = [eval(a) for a in tools_string if a != '']
  858. dict_elem['tooldia'] = tools_diameters[0] if tools_diameters else 0.0
  859. except Exception as e:
  860. self.app.log.debug("ToolDB.on_tool_add() --> %s" % str(e))
  861. return
  862. dict_elem['offset'] = 'Path'
  863. dict_elem['offset_value'] = 0.0
  864. dict_elem['type'] = 'Rough'
  865. dict_elem['tool_type'] = 'C1'
  866. dict_elem['data'] = default_data
  867. new_toolid = len(self.db_tool_dict) + 1
  868. self.db_tool_dict[new_toolid] = deepcopy(dict_elem)
  869. # add the new entry to the Tools DB table
  870. self.build_db_ui()
  871. self.callback_on_edited()
  872. self.app.inform.emit('[success] %s' % _("Tool added to DB."))
  873. def on_tool_copy(self):
  874. """
  875. Copy a selection of Tools in the Tools DB table
  876. :return:
  877. """
  878. new_tool_id = self.table_widget.rowCount() + 1
  879. for model_index in self.table_widget.selectionModel().selectedRows():
  880. # index = QtCore.QPersistentModelIndex(model_index)
  881. old_tool_id = self.table_widget.item(model_index.row(), 0).text()
  882. new_tool_id += 1
  883. for toolid, dict_val in list(self.db_tool_dict.items()):
  884. if int(old_tool_id) == int(toolid):
  885. self.db_tool_dict.update({
  886. new_tool_id: deepcopy(dict_val)
  887. })
  888. self.build_db_ui()
  889. self.callback_on_edited()
  890. self.app.inform.emit('[success] %s' % _("Tool copied from Tools DB."))
  891. def on_tool_delete(self):
  892. """
  893. Delete a selection of Tools in the Tools DB table
  894. :return:
  895. """
  896. for model_index in self.table_widget.selectionModel().selectedRows():
  897. # index = QtCore.QPersistentModelIndex(model_index)
  898. toolname_to_remove = self.table_widget.item(model_index.row(), 0).text()
  899. for toolid, dict_val in list(self.db_tool_dict.items()):
  900. if int(toolname_to_remove) == int(toolid):
  901. # remove from the storage
  902. self.db_tool_dict.pop(toolid, None)
  903. self.build_db_ui()
  904. self.callback_on_edited()
  905. self.app.inform.emit('[success] %s' % _("Tool removed from Tools DB."))
  906. def on_export_tools_db_file(self):
  907. self.app.report_usage("on_export_tools_db_file")
  908. self.app.log.debug("on_export_tools_db_file()")
  909. date = str(datetime.today()).rpartition('.')[0]
  910. date = ''.join(c for c in date if c not in ':-')
  911. date = date.replace(' ', '_')
  912. filter__ = "Text File (*.TXT);;All Files (*.*)"
  913. filename, _f = QtWidgets.QFileDialog.getSaveFileName(caption=_("Export Tools Database"),
  914. directory='{l_save}/FlatCAM_{n}_{date}'.format(
  915. l_save=str(self.app.get_last_save_folder()),
  916. n=_("Tools_Database"),
  917. date=date),
  918. filter=filter__)
  919. filename = str(filename)
  920. if filename == "":
  921. self.app.inform.emit('[WARNING_NOTCL] %s' % _("FlatCAM Tools DB export cancelled."))
  922. return
  923. else:
  924. try:
  925. f = open(filename, 'w')
  926. f.close()
  927. except PermissionError:
  928. self.app.inform.emit('[WARNING] %s' %
  929. _("Permission denied, saving not possible.\n"
  930. "Most likely another app is holding the file open and not accessible."))
  931. return
  932. except IOError:
  933. self.app.log.debug('Creating a new Tools DB file ...')
  934. f = open(filename, 'w')
  935. f.close()
  936. except Exception:
  937. e = sys.exc_info()[0]
  938. self.app.log.error("Could not load Tools DB file.")
  939. self.app.log.error(str(e))
  940. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Could not load Tools DB file."))
  941. return
  942. # Save update options
  943. try:
  944. # Save Tools DB in a file
  945. try:
  946. with open(filename, "w") as f:
  947. json.dump(self.db_tool_dict, f, default=to_dict, indent=2)
  948. except Exception as e:
  949. self.app.log.debug("App.on_save_tools_db() --> %s" % str(e))
  950. self.inform.emit('[ERROR_NOTCL] %s' % _("Failed to write Tools DB to file."))
  951. return
  952. except Exception:
  953. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Failed to write Tools DB to file."))
  954. return
  955. self.app.inform.emit('[success] %s: %s' % (_("Exported Tools DB to"), filename))
  956. def on_import_tools_db_file(self):
  957. self.app.report_usage("on_import_tools_db_file")
  958. self.app.log.debug("on_import_tools_db_file()")
  959. filter__ = "Text File (*.TXT);;All Files (*.*)"
  960. filename, _f = QtWidgets.QFileDialog.getOpenFileName(caption=_("Import FlatCAM Tools DB"), filter=filter__)
  961. if filename == "":
  962. self.app.inform.emit('[WARNING_NOTCL] %s' % _("FlatCAM Tools DB import cancelled."))
  963. else:
  964. try:
  965. with open(filename) as f:
  966. tools_in_db = f.read()
  967. except IOError:
  968. self.app.log.error("Could not load Tools DB file.")
  969. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Could not load Tools DB file."))
  970. return
  971. try:
  972. self.db_tool_dict = json.loads(tools_in_db)
  973. except Exception:
  974. e = sys.exc_info()[0]
  975. self.app.log.error(str(e))
  976. self.app.inform.emit('[ERROR] %s' % _("Failed to parse Tools DB file."))
  977. return
  978. self.app.inform.emit('[success] %s: %s' % (_("Loaded FlatCAM Tools DB from"), filename))
  979. self.build_db_ui()
  980. self.callback_on_edited()
  981. def on_save_tools_db(self, silent=False):
  982. self.app.log.debug("ToolsDB.on_save_button() --> Saving Tools Database to file.")
  983. filename = self.app.data_path + "/geo_tools_db.FlatDB"
  984. # Preferences save, update the color of the Tools DB Tab text
  985. for idx in range(self.app.ui.plot_tab_area.count()):
  986. if self.app.ui.plot_tab_area.tabText(idx) == _("Tools Database"):
  987. self.app.ui.plot_tab_area.tabBar.setTabTextColor(idx, QtGui.QColor('black'))
  988. # Save Tools DB in a file
  989. try:
  990. f = open(filename, "w")
  991. json.dump(self.db_tool_dict, f, default=to_dict, indent=2)
  992. f.close()
  993. except Exception as e:
  994. self.app.log.debug("ToolsDB.on_save_tools_db() --> %s" % str(e))
  995. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Failed to write Tools DB to file."))
  996. return
  997. if not silent:
  998. self.app.inform.emit('[success] %s' % _("Saved Tools DB."))
  999. def ui_connect(self):
  1000. try:
  1001. try:
  1002. self.table_widget.itemChanged.disconnect(self.callback_on_edited)
  1003. except (TypeError, AttributeError):
  1004. pass
  1005. self.table_widget.itemChanged.connect(self.callback_on_edited)
  1006. except AttributeError:
  1007. pass
  1008. for row in range(self.table_widget.rowCount()):
  1009. for col in range(self.table_widget.columnCount()):
  1010. # ComboBox
  1011. try:
  1012. try:
  1013. self.table_widget.cellWidget(row, col).currentIndexChanged.disconnect(self.callback_on_edited)
  1014. except (TypeError, AttributeError):
  1015. pass
  1016. self.table_widget.cellWidget(row, col).currentIndexChanged.connect(self.callback_on_edited)
  1017. except AttributeError:
  1018. pass
  1019. # CheckBox
  1020. try:
  1021. try:
  1022. self.table_widget.cellWidget(row, col).toggled.disconnect(self.callback_on_edited)
  1023. except (TypeError, AttributeError):
  1024. pass
  1025. self.table_widget.cellWidget(row, col).toggled.connect(self.callback_on_edited)
  1026. except AttributeError:
  1027. pass
  1028. # SpinBox, DoubleSpinBox
  1029. try:
  1030. try:
  1031. self.table_widget.cellWidget(row, col).valueChanged.disconnect(self.callback_on_edited)
  1032. except (TypeError, AttributeError):
  1033. pass
  1034. self.table_widget.cellWidget(row, col).valueChanged.connect(self.callback_on_edited)
  1035. except AttributeError:
  1036. pass
  1037. def ui_disconnect(self):
  1038. try:
  1039. self.table_widget.itemChanged.disconnect(self.callback_on_edited)
  1040. except (TypeError, AttributeError):
  1041. pass
  1042. for row in range(self.table_widget.rowCount()):
  1043. for col in range(self.table_widget.columnCount()):
  1044. # ComboBox
  1045. try:
  1046. self.table_widget.cellWidget(row, col).currentIndexChanged.disconnect(self.callback_on_edited)
  1047. except (TypeError, AttributeError):
  1048. pass
  1049. # CheckBox
  1050. try:
  1051. self.table_widget.cellWidget(row, col).toggled.disconnect(self.callback_on_edited)
  1052. except (TypeError, AttributeError):
  1053. pass
  1054. # SpinBox, DoubleSpinBox
  1055. try:
  1056. self.table_widget.cellWidget(row, col).valueChanged.disconnect(self.callback_on_edited)
  1057. except (TypeError, AttributeError):
  1058. pass
  1059. def callback_on_edited(self):
  1060. # update the dictionary storage self.db_tool_dict
  1061. self.db_tool_dict.clear()
  1062. dict_elem = {}
  1063. default_data = {}
  1064. for row in range(self.table_widget.rowCount()):
  1065. new_toolid = row + 1
  1066. for col in range(self.table_widget.columnCount()):
  1067. column_header_text = self.table_widget.horizontalHeaderItem(col).text()
  1068. if column_header_text == _('Tool Name'):
  1069. dict_elem['name'] = self.table_widget.item(row, col).text()
  1070. elif column_header_text == _('Tool Dia'):
  1071. dict_elem['tooldia'] = self.table_widget.cellWidget(row, col).get_value()
  1072. elif column_header_text == _('Tool Offset'):
  1073. dict_elem['offset'] = self.table_widget.cellWidget(row, col).get_value()
  1074. elif column_header_text == _('Custom Offset'):
  1075. dict_elem['offset_value'] = self.table_widget.cellWidget(row, col).get_value()
  1076. elif column_header_text == _('Tool Type'):
  1077. dict_elem['type'] = self.table_widget.cellWidget(row, col).get_value()
  1078. elif column_header_text == _('Tool Shape'):
  1079. dict_elem['tool_type'] = self.table_widget.cellWidget(row, col).get_value()
  1080. else:
  1081. if column_header_text == _('Cut Z'):
  1082. default_data['cutz'] = self.table_widget.cellWidget(row, col).get_value()
  1083. elif column_header_text == _('MultiDepth'):
  1084. default_data['multidepth'] = self.table_widget.cellWidget(row, col).get_value()
  1085. elif column_header_text == _('DPP'):
  1086. default_data['depthperpass'] = self.table_widget.cellWidget(row, col).get_value()
  1087. elif column_header_text == _('V-Dia'):
  1088. default_data['vtipdia'] = self.table_widget.cellWidget(row, col).get_value()
  1089. elif column_header_text == _('V-Angle'):
  1090. default_data['vtipangle'] = self.table_widget.cellWidget(row, col).get_value()
  1091. elif column_header_text == _('Travel Z'):
  1092. default_data['travelz'] = self.table_widget.cellWidget(row, col).get_value()
  1093. elif column_header_text == _('FR'):
  1094. default_data['feedrate'] = self.table_widget.cellWidget(row, col).get_value()
  1095. elif column_header_text == _('FR Z'):
  1096. default_data['feedrate_z'] = self.table_widget.cellWidget(row, col).get_value()
  1097. elif column_header_text == _('FR Rapids'):
  1098. default_data['feedrate_rapid'] = self.table_widget.cellWidget(row, col).get_value()
  1099. elif column_header_text == _('Spindle Speed'):
  1100. default_data['spindlespeed'] = self.table_widget.cellWidget(row, col).get_value()
  1101. elif column_header_text == _('Dwell'):
  1102. default_data['dwell'] = self.table_widget.cellWidget(row, col).get_value()
  1103. elif column_header_text == _('Dwelltime'):
  1104. default_data['dwelltime'] = self.table_widget.cellWidget(row, col).get_value()
  1105. elif column_header_text == _('Preprocessor'):
  1106. default_data['ppname_g'] = self.table_widget.cellWidget(row, col).get_value()
  1107. elif column_header_text == _('ExtraCut'):
  1108. default_data['extracut'] = self.table_widget.cellWidget(row, col).get_value()
  1109. elif column_header_text == _("E-Cut Length"):
  1110. default_data['extracut_length'] = self.table_widget.cellWidget(row, col).get_value()
  1111. elif column_header_text == _('Toolchange'):
  1112. default_data['toolchange'] = self.table_widget.cellWidget(row, col).get_value()
  1113. elif column_header_text == _('Toolchange XY'):
  1114. default_data['toolchangexy'] = self.table_widget.item(row, col).text()
  1115. elif column_header_text == _('Toolchange Z'):
  1116. default_data['toolchangez'] = self.table_widget.cellWidget(row, col).get_value()
  1117. elif column_header_text == _('Start Z'):
  1118. default_data['startz'] = float(self.table_widget.item(row, col).text()) \
  1119. if self.table_widget.item(row, col).text() is not '' else None
  1120. elif column_header_text == _('End Z'):
  1121. default_data['endz'] = self.table_widget.cellWidget(row, col).get_value()
  1122. dict_elem['data'] = default_data
  1123. self.db_tool_dict.update(
  1124. {
  1125. new_toolid: deepcopy(dict_elem)
  1126. }
  1127. )
  1128. self.callback_app()
  1129. def on_tool_requested_from_app(self):
  1130. if not self.table_widget.selectionModel().selectedRows():
  1131. self.app.inform.emit('[WARNING_NOTCL] %s...' % _("No Tool/row selected in the Tools Database table"))
  1132. return
  1133. model_index_list = self.table_widget.selectionModel().selectedRows()
  1134. for model_index in model_index_list:
  1135. selected_row = model_index.row()
  1136. tool_uid = selected_row + 1
  1137. for key in self.db_tool_dict.keys():
  1138. if str(key) == str(tool_uid):
  1139. selected_tool = self.db_tool_dict[key]
  1140. self.on_tool_request(tool=selected_tool)
  1141. def on_cancel_tool(self):
  1142. for idx in range(self.app.ui.plot_tab_area.count()):
  1143. if self.app.ui.plot_tab_area.tabText(idx) == _("Tools Database"):
  1144. wdg = self.app.ui.plot_tab_area.widget(idx)
  1145. wdg.deleteLater()
  1146. self.app.ui.plot_tab_area.removeTab(idx)
  1147. self.app.inform.emit('%s' % _("Cancelled adding tool from DB."))
  1148. def resize_new_tool_table_widget(self, min_size, max_size):
  1149. """
  1150. Resize the table widget responsible for adding new tool in the Tool Database
  1151. :param min_size: passed by rangeChanged signal or the self.new_tool_table_widget.horizontalScrollBar()
  1152. :param max_size: passed by rangeChanged signal or the self.new_tool_table_widget.horizontalScrollBar()
  1153. :return:
  1154. """
  1155. t_height = self.t_height
  1156. if max_size > min_size:
  1157. t_height = self.t_height + self.new_tool_table_widget.verticalScrollBar().height()
  1158. self.new_tool_table_widget.setMaximumHeight(t_height)
  1159. def closeEvent(self, QCloseEvent):
  1160. super().closeEvent(QCloseEvent)
  1161. class ToolsDB2(QtWidgets.QWidget):
  1162. mark_tools_rows = QtCore.pyqtSignal()
  1163. def __init__(self, app, callback_on_edited, callback_on_tool_request, parent=None):
  1164. super(ToolsDB2, self).__init__(parent)
  1165. self.app = app
  1166. self.decimals = self.app.decimals
  1167. self.callback_app = callback_on_edited
  1168. self.on_tool_request = callback_on_tool_request
  1169. self.offset_item_options = ["Path", "In", "Out", "Custom"]
  1170. self.type_item_options = ["Iso", "Rough", "Finish"]
  1171. self.tool_type_item_options = ["C1", "C2", "C3", "C4", "B", "V"]
  1172. '''
  1173. dict to hold all the tools in the Tools DB
  1174. format:
  1175. {
  1176. tool_id: {
  1177. 'name': 'new_tool'
  1178. 'tooldia': self.app.defaults["geometry_cnctooldia"]
  1179. 'offset': 'Path'
  1180. 'offset_value': 0.0
  1181. 'type': _('Rough'),
  1182. 'tool_type': 'C1'
  1183. 'data': dict()
  1184. }
  1185. }
  1186. '''
  1187. self.db_tool_dict = {}
  1188. # layouts
  1189. grid_layout = QtWidgets.QGridLayout()
  1190. grid_layout.setColumnStretch(0, 0)
  1191. grid_layout.setColumnStretch(1, 1)
  1192. self.setLayout(grid_layout)
  1193. tree_layout = QtWidgets.QVBoxLayout()
  1194. grid_layout.addLayout(tree_layout, 0, 0)
  1195. self.tree_widget = FCTree(columns=2, header_hidden=False, protected_column=[0])
  1196. self.tree_widget.setHeaderLabels(["ID", "Tool Name"])
  1197. self.tree_widget.setIndentation(0)
  1198. self.tree_widget.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
  1199. self.tree_widget.setSelectionMode(QtWidgets.QAbstractItemView.ExtendedSelection)
  1200. # set alternating colors
  1201. # self.tree_widget.setAlternatingRowColors(True)
  1202. # p = QtGui.QPalette()
  1203. # p.setColor(QtGui.QPalette.AlternateBase, QtGui.QColor(226, 237, 253) )
  1204. # self.tree_widget.setPalette(p)
  1205. self.tree_widget.setSizePolicy(QtWidgets.QSizePolicy.MinimumExpanding, QtWidgets.QSizePolicy.MinimumExpanding)
  1206. tree_layout.addWidget(self.tree_widget)
  1207. param_hlay = QtWidgets.QHBoxLayout()
  1208. param_area = QtWidgets.QScrollArea()
  1209. param_widget = QtWidgets.QWidget()
  1210. param_widget.setLayout(param_hlay)
  1211. param_area.setWidget(param_widget)
  1212. param_area.setWidgetResizable(True)
  1213. grid_layout.addWidget(param_area, 0, 1)
  1214. # ###########################################################################
  1215. # ############## The UI form ################################################
  1216. # ###########################################################################
  1217. self.basic_box = QtWidgets.QGroupBox()
  1218. self.basic_box.setStyleSheet("""
  1219. QGroupBox
  1220. {
  1221. font-size: 16px;
  1222. font-weight: bold;
  1223. }
  1224. """)
  1225. self.basic_vlay = QtWidgets.QVBoxLayout()
  1226. self.basic_box.setTitle(_("Basic Parameters"))
  1227. self.basic_box.setMinimumWidth(250)
  1228. self.advanced_box = QtWidgets.QGroupBox()
  1229. self.advanced_box.setStyleSheet("""
  1230. QGroupBox
  1231. {
  1232. font-size: 16px;
  1233. font-weight: bold;
  1234. }
  1235. """)
  1236. self.advanced_vlay = QtWidgets.QVBoxLayout()
  1237. self.advanced_box.setTitle(_("Advanced Parameters"))
  1238. self.advanced_box.setMinimumWidth(250)
  1239. self.basic_box.setLayout(self.basic_vlay)
  1240. self.advanced_box.setLayout(self.advanced_vlay)
  1241. param_hlay.addWidget(self.basic_box)
  1242. param_hlay.addWidget(self.advanced_box)
  1243. param_hlay.addStretch()
  1244. # ###########################################################################
  1245. # ############### BASIC UI form #############################################
  1246. # ###########################################################################
  1247. self.grid0 = QtWidgets.QGridLayout()
  1248. self.basic_vlay.addLayout(self.grid0)
  1249. self.grid0.setColumnStretch(0, 0)
  1250. self.grid0.setColumnStretch(1, 1)
  1251. self.basic_vlay.addStretch()
  1252. # Tool Name
  1253. self.name_label = QtWidgets.QLabel('%s:' % _('Tool Name'))
  1254. self.name_label.setToolTip(
  1255. _("Tool name.\n"
  1256. "This is not used in the app, it's function\n"
  1257. "is to serve as a note for the user."))
  1258. self.name_entry = FCEntry()
  1259. self.name_entry.setObjectName('gdb_name')
  1260. self.grid0.addWidget(self.name_label, 0, 0)
  1261. self.grid0.addWidget(self.name_entry, 0, 1)
  1262. # Tool Dia
  1263. self.dia_label = QtWidgets.QLabel('%s:' % _('Tool Dia'))
  1264. self.dia_label.setToolTip(
  1265. _("Tool Diameter."))
  1266. self.dia_entry = FCDoubleSpinner()
  1267. self.dia_entry.set_range(-9999.9999, 9999.9999)
  1268. self.dia_entry.set_precision(self.decimals)
  1269. self.dia_entry.setObjectName('gdb_dia')
  1270. self.grid0.addWidget(self.dia_label, 1, 0)
  1271. self.grid0.addWidget(self.dia_entry, 1, 1)
  1272. # Tool Shape
  1273. self.shape_label = QtWidgets.QLabel('%s:' % _('Tool Shape'))
  1274. self.shape_label.setToolTip(
  1275. _("Tool Shape. \n"
  1276. "Can be:\n"
  1277. "C1 ... C4 = circular tool with x flutes\n"
  1278. "B = ball tip milling tool\n"
  1279. "V = v-shape milling tool"))
  1280. self.shape_combo = FCComboBox()
  1281. self.shape_combo.addItems(["C1", "C2", "C3", "C4", "B", "V"])
  1282. self.shape_combo.setObjectName('gdb_shape')
  1283. self.grid0.addWidget(self.shape_label, 2, 0)
  1284. self.grid0.addWidget(self.shape_combo, 2, 1)
  1285. # Cut Z
  1286. self.cutz_label = QtWidgets.QLabel('%s:' % _("Cut Z"))
  1287. self.cutz_label.setToolTip(
  1288. _("Cutting Depth.\n"
  1289. "The depth at which to cut into material."))
  1290. self.cutz_entry = FCDoubleSpinner()
  1291. self.cutz_entry.set_range(-9999.9999, 9999.9999)
  1292. self.cutz_entry.set_precision(self.decimals)
  1293. self.cutz_entry.setObjectName('gdb_cutz')
  1294. self.grid0.addWidget(self.cutz_label, 4, 0)
  1295. self.grid0.addWidget(self.cutz_entry, 4, 1)
  1296. # Multi Depth
  1297. self.multidepth_label = QtWidgets.QLabel('%s:' % _("MultiDepth"))
  1298. self.multidepth_label.setToolTip(
  1299. _("Multi Depth.\n"
  1300. "Selecting this will allow cutting in multiple passes,\n"
  1301. "each pass adding a DPP parameter depth."))
  1302. self.multidepth_cb = FCCheckBox()
  1303. self.multidepth_cb.setObjectName('gdb_multidepth')
  1304. self.grid0.addWidget(self.multidepth_label, 5, 0)
  1305. self.grid0.addWidget(self.multidepth_cb, 5, 1)
  1306. # Depth Per Pass
  1307. self.dpp_label = QtWidgets.QLabel('%s:' % _("DPP"))
  1308. self.dpp_label.setToolTip(
  1309. _("DPP. Depth per Pass.\n"
  1310. "The value used to cut into material on each pass."))
  1311. self.multidepth_entry = FCDoubleSpinner()
  1312. self.multidepth_entry.set_range(-9999.9999, 9999.9999)
  1313. self.multidepth_entry.set_precision(self.decimals)
  1314. self.multidepth_entry.setObjectName('gdb_multidepth_entry')
  1315. self.grid0.addWidget(self.dpp_label, 7, 0)
  1316. self.grid0.addWidget(self.multidepth_entry, 7, 1)
  1317. # Travel Z
  1318. self.travelz_label = QtWidgets.QLabel('%s:' % _("Travel Z"))
  1319. self.travelz_label.setToolTip(
  1320. _("Clearance Height.\n"
  1321. "Height at which the milling bit will travel between cuts,\n"
  1322. "above the surface of the material, avoiding all fixtures."))
  1323. self.travelz_entry = FCDoubleSpinner()
  1324. self.travelz_entry.set_range(-9999.9999, 9999.9999)
  1325. self.travelz_entry.set_precision(self.decimals)
  1326. self.travelz_entry.setObjectName('gdb_travel')
  1327. self.grid0.addWidget(self.travelz_label, 9, 0)
  1328. self.grid0.addWidget(self.travelz_entry, 9, 1)
  1329. # Feedrate X-Y
  1330. self.frxy_label = QtWidgets.QLabel('%s:' % _("Feedrate X-Y"))
  1331. self.frxy_label.setToolTip(
  1332. _("Feedrate X-Y. Feedrate\n"
  1333. "The speed on XY plane used while cutting into material."))
  1334. self.frxy_entry = FCDoubleSpinner()
  1335. self.frxy_entry.set_range(-9999.9999, 9999.9999)
  1336. self.frxy_entry.set_precision(self.decimals)
  1337. self.frxy_entry.setObjectName('gdb_frxy')
  1338. self.grid0.addWidget(self.frxy_label, 12, 0)
  1339. self.grid0.addWidget(self.frxy_entry, 12, 1)
  1340. # Feedrate Z
  1341. self.frz_label = QtWidgets.QLabel('%s:' % _("Feedrate Z"))
  1342. self.frz_label.setToolTip(
  1343. _("Feedrate Z\n"
  1344. "The speed on Z plane."))
  1345. self.frz_entry = FCDoubleSpinner()
  1346. self.frz_entry.set_range(-9999.9999, 9999.9999)
  1347. self.frz_entry.set_precision(self.decimals)
  1348. self.frz_entry.setObjectName('gdb_frz')
  1349. self.grid0.addWidget(self.frz_label, 14, 0)
  1350. self.grid0.addWidget(self.frz_entry, 14, 1)
  1351. # Spindle Spped
  1352. self.spindle_label = QtWidgets.QLabel('%s:' % _("Spindle Speed"))
  1353. self.spindle_label.setToolTip(
  1354. _("Spindle Speed.\n"
  1355. "If it's left empty it will not be used.\n"
  1356. "The speed of the spindle in RPM."))
  1357. self.spindle_entry = FCDoubleSpinner()
  1358. self.spindle_entry.set_range(-9999.9999, 9999.9999)
  1359. self.spindle_entry.set_precision(self.decimals)
  1360. self.frz_entry.setObjectName('gdb_spindle')
  1361. self.grid0.addWidget(self.spindle_label, 15, 0)
  1362. self.grid0.addWidget(self.spindle_entry, 15, 1)
  1363. # Dwell
  1364. self.dwell_label = QtWidgets.QLabel('%s:' % _("Dwell"))
  1365. self.dwell_label.setToolTip(
  1366. _("Dwell.\n"
  1367. "Check this if a delay is needed to allow\n"
  1368. "the spindle motor to reach it's set speed."))
  1369. self.dwell_cb = FCCheckBox()
  1370. self.dwell_cb.setObjectName('gdb_dwell')
  1371. self.grid0.addWidget(self.dwell_label, 16, 0)
  1372. self.grid0.addWidget(self.dwell_cb, 16, 1)
  1373. # Dwell Time
  1374. self.dwelltime_label = QtWidgets.QLabel('%s:' % _("Dwelltime"))
  1375. self.dwelltime_label.setToolTip(
  1376. _("Dwell Time.\n"
  1377. "A delay used to allow the motor spindle reach it's set speed."))
  1378. self.dwelltime_entry = FCDoubleSpinner()
  1379. self.dwelltime_entry.set_range(0.0000, 9999.9999)
  1380. self.dwelltime_entry.set_precision(self.decimals)
  1381. self.dwelltime_entry.setObjectName('gdb_dwelltime')
  1382. self.grid0.addWidget(self.dwelltime_label, 17, 0)
  1383. self.grid0.addWidget(self.dwelltime_entry, 17, 1)
  1384. # ###########################################################################
  1385. # ############### ADVANCED UI form ##########################################
  1386. # ###########################################################################
  1387. self.grid1 = QtWidgets.QGridLayout()
  1388. self.advanced_vlay.addLayout(self.grid1)
  1389. self.grid1.setColumnStretch(0, 0)
  1390. self.grid1.setColumnStretch(1, 1)
  1391. self.advanced_vlay.addStretch()
  1392. # Tool Type
  1393. self.type_label = QtWidgets.QLabel('%s:' % _("Tool Type"))
  1394. self.type_label.setToolTip(
  1395. _("Tool Type.\n"
  1396. "Can be:\n"
  1397. "Iso = isolation cut\n"
  1398. "Rough = rough cut, low feedrate, multiple passes\n"
  1399. "Finish = finishing cut, high feedrate"))
  1400. self.type_combo = FCComboBox()
  1401. self.type_combo.addItems(["Iso", "Rough", "Finish"])
  1402. self.type_combo.setObjectName('gdb_type')
  1403. self.grid1.addWidget(self.type_label, 0, 0)
  1404. self.grid1.addWidget(self.type_combo, 0, 1)
  1405. # Tool Offset
  1406. self.tooloffset_label = QtWidgets.QLabel('%s:' % _('Tool Offset'))
  1407. self.tooloffset_label.setToolTip(
  1408. _("Tool Offset.\n"
  1409. "Can be of a few types:\n"
  1410. "Path = zero offset\n"
  1411. "In = offset inside by half of tool diameter\n"
  1412. "Out = offset outside by half of tool diameter\n"
  1413. "Custom = custom offset using the Custom Offset value"))
  1414. self.tooloffset_combo = FCComboBox()
  1415. self.tooloffset_combo.addItems(["Path", "In", "Out", "Custom"])
  1416. self.tooloffset_combo.setObjectName('gdb_tool_offset')
  1417. self.grid1.addWidget(self.tooloffset_label, 2, 0)
  1418. self.grid1.addWidget(self.tooloffset_combo, 2, 1)
  1419. # Custom Offset
  1420. self.custom_offset_label = QtWidgets.QLabel('%s:' % _("Custom Offset"))
  1421. self.custom_offset_label.setToolTip(
  1422. _("Custom Offset.\n"
  1423. "A value to be used as offset from the current path."))
  1424. self.custom_offset_entry = FCDoubleSpinner()
  1425. self.custom_offset_entry.set_range(-9999.9999, 9999.9999)
  1426. self.custom_offset_entry.set_precision(self.decimals)
  1427. self.custom_offset_entry.setObjectName('gdb_custom_offset')
  1428. self.grid1.addWidget(self.custom_offset_label, 5, 0)
  1429. self.grid1.addWidget(self.custom_offset_entry, 5, 1)
  1430. # V-Dia
  1431. self.vdia_label = QtWidgets.QLabel('%s:' % _("V-Dia"))
  1432. self.vdia_label.setToolTip(
  1433. _("V-Dia.\n"
  1434. "Diameter of the tip for V-Shape Tools."))
  1435. self.vdia_entry = FCDoubleSpinner()
  1436. self.vdia_entry.set_range(0.0000, 9999.9999)
  1437. self.vdia_entry.set_precision(self.decimals)
  1438. self.vdia_entry.setObjectName('gdb_vdia')
  1439. self.grid1.addWidget(self.vdia_label, 7, 0)
  1440. self.grid1.addWidget(self.vdia_entry, 7, 1)
  1441. # V-Angle
  1442. self.vangle_label = QtWidgets.QLabel('%s:' % _("V-Angle"))
  1443. self.vangle_label.setToolTip(
  1444. _("V-Agle.\n"
  1445. "Angle at the tip for the V-Shape Tools."))
  1446. self.vangle_entry = FCDoubleSpinner()
  1447. self.vangle_entry.set_range(-360.0, 360.0)
  1448. self.vangle_entry.set_precision(self.decimals)
  1449. self.vangle_entry.setObjectName('gdb_vangle')
  1450. self.grid1.addWidget(self.vangle_label, 8, 0)
  1451. self.grid1.addWidget(self.vangle_entry, 8, 1)
  1452. # Feedrate Rapids
  1453. self.frapids_label = QtWidgets.QLabel('%s:' % _("FR Rapids"))
  1454. self.frapids_label.setToolTip(
  1455. _("FR Rapids. Feedrate Rapids\n"
  1456. "Speed used while moving as fast as possible.\n"
  1457. "This is used only by some devices that can't use\n"
  1458. "the G0 g-code command. Mostly 3D printers."))
  1459. self.frapids_entry = FCDoubleSpinner()
  1460. self.frapids_entry.set_range(0.0000, 9999.9999)
  1461. self.frapids_entry.set_precision(self.decimals)
  1462. self.frapids_entry.setObjectName('gdb_frapids')
  1463. self.grid1.addWidget(self.frapids_label, 10, 0)
  1464. self.grid1.addWidget(self.frapids_entry, 10, 1)
  1465. # Extra Cut
  1466. self.ecut_label = QtWidgets.QLabel('%s:' % _("ExtraCut"))
  1467. self.ecut_label.setToolTip(
  1468. _("Extra Cut.\n"
  1469. "If checked, after a isolation is finished an extra cut\n"
  1470. "will be added where the start and end of isolation meet\n"
  1471. "such as that this point is covered by this extra cut to\n"
  1472. "ensure a complete isolation."))
  1473. self.ecut_cb = FCCheckBox()
  1474. self.ecut_cb.setObjectName('gdb_ecut')
  1475. self.grid1.addWidget(self.ecut_label, 12, 0)
  1476. self.grid1.addWidget(self.ecut_cb, 12, 1)
  1477. # Extra Cut Length
  1478. self.ecut_length_label = QtWidgets.QLabel('%s:' % _("E-Cut Length"))
  1479. self.ecut_length_label.setToolTip(
  1480. _("Extra Cut length.\n"
  1481. "If checked, after a isolation is finished an extra cut\n"
  1482. "will be added where the start and end of isolation meet\n"
  1483. "such as that this point is covered by this extra cut to\n"
  1484. "ensure a complete isolation. This is the length of\n"
  1485. "the extra cut."))
  1486. self.ecut_length_entry = FCDoubleSpinner()
  1487. self.ecut_length_entry.set_range(0.0000, 9999.9999)
  1488. self.ecut_length_entry.set_precision(self.decimals)
  1489. self.ecut_length_entry.setObjectName('gdb_ecut_length')
  1490. self.grid1.addWidget(self.ecut_length_label, 13, 0)
  1491. self.grid1.addWidget(self.ecut_length_entry, 13, 1)
  1492. # ####################################################################
  1493. # ####################################################################
  1494. # GUI for the lower part of the window
  1495. # ####################################################################
  1496. # ####################################################################
  1497. new_vlay = QtWidgets.QVBoxLayout()
  1498. grid_layout.addLayout(new_vlay, 1, 0, 1, 2)
  1499. self.buttons_frame = QtWidgets.QFrame()
  1500. self.buttons_frame.setContentsMargins(0, 0, 0, 0)
  1501. new_vlay.addWidget(self.buttons_frame)
  1502. self.buttons_box = QtWidgets.QHBoxLayout()
  1503. self.buttons_box.setContentsMargins(0, 0, 0, 0)
  1504. self.buttons_frame.setLayout(self.buttons_box)
  1505. self.buttons_frame.show()
  1506. add_entry_btn = FCButton(_("Add Tool in DB"))
  1507. add_entry_btn.setToolTip(
  1508. _("Add a new tool in the Tools Database.\n"
  1509. "It will be used in the Geometry UI.\n"
  1510. "You can edit it after it is added.")
  1511. )
  1512. self.buttons_box.addWidget(add_entry_btn)
  1513. # add_fct_entry_btn = FCButton(_("Add Paint/NCC Tool in DB"))
  1514. # add_fct_entry_btn.setToolTip(
  1515. # _("Add a new tool in the Tools Database.\n"
  1516. # "It will be used in the Paint/NCC Tools UI.\n"
  1517. # "You can edit it after it is added.")
  1518. # )
  1519. # self.buttons_box.addWidget(add_fct_entry_btn)
  1520. remove_entry_btn = FCButton(_("Delete Tool from DB"))
  1521. remove_entry_btn.setToolTip(
  1522. _("Remove a selection of tools in the Tools Database.")
  1523. )
  1524. self.buttons_box.addWidget(remove_entry_btn)
  1525. export_db_btn = FCButton(_("Export DB"))
  1526. export_db_btn.setToolTip(
  1527. _("Save the Tools Database to a custom text file.")
  1528. )
  1529. self.buttons_box.addWidget(export_db_btn)
  1530. import_db_btn = FCButton(_("Import DB"))
  1531. import_db_btn.setToolTip(
  1532. _("Load the Tools Database information's from a custom text file.")
  1533. )
  1534. self.buttons_box.addWidget(import_db_btn)
  1535. self.add_tool_from_db = FCButton(_("Add Tool from Tools DB"))
  1536. self.add_tool_from_db.setToolTip(
  1537. _("Add a new tool in the Tools Table of the\n"
  1538. "active Geometry object after selecting a tool\n"
  1539. "in the Tools Database.")
  1540. )
  1541. self.add_tool_from_db.hide()
  1542. self.cancel_tool_from_db = FCButton(_("Cancel"))
  1543. self.cancel_tool_from_db.hide()
  1544. hlay = QtWidgets.QHBoxLayout()
  1545. tree_layout.addLayout(hlay)
  1546. hlay.addWidget(self.add_tool_from_db)
  1547. hlay.addWidget(self.cancel_tool_from_db)
  1548. hlay.addStretch()
  1549. # ##############################################################################
  1550. # ##############################################################################
  1551. # ########## SETUP THE DICTIONARIES THAT HOLD THE WIDGETS #####################
  1552. # ##############################################################################
  1553. # ##############################################################################
  1554. self.form_fields = {
  1555. # Basic
  1556. "name": self.name_entry,
  1557. "tooldia": self.dia_entry,
  1558. "tool_type": self.shape_combo,
  1559. "cutz": self.cutz_entry,
  1560. "multidepth": self.multidepth_cb,
  1561. "depthperpass": self.multidepth_entry,
  1562. "travelz": self.travelz_entry,
  1563. "feedrate": self.frxy_entry,
  1564. "feedrate_z": self.frz_entry,
  1565. "spindlespeed": self.spindle_entry,
  1566. "dwell": self.dwell_cb,
  1567. "dwelltime": self.dwelltime_entry,
  1568. # Advanced
  1569. "type": self.type_combo,
  1570. "offset": self.tooloffset_combo,
  1571. "offset_value": self.custom_offset_entry,
  1572. "vtipdia": self.vdia_entry,
  1573. "vtipangle": self.vangle_entry,
  1574. "feedrate_rapid": self.frapids_entry,
  1575. "extracut": self.ecut_cb,
  1576. "extracut_length": self.ecut_length_entry
  1577. }
  1578. self.name2option = {
  1579. # Basic
  1580. "gdb_name": "name",
  1581. "gdb_dia": "tooldia",
  1582. "gdb_shape": "tool_type",
  1583. "gdb_cutz": "cutz",
  1584. "gdb_multidepth": "multidepth",
  1585. "gdb_multidepth_entry": "depthperpass",
  1586. "gdb_travel": "travelz",
  1587. "gdb_frxy": "feedrate",
  1588. "gdb_frz": "feedrate_z",
  1589. "gdb_spindle": "spindlespeed",
  1590. "gdb_dwell": "dwell",
  1591. "gdb_dwelltime": "dwelltime",
  1592. # Advanced
  1593. "gdb_type": "type",
  1594. "gdb_tool_offset": "offset",
  1595. "gdb_custom_offset": "offset_value",
  1596. "gdb_vdia": "vtipdia",
  1597. "gdb_vangle": "vtipangle",
  1598. "gdb_frapids": "feedrate_rapid",
  1599. "gdb_ecut": "extracut",
  1600. "gdb_ecut_length": "extracut_length"
  1601. }
  1602. self.current_toolid = None
  1603. # ##############################################################################
  1604. # ######################## SIGNALS #############################################
  1605. # ##############################################################################
  1606. add_entry_btn.clicked.connect(self.on_tool_add)
  1607. remove_entry_btn.clicked.connect(self.on_tool_delete)
  1608. export_db_btn.clicked.connect(self.on_export_tools_db_file)
  1609. import_db_btn.clicked.connect(self.on_import_tools_db_file)
  1610. # closebtn.clicked.connect(self.accept)
  1611. self.add_tool_from_db.clicked.connect(self.on_tool_requested_from_app)
  1612. self.cancel_tool_from_db.clicked.connect(self.on_cancel_tool)
  1613. # self.tree_widget.selectionModel().selectionChanged.connect(self.on_list_selection_change)
  1614. self.tree_widget.currentItemChanged.connect(self.on_list_selection_change)
  1615. self.tree_widget.itemChanged.connect(self.on_list_item_edited)
  1616. self.tree_widget.customContextMenuRequested.connect(self.on_menu_request)
  1617. self.setup_db_ui()
  1618. def on_menu_request(self, pos):
  1619. menu = QtWidgets.QMenu()
  1620. add_tool = menu.addAction(QtGui.QIcon(self.app.resource_location + '/plus16.png'), _("Add to DB"))
  1621. add_tool.triggered.connect(self.on_tool_add)
  1622. copy_tool = menu.addAction(QtGui.QIcon(self.app.resource_location + '/copy16.png'), _("Copy from DB"))
  1623. copy_tool.triggered.connect(self.on_tool_copy)
  1624. delete_tool = menu.addAction(QtGui.QIcon(self.app.resource_location + '/delete32.png'), _("Delete from DB"))
  1625. delete_tool.triggered.connect(self.on_tool_delete)
  1626. # tree_item = self.tree_widget.itemAt(pos)
  1627. menu.exec(self.tree_widget.viewport().mapToGlobal(pos))
  1628. def on_list_selection_change(self, current, previous):
  1629. # for idx in current.indexes():
  1630. # print(idx.data())
  1631. # print(current.text(0))
  1632. self.current_toolid = int(current.text(0))
  1633. self.storage_to_form(self.db_tool_dict[current.text(0)])
  1634. def on_list_item_edited(self, item, column):
  1635. if column == 0:
  1636. return
  1637. self.name_entry.set_value(item.text(1))
  1638. def storage_to_form(self, dict_storage):
  1639. for form_key in self.form_fields:
  1640. for storage_key in dict_storage:
  1641. if form_key == storage_key:
  1642. try:
  1643. self.form_fields[form_key].set_value(dict_storage[form_key])
  1644. except Exception as e:
  1645. print(str(e))
  1646. if storage_key == 'data':
  1647. for data_key in dict_storage[storage_key]:
  1648. if form_key == data_key:
  1649. try:
  1650. self.form_fields[form_key].set_value(dict_storage['data'][data_key])
  1651. except Exception as e:
  1652. print(str(e))
  1653. def form_to_storage(self, tool):
  1654. self.blockSignals(True)
  1655. widget_changed = self.sender()
  1656. wdg_objname = widget_changed.objectName()
  1657. option_changed = self.name2option[wdg_objname]
  1658. tooluid_item = int(tool)
  1659. for tooluid_key, tooluid_val in self.db_tool_dict.items():
  1660. if int(tooluid_key) == tooluid_item:
  1661. new_option_value = self.form_fields[option_changed].get_value()
  1662. if option_changed in tooluid_val:
  1663. tooluid_val[option_changed] = new_option_value
  1664. if option_changed in tooluid_val['data']:
  1665. tooluid_val['data'][option_changed] = new_option_value
  1666. self.blockSignals(False)
  1667. def setup_db_ui(self):
  1668. filename = self.app.data_path + '/geo_tools_db.FlatDB'
  1669. # load the database tools from the file
  1670. try:
  1671. with open(filename) as f:
  1672. tools = f.read()
  1673. except IOError:
  1674. self.app.log.error("Could not load tools DB file.")
  1675. self.app.inform.emit('[ERROR] %s' % _("Could not load Tools DB file."))
  1676. return
  1677. try:
  1678. self.db_tool_dict = json.loads(tools)
  1679. except Exception:
  1680. e = sys.exc_info()[0]
  1681. self.app.log.error(str(e))
  1682. self.app.inform.emit('[ERROR] %s' % _("Failed to parse Tools DB file."))
  1683. return
  1684. self.app.inform.emit('[success] %s: %s' % (_("Loaded FlatCAM Tools DB from"), filename))
  1685. self.build_db_ui()
  1686. def build_db_ui(self):
  1687. self.ui_disconnect()
  1688. nr_crt = 0
  1689. parent = self.tree_widget
  1690. self.tree_widget.blockSignals(True)
  1691. self.tree_widget.clear()
  1692. self.tree_widget.blockSignals(False)
  1693. for toolid, dict_val in self.db_tool_dict.items():
  1694. row = nr_crt
  1695. nr_crt += 1
  1696. t_name = dict_val['name']
  1697. try:
  1698. # self.add_tool_table_line(row, name=t_name, tooldict=dict_val)
  1699. self.tree_widget.blockSignals(True)
  1700. try:
  1701. self.tree_widget.addParentEditable(parent=parent, title=[str(row+1), t_name], editable=True)
  1702. except Exception as e:
  1703. print('FlatCAMCoomn.ToolDB2.build_db_ui() -> ', str(e))
  1704. self.tree_widget.blockSignals(False)
  1705. except Exception as e:
  1706. self.app.log.debug("ToolDB.build_db_ui.add_tool_table_line() --> %s" % str(e))
  1707. if self.current_toolid is None or self.current_toolid < 1:
  1708. if self.db_tool_dict:
  1709. self.storage_to_form(self.db_tool_dict['1'])
  1710. else:
  1711. self.storage_to_form(self.db_tool_dict[str(self.current_toolid)])
  1712. self.ui_connect()
  1713. def add_tool_table_line(self, row, name, tooldict):
  1714. data = tooldict['data']
  1715. def on_tool_add(self):
  1716. """
  1717. Add a tool in the DB Tool Table
  1718. :return: None
  1719. """
  1720. default_data = {}
  1721. default_data.update({
  1722. "cutz": float(self.app.defaults["geometry_cutz"]),
  1723. "multidepth": self.app.defaults["geometry_multidepth"],
  1724. "depthperpass": float(self.app.defaults["geometry_depthperpass"]),
  1725. "vtipdia": float(self.app.defaults["geometry_vtipdia"]),
  1726. "vtipangle": float(self.app.defaults["geometry_vtipangle"]),
  1727. "travelz": float(self.app.defaults["geometry_travelz"]),
  1728. "feedrate": float(self.app.defaults["geometry_feedrate"]),
  1729. "feedrate_z": float(self.app.defaults["geometry_feedrate_z"]),
  1730. "feedrate_rapid": float(self.app.defaults["geometry_feedrate_rapid"]),
  1731. "spindlespeed": self.app.defaults["geometry_spindlespeed"],
  1732. "dwell": self.app.defaults["geometry_dwell"],
  1733. "dwelltime": float(self.app.defaults["geometry_dwelltime"]),
  1734. "ppname_g": self.app.defaults["geometry_ppname_g"],
  1735. "extracut": self.app.defaults["geometry_extracut"],
  1736. "extracut_length": float(self.app.defaults["geometry_extracut_length"]),
  1737. "toolchange": self.app.defaults["geometry_toolchange"],
  1738. "toolchangexy": self.app.defaults["geometry_toolchangexy"],
  1739. "toolchangez": float(self.app.defaults["geometry_toolchangez"]),
  1740. "startz": self.app.defaults["geometry_startz"],
  1741. "endz": float(self.app.defaults["geometry_endz"])
  1742. })
  1743. dict_elem = {}
  1744. dict_elem['name'] = 'new_tool'
  1745. if type(self.app.defaults["geometry_cnctooldia"]) == float:
  1746. dict_elem['tooldia'] = self.app.defaults["geometry_cnctooldia"]
  1747. else:
  1748. try:
  1749. tools_string = self.app.defaults["geometry_cnctooldia"].split(",")
  1750. tools_diameters = [eval(a) for a in tools_string if a != '']
  1751. dict_elem['tooldia'] = tools_diameters[0] if tools_diameters else 0.0
  1752. except Exception as e:
  1753. self.app.log.debug("ToolDB.on_tool_add() --> %s" % str(e))
  1754. return
  1755. dict_elem['offset'] = 'Path'
  1756. dict_elem['offset_value'] = 0.0
  1757. dict_elem['type'] = 'Rough'
  1758. dict_elem['tool_type'] = 'C1'
  1759. dict_elem['data'] = default_data
  1760. new_toolid = len(self.db_tool_dict) + 1
  1761. self.db_tool_dict[str(new_toolid)] = deepcopy(dict_elem)
  1762. # add the new entry to the Tools DB table
  1763. self.update_storage()
  1764. self.build_db_ui()
  1765. self.app.inform.emit('[success] %s' % _("Tool added to DB."))
  1766. def on_tool_copy(self):
  1767. """
  1768. Copy a selection of Tools in the Tools DB table
  1769. :return:
  1770. """
  1771. new_tool_id = len(self.db_tool_dict)
  1772. for item in self.tree_widget.selectedItems():
  1773. old_tool_id = item.data(0, QtCore.Qt.DisplayRole)
  1774. for toolid, dict_val in list(self.db_tool_dict.items()):
  1775. if int(old_tool_id) == int(toolid):
  1776. new_tool_id += 1
  1777. new_key = str(new_tool_id)
  1778. self.db_tool_dict.update({
  1779. new_key: deepcopy(dict_val)
  1780. })
  1781. self.current_toolid = new_tool_id
  1782. self.update_storage()
  1783. self.build_db_ui()
  1784. self.app.inform.emit('[success] %s' % _("Tool copied from Tools DB."))
  1785. def on_tool_delete(self):
  1786. """
  1787. Delete a selection of Tools in the Tools DB table
  1788. :return:
  1789. """
  1790. for item in self.tree_widget.selectedItems():
  1791. toolname_to_remove = item.data(0, QtCore.Qt.DisplayRole)
  1792. for toolid, dict_val in list(self.db_tool_dict.items()):
  1793. if int(toolname_to_remove) == int(toolid):
  1794. # remove from the storage
  1795. self.db_tool_dict.pop(toolid, None)
  1796. self.current_toolid -= 1
  1797. self.update_storage()
  1798. self.build_db_ui()
  1799. self.app.inform.emit('[success] %s' % _("Tool removed from Tools DB."))
  1800. def on_export_tools_db_file(self):
  1801. self.app.report_usage("on_export_tools_db_file")
  1802. self.app.log.debug("on_export_tools_db_file()")
  1803. date = str(datetime.today()).rpartition('.')[0]
  1804. date = ''.join(c for c in date if c not in ':-')
  1805. date = date.replace(' ', '_')
  1806. filter__ = "Text File (*.TXT);;All Files (*.*)"
  1807. filename, _f = QtWidgets.QFileDialog.getSaveFileName(caption=_("Export Tools Database"),
  1808. directory='{l_save}/FlatCAM_{n}_{date}'.format(
  1809. l_save=str(self.app.get_last_save_folder()),
  1810. n=_("Tools_Database"),
  1811. date=date),
  1812. filter=filter__)
  1813. filename = str(filename)
  1814. if filename == "":
  1815. self.app.inform.emit('[WARNING_NOTCL] %s' % _("FlatCAM Tools DB export cancelled."))
  1816. return
  1817. else:
  1818. try:
  1819. f = open(filename, 'w')
  1820. f.close()
  1821. except PermissionError:
  1822. self.app.inform.emit('[WARNING] %s' %
  1823. _("Permission denied, saving not possible.\n"
  1824. "Most likely another app is holding the file open and not accessible."))
  1825. return
  1826. except IOError:
  1827. self.app.log.debug('Creating a new Tools DB file ...')
  1828. f = open(filename, 'w')
  1829. f.close()
  1830. except Exception:
  1831. e = sys.exc_info()[0]
  1832. self.app.log.error("Could not load Tools DB file.")
  1833. self.app.log.error(str(e))
  1834. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Could not load Tools DB file."))
  1835. return
  1836. # Save update options
  1837. try:
  1838. # Save Tools DB in a file
  1839. try:
  1840. with open(filename, "w") as f:
  1841. json.dump(self.db_tool_dict, f, default=to_dict, indent=2)
  1842. except Exception as e:
  1843. self.app.log.debug("App.on_save_tools_db() --> %s" % str(e))
  1844. self.inform.emit('[ERROR_NOTCL] %s' % _("Failed to write Tools DB to file."))
  1845. return
  1846. except Exception:
  1847. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Failed to write Tools DB to file."))
  1848. return
  1849. self.app.inform.emit('[success] %s: %s' % (_("Exported Tools DB to"), filename))
  1850. def on_import_tools_db_file(self):
  1851. self.app.report_usage("on_import_tools_db_file")
  1852. self.app.log.debug("on_import_tools_db_file()")
  1853. filter__ = "Text File (*.TXT);;All Files (*.*)"
  1854. filename, _f = QtWidgets.QFileDialog.getOpenFileName(caption=_("Import FlatCAM Tools DB"), filter=filter__)
  1855. if filename == "":
  1856. self.app.inform.emit('[WARNING_NOTCL] %s' % _("FlatCAM Tools DB import cancelled."))
  1857. else:
  1858. try:
  1859. with open(filename) as f:
  1860. tools_in_db = f.read()
  1861. except IOError:
  1862. self.app.log.error("Could not load Tools DB file.")
  1863. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Could not load Tools DB file."))
  1864. return
  1865. try:
  1866. self.db_tool_dict = json.loads(tools_in_db)
  1867. except Exception:
  1868. e = sys.exc_info()[0]
  1869. self.app.log.error(str(e))
  1870. self.app.inform.emit('[ERROR] %s' % _("Failed to parse Tools DB file."))
  1871. return
  1872. self.app.inform.emit('[success] %s: %s' % (_("Loaded FlatCAM Tools DB from"), filename))
  1873. self.build_db_ui()
  1874. self.update_storage()
  1875. def on_save_tools_db(self, silent=False):
  1876. self.app.log.debug("ToolsDB.on_save_button() --> Saving Tools Database to file.")
  1877. filename = self.app.data_path + "/geo_tools_db.FlatDB"
  1878. # Preferences save, update the color of the Tools DB Tab text
  1879. for idx in range(self.app.ui.plot_tab_area.count()):
  1880. if self.app.ui.plot_tab_area.tabText(idx) == _("Tools Database"):
  1881. self.app.ui.plot_tab_area.tabBar.setTabTextColor(idx, QtGui.QColor('black'))
  1882. # Save Tools DB in a file
  1883. try:
  1884. f = open(filename, "w")
  1885. json.dump(self.db_tool_dict, f, default=to_dict, indent=2)
  1886. f.close()
  1887. except Exception as e:
  1888. self.app.log.debug("ToolsDB.on_save_tools_db() --> %s" % str(e))
  1889. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Failed to write Tools DB to file."))
  1890. return
  1891. if not silent:
  1892. self.app.inform.emit('[success] %s' % _("Saved Tools DB."))
  1893. def ui_connect(self):
  1894. # make sure that we don't make multiple connections to the widgets
  1895. self.ui_disconnect()
  1896. self.name_entry.editingFinished.connect(self.update_tree_name)
  1897. for key in self.form_fields:
  1898. wdg = self.form_fields[key]
  1899. # FCEntry
  1900. if isinstance(wdg, FCEntry):
  1901. wdg.textChanged.connect(self.update_storage)
  1902. # ComboBox
  1903. if isinstance(wdg, FCComboBox):
  1904. wdg.currentIndexChanged.connect(self.update_storage)
  1905. # CheckBox
  1906. if isinstance(wdg, FCCheckBox):
  1907. wdg.toggled.connect(self.update_storage)
  1908. # SpinBox, DoubleSpinBox
  1909. if isinstance(wdg, FCSpinner) or isinstance(wdg, FCDoubleSpinner):
  1910. wdg.valueChanged.connect(self.update_storage)
  1911. def ui_disconnect(self):
  1912. try:
  1913. self.name_entry.editingFinished.disconnect(self.update_tree_name)
  1914. except (TypeError, AttributeError):
  1915. pass
  1916. for key in self.form_fields:
  1917. wdg = self.form_fields[key]
  1918. # FCEntry
  1919. if isinstance(wdg, FCEntry):
  1920. try:
  1921. wdg.textChanged.disconnect(self.update_storage)
  1922. except (TypeError, AttributeError):
  1923. pass
  1924. # ComboBox
  1925. if isinstance(wdg, FCComboBox):
  1926. try:
  1927. wdg.currentIndexChanged.disconnect(self.update_storage)
  1928. except (TypeError, AttributeError):
  1929. pass
  1930. # CheckBox
  1931. if isinstance(wdg, FCCheckBox):
  1932. try:
  1933. wdg.toggled.disconnect(self.update_storage)
  1934. except (TypeError, AttributeError):
  1935. pass
  1936. # SpinBox, DoubleSpinBox
  1937. if isinstance(wdg, FCSpinner) or isinstance(wdg, FCDoubleSpinner):
  1938. try:
  1939. wdg.valueChanged.disconnect(self.update_storage)
  1940. except (TypeError, AttributeError):
  1941. pass
  1942. def update_tree_name(self):
  1943. val = self.name_entry.get_value()
  1944. item = self.tree_widget.currentItem()
  1945. # I'm setting the value for the second column (designated by 1) because first column holds the ID
  1946. # and second column holds the Name (this behavior is set in the build_ui method)
  1947. item.setData(1, QtCore.Qt.DisplayRole, val)
  1948. def update_storage(self):
  1949. tool_id = str(self.current_toolid)
  1950. wdg = self.sender()
  1951. if wdg is None:
  1952. return
  1953. wdg_name = wdg.objectName()
  1954. if wdg_name == "gdb_name":
  1955. self.db_tool_dict[tool_id]['name'] = wdg.get_value()
  1956. elif wdg_name == "gdb_dia":
  1957. self.db_tool_dict[tool_id]['tooldia'] = wdg.get_value()
  1958. elif wdg_name == "gdb_tool_offset":
  1959. self.db_tool_dict[tool_id]['offset'] = wdg.get_value()
  1960. elif wdg_name == "gdb_custom_offset":
  1961. self.db_tool_dict[tool_id]['offset_value'] = wdg.get_value()
  1962. elif wdg_name == "gdb_type":
  1963. self.db_tool_dict[tool_id]['type'] = wdg.get_value()
  1964. elif wdg_name == "gdb_shape":
  1965. self.db_tool_dict[tool_id]['tool_type'] = wdg.get_value()
  1966. else:
  1967. if wdg_name == "gdb_cutz":
  1968. self.db_tool_dict[tool_id]['data']['cutz'] = wdg.get_value()
  1969. elif wdg_name == "gdb_multidepth":
  1970. self.db_tool_dict[tool_id]['data']['multidepth'] = wdg.get_value()
  1971. elif wdg_name == "gdb_multidepth_entry":
  1972. self.db_tool_dict[tool_id]['data']['depthperpass'] = wdg.get_value()
  1973. elif wdg_name == "gdb_travel":
  1974. self.db_tool_dict[tool_id]['data']['travelz'] = wdg.get_value()
  1975. elif wdg_name == "gdb_frxy":
  1976. self.db_tool_dict[tool_id]['data']['feedrate'] = wdg.get_value()
  1977. elif wdg_name == "gdb_frz":
  1978. self.db_tool_dict[tool_id]['data']['feedrate_z'] = wdg.get_value()
  1979. elif wdg_name == "gdb_spindle":
  1980. self.db_tool_dict[tool_id]['data']['spindlespeed'] = wdg.get_value()
  1981. elif wdg_name == "gdb_dwell":
  1982. self.db_tool_dict[tool_id]['data']['dwell'] = wdg.get_value()
  1983. elif wdg_name == "gdb_dwelltime":
  1984. self.db_tool_dict[tool_id]['data']['dwelltime'] = wdg.get_value()
  1985. elif wdg_name == "gdb_vdia":
  1986. self.db_tool_dict[tool_id]['data']['vtipdia'] = wdg.get_value()
  1987. elif wdg_name == "gdb_vangle":
  1988. self.db_tool_dict[tool_id]['data']['vtipangle'] = wdg.get_value()
  1989. elif wdg_name == "gdb_frapids":
  1990. self.db_tool_dict[tool_id]['data']['feedrate_rapid'] = wdg.get_value()
  1991. elif wdg_name == "gdb_ecut":
  1992. self.db_tool_dict[tool_id]['data']['extracut'] = wdg.get_value()
  1993. elif wdg_name == "gdb_ecut_length":
  1994. self.db_tool_dict[tool_id]['data']['extracut_length'] = wdg.get_value()
  1995. self.callback_app()
  1996. def on_tool_requested_from_app(self):
  1997. if not self.table_widget.selectionModel().selectedRows():
  1998. self.app.inform.emit('[WARNING_NOTCL] %s...' % _("No Tool/row selected in the Tools Database table"))
  1999. return
  2000. model_index_list = self.table_widget.selectionModel().selectedRows()
  2001. for model_index in model_index_list:
  2002. selected_row = model_index.row()
  2003. tool_uid = selected_row + 1
  2004. for key in self.db_tool_dict.keys():
  2005. if str(key) == str(tool_uid):
  2006. selected_tool = self.db_tool_dict[key]
  2007. self.on_tool_request(tool=selected_tool)
  2008. def on_cancel_tool(self):
  2009. for idx in range(self.app.ui.plot_tab_area.count()):
  2010. if self.app.ui.plot_tab_area.tabText(idx) == _("Tools Database"):
  2011. wdg = self.app.ui.plot_tab_area.widget(idx)
  2012. wdg.deleteLater()
  2013. self.app.ui.plot_tab_area.removeTab(idx)
  2014. self.app.inform.emit('%s' % _("Cancelled adding tool from DB."))
  2015. def resize_new_tool_table_widget(self, min_size, max_size):
  2016. """
  2017. Resize the table widget responsible for adding new tool in the Tool Database
  2018. :param min_size: passed by rangeChanged signal or the self.new_tool_table_widget.horizontalScrollBar()
  2019. :param max_size: passed by rangeChanged signal or the self.new_tool_table_widget.horizontalScrollBar()
  2020. :return:
  2021. """
  2022. t_height = self.t_height
  2023. if max_size > min_size:
  2024. t_height = self.t_height + self.new_tool_table_widget.verticalScrollBar().height()
  2025. self.new_tool_table_widget.setMaximumHeight(t_height)
  2026. def closeEvent(self, QCloseEvent):
  2027. super().closeEvent(QCloseEvent)
  2028. def color_variant(hex_color, bright_factor=1):
  2029. """
  2030. Takes a color in HEX format #FF00FF and produces a lighter or darker variant
  2031. :param hex_color: color to change
  2032. :param bright_factor: factor to change the color brightness [0 ... 1]
  2033. :return: modified color
  2034. """
  2035. if len(hex_color) != 7:
  2036. print("Color is %s, but needs to be in #FF00FF format. Returning original color." % hex_color)
  2037. return hex_color
  2038. if bright_factor > 1.0:
  2039. bright_factor = 1.0
  2040. if bright_factor < 0.0:
  2041. bright_factor = 0.0
  2042. rgb_hex = [hex_color[x:x + 2] for x in [1, 3, 5]]
  2043. new_rgb = []
  2044. for hex_value in rgb_hex:
  2045. # adjust each color channel and turn it into a INT suitable as argument for hex()
  2046. mod_color = round(int(hex_value, 16) * bright_factor)
  2047. # make sure that each color channel has two digits without the 0x prefix
  2048. mod_color_hex = str(hex(mod_color)[2:]).zfill(2)
  2049. new_rgb.append(mod_color_hex)
  2050. return "#" + "".join([i for i in new_rgb])