ObjectCollection.py 35 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000
  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 flatcamObjects.FlatCAMObj import FlatCAMObj
  17. from flatcamObjects.FlatCAMCNCJob import CNCJobObject
  18. from flatcamObjects.FlatCAMDocument import DocumentObject
  19. from flatcamObjects.FlatCAMExcellon import ExcellonObject
  20. from flatcamObjects.FlatCAMGeometry import GeometryObject
  21. from flatcamObjects.FlatCAMGerber import GerberObject
  22. from flatcamObjects.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 FlatCAMTranslation 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 GUI
  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. def promise(self, obj_name):
  268. log.debug("Object %s has been promised." % obj_name)
  269. self.promises.add(obj_name)
  270. def has_promises(self):
  271. return len(self.promises) > 0
  272. def plot_promise(self, plot_obj_name):
  273. self.plot_promises.add(plot_obj_name)
  274. def plot_remove_promise(self, plot_obj_name):
  275. if plot_obj_name in self.plot_promises:
  276. self.plot_promises.remove(plot_obj_name)
  277. def has_plot_promises(self):
  278. return len(self.plot_promises) > 0
  279. def on_mouse_down(self, event):
  280. log.debug("Mouse button pressed on list")
  281. def on_menu_request(self, pos):
  282. sel = len(self.view.selectedIndexes()) > 0
  283. self.app.ui.menuprojectenable.setEnabled(sel)
  284. self.app.ui.menuprojectdisable.setEnabled(sel)
  285. self.app.ui.menuprojectcolor.setEnabled(sel)
  286. self.app.ui.menuprojectviewsource.setEnabled(sel)
  287. self.app.ui.menuprojectcopy.setEnabled(sel)
  288. self.app.ui.menuprojectedit.setEnabled(sel)
  289. self.app.ui.menuprojectdelete.setEnabled(sel)
  290. self.app.ui.menuprojectsave.setEnabled(sel)
  291. self.app.ui.menuprojectproperties.setEnabled(sel)
  292. if sel:
  293. self.app.ui.menuprojectgeneratecnc.setVisible(True)
  294. self.app.ui.menuprojectedit.setVisible(True)
  295. self.app.ui.menuprojectsave.setVisible(True)
  296. self.app.ui.menuprojectviewsource.setVisible(True)
  297. self.app.ui.menuprojectcolor.setEnabled(False)
  298. for obj in self.get_selected():
  299. if type(obj) == GerberObject or type(obj) == ExcellonObject:
  300. self.app.ui.menuprojectcolor.setEnabled(True)
  301. if type(obj) != GeometryObject:
  302. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  303. if type(obj) != GeometryObject and type(obj) != ExcellonObject and type(obj) != GerberObject:
  304. self.app.ui.menuprojectedit.setVisible(False)
  305. if type(obj) != GerberObject and type(obj) != ExcellonObject and type(obj) != CNCJobObject:
  306. self.app.ui.menuprojectviewsource.setVisible(False)
  307. if type(obj) != GerberObject and type(obj) != GeometryObject and type(obj) != ExcellonObject and \
  308. type(obj) != CNCJobObject:
  309. # meaning for Scripts and for Document type of FlatCAM object
  310. self.app.ui.menuprojectenable.setVisible(False)
  311. self.app.ui.menuprojectdisable.setVisible(False)
  312. self.app.ui.menuprojectedit.setVisible(False)
  313. self.app.ui.menuprojectproperties.setVisible(False)
  314. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  315. else:
  316. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  317. self.app.ui.menuproject.popup(self.view.mapToGlobal(pos))
  318. def index(self, row, column=0, parent=None, *args, **kwargs):
  319. if not self.hasIndex(row, column, parent):
  320. return QtCore.QModelIndex()
  321. # if not parent.isValid():
  322. # parent_item = self.root_item
  323. # else:
  324. # parent_item = parent.internalPointer()
  325. parent_item = parent.internalPointer() if parent.isValid() else self.root_item
  326. child_item = parent_item.child(row)
  327. if child_item:
  328. return self.createIndex(row, column, child_item)
  329. else:
  330. return QtCore.QModelIndex()
  331. def parent(self, index=None):
  332. if not index.isValid():
  333. return QtCore.QModelIndex()
  334. parent_item = index.internalPointer().parent_item
  335. if parent_item == self.root_item:
  336. return QtCore.QModelIndex()
  337. return self.createIndex(parent_item.row(), 0, parent_item)
  338. def rowCount(self, index=None, *args, **kwargs):
  339. if index.column() > 0:
  340. return 0
  341. if not index.isValid():
  342. parent_item = self.root_item
  343. else:
  344. parent_item = index.internalPointer()
  345. return parent_item.child_count()
  346. def columnCount(self, index=None, *args, **kwargs):
  347. if index.isValid():
  348. return index.internalPointer().column_count()
  349. else:
  350. return self.root_item.column_count()
  351. def data(self, index, role=None):
  352. if not index.isValid():
  353. return None
  354. if role in [Qt.DisplayRole, Qt.EditRole]:
  355. obj = index.internalPointer().obj
  356. if obj:
  357. return obj.options["name"]
  358. else:
  359. return index.internalPointer().data(index.column())
  360. if role == Qt.ForegroundRole:
  361. color = QColor(self.app.defaults['global_proj_item_color'])
  362. color_disabled = QColor(self.app.defaults['global_proj_item_dis_color'])
  363. obj = index.internalPointer().obj
  364. if obj:
  365. return QtGui.QBrush(color) if obj.options["plot"] else QtGui.QBrush(color_disabled)
  366. else:
  367. return index.internalPointer().data(index.column())
  368. elif role == Qt.DecorationRole:
  369. icon = index.internalPointer().icon
  370. if icon:
  371. return icon
  372. else:
  373. return QtGui.QPixmap()
  374. elif role == Qt.ToolTipRole:
  375. try:
  376. obj = index.internalPointer().obj
  377. except AttributeError:
  378. return None
  379. if obj:
  380. text = obj.options['name']
  381. return text
  382. else:
  383. QtWidgets.QToolTip.hideText()
  384. return None
  385. else:
  386. return None
  387. def setData(self, index, data, role=None):
  388. if index.isValid():
  389. obj = index.internalPointer().obj
  390. if obj:
  391. old_name = deepcopy(obj.options['name'])
  392. new_name = str(data)
  393. if old_name != new_name and new_name != '':
  394. # rename the object
  395. obj.options["name"] = deepcopy(data)
  396. self.app.object_status_changed.emit(obj, 'rename', old_name)
  397. # update the SHELL auto-completer model data
  398. try:
  399. self.app.myKeywords.remove(old_name)
  400. self.app.myKeywords.append(new_name)
  401. self.app.shell._edit.set_model_data(self.app.myKeywords)
  402. except Exception as e:
  403. log.debug(
  404. "setData() --> Could not remove the old object name from auto-completer model list. %s" %
  405. str(e))
  406. # obj.build_ui()
  407. self.app.inform.emit(_("Object renamed from <b>{old}</b> to <b>{new}</b>").format(old=old_name,
  408. new=new_name))
  409. self.dataChanged.emit(index, index)
  410. return True
  411. else:
  412. return False
  413. def supportedDropActions(self):
  414. return Qt.MoveAction
  415. def flags(self, index):
  416. default_flags = QtCore.QAbstractItemModel.flags(self, index)
  417. if not index.isValid():
  418. return Qt.ItemIsEnabled | default_flags
  419. # Prevent groups from selection
  420. try:
  421. if not index.internalPointer().obj:
  422. return Qt.ItemIsEnabled
  423. else:
  424. return Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsEditable | \
  425. Qt.ItemIsDragEnabled | Qt.ItemIsDropEnabled
  426. except AttributeError:
  427. return Qt.ItemIsEnabled
  428. # return QtWidgets.QAbstractItemModel.flags(self, index)
  429. def append(self, obj, active=False, to_index=None):
  430. log.debug(str(inspect.stack()[1][3]) + " --> OC.append()")
  431. name = obj.options["name"]
  432. # Check promises and clear if exists
  433. if name in self.promises:
  434. self.promises.remove(name)
  435. # log.debug("Promised object %s became available." % name)
  436. # log.debug("%d promised objects remaining." % len(self.promises))
  437. # Prevent same name
  438. while name in self.get_names():
  439. # ## Create a new name
  440. # Ends with number?
  441. log.debug("new_object(): Object name (%s) exists, changing." % name)
  442. match = re.search(r'(.*[^\d])?(\d+)$', name)
  443. if match: # Yes: Increment the number!
  444. base = match.group(1) or ''
  445. num = int(match.group(2))
  446. name = base + str(num + 1)
  447. else: # No: add a number!
  448. name += "_1"
  449. obj.options["name"] = name
  450. obj.set_ui(obj.ui_type(app=self.app))
  451. # Required before appending (Qt MVC)
  452. group = self.group_items[obj.kind]
  453. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  454. if to_index is None:
  455. self.beginInsertRows(group_index, group.child_count(), group.child_count())
  456. # Append new item
  457. obj.item = TreeItem(None, self.icons[obj.kind], obj, group)
  458. # Required after appending (Qt MVC)
  459. self.endInsertRows()
  460. else:
  461. self.beginInsertRows(group_index, to_index.row()-1, to_index.row()-1)
  462. # Append new item
  463. obj.item = TreeItem(None, self.icons[obj.kind], obj, group)
  464. # Required after appending (Qt MVC)
  465. self.endInsertRows()
  466. # Expand group
  467. if group.child_count() == 1:
  468. self.view.setExpanded(group_index, True)
  469. self.app.should_we_save = True
  470. self.app.object_status_changed.emit(obj, 'append', name)
  471. # decide if to show or hide the Notebook side of the screen
  472. if self.app.defaults["global_project_autohide"] is True:
  473. # always open the notebook on object added to collection
  474. self.app.ui.splitter.setSizes([1, 1])
  475. def get_names(self):
  476. """
  477. Gets a list of the names of all objects in the collection.
  478. :return: List of names.
  479. :rtype: list
  480. """
  481. # log.debug(str(inspect.stack()[1][3]) + " --> OC.get_names()")
  482. return [x.options['name'] for x in self.get_list()]
  483. def get_bounds(self):
  484. """
  485. Finds coordinates bounding all objects in the collection.
  486. :return: [xmin, ymin, xmax, ymax]
  487. :rtype: list
  488. """
  489. log.debug(str(inspect.stack()[1][3]) + "--> OC.get_bounds()")
  490. # TODO: Move the operation out of here.
  491. xmin = Inf
  492. ymin = Inf
  493. xmax = -Inf
  494. ymax = -Inf
  495. # for obj in self.object_list:
  496. for obj in self.get_list():
  497. try:
  498. gxmin, gymin, gxmax, gymax = obj.bounds()
  499. xmin = min([xmin, gxmin])
  500. ymin = min([ymin, gymin])
  501. xmax = max([xmax, gxmax])
  502. ymax = max([ymax, gymax])
  503. except Exception as e:
  504. log.warning("DEV WARNING: Tried to get bounds of empty geometry. %s" % str(e))
  505. return [xmin, ymin, xmax, ymax]
  506. def get_by_name(self, name, isCaseSensitive=None):
  507. """
  508. Fetches the FlatCAMObj with the given `name`.
  509. :param name: The name of the object.
  510. :type name: str
  511. :param isCaseSensitive: whether searching of the object is done by name where the name is case sensitive
  512. :return: The requested object or None if no such object.
  513. :rtype: FlatCAMObj or None
  514. """
  515. # log.debug(str(inspect.stack()[1][3]) + "--> OC.get_by_name()")
  516. if isCaseSensitive is None or isCaseSensitive is True:
  517. for obj in self.get_list():
  518. if obj.options['name'] == name:
  519. return obj
  520. else:
  521. for obj in self.get_list():
  522. if obj.options['name'].lower() == name.lower():
  523. return obj
  524. return None
  525. def delete_active(self, select_project=True):
  526. selections = self.view.selectedIndexes()
  527. if len(selections) == 0:
  528. return
  529. active = selections[0].internalPointer()
  530. group = active.parent_item
  531. # send signal with the object that is deleted
  532. # self.app.object_status_changed.emit(active.obj, 'delete')
  533. # some objects add a Tab on creation, close it here
  534. for idx in range(self.app.ui.plot_tab_area.count()):
  535. if self.app.ui.plot_tab_area.widget(idx).objectName() == active.obj.options['name']:
  536. self.app.ui.plot_tab_area.removeTab(idx)
  537. break
  538. # update the SHELL auto-completer model data
  539. name = active.obj.options['name']
  540. try:
  541. self.app.myKeywords.remove(name)
  542. self.app.shell._edit.set_model_data(self.app.myKeywords)
  543. # this is not needed any more because now the code editor is created on demand
  544. # self.app.ui.code_editor.set_model_data(self.app.myKeywords)
  545. except Exception as e:
  546. log.debug(
  547. "delete_active() --> Could not remove the old object name from auto-completer model list. %s" % str(e))
  548. self.app.object_status_changed.emit(active.obj, 'delete', name)
  549. # ############ OBJECT DELETION FROM MODEL STARTS HERE ####################
  550. self.beginRemoveRows(self.index(group.row(), 0, QtCore.QModelIndex()), active.row(), active.row())
  551. group.remove_child(active)
  552. # after deletion of object store the current list of objects into the self.app.all_objects_list
  553. self.app.all_objects_list = self.get_list()
  554. self.endRemoveRows()
  555. # ############ OBJECT DELETION FROM MODEL STOPS HERE ####################
  556. if self.app.is_legacy is False:
  557. self.app.plotcanvas.redraw()
  558. if select_project:
  559. # always go to the Project Tab after object deletion as it may be done with a shortcut key
  560. self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
  561. self.app.should_we_save = True
  562. # decide if to show or hide the Notebook side of the screen
  563. if self.app.defaults["global_project_autohide"] is True:
  564. # hide the notebook if there are no objects in the collection
  565. if not self.get_list():
  566. self.app.ui.splitter.setSizes([0, 1])
  567. def delete_by_name(self, name, select_project=True):
  568. obj = self.get_by_name(name=name)
  569. item = obj.item
  570. group = self.group_items[obj.kind]
  571. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  572. item_index = self.index(item.row(), 0, group_index)
  573. deleted = item_index.internalPointer()
  574. group = deleted.parent_item
  575. # some objects add a Tab on creation, close it here
  576. for idx in range(self.app.ui.plot_tab_area.count()):
  577. if self.app.ui.plot_tab_area.widget(idx).objectName() == deleted.obj.options['name']:
  578. self.app.ui.plot_tab_area.removeTab(idx)
  579. break
  580. # update the SHELL auto-completer model data
  581. name = deleted.obj.options['name']
  582. try:
  583. self.app.myKeywords.remove(name)
  584. self.app.shell._edit.set_model_data(self.app.myKeywords)
  585. # this is not needed any more because now the code editor is created on demand
  586. # self.app.ui.code_editor.set_model_data(self.app.myKeywords)
  587. except Exception as e:
  588. log.debug(
  589. "delete_by_name() --> Could not remove the old object name from auto-completer model list. %s" % str(e))
  590. self.app.object_status_changed.emit(deleted.obj, 'delete', name)
  591. # ############ OBJECT DELETION FROM MODEL STARTS HERE ####################
  592. self.beginRemoveRows(self.index(group.row(), 0, QtCore.QModelIndex()), deleted.row(), deleted.row())
  593. group.remove_child(deleted)
  594. # after deletion of object store the current list of objects into the self.app.all_objects_list
  595. self.update_list_signal.emit()
  596. self.endRemoveRows()
  597. # ############ OBJECT DELETION FROM MODEL STOPS HERE ####################
  598. if self.app.is_legacy is False:
  599. self.app.plotcanvas.redraw()
  600. if select_project:
  601. # always go to the Project Tab after object deletion as it may be done with a shortcut key
  602. self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
  603. self.app.should_we_save = True
  604. # decide if to show or hide the Notebook side of the screen
  605. if self.app.defaults["global_project_autohide"] is True:
  606. # hide the notebook if there are no objects in the collection
  607. if not self.get_list():
  608. self.app.ui.splitter.setSizes([0, 1])
  609. def on_update_list_signal(self):
  610. self.app.all_objects_list = self.get_list()
  611. def delete_all(self):
  612. log.debug(str(inspect.stack()[1][3]) + "--> OC.delete_all()")
  613. self.app.object_status_changed.emit(None, 'delete_all', '')
  614. try:
  615. self.app.all_objects_list.clear()
  616. self.app.geo_editor.clear()
  617. self.app.exc_editor.clear()
  618. self.app.dblsidedtool.reset_fields()
  619. self.app.panelize_tool.reset_fields()
  620. self.app.cutout_tool.reset_fields()
  621. self.app.film_tool.reset_fields()
  622. self.beginResetModel()
  623. self.checked_indexes = []
  624. for group in self.root_item.child_items:
  625. group.remove_children()
  626. self.endResetModel()
  627. self.app.plotcanvas.redraw()
  628. except Exception as e:
  629. log.debug("ObjectCollection.delete_all() --> %s" % str(e))
  630. def get_active(self):
  631. """
  632. Returns the active object or None
  633. :return: FlatCAMObj or None
  634. """
  635. selections = self.view.selectedIndexes()
  636. if len(selections) == 0:
  637. return None
  638. return selections[0].internalPointer().obj
  639. def get_selected(self):
  640. """
  641. Returns list of objects selected in the view.
  642. :return: List of objects
  643. """
  644. return [sel.internalPointer().obj for sel in self.view.selectedIndexes()]
  645. def get_non_selected(self):
  646. """
  647. Returns list of objects non-selected in the view.
  648. :return: List of objects
  649. """
  650. obj_list = self.get_list()
  651. for sel in self.get_selected():
  652. obj_list.remove(sel)
  653. return obj_list
  654. def set_active(self, name):
  655. """
  656. Selects object by name from the project list. This triggers the
  657. list_selection_changed event and call on_list_selection_changed.
  658. :param name: Name of the FlatCAM Object
  659. :return: None
  660. """
  661. try:
  662. obj = self.get_by_name(name)
  663. item = obj.item
  664. group = self.group_items[obj.kind]
  665. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  666. item_index = self.index(item.row(), 0, group_index)
  667. self.view.selectionModel().select(item_index, QtCore.QItemSelectionModel.Select)
  668. except Exception as e:
  669. log.error("[ERROR] Cause: %s" % str(e))
  670. raise
  671. def set_all_active(self):
  672. """
  673. Select all objects from the project list. This triggers the
  674. list_selection_changed event and call on_list_selection_changed.
  675. :return: None
  676. """
  677. for name in self.get_names():
  678. self.set_active(name)
  679. def set_exclusive_active(self, name):
  680. """
  681. Make the object with the name in parameters the only selected object
  682. :param name: name of object to be selected and made the only active object
  683. :return: None
  684. """
  685. self.set_all_inactive()
  686. self.set_active(name)
  687. def set_inactive(self, name):
  688. """
  689. Unselect object by name from the project list. This triggers the
  690. list_selection_changed event and call on_list_selection_changed.
  691. :param name: Name of the FlatCAM Object
  692. :return: None
  693. """
  694. # log.debug("ObjectCollection.set_inactive()")
  695. obj = self.get_by_name(name)
  696. item = obj.item
  697. group = self.group_items[obj.kind]
  698. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  699. item_index = self.index(item.row(), 0, group_index)
  700. self.view.selectionModel().select(item_index, QtCore.QItemSelectionModel.Deselect)
  701. def set_all_inactive(self):
  702. """
  703. Unselect all objects from the project list. This triggers the
  704. list_selection_changed event and call on_list_selection_changed.
  705. :return: None
  706. """
  707. for name in self.get_names():
  708. self.set_inactive(name)
  709. def on_list_selection_change(self, current, previous):
  710. """
  711. :param current: Current selected item
  712. :param previous: Previously selected item
  713. :return:
  714. """
  715. # log.debug("on_list_selection_change()")
  716. # log.debug("Current: %s, Previous %s" % (str(current), str(previous)))
  717. try:
  718. obj = current.indexes()[0].internalPointer().obj
  719. self.item_selected.emit(obj.options['name'])
  720. if obj.kind == 'gerber':
  721. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  722. color='green',
  723. name=str(obj.options['name']),
  724. tx=_("selected"))
  725. )
  726. elif obj.kind == 'excellon':
  727. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  728. color='brown',
  729. name=str(obj.options['name']),
  730. tx=_("selected"))
  731. )
  732. elif obj.kind == 'cncjob':
  733. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  734. color='blue',
  735. name=str(obj.options['name']),
  736. tx=_("selected"))
  737. )
  738. elif obj.kind == 'geometry':
  739. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  740. color='red',
  741. name=str(obj.options['name']),
  742. tx=_("selected"))
  743. )
  744. elif obj.kind == 'script':
  745. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  746. color='orange',
  747. name=str(obj.options['name']),
  748. tx=_("selected"))
  749. )
  750. elif obj.kind == 'document':
  751. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  752. color='darkCyan',
  753. name=str(obj.options['name']),
  754. tx=_("selected"))
  755. )
  756. except IndexError:
  757. self.item_selected.emit('none')
  758. # log.debug("on_list_selection_change(): Index Error (Nothing selected?)")
  759. self.app.inform.emit('')
  760. try:
  761. self.app.ui.selected_scroll_area.takeWidget()
  762. except Exception as e:
  763. log.debug("Nothing to remove. %s" % str(e))
  764. self.app.setup_component_editor()
  765. return
  766. if obj:
  767. obj.build_ui()
  768. def on_item_activated(self, index):
  769. """
  770. Double-click or Enter on item.
  771. :param index: Index of the item in the list.
  772. :return: None
  773. """
  774. a_idx = index.internalPointer().obj
  775. if a_idx is None:
  776. return
  777. else:
  778. try:
  779. a_idx.build_ui()
  780. except Exception as e:
  781. self.app.inform.emit('[ERROR] %s: %s' % (_("Cause of error"), str(e)))
  782. raise
  783. def get_list(self):
  784. """
  785. Will return a list of all objects currently opened.
  786. :return:
  787. """
  788. obj_list = []
  789. for group in self.root_item.child_items:
  790. for item in group.child_items:
  791. obj_list.append(item.obj)
  792. return obj_list
  793. def update_view(self):
  794. self.dataChanged.emit(QtCore.QModelIndex(), QtCore.QModelIndex())