ObjectCollection.py 43 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199
  1. # ##########################################################
  2. # FlatCAM: 2D Post-processing for Manufacturing #
  3. # http://flatcam.org #
  4. # Author: Juan Pablo Caram (c) #
  5. # Date: 2/5/2014 #
  6. # MIT Licence #
  7. # ##########################################################
  8. # ##########################################################
  9. # File modified by: Dennis Hayrullin #
  10. # File modified by: Marius Stanciu #
  11. # ##########################################################
  12. from PyQt5 import QtGui, QtCore, QtWidgets
  13. from PyQt5.QtCore import Qt, QSettings
  14. from PyQt5.QtGui import QColor
  15. # from PyQt5.QtCore import QModelIndex
  16. from AppObjects.FlatCAMObj import FlatCAMObj
  17. from AppObjects.FlatCAMCNCJob import CNCJobObject
  18. from AppObjects.FlatCAMDocument import DocumentObject
  19. from AppObjects.FlatCAMExcellon import ExcellonObject
  20. from AppObjects.FlatCAMGeometry import GeometryObject
  21. from AppObjects.FlatCAMGerber import GerberObject
  22. from AppObjects.FlatCAMScript import ScriptObject
  23. import inspect # TODO: Remove
  24. import re
  25. import logging
  26. from copy import deepcopy
  27. from numpy import Inf
  28. import gettext
  29. import AppTranslation as fcTranslate
  30. import builtins
  31. fcTranslate.apply_language('strings')
  32. if '_' not in builtins.__dict__:
  33. _ = gettext.gettext
  34. log = logging.getLogger('base')
  35. class KeySensitiveListView(QtWidgets.QTreeView):
  36. """
  37. QtGui.QListView extended to emit a signal on key press.
  38. """
  39. def __init__(self, app, parent=None):
  40. super(KeySensitiveListView, self).__init__(parent)
  41. self.setHeaderHidden(True)
  42. self.setEditTriggers(QtWidgets.QTreeView.SelectedClicked)
  43. # self.setRootIsDecorated(False)
  44. # self.setExpandsOnDoubleClick(False)
  45. # Enable dragging and dropping onto the AppGUI
  46. self.setAcceptDrops(True)
  47. self.filename = ""
  48. self.app = app
  49. # Enabling Drag and Drop for the items in the Project Tab
  50. # Example: https://github.com/d1vanov/PyQt5-reorderable-list-model/blob/master/reorderable_list_model.py
  51. # https://github.com/jimmykuu/PyQt-PySide-Cookbook/blob/master/tree/drop_indicator.md
  52. # self.setDragEnabled(True)
  53. # self.viewport().setAcceptDrops(True)
  54. # self.setDropIndicatorShown(True)
  55. # self.DragDropMode(QtWidgets.QAbstractItemView.InternalMove)
  56. # self.current_idx = None
  57. # self.current_group = None
  58. # self.dropped_obj = None
  59. keyPressed = QtCore.pyqtSignal(int)
  60. def keyPressEvent(self, event):
  61. # super(KeySensitiveListView, self).keyPressEvent(event)
  62. self.keyPressed.emit(event.key())
  63. def dragEnterEvent(self, event):
  64. # if event.source():
  65. # self.current_idx = self.currentIndex()
  66. # self.current_group = self.model().group_items[self.current_idx.internalPointer().obj.kind]
  67. # self.dropped_obj = self.current_idx.internalPointer().obj
  68. if event.mimeData().hasUrls:
  69. event.accept()
  70. else:
  71. event.ignore()
  72. def dragMoveEvent(self, event):
  73. self.setDropIndicatorShown(True)
  74. if event.mimeData().hasUrls:
  75. event.accept()
  76. else:
  77. event.ignore()
  78. def dropEvent(self, event):
  79. drop_indicator = self.dropIndicatorPosition()
  80. # if event.source():
  81. # new_index = self.indexAt(event.pos())
  82. # new_group = self.model().group_items[new_index.internalPointer().obj.kind]
  83. # if self.current_group == new_group:
  84. #
  85. # # delete it from the model
  86. # deleted_obj_name = self.dropped_obj.options['name']
  87. # self.model().delete_by_name(deleted_obj_name)
  88. #
  89. # # add the object to the new index
  90. # self.model().append(self.dropped_obj, to_index=new_index)
  91. #
  92. # return
  93. m = event.mimeData()
  94. if m.hasUrls:
  95. event.accept()
  96. for url in m.urls():
  97. self.filename = str(url.toLocalFile())
  98. # file drop from outside application
  99. if drop_indicator == QtWidgets.QAbstractItemView.OnItem:
  100. if self.filename == "":
  101. self.app.inform.emit(_("Cancelled."))
  102. else:
  103. if self.filename.lower().rpartition('.')[-1] in self.app.grb_list:
  104. self.app.worker_task.emit({'fcn': self.app.open_gerber,
  105. 'params': [self.filename]})
  106. else:
  107. event.ignore()
  108. if self.filename.lower().rpartition('.')[-1] in self.app.exc_list:
  109. self.app.worker_task.emit({'fcn': self.app.open_excellon,
  110. 'params': [self.filename]})
  111. else:
  112. event.ignore()
  113. if self.filename.lower().rpartition('.')[-1] in self.app.gcode_list:
  114. self.app.worker_task.emit({'fcn': self.app.open_gcode,
  115. 'params': [self.filename]})
  116. else:
  117. event.ignore()
  118. if self.filename.lower().rpartition('.')[-1] in self.app.svg_list:
  119. object_type = 'geometry'
  120. self.app.worker_task.emit({'fcn': self.app.import_svg,
  121. 'params': [self.filename, object_type, None]})
  122. if self.filename.lower().rpartition('.')[-1] in self.app.dxf_list:
  123. object_type = 'geometry'
  124. self.app.worker_task.emit({'fcn': self.app.import_dxf,
  125. 'params': [self.filename, object_type, None]})
  126. if self.filename.lower().rpartition('.')[-1] in self.app.prj_list:
  127. # self.app.open_project() is not Thread Safe
  128. self.app.open_project(self.filename)
  129. else:
  130. event.ignore()
  131. else:
  132. pass
  133. else:
  134. event.ignore()
  135. class TreeItem(KeySensitiveListView):
  136. """
  137. Item of a tree model
  138. """
  139. def __init__(self, data, icon=None, obj=None, parent_item=None):
  140. super(TreeItem, self).__init__(parent_item)
  141. self.parent_item = parent_item
  142. self.item_data = data # Columns string data
  143. self.icon = icon # Decoration
  144. self.obj = obj # FlatCAMObj
  145. self.child_items = []
  146. if parent_item:
  147. parent_item.append_child(self)
  148. def append_child(self, item):
  149. self.child_items.append(item)
  150. item.set_parent_item(self)
  151. def remove_child(self, item):
  152. child = self.child_items.pop(self.child_items.index(item))
  153. child.obj.clear(True)
  154. child.obj.delete()
  155. del child.obj
  156. del child
  157. def remove_children(self):
  158. for child in self.child_items:
  159. child.obj.clear()
  160. child.obj.delete()
  161. del child.obj
  162. del child
  163. self.child_items = []
  164. def child(self, row):
  165. return self.child_items[row]
  166. def child_count(self):
  167. return len(self.child_items)
  168. def column_count(self):
  169. return len(self.item_data)
  170. def data(self, column):
  171. return self.item_data[column]
  172. def row(self):
  173. return self.parent_item.child_items.index(self)
  174. def set_parent_item(self, parent_item):
  175. self.parent_item = parent_item
  176. def __del__(self):
  177. del self.icon
  178. class ObjectCollection(QtCore.QAbstractItemModel):
  179. """
  180. Object storage and management.
  181. """
  182. groups = [
  183. ("gerber", "Gerber"),
  184. ("excellon", "Excellon"),
  185. ("geometry", "Geometry"),
  186. ("cncjob", "CNC Job"),
  187. ("script", "Scripts"),
  188. ("document", "Document"),
  189. ]
  190. classdict = {
  191. "gerber": GerberObject,
  192. "excellon": ExcellonObject,
  193. "cncjob": CNCJobObject,
  194. "geometry": GeometryObject,
  195. "script": ScriptObject,
  196. "document": DocumentObject
  197. }
  198. icon_files = {
  199. "gerber": "assets/resources/flatcam_icon16.png",
  200. "excellon": "assets/resources/drill16.png",
  201. "cncjob": "assets/resources/cnc16.png",
  202. "geometry": "assets/resources/geometry16.png",
  203. "script": "assets/resources/script_new16.png",
  204. "document": "assets/resources/notes16_1.png"
  205. }
  206. # will emit the name of the object that was just selected
  207. item_selected = QtCore.pyqtSignal(str)
  208. update_list_signal = QtCore.pyqtSignal()
  209. root_item = None
  210. # app = None
  211. def __init__(self, app, parent=None):
  212. QtCore.QAbstractItemModel.__init__(self)
  213. self.app = app
  214. # ## Icons for the list view
  215. self.icons = {}
  216. for kind in ObjectCollection.icon_files:
  217. self.icons[kind] = QtGui.QPixmap(
  218. ObjectCollection.icon_files[kind].replace('assets/resources', self.app.resource_location))
  219. # Create root tree view item
  220. self.root_item = TreeItem(["root"])
  221. # Create group items
  222. self.group_items = {}
  223. for kind, title in ObjectCollection.groups:
  224. item = TreeItem([title], self.icons[kind])
  225. self.group_items[kind] = item
  226. self.root_item.append_child(item)
  227. # Create test sub-items
  228. # for i in self.root_item.m_child_items:
  229. # print i.data(0)
  230. # i.append_child(TreeItem(["empty"]))
  231. # ## Data # ##
  232. self.checked_indexes = []
  233. # Names of objects that are expected to become available.
  234. # For example, when the creation of a new object will run
  235. # in the background and will complete some time in the
  236. # future. This is a way to reserve the name and to let other
  237. # tasks know that they have to wait until available.
  238. self.promises = set()
  239. # same as above only for objects that are plotted
  240. self.plot_promises = set()
  241. # ## View
  242. self.view = KeySensitiveListView(self.app)
  243. self.view.setModel(self)
  244. self.view.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
  245. self.view.setSelectionMode(QtWidgets.QAbstractItemView.ExtendedSelection)
  246. # self.view.setDragDropMode(QtWidgets.QAbstractItemView.InternalMove)
  247. # self.view.setDragEnabled(True)
  248. # self.view.setAcceptDrops(True)
  249. # self.view.setDropIndicatorShown(True)
  250. settings = QSettings("Open Source", "FlatCAM")
  251. if settings.contains("notebook_font_size"):
  252. fsize = settings.value('notebook_font_size', type=int)
  253. else:
  254. fsize = 12
  255. font = QtGui.QFont()
  256. font.setPixelSize(fsize)
  257. font.setFamily("Seagoe UI")
  258. self.view.setFont(font)
  259. # ## GUI Events
  260. self.view.selectionModel().selectionChanged.connect(self.on_list_selection_change)
  261. # self.view.activated.connect(self.on_item_activated)
  262. self.view.keyPressed.connect(self.app.ui.keyPressEvent)
  263. # self.view.clicked.connect(self.on_mouse_down)
  264. self.view.customContextMenuRequested.connect(self.on_menu_request)
  265. self.click_modifier = None
  266. self.update_list_signal.connect(self.on_update_list_signal)
  267. self.view.activated.connect(self.on_row_activated)
  268. self.item_selected.connect(self.on_row_selected)
  269. def promise(self, obj_name):
  270. log.debug("Object %s has been promised." % obj_name)
  271. self.promises.add(obj_name)
  272. def has_promises(self):
  273. return len(self.promises) > 0
  274. def plot_promise(self, plot_obj_name):
  275. self.plot_promises.add(plot_obj_name)
  276. def plot_remove_promise(self, plot_obj_name):
  277. if plot_obj_name in self.plot_promises:
  278. self.plot_promises.remove(plot_obj_name)
  279. def has_plot_promises(self):
  280. return len(self.plot_promises) > 0
  281. def on_mouse_down(self, event):
  282. log.debug("Mouse button pressed on list")
  283. def on_menu_request(self, pos):
  284. sel = len(self.view.selectedIndexes()) > 0
  285. self.app.ui.menuprojectenable.setEnabled(sel)
  286. self.app.ui.menuprojectdisable.setEnabled(sel)
  287. self.app.ui.menuprojectcolor.setEnabled(sel)
  288. self.app.ui.menuprojectviewsource.setEnabled(sel)
  289. self.app.ui.menuprojectcopy.setEnabled(sel)
  290. self.app.ui.menuprojectedit.setEnabled(sel)
  291. self.app.ui.menuprojectdelete.setEnabled(sel)
  292. self.app.ui.menuprojectsave.setEnabled(sel)
  293. self.app.ui.menuprojectproperties.setEnabled(sel)
  294. if sel:
  295. self.app.ui.menuprojectgeneratecnc.setVisible(True)
  296. self.app.ui.menuprojectedit.setVisible(True)
  297. self.app.ui.menuprojectsave.setVisible(True)
  298. self.app.ui.menuprojectviewsource.setVisible(True)
  299. self.app.ui.menuprojectcolor.setEnabled(False)
  300. for obj in self.get_selected():
  301. if type(obj) == GerberObject or type(obj) == ExcellonObject:
  302. self.app.ui.menuprojectcolor.setEnabled(True)
  303. if type(obj) != GeometryObject:
  304. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  305. # if type(obj) != GeometryObject and type(obj) != ExcellonObject and type(obj) != GerberObject or \
  306. # type(obj) != CNCJobObject:
  307. # self.app.ui.menuprojectedit.setVisible(False)
  308. if type(obj) != GerberObject and type(obj) != ExcellonObject and type(obj) != CNCJobObject:
  309. self.app.ui.menuprojectviewsource.setVisible(False)
  310. if type(obj) != GerberObject and type(obj) != GeometryObject and type(obj) != ExcellonObject and \
  311. type(obj) != CNCJobObject:
  312. # meaning for Scripts and for Document type of FlatCAM object
  313. self.app.ui.menuprojectenable.setVisible(False)
  314. self.app.ui.menuprojectdisable.setVisible(False)
  315. self.app.ui.menuprojectedit.setVisible(False)
  316. self.app.ui.menuprojectproperties.setVisible(False)
  317. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  318. else:
  319. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  320. self.app.ui.menuproject.popup(self.view.mapToGlobal(pos))
  321. def index(self, row, column=0, parent=None, *args, **kwargs):
  322. if not self.hasIndex(row, column, parent):
  323. return QtCore.QModelIndex()
  324. # if not parent.isValid():
  325. # parent_item = self.root_item
  326. # else:
  327. # parent_item = parent.internalPointer()
  328. parent_item = parent.internalPointer() if parent.isValid() else self.root_item
  329. child_item = parent_item.child(row)
  330. if child_item:
  331. return self.createIndex(row, column, child_item)
  332. else:
  333. return QtCore.QModelIndex()
  334. def parent(self, index=None):
  335. if not index.isValid():
  336. return QtCore.QModelIndex()
  337. parent_item = index.internalPointer().parent_item
  338. if parent_item == self.root_item:
  339. return QtCore.QModelIndex()
  340. return self.createIndex(parent_item.row(), 0, parent_item)
  341. def rowCount(self, index=None, *args, **kwargs):
  342. if index.column() > 0:
  343. return 0
  344. if not index.isValid():
  345. parent_item = self.root_item
  346. else:
  347. parent_item = index.internalPointer()
  348. return parent_item.child_count()
  349. def columnCount(self, index=None, *args, **kwargs):
  350. if index.isValid():
  351. return index.internalPointer().column_count()
  352. else:
  353. return self.root_item.column_count()
  354. def data(self, index, role=None):
  355. if not index.isValid():
  356. return None
  357. if role in [Qt.DisplayRole, Qt.EditRole]:
  358. obj = index.internalPointer().obj
  359. if obj:
  360. return obj.options["name"]
  361. else:
  362. return index.internalPointer().data(index.column())
  363. if role == Qt.ForegroundRole:
  364. color = QColor(self.app.defaults['global_proj_item_color'])
  365. color_disabled = QColor(self.app.defaults['global_proj_item_dis_color'])
  366. obj = index.internalPointer().obj
  367. if obj:
  368. return QtGui.QBrush(color) if obj.options["plot"] else QtGui.QBrush(color_disabled)
  369. else:
  370. return index.internalPointer().data(index.column())
  371. elif role == Qt.DecorationRole:
  372. icon = index.internalPointer().icon
  373. if icon:
  374. return icon
  375. else:
  376. return QtGui.QPixmap()
  377. elif role == Qt.ToolTipRole:
  378. try:
  379. obj = index.internalPointer().obj
  380. except AttributeError:
  381. return None
  382. if obj:
  383. text = obj.options['name']
  384. return text
  385. else:
  386. QtWidgets.QToolTip.hideText()
  387. return None
  388. else:
  389. return None
  390. def setData(self, index, data, role=None):
  391. if index.isValid():
  392. obj = index.internalPointer().obj
  393. if obj:
  394. old_name = deepcopy(obj.options['name'])
  395. new_name = str(data)
  396. if old_name != new_name and new_name != '':
  397. # rename the object
  398. obj.options["name"] = deepcopy(data)
  399. self.app.object_status_changed.emit(obj, 'rename', old_name)
  400. # update the SHELL auto-completer model data
  401. try:
  402. self.app.myKeywords.remove(old_name)
  403. self.app.myKeywords.append(new_name)
  404. self.app.shell._edit.set_model_data(self.app.myKeywords)
  405. except Exception as e:
  406. log.debug(
  407. "setData() --> Could not remove the old object name from auto-completer model list. %s" %
  408. str(e))
  409. # obj.build_ui()
  410. self.app.inform.emit(_("Object renamed from <b>{old}</b> to <b>{new}</b>").format(old=old_name,
  411. new=new_name))
  412. self.dataChanged.emit(index, index)
  413. return True
  414. else:
  415. return False
  416. def supportedDropActions(self):
  417. return Qt.MoveAction
  418. def flags(self, index):
  419. default_flags = QtCore.QAbstractItemModel.flags(self, index)
  420. if not index.isValid():
  421. return Qt.ItemIsEnabled | default_flags
  422. # Prevent groups from selection
  423. try:
  424. if not index.internalPointer().obj:
  425. return Qt.ItemIsEnabled
  426. else:
  427. return Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsEditable | \
  428. Qt.ItemIsDragEnabled | Qt.ItemIsDropEnabled
  429. except AttributeError:
  430. return Qt.ItemIsEnabled
  431. # return QtWidgets.QAbstractItemModel.flags(self, index)
  432. def append(self, obj, active=False, to_index=None):
  433. log.debug(str(inspect.stack()[1][3]) + " --> OC.append()")
  434. name = obj.options["name"]
  435. # Check promises and clear if exists
  436. if name in self.promises:
  437. self.promises.remove(name)
  438. # log.debug("Promised object %s became available." % name)
  439. # log.debug("%d promised objects remaining." % len(self.promises))
  440. # Prevent same name
  441. while name in self.get_names():
  442. # ## Create a new name
  443. # Ends with number?
  444. log.debug("app_obj.new_object(): Object name (%s) exists, changing." % name)
  445. match = re.search(r'(.*[^\d])?(\d+)$', name)
  446. if match: # Yes: Increment the number!
  447. base = match.group(1) or ''
  448. num = int(match.group(2))
  449. name = base + str(num + 1)
  450. else: # No: add a number!
  451. name += "_1"
  452. obj.options["name"] = name
  453. obj.set_ui(obj.ui_type(app=self.app))
  454. # Required before appending (Qt MVC)
  455. group = self.group_items[obj.kind]
  456. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  457. if to_index is None:
  458. self.beginInsertRows(group_index, group.child_count(), group.child_count())
  459. # Append new item
  460. obj.item = TreeItem(None, self.icons[obj.kind], obj, group)
  461. # Required after appending (Qt MVC)
  462. self.endInsertRows()
  463. else:
  464. self.beginInsertRows(group_index, to_index.row()-1, to_index.row()-1)
  465. # Append new item
  466. obj.item = TreeItem(None, self.icons[obj.kind], obj, group)
  467. # Required after appending (Qt MVC)
  468. self.endInsertRows()
  469. # Expand group
  470. if group.child_count() == 1:
  471. self.view.setExpanded(group_index, True)
  472. self.app.should_we_save = True
  473. self.app.object_status_changed.emit(obj, 'append', name)
  474. # decide if to show or hide the Notebook side of the screen
  475. if self.app.defaults["global_project_autohide"] is True:
  476. # always open the notebook on object added to collection
  477. self.app.ui.splitter.setSizes([1, 1])
  478. def get_names(self):
  479. """
  480. Gets a list of the names of all objects in the collection.
  481. :return: List of names.
  482. :rtype: list
  483. """
  484. # log.debug(str(inspect.stack()[1][3]) + " --> OC.get_names()")
  485. return [x.options['name'] for x in self.get_list()]
  486. def get_bounds(self):
  487. """
  488. Finds coordinates bounding all objects in the collection.
  489. :return: [xmin, ymin, xmax, ymax]
  490. :rtype: list
  491. """
  492. log.debug(str(inspect.stack()[1][3]) + "--> OC.get_bounds()")
  493. # TODO: Move the operation out of here.
  494. xmin = Inf
  495. ymin = Inf
  496. xmax = -Inf
  497. ymax = -Inf
  498. # for obj in self.object_list:
  499. for obj in self.get_list():
  500. try:
  501. gxmin, gymin, gxmax, gymax = obj.bounds()
  502. xmin = min([xmin, gxmin])
  503. ymin = min([ymin, gymin])
  504. xmax = max([xmax, gxmax])
  505. ymax = max([ymax, gymax])
  506. except Exception as e:
  507. log.warning("DEV WARNING: Tried to get bounds of empty geometry. %s" % str(e))
  508. return [xmin, ymin, xmax, ymax]
  509. def get_by_name(self, name, isCaseSensitive=None):
  510. """
  511. Fetches the FlatCAMObj with the given `name`.
  512. :param name: The name of the object.
  513. :type name: str
  514. :param isCaseSensitive: whether searching of the object is done by name where the name is case sensitive
  515. :return: The requested object or None if no such object.
  516. :rtype: FlatCAMObj or None
  517. """
  518. # log.debug(str(inspect.stack()[1][3]) + "--> OC.get_by_name()")
  519. if isCaseSensitive is None or isCaseSensitive is True:
  520. for obj in self.get_list():
  521. if obj.options['name'] == name:
  522. return obj
  523. else:
  524. for obj in self.get_list():
  525. if obj.options['name'].lower() == name.lower():
  526. return obj
  527. return None
  528. def delete_active(self, select_project=True):
  529. selections = self.view.selectedIndexes()
  530. if len(selections) == 0:
  531. return
  532. active = selections[0].internalPointer()
  533. group = active.parent_item
  534. # send signal with the object that is deleted
  535. # self.app.object_status_changed.emit(active.obj, 'delete')
  536. # some objects add a Tab on creation, close it here
  537. for idx in range(self.app.ui.plot_tab_area.count()):
  538. if self.app.ui.plot_tab_area.widget(idx).objectName() == active.obj.options['name']:
  539. self.app.ui.plot_tab_area.removeTab(idx)
  540. break
  541. # update the SHELL auto-completer model data
  542. name = active.obj.options['name']
  543. try:
  544. self.app.myKeywords.remove(name)
  545. self.app.shell._edit.set_model_data(self.app.myKeywords)
  546. # this is not needed any more because now the code editor is created on demand
  547. # self.app.ui.code_editor.set_model_data(self.app.myKeywords)
  548. except Exception as e:
  549. log.debug(
  550. "delete_active() --> Could not remove the old object name from auto-completer model list. %s" % str(e))
  551. self.app.object_status_changed.emit(active.obj, 'delete', name)
  552. # ############ OBJECT DELETION FROM MODEL STARTS HERE ####################
  553. self.beginRemoveRows(self.index(group.row(), 0, QtCore.QModelIndex()), active.row(), active.row())
  554. group.remove_child(active)
  555. # after deletion of object store the current list of objects into the self.app.all_objects_list
  556. self.app.all_objects_list = self.get_list()
  557. self.endRemoveRows()
  558. # ############ OBJECT DELETION FROM MODEL STOPS HERE ####################
  559. if self.app.is_legacy is False:
  560. self.app.plotcanvas.redraw()
  561. if select_project:
  562. # always go to the Project Tab after object deletion as it may be done with a shortcut key
  563. self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
  564. self.app.should_we_save = True
  565. # decide if to show or hide the Notebook side of the screen
  566. if self.app.defaults["global_project_autohide"] is True:
  567. # hide the notebook if there are no objects in the collection
  568. if not self.get_list():
  569. self.app.ui.splitter.setSizes([0, 1])
  570. def delete_by_name(self, name, select_project=True):
  571. obj = self.get_by_name(name=name)
  572. item = obj.item
  573. group = self.group_items[obj.kind]
  574. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  575. item_index = self.index(item.row(), 0, group_index)
  576. deleted = item_index.internalPointer()
  577. group = deleted.parent_item
  578. # some objects add a Tab on creation, close it here
  579. for idx in range(self.app.ui.plot_tab_area.count()):
  580. if self.app.ui.plot_tab_area.widget(idx).objectName() == deleted.obj.options['name']:
  581. self.app.ui.plot_tab_area.removeTab(idx)
  582. break
  583. # update the SHELL auto-completer model data
  584. name = deleted.obj.options['name']
  585. try:
  586. self.app.myKeywords.remove(name)
  587. self.app.shell._edit.set_model_data(self.app.myKeywords)
  588. # this is not needed any more because now the code editor is created on demand
  589. # self.app.ui.code_editor.set_model_data(self.app.myKeywords)
  590. except Exception as e:
  591. log.debug(
  592. "delete_by_name() --> Could not remove the old object name from auto-completer model list. %s" % str(e))
  593. self.app.object_status_changed.emit(deleted.obj, 'delete', name)
  594. # ############ OBJECT DELETION FROM MODEL STARTS HERE ####################
  595. self.beginRemoveRows(self.index(group.row(), 0, QtCore.QModelIndex()), deleted.row(), deleted.row())
  596. group.remove_child(deleted)
  597. # after deletion of object store the current list of objects into the self.app.all_objects_list
  598. self.update_list_signal.emit()
  599. self.endRemoveRows()
  600. # ############ OBJECT DELETION FROM MODEL STOPS HERE ####################
  601. if self.app.is_legacy is False:
  602. self.app.plotcanvas.redraw()
  603. if select_project:
  604. # always go to the Project Tab after object deletion as it may be done with a shortcut key
  605. self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
  606. self.app.should_we_save = True
  607. # decide if to show or hide the Notebook side of the screen
  608. if self.app.defaults["global_project_autohide"] is True:
  609. # hide the notebook if there are no objects in the collection
  610. if not self.get_list():
  611. self.app.ui.splitter.setSizes([0, 1])
  612. def on_update_list_signal(self):
  613. self.app.all_objects_list = self.get_list()
  614. def delete_all(self):
  615. log.debug(str(inspect.stack()[1][3]) + "--> OC.delete_all()")
  616. self.app.object_status_changed.emit(None, 'delete_all', '')
  617. try:
  618. self.app.all_objects_list.clear()
  619. self.app.geo_editor.clear()
  620. self.app.exc_editor.clear()
  621. self.app.dblsidedtool.reset_fields()
  622. self.app.panelize_tool.reset_fields()
  623. self.app.cutout_tool.reset_fields()
  624. self.app.film_tool.reset_fields()
  625. self.beginResetModel()
  626. self.checked_indexes = []
  627. for group in self.root_item.child_items:
  628. group.remove_children()
  629. self.endResetModel()
  630. self.app.plotcanvas.redraw()
  631. except Exception as e:
  632. log.debug("ObjectCollection.delete_all() --> %s" % str(e))
  633. def get_active(self):
  634. """
  635. Returns the active object or None
  636. :return: FlatCAMObj or None
  637. """
  638. selections = self.view.selectedIndexes()
  639. if len(selections) == 0:
  640. return None
  641. return selections[0].internalPointer().obj
  642. def get_selected(self):
  643. """
  644. Returns list of objects selected in the view.
  645. :return: List of objects
  646. """
  647. return [sel.internalPointer().obj for sel in self.view.selectedIndexes()]
  648. def get_non_selected(self):
  649. """
  650. Returns list of objects non-selected in the view.
  651. :return: List of objects
  652. """
  653. obj_list = self.get_list()
  654. for sel in self.get_selected():
  655. obj_list.remove(sel)
  656. return obj_list
  657. def set_active(self, name):
  658. """
  659. Selects object by name from the project list. This triggers the
  660. list_selection_changed event and call on_list_selection_changed.
  661. :param name: Name of the FlatCAM Object
  662. :return: None
  663. """
  664. try:
  665. obj = self.get_by_name(name)
  666. item = obj.item
  667. group = self.group_items[obj.kind]
  668. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  669. item_index = self.index(item.row(), 0, group_index)
  670. self.view.selectionModel().select(item_index, QtCore.QItemSelectionModel.Select)
  671. except Exception as e:
  672. log.error("[ERROR] Cause: %s" % str(e))
  673. raise
  674. def set_all_active(self):
  675. """
  676. Select all objects from the project list. This triggers the
  677. list_selection_changed event and call on_list_selection_changed.
  678. :return: None
  679. """
  680. for name in self.get_names():
  681. self.set_active(name)
  682. def set_exclusive_active(self, name):
  683. """
  684. Make the object with the name in parameters the only selected object
  685. :param name: name of object to be selected and made the only active object
  686. :return: None
  687. """
  688. self.set_all_inactive()
  689. self.set_active(name)
  690. def set_inactive(self, name):
  691. """
  692. Unselect object by name from the project list. This triggers the
  693. list_selection_changed event and call on_list_selection_changed.
  694. :param name: Name of the FlatCAM Object
  695. :return: None
  696. """
  697. # log.debug("ObjectCollection.set_inactive()")
  698. obj = self.get_by_name(name)
  699. item = obj.item
  700. group = self.group_items[obj.kind]
  701. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  702. item_index = self.index(item.row(), 0, group_index)
  703. self.view.selectionModel().select(item_index, QtCore.QItemSelectionModel.Deselect)
  704. def set_all_inactive(self):
  705. """
  706. Unselect all objects from the project list. This triggers the
  707. list_selection_changed event and call on_list_selection_changed.
  708. :return: None
  709. """
  710. for name in self.get_names():
  711. self.set_inactive(name)
  712. def on_list_selection_change(self, current, previous):
  713. """
  714. :param current: Current selected item
  715. :param previous: Previously selected item
  716. :return:
  717. """
  718. # log.debug("on_list_selection_change()")
  719. # log.debug("Current: %s, Previous %s" % (str(current), str(previous)))
  720. try:
  721. obj = current.indexes()[0].internalPointer().obj
  722. self.item_selected.emit(obj.options['name'])
  723. if obj.kind == 'gerber':
  724. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  725. color='green',
  726. name=str(obj.options['name']),
  727. tx=_("selected"))
  728. )
  729. elif obj.kind == 'excellon':
  730. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  731. color='brown',
  732. name=str(obj.options['name']),
  733. tx=_("selected"))
  734. )
  735. elif obj.kind == 'cncjob':
  736. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  737. color='blue',
  738. name=str(obj.options['name']),
  739. tx=_("selected"))
  740. )
  741. elif obj.kind == 'geometry':
  742. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  743. color='red',
  744. name=str(obj.options['name']),
  745. tx=_("selected"))
  746. )
  747. elif obj.kind == 'script':
  748. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  749. color='orange',
  750. name=str(obj.options['name']),
  751. tx=_("selected"))
  752. )
  753. elif obj.kind == 'document':
  754. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  755. color='darkCyan',
  756. name=str(obj.options['name']),
  757. tx=_("selected"))
  758. )
  759. except IndexError:
  760. self.item_selected.emit('none')
  761. # log.debug("on_list_selection_change(): Index Error (Nothing selected?)")
  762. self.app.inform.emit('')
  763. try:
  764. self.app.ui.selected_scroll_area.takeWidget()
  765. except Exception as e:
  766. log.debug("Nothing to remove. %s" % str(e))
  767. self.app.setup_component_editor()
  768. return
  769. if obj:
  770. obj.build_ui()
  771. def on_item_activated(self, index):
  772. """
  773. Double-click or Enter on item.
  774. :param index: Index of the item in the list.
  775. :return: None
  776. """
  777. a_idx = index.internalPointer().obj
  778. if a_idx is None:
  779. return
  780. else:
  781. try:
  782. a_idx.build_ui()
  783. except Exception as e:
  784. self.app.inform.emit('[ERROR] %s: %s' % (_("Cause of error"), str(e)))
  785. raise
  786. def get_list(self):
  787. """
  788. Will return a list of all objects currently opened. Except FlatCAMScript and FlatCAMDocuments
  789. :return:
  790. """
  791. obj_list = []
  792. for group in self.root_item.child_items:
  793. for item in group.child_items:
  794. obj_list.append(item.obj)
  795. return obj_list
  796. def update_view(self):
  797. self.dataChanged.emit(QtCore.QModelIndex(), QtCore.QModelIndex())
  798. def on_row_activated(self, index):
  799. if index.isValid():
  800. if index.internalPointer().parent_item != self.root_item:
  801. self.app.ui.notebook.setCurrentWidget(self.app.ui.selected_tab)
  802. self.on_item_activated(index)
  803. def on_row_selected(self, obj_name):
  804. """
  805. This is a special string; when received it will make all Menu -> Objects entries unchecked
  806. It mean we clicked outside of the items and deselected all
  807. :param obj_name:
  808. :return:
  809. """
  810. if obj_name == 'none':
  811. for act in self.app.ui.menuobjects.actions():
  812. act.setChecked(False)
  813. return
  814. # get the name of the selected objects and add them to a list
  815. name_list = []
  816. for obj in self.get_selected():
  817. name_list.append(obj.options['name'])
  818. # set all actions as unchecked but the ones selected make them checked
  819. for act in self.app.ui.menuobjects.actions():
  820. act.setChecked(False)
  821. if act.text() in name_list:
  822. act.setChecked(True)
  823. def on_collection_updated(self, obj, state, old_name):
  824. """
  825. Create a menu from the object loaded in the collection.
  826. :param obj: object that was changed (added, deleted, renamed)
  827. :param state: what was done with the object. Can be: added, deleted, delete_all, renamed
  828. :param old_name: the old name of the object before the action that triggered this slot happened
  829. :return: None
  830. """
  831. icon_files = {
  832. "gerber": self.app.resource_location + "/flatcam_icon16.png",
  833. "excellon": self.app.resource_location + "/drill16.png",
  834. "cncjob": self.app.resource_location + "/cnc16.png",
  835. "geometry": self.app.resource_location + "/geometry16.png",
  836. "script": self.app.resource_location + "/script_new16.png",
  837. "document": self.app.resource_location + "/notes16_1.png"
  838. }
  839. if state == 'append':
  840. for act in self.app.ui.menuobjects.actions():
  841. try:
  842. act.triggered.disconnect()
  843. except TypeError:
  844. pass
  845. self.app.ui.menuobjects.clear()
  846. gerber_list = []
  847. exc_list = []
  848. cncjob_list = []
  849. geo_list = []
  850. script_list = []
  851. doc_list = []
  852. for name in self.get_names():
  853. obj_named = self.get_by_name(name)
  854. if obj_named.kind == 'gerber':
  855. gerber_list.append(name)
  856. elif obj_named.kind == 'excellon':
  857. exc_list.append(name)
  858. elif obj_named.kind == 'cncjob':
  859. cncjob_list.append(name)
  860. elif obj_named.kind == 'geometry':
  861. geo_list.append(name)
  862. elif obj_named.kind == 'script':
  863. script_list.append(name)
  864. elif obj_named.kind == 'document':
  865. doc_list.append(name)
  866. def add_act(o_name):
  867. obj_for_icon = self.get_by_name(o_name)
  868. add_action = QtWidgets.QAction(parent=self.app.ui.menuobjects)
  869. add_action.setCheckable(True)
  870. add_action.setText(o_name)
  871. add_action.setIcon(QtGui.QIcon(icon_files[obj_for_icon.kind]))
  872. add_action.triggered.connect(
  873. lambda: self.set_active(o_name) if add_action.isChecked() is True else
  874. self.set_inactive(o_name))
  875. self.app.ui.menuobjects.addAction(add_action)
  876. for name in gerber_list:
  877. add_act(name)
  878. self.app.ui.menuobjects.addSeparator()
  879. for name in exc_list:
  880. add_act(name)
  881. self.app.ui.menuobjects.addSeparator()
  882. for name in cncjob_list:
  883. add_act(name)
  884. self.app.ui.menuobjects.addSeparator()
  885. for name in geo_list:
  886. add_act(name)
  887. self.app.ui.menuobjects.addSeparator()
  888. for name in script_list:
  889. add_act(name)
  890. self.app.ui.menuobjects.addSeparator()
  891. for name in doc_list:
  892. add_act(name)
  893. self.app.ui.menuobjects.addSeparator()
  894. self.app.ui.menuobjects_selall = self.app.ui.menuobjects.addAction(
  895. QtGui.QIcon(self.app.resource_location + '/select_all.png'),
  896. _('Select All')
  897. )
  898. self.app.ui.menuobjects_unselall = self.app.ui.menuobjects.addAction(
  899. QtGui.QIcon(self.app.resource_location + '/deselect_all32.png'),
  900. _('Deselect All')
  901. )
  902. self.app.ui.menuobjects_selall.triggered.connect(lambda: self.on_objects_selection(True))
  903. self.app.ui.menuobjects_unselall.triggered.connect(lambda: self.on_objects_selection(False))
  904. elif state == 'delete':
  905. for act in self.app.ui.menuobjects.actions():
  906. if act.text() == obj.options['name']:
  907. try:
  908. act.triggered.disconnect()
  909. except TypeError:
  910. pass
  911. self.app.ui.menuobjects.removeAction(act)
  912. break
  913. elif state == 'rename':
  914. for act in self.app.ui.menuobjects.actions():
  915. if act.text() == old_name:
  916. add_action = QtWidgets.QAction(parent=self.app.ui.menuobjects)
  917. add_action.setText(obj.options['name'])
  918. add_action.setIcon(QtGui.QIcon(icon_files[obj.kind]))
  919. add_action.triggered.connect(
  920. lambda: self.set_active(obj.options['name']) if add_action.isChecked() is True else
  921. self.set_inactive(obj.options['name']))
  922. self.app.ui.menuobjects.insertAction(act, add_action)
  923. try:
  924. act.triggered.disconnect()
  925. except TypeError:
  926. pass
  927. self.app.ui.menuobjects.removeAction(act)
  928. break
  929. elif state == 'delete_all':
  930. for act in self.app.ui.menuobjects.actions():
  931. try:
  932. act.triggered.disconnect()
  933. except TypeError:
  934. pass
  935. self.app.ui.menuobjects.clear()
  936. self.app.ui.menuobjects.addSeparator()
  937. self.app.ui.menuobjects_selall = self.app.ui.menuobjects.addAction(
  938. QtGui.QIcon(self.app.resource_location + '/select_all.png'),
  939. _('Select All')
  940. )
  941. self.app.ui.menuobjects_unselall = self.app.ui.menuobjects.addAction(
  942. QtGui.QIcon(self.app.resource_location + '/deselect_all32.png'),
  943. _('Deselect All')
  944. )
  945. self.app.ui.menuobjects_selall.triggered.connect(lambda: self.on_objects_selection(True))
  946. self.app.ui.menuobjects_unselall.triggered.connect(lambda: self.on_objects_selection(False))
  947. def on_objects_selection(self, on_off):
  948. obj_list = self.get_names()
  949. if on_off is True:
  950. self.set_all_active()
  951. for act in self.app.ui.menuobjects.actions():
  952. try:
  953. act.setChecked(True)
  954. except Exception:
  955. pass
  956. if obj_list:
  957. self.app.inform.emit('[selected] %s' % _("All objects are selected."))
  958. else:
  959. self.set_all_inactive()
  960. for act in self.app.ui.menuobjects.actions():
  961. try:
  962. act.setChecked(False)
  963. except Exception:
  964. pass
  965. if obj_list:
  966. self.app.inform.emit('%s' % _("Objects selection is cleared."))
  967. else:
  968. self.app.inform.emit('')