FlatCAMCommon.py 55 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328
  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
  14. from camlib import to_dict
  15. import sys
  16. import webbrowser
  17. import json
  18. from copy import deepcopy
  19. from datetime import datetime
  20. import gettext
  21. import FlatCAMTranslation as fcTranslate
  22. import builtins
  23. fcTranslate.apply_language('strings')
  24. if '_' not in builtins.__dict__:
  25. _ = gettext.gettext
  26. class LoudDict(dict):
  27. """
  28. A Dictionary with a callback for
  29. item changes.
  30. """
  31. def __init__(self, *args, **kwargs):
  32. dict.__init__(self, *args, **kwargs)
  33. self.callback = lambda x: None
  34. def __setitem__(self, key, value):
  35. """
  36. Overridden __setitem__ method. Will emit 'changed(QString)'
  37. if the item was changed, with key as parameter.
  38. """
  39. if key in self and self.__getitem__(key) == value:
  40. return
  41. dict.__setitem__(self, key, value)
  42. self.callback(key)
  43. def update(self, *args, **kwargs):
  44. if len(args) > 1:
  45. raise TypeError("update expected at most 1 arguments, got %d" % len(args))
  46. other = dict(*args, **kwargs)
  47. for key in other:
  48. self[key] = other[key]
  49. def set_change_callback(self, callback):
  50. """
  51. Assigns a function as callback on item change. The callback
  52. will receive the key of the object that was changed.
  53. :param callback: Function to call on item change.
  54. :type callback: func
  55. :return: None
  56. """
  57. self.callback = callback
  58. class FCSignal:
  59. """
  60. Taken from here: https://blog.abstractfactory.io/dynamic-signals-in-pyqt/
  61. """
  62. def __init__(self):
  63. self.__subscribers = []
  64. def emit(self, *args, **kwargs):
  65. for subs in self.__subscribers:
  66. subs(*args, **kwargs)
  67. def connect(self, func):
  68. self.__subscribers.append(func)
  69. def disconnect(self, func):
  70. try:
  71. self.__subscribers.remove(func)
  72. except ValueError:
  73. print('Warning: function %s not removed '
  74. 'from signal %s' % (func, self))
  75. class BookmarkManager(QtWidgets.QWidget):
  76. mark_rows = QtCore.pyqtSignal()
  77. def __init__(self, app, storage, parent=None):
  78. super(BookmarkManager, self).__init__(parent)
  79. self.app = app
  80. assert isinstance(storage, dict), "Storage argument is not a dictionary"
  81. self.bm_dict = deepcopy(storage)
  82. # Icon and title
  83. # self.setWindowIcon(parent.app_icon)
  84. # self.setWindowTitle(_("Bookmark Manager"))
  85. # self.resize(600, 400)
  86. # title = QtWidgets.QLabel(
  87. # "<font size=8><B>FlatCAM</B></font><BR>"
  88. # )
  89. # title.setOpenExternalLinks(True)
  90. # layouts
  91. layout = QtWidgets.QVBoxLayout()
  92. self.setLayout(layout)
  93. table_hlay = QtWidgets.QHBoxLayout()
  94. layout.addLayout(table_hlay)
  95. self.table_widget = FCTable(drag_drop=True, protected_rows=[0, 1])
  96. self.table_widget.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
  97. table_hlay.addWidget(self.table_widget)
  98. self.table_widget.setColumnCount(3)
  99. self.table_widget.setColumnWidth(0, 20)
  100. self.table_widget.setHorizontalHeaderLabels(
  101. [
  102. '#',
  103. _('Title'),
  104. _('Web Link')
  105. ]
  106. )
  107. self.table_widget.horizontalHeaderItem(0).setToolTip(
  108. _("Index.\n"
  109. "The rows in gray color will populate the Bookmarks menu.\n"
  110. "The number of gray colored rows is set in Preferences."))
  111. self.table_widget.horizontalHeaderItem(1).setToolTip(
  112. _("Description of the link that is set as an menu action.\n"
  113. "Try to keep it short because it is installed as a menu item."))
  114. self.table_widget.horizontalHeaderItem(2).setToolTip(
  115. _("Web Link. E.g: https://your_website.org "))
  116. # pal = QtGui.QPalette()
  117. # pal.setColor(QtGui.QPalette.Background, Qt.white)
  118. # New Bookmark
  119. new_vlay = QtWidgets.QVBoxLayout()
  120. layout.addLayout(new_vlay)
  121. new_title_lbl = QtWidgets.QLabel('<b>%s</b>' % _("New Bookmark"))
  122. new_vlay.addWidget(new_title_lbl)
  123. form0 = QtWidgets.QFormLayout()
  124. new_vlay.addLayout(form0)
  125. title_lbl = QtWidgets.QLabel('%s:' % _("Title"))
  126. self.title_entry = FCEntry()
  127. form0.addRow(title_lbl, self.title_entry)
  128. link_lbl = QtWidgets.QLabel('%s:' % _("Web Link"))
  129. self.link_entry = FCEntry()
  130. self.link_entry.set_value('http://')
  131. form0.addRow(link_lbl, self.link_entry)
  132. # Buttons Layout
  133. button_hlay = QtWidgets.QHBoxLayout()
  134. layout.addLayout(button_hlay)
  135. add_entry_btn = FCButton(_("Add Entry"))
  136. remove_entry_btn = FCButton(_("Remove Entry"))
  137. export_list_btn = FCButton(_("Export List"))
  138. import_list_btn = FCButton(_("Import List"))
  139. # closebtn = QtWidgets.QPushButton(_("Close"))
  140. # button_hlay.addStretch()
  141. button_hlay.addWidget(add_entry_btn)
  142. button_hlay.addWidget(remove_entry_btn)
  143. button_hlay.addWidget(export_list_btn)
  144. button_hlay.addWidget(import_list_btn)
  145. # button_hlay.addWidget(closebtn)
  146. # ##############################################################################
  147. # ######################## SIGNALS #############################################
  148. # ##############################################################################
  149. add_entry_btn.clicked.connect(self.on_add_entry)
  150. remove_entry_btn.clicked.connect(self.on_remove_entry)
  151. export_list_btn.clicked.connect(self.on_export_bookmarks)
  152. import_list_btn.clicked.connect(self.on_import_bookmarks)
  153. self.title_entry.returnPressed.connect(self.on_add_entry)
  154. self.link_entry.returnPressed.connect(self.on_add_entry)
  155. # closebtn.clicked.connect(self.accept)
  156. self.table_widget.drag_drop_sig.connect(self.mark_table_rows_for_actions)
  157. self.build_bm_ui()
  158. def build_bm_ui(self):
  159. self.table_widget.setRowCount(len(self.bm_dict))
  160. nr_crt = 0
  161. sorted_bookmarks = sorted(list(self.bm_dict.items()), key=lambda x: int(x[0]))
  162. for entry, bookmark in sorted_bookmarks:
  163. row = nr_crt
  164. nr_crt += 1
  165. title = bookmark[0]
  166. weblink = bookmark[1]
  167. id_item = QtWidgets.QTableWidgetItem('%d' % int(nr_crt))
  168. # id.setFlags(QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
  169. self.table_widget.setItem(row, 0, id_item) # Tool name/id
  170. title_item = QtWidgets.QTableWidgetItem(title)
  171. self.table_widget.setItem(row, 1, title_item)
  172. weblink_txt = QtWidgets.QTextBrowser()
  173. weblink_txt.setOpenExternalLinks(True)
  174. weblink_txt.setFrameStyle(QtWidgets.QFrame.NoFrame)
  175. weblink_txt.document().setDefaultStyleSheet("a{ text-decoration: none; }")
  176. weblink_txt.setHtml('<a href=%s>%s</a>' % (weblink, weblink))
  177. self.table_widget.setCellWidget(row, 2, weblink_txt)
  178. vertical_header = self.table_widget.verticalHeader()
  179. vertical_header.hide()
  180. horizontal_header = self.table_widget.horizontalHeader()
  181. horizontal_header.setMinimumSectionSize(10)
  182. horizontal_header.setDefaultSectionSize(70)
  183. horizontal_header.setSectionResizeMode(0, QtWidgets.QHeaderView.Fixed)
  184. horizontal_header.resizeSection(0, 20)
  185. horizontal_header.setSectionResizeMode(1, QtWidgets.QHeaderView.ResizeToContents)
  186. horizontal_header.setSectionResizeMode(2, QtWidgets.QHeaderView.Stretch)
  187. self.mark_table_rows_for_actions()
  188. self.app.defaults["global_bookmarks"].clear()
  189. for key, val in self.bm_dict.items():
  190. self.app.defaults["global_bookmarks"][key] = deepcopy(val)
  191. def on_add_entry(self, **kwargs):
  192. """
  193. Add a entry in the Bookmark Table and in the menu actions
  194. :return: None
  195. """
  196. if 'title' in kwargs:
  197. title = kwargs['title']
  198. else:
  199. title = self.title_entry.get_value()
  200. if title == '':
  201. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Title entry is empty."))
  202. return 'fail'
  203. if 'link' is kwargs:
  204. link = kwargs['link']
  205. else:
  206. link = self.link_entry.get_value()
  207. if link == 'http://':
  208. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Web link entry is empty."))
  209. return 'fail'
  210. # if 'http' not in link or 'https' not in link:
  211. # link = 'http://' + link
  212. for bookmark in self.bm_dict.values():
  213. if title == bookmark[0] or link == bookmark[1]:
  214. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Either the Title or the Weblink already in the table."))
  215. return 'fail'
  216. # for some reason if the last char in the weblink is a slash it does not make the link clickable
  217. # so I remove it
  218. if link[-1] == '/':
  219. link = link[:-1]
  220. # add the new entry to storage
  221. new_entry = len(self.bm_dict) + 1
  222. self.bm_dict[str(new_entry)] = [title, link]
  223. # add the link to the menu but only if it is within the set limit
  224. bm_limit = int(self.app.defaults["global_bookmarks_limit"])
  225. if len(self.bm_dict) < bm_limit:
  226. act = QtWidgets.QAction(parent=self.app.ui.menuhelp_bookmarks)
  227. act.setText(title)
  228. act.setIcon(QtGui.QIcon('share/link16.png'))
  229. act.triggered.connect(lambda: webbrowser.open(link))
  230. self.app.ui.menuhelp_bookmarks.insertAction(self.app.ui.menuhelp_bookmarks_manager, act)
  231. self.app.inform.emit('[success] %s' % _("Bookmark added."))
  232. # add the new entry to the bookmark manager table
  233. self.build_bm_ui()
  234. def on_remove_entry(self):
  235. """
  236. Remove an Entry in the Bookmark table and from the menu actions
  237. :return:
  238. """
  239. index_list = []
  240. for model_index in self.table_widget.selectionModel().selectedRows():
  241. index = QtCore.QPersistentModelIndex(model_index)
  242. index_list.append(index)
  243. title_to_remove = self.table_widget.item(model_index.row(), 1).text()
  244. if title_to_remove == 'FlatCAM' or title_to_remove == 'Backup Site':
  245. self.app.inform.emit('[WARNING_NOTCL] %s.' % _("This bookmark can not be removed"))
  246. self.build_bm_ui()
  247. return
  248. else:
  249. for k, bookmark in list(self.bm_dict.items()):
  250. if title_to_remove == bookmark[0]:
  251. # remove from the storage
  252. self.bm_dict.pop(k, None)
  253. for act in self.app.ui.menuhelp_bookmarks.actions():
  254. if act.text() == title_to_remove:
  255. # disconnect the signal
  256. try:
  257. act.triggered.disconnect()
  258. except TypeError:
  259. pass
  260. # remove the action from the menu
  261. self.app.ui.menuhelp_bookmarks.removeAction(act)
  262. # house keeping: it pays to have keys increased by one
  263. new_key = 0
  264. new_dict = dict()
  265. for k, v in self.bm_dict.items():
  266. # we start with key 1 so we can use the len(self.bm_dict)
  267. # when adding bookmarks (keys in bm_dict)
  268. new_key += 1
  269. new_dict[str(new_key)] = v
  270. self.bm_dict = deepcopy(new_dict)
  271. new_dict.clear()
  272. self.app.inform.emit('[success] %s' % _("Bookmark removed."))
  273. # for index in index_list:
  274. # self.table_widget.model().removeRow(index.row())
  275. self.build_bm_ui()
  276. def on_export_bookmarks(self):
  277. self.app.report_usage("on_export_bookmarks")
  278. self.app.log.debug("on_export_bookmarks()")
  279. date = str(datetime.today()).rpartition('.')[0]
  280. date = ''.join(c for c in date if c not in ':-')
  281. date = date.replace(' ', '_')
  282. filter__ = "Text File (*.TXT);;All Files (*.*)"
  283. filename, _f = QtWidgets.QFileDialog.getSaveFileName(caption=_("Export FlatCAM Bookmarks"),
  284. directory='{l_save}/FlatCAM_{n}_{date}'.format(
  285. l_save=str(self.app.get_last_save_folder()),
  286. n=_("Bookmarks"),
  287. date=date),
  288. filter=filter__)
  289. filename = str(filename)
  290. if filename == "":
  291. self.app.inform.emit('[WARNING_NOTCL] %s' % _("FlatCAM bookmarks export cancelled."))
  292. return
  293. else:
  294. try:
  295. f = open(filename, 'w')
  296. f.close()
  297. except PermissionError:
  298. self.app.inform.emit('[WARNING] %s' %
  299. _("Permission denied, saving not possible.\n"
  300. "Most likely another app is holding the file open and not accessible."))
  301. return
  302. except IOError:
  303. self.app.log.debug('Creating a new bookmarks file ...')
  304. f = open(filename, 'w')
  305. f.close()
  306. except Exception:
  307. e = sys.exc_info()[0]
  308. self.app.log.error("Could not load defaults file.")
  309. self.app.log.error(str(e))
  310. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Could not load bookmarks file."))
  311. return
  312. # Save Bookmarks to a file
  313. try:
  314. with open(filename, "w") as f:
  315. for title, link in self.bm_dict.items():
  316. line2write = str(title) + ':' + str(link) + '\n'
  317. f.write(line2write)
  318. except Exception:
  319. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Failed to write bookmarks to file."))
  320. return
  321. self.app.inform.emit('[success] %s: %s' % (_("Exported bookmarks to"), filename))
  322. def on_import_bookmarks(self):
  323. self.app.log.debug("on_import_bookmarks()")
  324. filter_ = "Text File (*.txt);;All Files (*.*)"
  325. filename, _f = QtWidgets.QFileDialog.getOpenFileName(caption=_("Import FlatCAM Bookmarks"), filter=filter_)
  326. filename = str(filename)
  327. if filename == "":
  328. self.app.inform.emit('[WARNING_NOTCL] %s' % _("FlatCAM bookmarks import cancelled."))
  329. else:
  330. try:
  331. with open(filename) as f:
  332. bookmarks = f.readlines()
  333. except IOError:
  334. self.app.log.error("Could not load bookmarks file.")
  335. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Could not load bookmarks file."))
  336. return
  337. for line in bookmarks:
  338. proc_line = line.replace(' ', '').partition(':')
  339. self.on_add_entry(title=proc_line[0], link=proc_line[2])
  340. self.app.inform.emit('[success] %s: %s' % (_("Imported Bookmarks from"), filename))
  341. def mark_table_rows_for_actions(self):
  342. for row in range(self.table_widget.rowCount()):
  343. item_to_paint = self.table_widget.item(row, 0)
  344. if row < self.app.defaults["global_bookmarks_limit"]:
  345. item_to_paint.setBackground(QtGui.QColor('gray'))
  346. # item_to_paint.setForeground(QtGui.QColor('black'))
  347. else:
  348. item_to_paint.setBackground(QtGui.QColor('white'))
  349. # item_to_paint.setForeground(QtGui.QColor('black'))
  350. def rebuild_actions(self):
  351. # rebuild the storage to reflect the order of the lines
  352. self.bm_dict.clear()
  353. for row in range(self.table_widget.rowCount()):
  354. title = self.table_widget.item(row, 1).text()
  355. wlink = self.table_widget.cellWidget(row, 2).toPlainText()
  356. entry = int(row) + 1
  357. self.bm_dict.update(
  358. {
  359. str(entry): [title, wlink]
  360. }
  361. )
  362. self.app.install_bookmarks(book_dict=self.bm_dict)
  363. # def accept(self):
  364. # self.rebuild_actions()
  365. # super().accept()
  366. def closeEvent(self, QCloseEvent):
  367. self.rebuild_actions()
  368. super().closeEvent(QCloseEvent)
  369. class ToolsDB(QtWidgets.QWidget):
  370. mark_tools_rows = QtCore.pyqtSignal()
  371. def __init__(self, app, callback_on_edited, callback_on_tool_request, parent=None):
  372. super(ToolsDB, self).__init__(parent)
  373. self.app = app
  374. self.decimals = 4
  375. self.callback_app = callback_on_edited
  376. self.on_tool_request = callback_on_tool_request
  377. self.offset_item_options = ["Path", "In", "Out", "Custom"]
  378. self.type_item_options = [_("Iso"), _("Rough"), _("Finish")]
  379. self.tool_type_item_options = ["C1", "C2", "C3", "C4", "B", "V"]
  380. '''
  381. dict to hold all the tools in the Tools DB
  382. format:
  383. {
  384. tool_id: {
  385. 'name': 'new_tool'
  386. 'tooldia': self.app.defaults["geometry_cnctooldia"]
  387. 'offset': 'Path'
  388. 'offset_value': 0.0
  389. 'type': _('Rough'),
  390. 'tool_type': 'C1'
  391. 'data': dict()
  392. }
  393. }
  394. '''
  395. self.db_tool_dict = dict()
  396. # layouts
  397. layout = QtWidgets.QVBoxLayout()
  398. self.setLayout(layout)
  399. table_hlay = QtWidgets.QHBoxLayout()
  400. layout.addLayout(table_hlay)
  401. self.table_widget = FCTable(drag_drop=True)
  402. self.table_widget.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
  403. table_hlay.addWidget(self.table_widget)
  404. self.table_widget.setColumnCount(26)
  405. # self.table_widget.setColumnWidth(0, 20)
  406. self.table_widget.setHorizontalHeaderLabels(
  407. [
  408. '#',
  409. _("Tool Name"),
  410. _("Tool Dia"),
  411. _("Tool Offset"),
  412. _("Custom Offset"),
  413. _("Tool Type"),
  414. _("Tool Shape"),
  415. _("Cut Z"),
  416. _("MultiDepth"),
  417. _("DPP"),
  418. _("V-Dia"),
  419. _("V-Angle"),
  420. _("Travel Z"),
  421. _("FR"),
  422. _("FR Z"),
  423. _("FR Rapids"),
  424. _("Spindle Speed"),
  425. _("Dwell"),
  426. _("Dwelltime"),
  427. _("Preprocessor"),
  428. _("ExtraCut"),
  429. _("Toolchange"),
  430. _("Toolchange XY"),
  431. _("Toolchange Z"),
  432. _("Start Z"),
  433. _("End Z"),
  434. ]
  435. )
  436. self.table_widget.horizontalHeaderItem(0).setToolTip(
  437. _("Tool Index."))
  438. self.table_widget.horizontalHeaderItem(1).setToolTip(
  439. _("Tool name.\n"
  440. "This is not used in the app, it's function\n"
  441. "is to serve as a note for the user."))
  442. self.table_widget.horizontalHeaderItem(2).setToolTip(
  443. _("Tool Diameter."))
  444. self.table_widget.horizontalHeaderItem(3).setToolTip(
  445. _("Tool Offset.\n"
  446. "Can be of a few types:\n"
  447. "Path = zero offset\n"
  448. "In = offset inside by half of tool diameter\n"
  449. "Out = offset outside by half of tool diameter\n"
  450. "Custom = custom offset using the Custom Offset value"))
  451. self.table_widget.horizontalHeaderItem(4).setToolTip(
  452. _("Custom Offset.\n"
  453. "A value to be used as offset from the current path."))
  454. self.table_widget.horizontalHeaderItem(5).setToolTip(
  455. _("Tool Type.\n"
  456. "Can be:\n"
  457. "Iso = isolation cut\n"
  458. "Rough = rough cut, low feedrate, multiple passes\n"
  459. "Finish = finishing cut, high feedrate"))
  460. self.table_widget.horizontalHeaderItem(6).setToolTip(
  461. _("Tool Shape. \n"
  462. "Can be:\n"
  463. "C1 ... C4 = circular tool with x flutes\n"
  464. "B = ball tip milling tool\n"
  465. "V = v-shape milling tool"))
  466. self.table_widget.horizontalHeaderItem(7).setToolTip(
  467. _("Cutting Depth.\n"
  468. "The depth at which to cut into material."))
  469. self.table_widget.horizontalHeaderItem(8).setToolTip(
  470. _("Multi Depth.\n"
  471. "Selecting this will allow cutting in multiple passes,\n"
  472. "each pass adding a DPP parameter depth."))
  473. self.table_widget.horizontalHeaderItem(9).setToolTip(
  474. _("DPP. Depth per Pass.\n"
  475. "The value used to cut into material on each pass."))
  476. self.table_widget.horizontalHeaderItem(10).setToolTip(
  477. _("V-Dia.\n"
  478. "Diameter of the tip for V-Shape Tools."))
  479. self.table_widget.horizontalHeaderItem(11).setToolTip(
  480. _("V-Agle.\n"
  481. "Angle at the tip for the V-Shape Tools."))
  482. self.table_widget.horizontalHeaderItem(12).setToolTip(
  483. _("Clearance Height.\n"
  484. "Height at which the milling bit will travel between cuts,\n"
  485. "above the surface of the material, avoiding all fixtures."))
  486. self.table_widget.horizontalHeaderItem(13).setToolTip(
  487. _("FR. Feedrate\n"
  488. "The speed on XY plane used while cutting into material."))
  489. self.table_widget.horizontalHeaderItem(14).setToolTip(
  490. _("FR Z. Feedrate Z\n"
  491. "The speed on Z plane."))
  492. self.table_widget.horizontalHeaderItem(15).setToolTip(
  493. _("FR Rapids. Feedrate Rapids\n"
  494. "Speed used while moving as fast as possible.\n"
  495. "This is used only by some devices that can't use\n"
  496. "the G0 g-code command. Mostly 3D printers."))
  497. self.table_widget.horizontalHeaderItem(16).setToolTip(
  498. _("Spindle Speed.\n"
  499. "If it's left empty it will not be used.\n"
  500. "The speed of the spindle in RPM."))
  501. self.table_widget.horizontalHeaderItem(17).setToolTip(
  502. _("Dwell.\n"
  503. "Check this if a delay is needed to allow\n"
  504. "the spindle motor to reach it's set speed."))
  505. self.table_widget.horizontalHeaderItem(18).setToolTip(
  506. _("Dwell Time.\n"
  507. "A delay used to allow the motor spindle reach it's set speed."))
  508. self.table_widget.horizontalHeaderItem(19).setToolTip(
  509. _("Preprocessor.\n"
  510. "A selection of files that will alter the generated G-code\n"
  511. "to fit for a number of use cases."))
  512. self.table_widget.horizontalHeaderItem(20).setToolTip(
  513. _("Extra Cut.\n"
  514. "If checked, after a isolation is finished an extra cut\n"
  515. "will be added where the start and end of isolation meet\n"
  516. "such as that this point is covered by this extra cut to\n"
  517. "ensure a complete isolation."))
  518. self.table_widget.horizontalHeaderItem(21).setToolTip(
  519. _("Toolchange.\n"
  520. "It will create a toolchange event.\n"
  521. "The kind of toolchange is determined by\n"
  522. "the preprocessor file."))
  523. self.table_widget.horizontalHeaderItem(22).setToolTip(
  524. _("Toolchange XY.\n"
  525. "A set of coordinates in the format (x, y).\n"
  526. "Will determine the cartesian position of the point\n"
  527. "where the tool change event take place."))
  528. self.table_widget.horizontalHeaderItem(23).setToolTip(
  529. _("Toolchange Z.\n"
  530. "The position on Z plane where the tool change event take place."))
  531. self.table_widget.horizontalHeaderItem(24).setToolTip(
  532. _("Start Z.\n"
  533. "If it's left empty it will not be used.\n"
  534. "A position on Z plane to move immediately after job start."))
  535. self.table_widget.horizontalHeaderItem(25).setToolTip(
  536. _("End Z.\n"
  537. "A position on Z plane to move immediately after job stop."))
  538. # pal = QtGui.QPalette()
  539. # pal.setColor(QtGui.QPalette.Background, Qt.white)
  540. # New Bookmark
  541. new_vlay = QtWidgets.QVBoxLayout()
  542. layout.addLayout(new_vlay)
  543. # new_tool_lbl = QtWidgets.QLabel('<b>%s</b>' % _("New Tool"))
  544. # new_vlay.addWidget(new_tool_lbl, alignment=QtCore.Qt.AlignBottom)
  545. self.buttons_frame = QtWidgets.QFrame()
  546. self.buttons_frame.setContentsMargins(0, 0, 0, 0)
  547. layout.addWidget(self.buttons_frame)
  548. self.buttons_box = QtWidgets.QHBoxLayout()
  549. self.buttons_box.setContentsMargins(0, 0, 0, 0)
  550. self.buttons_frame.setLayout(self.buttons_box)
  551. self.buttons_frame.show()
  552. add_entry_btn = FCButton(_("Add Tool to Tools DB"))
  553. add_entry_btn.setToolTip(
  554. _("Add a new tool in the Tools Database.\n"
  555. "You can edit it after it is added.")
  556. )
  557. remove_entry_btn = FCButton(_("Remove Tool from Tools DB"))
  558. remove_entry_btn.setToolTip(
  559. _("Remove a selection of tools in the Tools Database.")
  560. )
  561. export_db_btn = FCButton(_("Export Tool DB"))
  562. export_db_btn.setToolTip(
  563. _("Save the Tools Database to a custom text file.")
  564. )
  565. import_db_btn = FCButton(_("Import Tool DB"))
  566. import_db_btn.setToolTip(
  567. _("Load the Tools Database information's from a custom text file.")
  568. )
  569. # button_hlay.addStretch()
  570. self.buttons_box.addWidget(add_entry_btn)
  571. self.buttons_box.addWidget(remove_entry_btn)
  572. self.buttons_box.addWidget(export_db_btn)
  573. self.buttons_box.addWidget(import_db_btn)
  574. # self.buttons_box.addWidget(closebtn)
  575. self.add_tool_from_db = FCButton(_("Add Tool from Tools DB"))
  576. self.add_tool_from_db.setToolTip(
  577. _("Add a new tool in the Tools Table of the\n"
  578. "active Geometry object after selecting a tool\n"
  579. "in the Tools Database.")
  580. )
  581. self.add_tool_from_db.hide()
  582. self.cancel_tool_from_db = FCButton(_("Cancel"))
  583. self.cancel_tool_from_db.hide()
  584. hlay = QtWidgets.QHBoxLayout()
  585. layout.addLayout(hlay)
  586. hlay.addWidget(self.add_tool_from_db)
  587. hlay.addWidget(self.cancel_tool_from_db)
  588. hlay.addStretch()
  589. # ##############################################################################
  590. # ######################## SIGNALS #############################################
  591. # ##############################################################################
  592. add_entry_btn.clicked.connect(self.on_tool_add)
  593. remove_entry_btn.clicked.connect(self.on_tool_delete)
  594. export_db_btn.clicked.connect(self.on_export_tools_db_file)
  595. import_db_btn.clicked.connect(self.on_import_tools_db_file)
  596. # closebtn.clicked.connect(self.accept)
  597. self.add_tool_from_db.clicked.connect(self.on_tool_requested_from_app)
  598. self.cancel_tool_from_db.clicked.connect(self.on_cancel_tool)
  599. self.setup_db_ui()
  600. def setup_db_ui(self):
  601. filename = self.app.data_path + '/tools_db.FlatConfig'
  602. # load the database tools from the file
  603. try:
  604. with open(filename) as f:
  605. tools = f.read()
  606. except IOError:
  607. self.app.log.error("Could not load tools DB file.")
  608. self.app.inform.emit('[ERROR] %s' % _("Could not load Tools DB file."))
  609. return
  610. try:
  611. self.db_tool_dict = json.loads(tools)
  612. except Exception:
  613. e = sys.exc_info()[0]
  614. self.app.log.error(str(e))
  615. self.app.inform.emit('[ERROR] %s' % _("Failed to parse Tools DB file."))
  616. return
  617. self.app.inform.emit('[success] %s: %s' % (_("Loaded FlatCAM Tools DB from"), filename))
  618. self.build_db_ui()
  619. self.table_widget.setupContextMenu()
  620. self.table_widget.addContextMenu(
  621. _("Add to DB"), self.on_tool_add, icon=QtGui.QIcon("share/plus16.png"))
  622. self.table_widget.addContextMenu(
  623. _("Copy from DB"), self.on_tool_copy, icon=QtGui.QIcon("share/copy16.png"))
  624. self.table_widget.addContextMenu(
  625. _("Delete from DB"), self.on_tool_delete, icon=QtGui.QIcon("share/delete32.png"))
  626. def build_db_ui(self):
  627. self.ui_disconnect()
  628. self.table_widget.setRowCount(len(self.db_tool_dict))
  629. nr_crt = 0
  630. for toolid, dict_val in self.db_tool_dict.items():
  631. row = nr_crt
  632. nr_crt += 1
  633. t_name = dict_val['name']
  634. self.add_tool_table_line(row, name=t_name, widget=self.table_widget, tooldict=dict_val)
  635. vertical_header = self.table_widget.verticalHeader()
  636. vertical_header.hide()
  637. horizontal_header = self.table_widget.horizontalHeader()
  638. horizontal_header.setMinimumSectionSize(10)
  639. horizontal_header.setDefaultSectionSize(70)
  640. self.table_widget.setSizeAdjustPolicy(QtWidgets.QAbstractScrollArea.AdjustToContents)
  641. for x in range(27):
  642. self.table_widget.resizeColumnToContents(x)
  643. horizontal_header.setSectionResizeMode(0, QtWidgets.QHeaderView.Fixed)
  644. # horizontal_header.setSectionResizeMode(1, QtWidgets.QHeaderView.Stretch)
  645. # horizontal_header.setSectionResizeMode(13, QtWidgets.QHeaderView.Fixed)
  646. horizontal_header.resizeSection(0, 20)
  647. # horizontal_header.setSectionResizeMode(1, QtWidgets.QHeaderView.ResizeToContents)
  648. # horizontal_header.setSectionResizeMode(2, QtWidgets.QHeaderView.Stretch)
  649. self.ui_connect()
  650. def add_tool_table_line(self, row, name, widget, tooldict):
  651. data = tooldict['data']
  652. nr_crt = row + 1
  653. id_item = QtWidgets.QTableWidgetItem('%d' % int(nr_crt))
  654. # id_item.setFlags(QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
  655. flags = id_item.flags() & ~QtCore.Qt.ItemIsEditable
  656. id_item.setFlags(flags)
  657. widget.setItem(row, 0, id_item) # Tool name/id
  658. tool_name_item = QtWidgets.QTableWidgetItem(name)
  659. widget.setItem(row, 1, tool_name_item)
  660. dia_item = FCDoubleSpinner()
  661. dia_item.set_precision(self.decimals)
  662. dia_item.setSingleStep(0.1)
  663. dia_item.set_range(0.0, 9999.9999)
  664. dia_item.set_value(float(tooldict['tooldia']))
  665. widget.setCellWidget(row, 2, dia_item)
  666. tool_offset_item = FCComboBox()
  667. for item in self.offset_item_options:
  668. tool_offset_item.addItem(item)
  669. tool_offset_item.set_value(tooldict['offset'])
  670. widget.setCellWidget(row, 3, tool_offset_item)
  671. c_offset_item = FCDoubleSpinner()
  672. c_offset_item.set_precision(self.decimals)
  673. c_offset_item.setSingleStep(0.1)
  674. c_offset_item.set_range(-9999.9999, 9999.9999)
  675. c_offset_item.set_value(float(tooldict['offset_value']))
  676. widget.setCellWidget(row, 4, c_offset_item)
  677. tt_item = FCComboBox()
  678. for item in self.type_item_options:
  679. tt_item.addItem(item)
  680. tt_item.set_value(tooldict['type'])
  681. widget.setCellWidget(row, 5, tt_item)
  682. tshape_item = FCComboBox()
  683. for item in self.tool_type_item_options:
  684. tshape_item.addItem(item)
  685. tshape_item.set_value(tooldict['tool_type'])
  686. widget.setCellWidget(row, 6, tshape_item)
  687. cutz_item = FCDoubleSpinner()
  688. cutz_item.set_precision(self.decimals)
  689. cutz_item.setSingleStep(0.1)
  690. if self.app.defaults['global_machinist_setting']:
  691. cutz_item.set_range(-9999.9999, 9999.9999)
  692. else:
  693. cutz_item.set_range(-9999.9999, -0.0000)
  694. cutz_item.set_value(float(data['cutz']))
  695. widget.setCellWidget(row, 7, cutz_item)
  696. multidepth_item = FCCheckBox()
  697. multidepth_item.set_value(data['multidepth'])
  698. widget.setCellWidget(row, 8, multidepth_item)
  699. depth_per_pass_item = FCDoubleSpinner()
  700. depth_per_pass_item.set_precision(self.decimals)
  701. depth_per_pass_item.setSingleStep(0.1)
  702. depth_per_pass_item.set_range(0.0, 9999.9999)
  703. depth_per_pass_item.set_value(float(data['depthperpass']))
  704. widget.setCellWidget(row, 9, depth_per_pass_item)
  705. vtip_dia_item = FCDoubleSpinner()
  706. vtip_dia_item.set_precision(self.decimals)
  707. vtip_dia_item.setSingleStep(0.1)
  708. vtip_dia_item.set_range(0.0, 9999.9999)
  709. vtip_dia_item.set_value(float(data['vtipdia']))
  710. widget.setCellWidget(row, 10, vtip_dia_item)
  711. vtip_angle_item = FCDoubleSpinner()
  712. vtip_angle_item.set_precision(self.decimals)
  713. vtip_angle_item.setSingleStep(0.1)
  714. vtip_angle_item.set_range(-360.0, 360.0)
  715. vtip_angle_item.set_value(float(data['vtipangle']))
  716. widget.setCellWidget(row, 11, vtip_angle_item)
  717. travelz_item = FCDoubleSpinner()
  718. travelz_item.set_precision(self.decimals)
  719. travelz_item.setSingleStep(0.1)
  720. if self.app.defaults['global_machinist_setting']:
  721. travelz_item.set_range(-9999.9999, 9999.9999)
  722. else:
  723. travelz_item.set_range(0.0000, 9999.9999)
  724. travelz_item.set_value(float(data['travelz']))
  725. widget.setCellWidget(row, 12, travelz_item)
  726. fr_item = FCDoubleSpinner()
  727. fr_item.set_precision(self.decimals)
  728. fr_item.set_range(0.0, 9999.9999)
  729. fr_item.set_value(float(data['feedrate']))
  730. widget.setCellWidget(row, 13, fr_item)
  731. frz_item = FCDoubleSpinner()
  732. frz_item.set_precision(self.decimals)
  733. frz_item.set_range(0.0, 9999.9999)
  734. frz_item.set_value(float(data['feedrate_z']))
  735. widget.setCellWidget(row, 14, frz_item)
  736. frrapids_item = FCDoubleSpinner()
  737. frrapids_item.set_precision(self.decimals)
  738. frrapids_item.set_range(0.0, 9999.9999)
  739. frrapids_item.set_value(float(data['feedrate_rapid']))
  740. widget.setCellWidget(row, 15, frrapids_item)
  741. spindlespeed_item = QtWidgets.QTableWidgetItem(str(data['spindlespeed']) if data['spindlespeed'] else '')
  742. widget.setItem(row, 16, spindlespeed_item)
  743. dwell_item = FCCheckBox()
  744. dwell_item.set_value(data['dwell'])
  745. widget.setCellWidget(row, 17, dwell_item)
  746. dwelltime_item = FCDoubleSpinner()
  747. dwelltime_item.set_precision(self.decimals)
  748. dwelltime_item.set_range(0.0, 9999.9999)
  749. dwelltime_item.set_value(float(data['dwelltime']))
  750. widget.setCellWidget(row, 18, dwelltime_item)
  751. pp_item = FCComboBox()
  752. for item in self.app.preprocessors:
  753. pp_item.addItem(item)
  754. pp_item.set_value(data['ppname_g'])
  755. widget.setCellWidget(row, 19, pp_item)
  756. ecut_item = FCCheckBox()
  757. ecut_item.set_value(data['extracut'])
  758. widget.setCellWidget(row, 20, ecut_item)
  759. toolchange_item = FCCheckBox()
  760. toolchange_item.set_value(data['toolchange'])
  761. widget.setCellWidget(row, 21, toolchange_item)
  762. toolchangexy_item = QtWidgets.QTableWidgetItem(str(data['toolchangexy']) if data['toolchangexy'] else '')
  763. widget.setItem(row, 22, toolchangexy_item)
  764. toolchangez_item = FCDoubleSpinner()
  765. toolchangez_item.set_precision(self.decimals)
  766. toolchangez_item.setSingleStep(0.1)
  767. if self.app.defaults['global_machinist_setting']:
  768. toolchangez_item.set_range(-9999.9999, 9999.9999)
  769. else:
  770. toolchangez_item.set_range(0.0000, 9999.9999)
  771. toolchangez_item.set_value(float(data['toolchangez']))
  772. widget.setCellWidget(row, 23, toolchangez_item)
  773. startz_item = QtWidgets.QTableWidgetItem(str(data['startz']) if data['startz'] else '')
  774. widget.setItem(row, 24, startz_item)
  775. endz_item = FCDoubleSpinner()
  776. endz_item.set_precision(self.decimals)
  777. endz_item.setSingleStep(0.1)
  778. if self.app.defaults['global_machinist_setting']:
  779. endz_item.set_range(-9999.9999, 9999.9999)
  780. else:
  781. endz_item.set_range(0.0000, 9999.9999)
  782. endz_item.set_value(float(data['endz']))
  783. widget.setCellWidget(row, 25, endz_item)
  784. def on_tool_add(self):
  785. """
  786. Add a tool in the DB Tool Table
  787. :return: None
  788. """
  789. new_toolid = len(self.db_tool_dict) + 1
  790. dict_elem = dict()
  791. default_data = dict()
  792. default_data.update({
  793. "cutz": float(self.app.defaults["geometry_cutz"]),
  794. "multidepth": self.app.defaults["geometry_multidepth"],
  795. "depthperpass": float(self.app.defaults["geometry_depthperpass"]),
  796. "vtipdia": float(self.app.defaults["geometry_vtipdia"]),
  797. "vtipangle": float(self.app.defaults["geometry_vtipangle"]),
  798. "travelz": float(self.app.defaults["geometry_travelz"]),
  799. "feedrate": float(self.app.defaults["geometry_feedrate"]),
  800. "feedrate_z": float(self.app.defaults["geometry_feedrate_z"]),
  801. "feedrate_rapid": float(self.app.defaults["geometry_feedrate_rapid"]),
  802. "spindlespeed": self.app.defaults["geometry_spindlespeed"],
  803. "dwell": self.app.defaults["geometry_dwell"],
  804. "dwelltime": float(self.app.defaults["geometry_dwelltime"]),
  805. "ppname_g": self.app.defaults["geometry_ppname_g"],
  806. "extracut": self.app.defaults["geometry_extracut"],
  807. "toolchange": self.app.defaults["geometry_toolchange"],
  808. "toolchangexy": self.app.defaults["geometry_toolchangexy"],
  809. "toolchangez": float(self.app.defaults["geometry_toolchangez"]),
  810. "startz": self.app.defaults["geometry_startz"],
  811. "endz": float(self.app.defaults["geometry_endz"])
  812. })
  813. dict_elem['name'] = 'new_tool'
  814. dict_elem['tooldia'] = self.app.defaults["geometry_cnctooldia"]
  815. dict_elem['offset'] = 'Path'
  816. dict_elem['offset_value'] = 0.0
  817. dict_elem['type'] = _('Rough')
  818. dict_elem['tool_type'] = 'C1'
  819. dict_elem['data'] = default_data
  820. self.db_tool_dict.update(
  821. {
  822. new_toolid: deepcopy(dict_elem)
  823. }
  824. )
  825. # add the new entry to the Tools DB table
  826. self.build_db_ui()
  827. self.callback_on_edited()
  828. self.app.inform.emit('[success] %s' % _("Tool added to DB."))
  829. def on_tool_copy(self):
  830. """
  831. Copy a selection of Tools in the Tools DB table
  832. :return:
  833. """
  834. new_tool_id = self.table_widget.rowCount() + 1
  835. for model_index in self.table_widget.selectionModel().selectedRows():
  836. # index = QtCore.QPersistentModelIndex(model_index)
  837. old_tool_id = self.table_widget.item(model_index.row(), 0).text()
  838. new_tool_id += 1
  839. for toolid, dict_val in list(self.db_tool_dict.items()):
  840. if int(old_tool_id) == int(toolid):
  841. self.db_tool_dict.update({
  842. new_tool_id: deepcopy(dict_val)
  843. })
  844. self.build_db_ui()
  845. self.callback_on_edited()
  846. self.app.inform.emit('[success] %s' % _("Tool copied from Tools DB."))
  847. def on_tool_delete(self):
  848. """
  849. Delete a selection of Tools in the Tools DB table
  850. :return:
  851. """
  852. for model_index in self.table_widget.selectionModel().selectedRows():
  853. # index = QtCore.QPersistentModelIndex(model_index)
  854. toolname_to_remove = self.table_widget.item(model_index.row(), 0).text()
  855. for toolid, dict_val in list(self.db_tool_dict.items()):
  856. if int(toolname_to_remove) == int(toolid):
  857. # remove from the storage
  858. self.db_tool_dict.pop(toolid, None)
  859. self.build_db_ui()
  860. self.callback_on_edited()
  861. self.app.inform.emit('[success] %s' % _("Tool removed from Tools DB."))
  862. def on_export_tools_db_file(self):
  863. self.app.report_usage("on_export_tools_db_file")
  864. self.app.log.debug("on_export_tools_db_file()")
  865. date = str(datetime.today()).rpartition('.')[0]
  866. date = ''.join(c for c in date if c not in ':-')
  867. date = date.replace(' ', '_')
  868. filter__ = "Text File (*.TXT);;All Files (*.*)"
  869. filename, _f = QtWidgets.QFileDialog.getSaveFileName(caption=_("Export Tools Database"),
  870. directory='{l_save}/FlatCAM_{n}_{date}'.format(
  871. l_save=str(self.app.get_last_save_folder()),
  872. n=_("Tools_Database"),
  873. date=date),
  874. filter=filter__)
  875. filename = str(filename)
  876. if filename == "":
  877. self.app.inform.emit('[WARNING_NOTCL] %s' % _("FlatCAM Tools DB export cancelled."))
  878. return
  879. else:
  880. try:
  881. f = open(filename, 'w')
  882. f.close()
  883. except PermissionError:
  884. self.app.inform.emit('[WARNING] %s' %
  885. _("Permission denied, saving not possible.\n"
  886. "Most likely another app is holding the file open and not accessible."))
  887. return
  888. except IOError:
  889. self.app.log.debug('Creating a new Tools DB file ...')
  890. f = open(filename, 'w')
  891. f.close()
  892. except Exception:
  893. e = sys.exc_info()[0]
  894. self.app.log.error("Could not load Tools DB file.")
  895. self.app.log.error(str(e))
  896. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Could not load Tools DB file."))
  897. return
  898. # Save update options
  899. try:
  900. # Save Tools DB in a file
  901. try:
  902. with open(filename, "w") as f:
  903. json.dump(self.db_tool_dict, f, default=to_dict, indent=2)
  904. except Exception as e:
  905. self.app.log.debug("App.on_save_tools_db() --> %s" % str(e))
  906. self.inform.emit('[ERROR_NOTCL] %s' % _("Failed to write Tools DB to file."))
  907. return
  908. except Exception:
  909. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Failed to write Tools DB to file."))
  910. return
  911. self.app.inform.emit('[success] %s: %s' % (_("Exported Tools DB to"), filename))
  912. def on_import_tools_db_file(self):
  913. self.app.report_usage("on_import_tools_db_file")
  914. self.app.log.debug("on_import_tools_db_file()")
  915. filter__ = "Text File (*.TXT);;All Files (*.*)"
  916. filename, _f = QtWidgets.QFileDialog.getOpenFileName(caption=_("Import FlatCAM Tools DB"), filter=filter__)
  917. if filename == "":
  918. self.app.inform.emit('[WARNING_NOTCL] %s' % _("FlatCAM Tools DB import cancelled."))
  919. else:
  920. try:
  921. with open(filename) as f:
  922. tools_in_db = f.read()
  923. except IOError:
  924. self.app.log.error("Could not load Tools DB file.")
  925. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Could not load Tools DB file."))
  926. return
  927. try:
  928. self.db_tool_dict = json.loads(tools_in_db)
  929. except Exception:
  930. e = sys.exc_info()[0]
  931. self.app.log.error(str(e))
  932. self.app.inform.emit('[ERROR] %s' % _("Failed to parse Tools DB file."))
  933. return
  934. self.app.inform.emit('[success] %s: %s' % (_("Loaded FlatCAM Tools DB from"), filename))
  935. self.build_db_ui()
  936. self.callback_on_edited()
  937. def on_save_tools_db(self, silent=False):
  938. self.app.log.debug("ToolsDB.on_save_button() --> Saving Tools Database to file.")
  939. filename = self.app.data_path + "/tools_db.FlatConfig"
  940. # Preferences save, update the color of the Tools DB Tab text
  941. for idx in range(self.app.ui.plot_tab_area.count()):
  942. if self.app.ui.plot_tab_area.tabText(idx) == _("Tools Database"):
  943. self.app.ui.plot_tab_area.tabBar.setTabTextColor(idx, QtGui.QColor('black'))
  944. # Save Tools DB in a file
  945. try:
  946. f = open(filename, "w")
  947. json.dump(self.db_tool_dict, f, default=to_dict, indent=2)
  948. f.close()
  949. except Exception as e:
  950. self.app.log.debug("ToolsDB.on_save_tools_db() --> %s" % str(e))
  951. self.app.inform.emit('[ERROR_NOTCL] %s' % _("Failed to write Tools DB to file."))
  952. return
  953. if not silent:
  954. self.app.inform.emit('[success] %s' % _("Saved Tools DB."))
  955. def ui_connect(self):
  956. try:
  957. try:
  958. self.table_widget.itemChanged.disconnect(self.callback_on_edited)
  959. except (TypeError, AttributeError):
  960. pass
  961. self.table_widget.itemChanged.connect(self.callback_on_edited)
  962. except AttributeError:
  963. pass
  964. for row in range(self.table_widget.rowCount()):
  965. for col in range(self.table_widget.columnCount()):
  966. # ComboBox
  967. try:
  968. try:
  969. self.table_widget.cellWidget(row, col).currentIndexChanged.disconnect(self.callback_on_edited)
  970. except (TypeError, AttributeError):
  971. pass
  972. self.table_widget.cellWidget(row, col).currentIndexChanged.connect(self.callback_on_edited)
  973. except AttributeError:
  974. pass
  975. # CheckBox
  976. try:
  977. try:
  978. self.table_widget.cellWidget(row, col).toggled.disconnect(self.callback_on_edited)
  979. except (TypeError, AttributeError):
  980. pass
  981. self.table_widget.cellWidget(row, col).toggled.connect(self.callback_on_edited)
  982. except AttributeError:
  983. pass
  984. # SpinBox, DoubleSpinBox
  985. try:
  986. try:
  987. self.table_widget.cellWidget(row, col).valueChanged.disconnect(self.callback_on_edited)
  988. except (TypeError, AttributeError):
  989. pass
  990. self.table_widget.cellWidget(row, col).valueChanged.connect(self.callback_on_edited)
  991. except AttributeError:
  992. pass
  993. def ui_disconnect(self):
  994. try:
  995. self.table_widget.itemChanged.disconnect(self.callback_on_edited)
  996. except (TypeError, AttributeError):
  997. pass
  998. for row in range(self.table_widget.rowCount()):
  999. for col in range(self.table_widget.columnCount()):
  1000. # ComboBox
  1001. try:
  1002. self.table_widget.cellWidget(row, col).currentIndexChanged.disconnect(self.callback_on_edited)
  1003. except (TypeError, AttributeError):
  1004. pass
  1005. # CheckBox
  1006. try:
  1007. self.table_widget.cellWidget(row, col).toggled.disconnect(self.callback_on_edited)
  1008. except (TypeError, AttributeError):
  1009. pass
  1010. # SpinBox, DoubleSpinBox
  1011. try:
  1012. self.table_widget.cellWidget(row, col).valueChanged.disconnect(self.callback_on_edited)
  1013. except (TypeError, AttributeError):
  1014. pass
  1015. def callback_on_edited(self):
  1016. # update the dictionary storage self.db_tool_dict
  1017. self.db_tool_dict.clear()
  1018. dict_elem = dict()
  1019. default_data = dict()
  1020. for row in range(self.table_widget.rowCount()):
  1021. new_toolid = row + 1
  1022. for col in range(self.table_widget.columnCount()):
  1023. column_header_text = self.table_widget.horizontalHeaderItem(col).text()
  1024. if column_header_text == 'Tool Name':
  1025. dict_elem['name'] = self.table_widget.item(row, col).text()
  1026. elif column_header_text == 'Tool Dia':
  1027. dict_elem['tooldia'] = self.table_widget.cellWidget(row, col).get_value()
  1028. elif column_header_text == 'Tool Offset':
  1029. dict_elem['offset'] = self.table_widget.cellWidget(row, col).get_value()
  1030. elif column_header_text == 'Custom Offset':
  1031. dict_elem['offset_value'] = self.table_widget.cellWidget(row, col).get_value()
  1032. elif column_header_text == 'Tool Type':
  1033. dict_elem['type'] = self.table_widget.cellWidget(row, col).get_value()
  1034. elif column_header_text == 'Tool Shape':
  1035. dict_elem['tool_type'] = self.table_widget.cellWidget(row, col).get_value()
  1036. else:
  1037. if column_header_text == 'Cut Z':
  1038. default_data['cutz'] = self.table_widget.cellWidget(row, col).get_value()
  1039. elif column_header_text == 'MultiDepth':
  1040. default_data['multidepth'] = self.table_widget.cellWidget(row, col).get_value()
  1041. elif column_header_text == 'DPP':
  1042. default_data['depthperpass'] = self.table_widget.cellWidget(row, col).get_value()
  1043. elif column_header_text == 'V-Dia':
  1044. default_data['vtipdia'] = self.table_widget.cellWidget(row, col).get_value()
  1045. elif column_header_text == 'V-Angle':
  1046. default_data['vtipangle'] = self.table_widget.cellWidget(row, col).get_value()
  1047. elif column_header_text == 'Travel Z':
  1048. default_data['travelz'] = self.table_widget.cellWidget(row, col).get_value()
  1049. elif column_header_text == 'FR':
  1050. default_data['feedrate'] = self.table_widget.cellWidget(row, col).get_value()
  1051. elif column_header_text == 'FR Z':
  1052. default_data['feedrate_z'] = self.table_widget.cellWidget(row, col).get_value()
  1053. elif column_header_text == 'FR Rapids':
  1054. default_data['feedrate_rapid'] = self.table_widget.cellWidget(row, col).get_value()
  1055. elif column_header_text == 'Spindle Speed':
  1056. default_data['spindlespeed'] = float(self.table_widget.item(row, col).text()) \
  1057. if self.table_widget.item(row, col).text() is not '' else None
  1058. elif column_header_text == 'Dwell':
  1059. default_data['dwell'] = self.table_widget.cellWidget(row, col).get_value()
  1060. elif column_header_text == 'Dwelltime':
  1061. default_data['dwelltime'] = self.table_widget.cellWidget(row, col).get_value()
  1062. elif column_header_text == 'Preprocessor':
  1063. default_data['ppname_g'] = self.table_widget.cellWidget(row, col).get_value()
  1064. elif column_header_text == 'ExtraCut':
  1065. default_data['extracut'] = self.table_widget.cellWidget(row, col).get_value()
  1066. elif column_header_text == 'Toolchange':
  1067. default_data['toolchange'] = self.table_widget.cellWidget(row, col).get_value()
  1068. elif column_header_text == 'Toolchange XY':
  1069. default_data['toolchangexy'] = self.table_widget.item(row, col).text()
  1070. elif column_header_text == 'Toolchange Z':
  1071. default_data['toolchangez'] = self.table_widget.cellWidget(row, col).get_value()
  1072. elif column_header_text == 'Start Z':
  1073. default_data['startz'] = float(self.table_widget.item(row, col).text()) \
  1074. if self.table_widget.item(row, col).text() is not '' else None
  1075. elif column_header_text == 'End Z':
  1076. default_data['endz'] = self.table_widget.cellWidget(row, col).get_value()
  1077. dict_elem['data'] = default_data
  1078. self.db_tool_dict.update(
  1079. {
  1080. new_toolid: deepcopy(dict_elem)
  1081. }
  1082. )
  1083. self.callback_app()
  1084. def on_tool_requested_from_app(self):
  1085. if not self.table_widget.selectionModel().selectedRows():
  1086. self.app.inform.emit('[WARNING_NOTCL] %s...' % _("No Tool/row selected in the Tools Database table"))
  1087. return
  1088. model_index_list = self.table_widget.selectionModel().selectedRows()
  1089. for model_index in model_index_list:
  1090. selected_row = model_index.row()
  1091. tool_uid = selected_row + 1
  1092. for key in self.db_tool_dict.keys():
  1093. if str(key) == str(tool_uid):
  1094. selected_tool = self.db_tool_dict[key]
  1095. self.on_tool_request(tool=selected_tool)
  1096. def on_cancel_tool(self):
  1097. for idx in range(self.app.ui.plot_tab_area.count()):
  1098. if self.app.ui.plot_tab_area.tabText(idx) == _("Tools Database"):
  1099. wdg = self.app.ui.plot_tab_area.widget(idx)
  1100. wdg.deleteLater()
  1101. self.app.ui.plot_tab_area.removeTab(idx)
  1102. self.app.inform.emit('%s' % _("Cancelled adding tool from DB."))
  1103. def resize_new_tool_table_widget(self, min_size, max_size):
  1104. """
  1105. Resize the table widget responsible for adding new tool in the Tool Database
  1106. :param min_size: passed by rangeChanged signal or the self.new_tool_table_widget.horizontalScrollBar()
  1107. :param max_size: passed by rangeChanged signal or the self.new_tool_table_widget.horizontalScrollBar()
  1108. :return:
  1109. """
  1110. t_height = self.t_height
  1111. if max_size > min_size:
  1112. t_height = self.t_height + self.new_tool_table_widget.verticalScrollBar().height()
  1113. self.new_tool_table_widget.setMaximumHeight(t_height)
  1114. def closeEvent(self, QCloseEvent):
  1115. super().closeEvent(QCloseEvent)