ObjectCollection.py 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842
  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 FlatCAMObj import FlatCAMGerber, FlatCAMGeometry, FlatCAMExcellon, FlatCAMCNCjob, FlatCAMDocument, FlatCAMScript
  17. import inspect # TODO: Remove
  18. import FlatCAMApp
  19. import re
  20. import logging
  21. import collections
  22. from copy import deepcopy
  23. from numpy import Inf
  24. import gettext
  25. import FlatCAMTranslation as fcTranslate
  26. import builtins
  27. fcTranslate.apply_language('strings')
  28. if '_' not in builtins.__dict__:
  29. _ = gettext.gettext
  30. log = logging.getLogger('base')
  31. class KeySensitiveListView(QtWidgets.QTreeView):
  32. """
  33. QtGui.QListView extended to emit a signal on key press.
  34. """
  35. def __init__(self, app, parent=None):
  36. super(KeySensitiveListView, self).__init__(parent)
  37. self.setHeaderHidden(True)
  38. self.setEditTriggers(QtWidgets.QTreeView.SelectedClicked)
  39. # self.setRootIsDecorated(False)
  40. # self.setExpandsOnDoubleClick(False)
  41. # Enable dragging and dropping onto the GUI
  42. self.setAcceptDrops(True)
  43. self.filename = ""
  44. self.app = app
  45. keyPressed = QtCore.pyqtSignal(int)
  46. def keyPressEvent(self, event):
  47. # super(KeySensitiveListView, self).keyPressEvent(event)
  48. self.keyPressed.emit(event.key())
  49. def dragEnterEvent(self, event):
  50. if event.mimeData().hasUrls:
  51. event.accept()
  52. else:
  53. event.ignore()
  54. def dragMoveEvent(self, event):
  55. self.setDropIndicatorShown(True)
  56. if event.mimeData().hasUrls:
  57. event.accept()
  58. else:
  59. event.ignore()
  60. def dropEvent(self, event):
  61. drop_indicator = self.dropIndicatorPosition()
  62. m = event.mimeData()
  63. if m.hasUrls:
  64. event.accept()
  65. for url in m.urls():
  66. self.filename = str(url.toLocalFile())
  67. # file drop from outside application
  68. if drop_indicator == QtWidgets.QAbstractItemView.OnItem:
  69. if self.filename == "":
  70. self.app.inform.emit(_("Open cancelled."))
  71. else:
  72. if self.filename.lower().rpartition('.')[-1] in self.app.grb_list:
  73. self.app.worker_task.emit({'fcn': self.app.open_gerber,
  74. 'params': [self.filename]})
  75. else:
  76. event.ignore()
  77. if self.filename.lower().rpartition('.')[-1] in self.app.exc_list:
  78. self.app.worker_task.emit({'fcn': self.app.open_excellon,
  79. 'params': [self.filename]})
  80. else:
  81. event.ignore()
  82. if self.filename.lower().rpartition('.')[-1] in self.app.gcode_list:
  83. self.app.worker_task.emit({'fcn': self.app.open_gcode,
  84. 'params': [self.filename]})
  85. else:
  86. event.ignore()
  87. if self.filename.lower().rpartition('.')[-1] in self.app.svg_list:
  88. object_type = 'geometry'
  89. self.app.worker_task.emit({'fcn': self.app.import_svg,
  90. 'params': [self.filename, object_type, None]})
  91. if self.filename.lower().rpartition('.')[-1] in self.app.dxf_list:
  92. object_type = 'geometry'
  93. self.app.worker_task.emit({'fcn': self.app.import_dxf,
  94. 'params': [self.filename, object_type, None]})
  95. if self.filename.lower().rpartition('.')[-1] in self.app.prj_list:
  96. # self.app.open_project() is not Thread Safe
  97. self.app.open_project(self.filename)
  98. else:
  99. event.ignore()
  100. else:
  101. pass
  102. else:
  103. event.ignore()
  104. class TreeItem(KeySensitiveListView):
  105. """
  106. Item of a tree model
  107. """
  108. def __init__(self, data, icon=None, obj=None, parent_item=None):
  109. super(TreeItem, self).__init__(parent_item)
  110. self.parent_item = parent_item
  111. self.item_data = data # Columns string data
  112. self.icon = icon # Decoration
  113. self.obj = obj # FlatCAMObj
  114. self.child_items = []
  115. if parent_item:
  116. parent_item.append_child(self)
  117. def append_child(self, item):
  118. self.child_items.append(item)
  119. item.set_parent_item(self)
  120. def remove_child(self, item):
  121. child = self.child_items.pop(self.child_items.index(item))
  122. child.obj.clear(True)
  123. child.obj.delete()
  124. del child.obj
  125. del child
  126. def remove_children(self):
  127. for child in self.child_items:
  128. child.obj.clear()
  129. child.obj.delete()
  130. del child.obj
  131. del child
  132. self.child_items = []
  133. def child(self, row):
  134. return self.child_items[row]
  135. def child_count(self):
  136. return len(self.child_items)
  137. def column_count(self):
  138. return len(self.item_data)
  139. def data(self, column):
  140. return self.item_data[column]
  141. def row(self):
  142. return self.parent_item.child_items.index(self)
  143. def set_parent_item(self, parent_item):
  144. self.parent_item = parent_item
  145. def __del__(self):
  146. del self.icon
  147. class ObjectCollection(QtCore.QAbstractItemModel):
  148. """
  149. Object storage and management.
  150. """
  151. groups = [
  152. ("gerber", "Gerber"),
  153. ("excellon", "Excellon"),
  154. ("geometry", "Geometry"),
  155. ("cncjob", "CNC Job"),
  156. ("script", "Scripts"),
  157. ("document", "Document"),
  158. ]
  159. classdict = {
  160. "gerber": FlatCAMGerber,
  161. "excellon": FlatCAMExcellon,
  162. "cncjob": FlatCAMCNCjob,
  163. "geometry": FlatCAMGeometry,
  164. "script": FlatCAMScript,
  165. "document": FlatCAMDocument
  166. }
  167. icon_files = {
  168. "gerber": "share/flatcam_icon16.png",
  169. "excellon": "share/drill16.png",
  170. "cncjob": "share/cnc16.png",
  171. "geometry": "share/geometry16.png",
  172. "script": "share/script_new16.png",
  173. "document": "share/notes16_1.png"
  174. }
  175. root_item = None
  176. # app = None
  177. def __init__(self, app, parent=None):
  178. QtCore.QAbstractItemModel.__init__(self)
  179. self.app = app
  180. # ## Icons for the list view
  181. self.icons = {}
  182. for kind in ObjectCollection.icon_files:
  183. self.icons[kind] = QtGui.QPixmap(ObjectCollection.icon_files[kind])
  184. # Create root tree view item
  185. self.root_item = TreeItem(["root"])
  186. # Create group items
  187. self.group_items = {}
  188. for kind, title in ObjectCollection.groups:
  189. item = TreeItem([title], self.icons[kind])
  190. self.group_items[kind] = item
  191. self.root_item.append_child(item)
  192. # Create test sub-items
  193. # for i in self.root_item.m_child_items:
  194. # print i.data(0)
  195. # i.append_child(TreeItem(["empty"]))
  196. # ## Data # ##
  197. self.checked_indexes = []
  198. # Names of objects that are expected to become available.
  199. # For example, when the creation of a new object will run
  200. # in the background and will complete some time in the
  201. # future. This is a way to reserve the name and to let other
  202. # tasks know that they have to wait until available.
  203. self.promises = set()
  204. # same as above only for objects that are plotted
  205. self.plot_promises = set()
  206. # ## View
  207. self.view = KeySensitiveListView(self.app)
  208. self.view.setModel(self)
  209. self.view.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
  210. self.view.setSelectionMode(QtWidgets.QAbstractItemView.ExtendedSelection)
  211. # self.view.setDragDropMode(QtWidgets.QAbstractItemView.InternalMove)
  212. # self.view.setDragEnabled(True)
  213. # self.view.setAcceptDrops(True)
  214. # self.view.setDropIndicatorShown(True)
  215. settings = QSettings("Open Source", "FlatCAM")
  216. if settings.contains("notebook_font_size"):
  217. fsize = settings.value('notebook_font_size', type=int)
  218. else:
  219. fsize = 12
  220. font = QtGui.QFont()
  221. font.setPixelSize(fsize)
  222. font.setFamily("Seagoe UI")
  223. self.view.setFont(font)
  224. # ## GUI Events
  225. self.view.selectionModel().selectionChanged.connect(self.on_list_selection_change)
  226. # self.view.activated.connect(self.on_item_activated)
  227. self.view.keyPressed.connect(self.app.ui.keyPressEvent)
  228. # self.view.clicked.connect(self.on_mouse_down)
  229. self.view.customContextMenuRequested.connect(self.on_menu_request)
  230. self.click_modifier = None
  231. def promise(self, obj_name):
  232. FlatCAMApp.App.log.debug("Object %s has been promised." % obj_name)
  233. self.promises.add(obj_name)
  234. def has_promises(self):
  235. return len(self.promises) > 0
  236. def plot_promise(self, plot_obj_name):
  237. self.plot_promises.add(plot_obj_name)
  238. def plot_remove_promise(self, plot_obj_name):
  239. if plot_obj_name in self.plot_promises:
  240. self.plot_promises.remove(plot_obj_name)
  241. def has_plot_promises(self):
  242. return len(self.plot_promises) > 0
  243. def on_mouse_down(self, event):
  244. FlatCAMApp.App.log.debug("Mouse button pressed on list")
  245. def on_menu_request(self, pos):
  246. sel = len(self.view.selectedIndexes()) > 0
  247. self.app.ui.menuprojectenable.setEnabled(sel)
  248. self.app.ui.menuprojectdisable.setEnabled(sel)
  249. self.app.ui.menuprojectviewsource.setEnabled(sel)
  250. self.app.ui.menuprojectcopy.setEnabled(sel)
  251. self.app.ui.menuprojectedit.setEnabled(sel)
  252. self.app.ui.menuprojectdelete.setEnabled(sel)
  253. self.app.ui.menuprojectsave.setEnabled(sel)
  254. self.app.ui.menuprojectproperties.setEnabled(sel)
  255. if sel:
  256. self.app.ui.menuprojectgeneratecnc.setVisible(True)
  257. self.app.ui.menuprojectedit.setVisible(True)
  258. self.app.ui.menuprojectsave.setVisible(True)
  259. self.app.ui.menuprojectviewsource.setVisible(True)
  260. for obj in self.get_selected():
  261. if type(obj) != FlatCAMGeometry:
  262. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  263. if type(obj) != FlatCAMGeometry and type(obj) != FlatCAMExcellon and type(obj) != FlatCAMGerber:
  264. self.app.ui.menuprojectedit.setVisible(False)
  265. if type(obj) != FlatCAMGerber and type(obj) != FlatCAMExcellon and type(obj) != FlatCAMCNCjob:
  266. self.app.ui.menuprojectviewsource.setVisible(False)
  267. if type(obj) != FlatCAMGerber and type(obj) != FlatCAMGeometry and type(obj) != FlatCAMExcellon and \
  268. type(obj) != FlatCAMCNCjob:
  269. # meaning for Scripts and for Document type of FlatCAM object
  270. self.app.ui.menuprojectenable.setVisible(False)
  271. self.app.ui.menuprojectdisable.setVisible(False)
  272. self.app.ui.menuprojectedit.setVisible(False)
  273. self.app.ui.menuprojectproperties.setVisible(False)
  274. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  275. else:
  276. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  277. self.app.ui.menuproject.popup(self.view.mapToGlobal(pos))
  278. def index(self, row, column=0, parent=None, *args, **kwargs):
  279. if not self.hasIndex(row, column, parent):
  280. return QtCore.QModelIndex()
  281. # if not parent.isValid():
  282. # parent_item = self.root_item
  283. # else:
  284. # parent_item = parent.internalPointer()
  285. parent_item = parent.internalPointer() if parent.isValid() else self.root_item
  286. child_item = parent_item.child(row)
  287. if child_item:
  288. return self.createIndex(row, column, child_item)
  289. else:
  290. return QtCore.QModelIndex()
  291. def parent(self, index=None):
  292. if not index.isValid():
  293. return QtCore.QModelIndex()
  294. parent_item = index.internalPointer().parent_item
  295. if parent_item == self.root_item:
  296. return QtCore.QModelIndex()
  297. return self.createIndex(parent_item.row(), 0, parent_item)
  298. def rowCount(self, index=None, *args, **kwargs):
  299. if index.column() > 0:
  300. return 0
  301. if not index.isValid():
  302. parent_item = self.root_item
  303. else:
  304. parent_item = index.internalPointer()
  305. return parent_item.child_count()
  306. def columnCount(self, index=None, *args, **kwargs):
  307. if index.isValid():
  308. return index.internalPointer().column_count()
  309. else:
  310. return self.root_item.column_count()
  311. def data(self, index, role=None):
  312. if not index.isValid():
  313. return None
  314. if role in [Qt.DisplayRole, Qt.EditRole]:
  315. obj = index.internalPointer().obj
  316. if obj:
  317. return obj.options["name"]
  318. else:
  319. return index.internalPointer().data(index.column())
  320. if role == Qt.ForegroundRole:
  321. color = QColor(self.app.defaults['global_proj_item_color'])
  322. color_disabled = QColor(self.app.defaults['global_proj_item_dis_color'])
  323. obj = index.internalPointer().obj
  324. if obj:
  325. return QtGui.QBrush(color) if obj.options["plot"] else QtGui.QBrush(color_disabled)
  326. else:
  327. return index.internalPointer().data(index.column())
  328. elif role == Qt.DecorationRole:
  329. icon = index.internalPointer().icon
  330. if icon:
  331. return icon
  332. else:
  333. return QtGui.QPixmap()
  334. else:
  335. return None
  336. def setData(self, index, data, role=None):
  337. if index.isValid():
  338. obj = index.internalPointer().obj
  339. if obj:
  340. old_name = deepcopy(obj.options['name'])
  341. new_name = str(data)
  342. if old_name != new_name and new_name != '':
  343. # rename the object
  344. obj.options["name"] = deepcopy(data)
  345. self.app.object_status_changed.emit(obj, 'rename', old_name)
  346. # update the SHELL auto-completer model data
  347. try:
  348. self.app.myKeywords.remove(old_name)
  349. self.app.myKeywords.append(new_name)
  350. self.app.shell._edit.set_model_data(self.app.myKeywords)
  351. except Exception as e:
  352. log.debug(
  353. "setData() --> Could not remove the old object name from auto-completer model list. %s" %
  354. str(e))
  355. # obj.build_ui()
  356. self.app.inform.emit(_("Object renamed from <b>{old}</b> to <b>{new}</b>").format(old=old_name,
  357. new=new_name))
  358. return True
  359. def supportedDropActions(self):
  360. return Qt.MoveAction
  361. def flags(self, index):
  362. default_flags = QtCore.QAbstractItemModel.flags(self, index)
  363. if not index.isValid():
  364. return Qt.ItemIsEnabled | default_flags
  365. # Prevent groups from selection
  366. if not index.internalPointer().obj:
  367. return Qt.ItemIsEnabled
  368. else:
  369. return Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsEditable | \
  370. Qt.ItemIsDragEnabled | Qt.ItemIsDropEnabled
  371. # return QtWidgets.QAbstractItemModel.flags(self, index)
  372. def append(self, obj, active=False):
  373. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + " --> OC.append()")
  374. name = obj.options["name"]
  375. # Check promises and clear if exists
  376. if name in self.promises:
  377. self.promises.remove(name)
  378. # FlatCAMApp.App.log.debug("Promised object %s became available." % name)
  379. # FlatCAMApp.App.log.debug("%d promised objects remaining." % len(self.promises))
  380. # Prevent same name
  381. while name in self.get_names():
  382. # ## Create a new name
  383. # Ends with number?
  384. FlatCAMApp.App.log.debug("new_object(): Object name (%s) exists, changing." % name)
  385. match = re.search(r'(.*[^\d])?(\d+)$', name)
  386. if match: # Yes: Increment the number!
  387. base = match.group(1) or ''
  388. num = int(match.group(2))
  389. name = base + str(num + 1)
  390. else: # No: add a number!
  391. name += "_1"
  392. obj.options["name"] = name
  393. obj.set_ui(obj.ui_type())
  394. # Required before appending (Qt MVC)
  395. group = self.group_items[obj.kind]
  396. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  397. self.beginInsertRows(group_index, group.child_count(), group.child_count())
  398. # Append new item
  399. obj.item = TreeItem(None, self.icons[obj.kind], obj, group)
  400. # Required after appending (Qt MVC)
  401. self.endInsertRows()
  402. # Expand group
  403. if group.child_count() is 1:
  404. self.view.setExpanded(group_index, True)
  405. self.app.should_we_save = True
  406. self.app.object_status_changed.emit(obj, 'append', name)
  407. # decide if to show or hide the Notebook side of the screen
  408. if self.app.defaults["global_project_autohide"] is True:
  409. # always open the notebook on object added to collection
  410. self.app.ui.splitter.setSizes([1, 1])
  411. def get_names(self):
  412. """
  413. Gets a list of the names of all objects in the collection.
  414. :return: List of names.
  415. :rtype: list
  416. """
  417. # FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + " --> OC.get_names()")
  418. return [x.options['name'] for x in self.get_list()]
  419. def get_bounds(self):
  420. """
  421. Finds coordinates bounding all objects in the collection.
  422. :return: [xmin, ymin, xmax, ymax]
  423. :rtype: list
  424. """
  425. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + "--> OC.get_bounds()")
  426. # TODO: Move the operation out of here.
  427. xmin = Inf
  428. ymin = Inf
  429. xmax = -Inf
  430. ymax = -Inf
  431. # for obj in self.object_list:
  432. for obj in self.get_list():
  433. try:
  434. gxmin, gymin, gxmax, gymax = obj.bounds()
  435. xmin = min([xmin, gxmin])
  436. ymin = min([ymin, gymin])
  437. xmax = max([xmax, gxmax])
  438. ymax = max([ymax, gymax])
  439. except Exception as e:
  440. FlatCAMApp.App.log.warning("DEV WARNING: Tried to get bounds of empty geometry. %s" % str(e))
  441. return [xmin, ymin, xmax, ymax]
  442. def get_by_name(self, name, isCaseSensitive=None):
  443. """
  444. Fetches the FlatCAMObj with the given `name`.
  445. :param name: The name of the object.
  446. :type name: str
  447. :param isCaseSensitive: whether searching of the object is done by name where the name is case sensitive
  448. :return: The requested object or None if no such object.
  449. :rtype: FlatCAMObj or None
  450. """
  451. # FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + "--> OC.get_by_name()")
  452. if isCaseSensitive is None or isCaseSensitive is True:
  453. for obj in self.get_list():
  454. if obj.options['name'] == name:
  455. return obj
  456. else:
  457. for obj in self.get_list():
  458. if obj.options['name'].lower() == name.lower():
  459. return obj
  460. return None
  461. def delete_active(self, select_project=True):
  462. selections = self.view.selectedIndexes()
  463. if len(selections) == 0:
  464. return
  465. active = selections[0].internalPointer()
  466. group = active.parent_item
  467. # send signal with the object that is deleted
  468. # self.app.object_status_changed.emit(active.obj, 'delete')
  469. # some objects add a Tab on creation, close it here
  470. for idx in range(self.app.ui.plot_tab_area.count()):
  471. if self.app.ui.plot_tab_area.widget(idx).objectName() == active.obj.options['name']:
  472. self.app.ui.plot_tab_area.removeTab(idx)
  473. break
  474. # update the SHELL auto-completer model data
  475. name = active.obj.options['name']
  476. try:
  477. self.app.myKeywords.remove(name)
  478. self.app.shell._edit.set_model_data(self.app.myKeywords)
  479. # this is not needed any more because now the code editor is created on demand
  480. # self.app.ui.code_editor.set_model_data(self.app.myKeywords)
  481. except Exception as e:
  482. log.debug(
  483. "delete_active() --> Could not remove the old object name from auto-completer model list. %s" % str(e))
  484. self.app.object_status_changed.emit(active.obj, 'delete', name)
  485. # ############ OBJECT DELETION FROM MODEL STARTS HERE ####################
  486. self.beginRemoveRows(self.index(group.row(), 0, QtCore.QModelIndex()), active.row(), active.row())
  487. group.remove_child(active)
  488. # after deletion of object store the current list of objects into the self.app.all_objects_list
  489. self.app.all_objects_list = self.get_list()
  490. self.endRemoveRows()
  491. # ############ OBJECT DELETION FROM MODEL STOPS HERE ####################
  492. if self.app.is_legacy is False:
  493. self.app.plotcanvas.redraw()
  494. if select_project:
  495. # always go to the Project Tab after object deletion as it may be done with a shortcut key
  496. self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
  497. self.app.should_we_save = True
  498. # decide if to show or hide the Notebook side of the screen
  499. if self.app.defaults["global_project_autohide"] is True:
  500. # hide the notebook if there are no objects in the collection
  501. if not self.get_list():
  502. self.app.ui.splitter.setSizes([0, 1])
  503. def delete_all(self):
  504. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + "--> OC.delete_all()")
  505. self.app.object_status_changed.emit(None, 'delete_all', '')
  506. try:
  507. self.app.all_objects_list.clear()
  508. self.app.geo_editor.clear()
  509. self.app.exc_editor.clear()
  510. self.app.dblsidedtool.reset_fields()
  511. self.app.panelize_tool.reset_fields()
  512. self.app.cutout_tool.reset_fields()
  513. self.app.film_tool.reset_fields()
  514. self.beginResetModel()
  515. self.checked_indexes = []
  516. for group in self.root_item.child_items:
  517. group.remove_children()
  518. self.endResetModel()
  519. self.app.plotcanvas.redraw()
  520. except Exception as e:
  521. log.debug("ObjectCollection.delete_all() --> %s" % str(e))
  522. def get_active(self):
  523. """
  524. Returns the active object or None
  525. :return: FlatCAMObj or None
  526. """
  527. selections = self.view.selectedIndexes()
  528. if len(selections) == 0:
  529. return None
  530. return selections[0].internalPointer().obj
  531. def get_selected(self):
  532. """
  533. Returns list of objects selected in the view.
  534. :return: List of objects
  535. """
  536. return [sel.internalPointer().obj for sel in self.view.selectedIndexes()]
  537. def get_non_selected(self):
  538. """
  539. Returns list of objects non-selected in the view.
  540. :return: List of objects
  541. """
  542. obj_list = self.get_list()
  543. for sel in self.get_selected():
  544. obj_list.remove(sel)
  545. return obj_list
  546. def set_active(self, name):
  547. """
  548. Selects object by name from the project list. This triggers the
  549. list_selection_changed event and call on_list_selection_changed.
  550. :param name: Name of the FlatCAM Object
  551. :return: None
  552. """
  553. try:
  554. obj = self.get_by_name(name)
  555. item = obj.item
  556. group = self.group_items[obj.kind]
  557. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  558. item_index = self.index(item.row(), 0, group_index)
  559. self.view.selectionModel().select(item_index, QtCore.QItemSelectionModel.Select)
  560. except Exception as e:
  561. log.error("[ERROR] Cause: %s" % str(e))
  562. raise
  563. def set_all_active(self):
  564. """
  565. Select all objects from the project list. This triggers the
  566. list_selection_changed event and call on_list_selection_changed.
  567. :return: None
  568. """
  569. for name in self.get_names():
  570. self.set_active(name)
  571. def set_exclusive_active(self, name):
  572. """
  573. Make the object with the name in parameters the only selected object
  574. :param name: name of object to be selected and made the only active object
  575. :return: None
  576. """
  577. self.set_all_inactive()
  578. self.set_active(name)
  579. def set_inactive(self, name):
  580. """
  581. Unselect object by name from the project list. This triggers the
  582. list_selection_changed event and call on_list_selection_changed.
  583. :param name: Name of the FlatCAM Object
  584. :return: None
  585. """
  586. # log.debug("ObjectCollection.set_inactive()")
  587. obj = self.get_by_name(name)
  588. item = obj.item
  589. group = self.group_items[obj.kind]
  590. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  591. item_index = self.index(item.row(), 0, group_index)
  592. self.view.selectionModel().select(item_index, QtCore.QItemSelectionModel.Deselect)
  593. def set_all_inactive(self):
  594. """
  595. Unselect all objects from the project list. This triggers the
  596. list_selection_changed event and call on_list_selection_changed.
  597. :return: None
  598. """
  599. for name in self.get_names():
  600. self.set_inactive(name)
  601. def on_list_selection_change(self, current, previous):
  602. # FlatCAMApp.App.log.debug("on_list_selection_change()")
  603. # FlatCAMApp.App.log.debug("Current: %s, Previous %s" % (str(current), str(previous)))
  604. try:
  605. obj = current.indexes()[0].internalPointer().obj
  606. if obj.kind == 'gerber':
  607. self.app.inform.emit(_('[selected]<span style="color:{color};">{name}</span> selected').format(
  608. color='green', name=str(obj.options['name'])))
  609. elif obj.kind == 'excellon':
  610. self.app.inform.emit(_('[selected]<span style="color:{color};">{name}</span> selected').format(
  611. color='brown', name=str(obj.options['name'])))
  612. elif obj.kind == 'cncjob':
  613. self.app.inform.emit(_('[selected]<span style="color:{color};">{name}</span> selected').format(
  614. color='blue', name=str(obj.options['name'])))
  615. elif obj.kind == 'geometry':
  616. self.app.inform.emit(_('[selected]<span style="color:{color};">{name}</span> selected').format(
  617. color='red', name=str(obj.options['name'])))
  618. elif obj.kind == 'script':
  619. self.app.inform.emit(_('[selected]<span style="color:{color};">{name}</span> selected').format(
  620. color='orange', name=str(obj.options['name'])))
  621. elif obj.kind == 'document':
  622. self.app.inform.emit(_('[selected]<span style="color:{color};">{name}</span> selected').format(
  623. color='darkCyan', name=str(obj.options['name'])))
  624. except IndexError:
  625. # FlatCAMApp.App.log.debug("on_list_selection_change(): Index Error (Nothing selected?)")
  626. self.app.inform.emit('')
  627. try:
  628. self.app.ui.selected_scroll_area.takeWidget()
  629. except Exception as e:
  630. FlatCAMApp.App.log.debug("Nothing to remove. %s" % str(e))
  631. self.app.setup_component_editor()
  632. return
  633. if obj:
  634. obj.build_ui()
  635. def on_item_activated(self, index):
  636. """
  637. Double-click or Enter on item.
  638. :param index: Index of the item in the list.
  639. :return: None
  640. """
  641. a_idx = index.internalPointer().obj
  642. if a_idx is None:
  643. return
  644. else:
  645. try:
  646. a_idx.build_ui()
  647. except Exception as e:
  648. self.app.inform.emit('[ERROR] %s: %s' %
  649. (_("Cause of error"), str(e)))
  650. raise
  651. def get_list(self):
  652. obj_list = []
  653. for group in self.root_item.child_items:
  654. for item in group.child_items:
  655. obj_list.append(item.obj)
  656. return obj_list
  657. def update_view(self):
  658. self.dataChanged.emit(QtCore.QModelIndex(), QtCore.QModelIndex())