ObjectCollection.py 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865
  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. # will emit the name of the object that was just selected
  176. item_selected = QtCore.pyqtSignal(str)
  177. root_item = None
  178. # app = None
  179. def __init__(self, app, parent=None):
  180. QtCore.QAbstractItemModel.__init__(self)
  181. self.app = app
  182. # ## Icons for the list view
  183. self.icons = {}
  184. for kind in ObjectCollection.icon_files:
  185. self.icons[kind] = QtGui.QPixmap(
  186. ObjectCollection.icon_files[kind].replace('share', self.app.resource_location))
  187. # Create root tree view item
  188. self.root_item = TreeItem(["root"])
  189. # Create group items
  190. self.group_items = {}
  191. for kind, title in ObjectCollection.groups:
  192. item = TreeItem([title], self.icons[kind])
  193. self.group_items[kind] = item
  194. self.root_item.append_child(item)
  195. # Create test sub-items
  196. # for i in self.root_item.m_child_items:
  197. # print i.data(0)
  198. # i.append_child(TreeItem(["empty"]))
  199. # ## Data # ##
  200. self.checked_indexes = []
  201. # Names of objects that are expected to become available.
  202. # For example, when the creation of a new object will run
  203. # in the background and will complete some time in the
  204. # future. This is a way to reserve the name and to let other
  205. # tasks know that they have to wait until available.
  206. self.promises = set()
  207. # same as above only for objects that are plotted
  208. self.plot_promises = set()
  209. # ## View
  210. self.view = KeySensitiveListView(self.app)
  211. self.view.setModel(self)
  212. self.view.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
  213. self.view.setSelectionMode(QtWidgets.QAbstractItemView.ExtendedSelection)
  214. # self.view.setDragDropMode(QtWidgets.QAbstractItemView.InternalMove)
  215. # self.view.setDragEnabled(True)
  216. # self.view.setAcceptDrops(True)
  217. # self.view.setDropIndicatorShown(True)
  218. settings = QSettings("Open Source", "FlatCAM")
  219. if settings.contains("notebook_font_size"):
  220. fsize = settings.value('notebook_font_size', type=int)
  221. else:
  222. fsize = 12
  223. font = QtGui.QFont()
  224. font.setPixelSize(fsize)
  225. font.setFamily("Seagoe UI")
  226. self.view.setFont(font)
  227. # ## GUI Events
  228. self.view.selectionModel().selectionChanged.connect(self.on_list_selection_change)
  229. # self.view.activated.connect(self.on_item_activated)
  230. self.view.keyPressed.connect(self.app.ui.keyPressEvent)
  231. # self.view.clicked.connect(self.on_mouse_down)
  232. self.view.customContextMenuRequested.connect(self.on_menu_request)
  233. self.click_modifier = None
  234. def promise(self, obj_name):
  235. FlatCAMApp.App.log.debug("Object %s has been promised." % obj_name)
  236. self.promises.add(obj_name)
  237. def has_promises(self):
  238. return len(self.promises) > 0
  239. def plot_promise(self, plot_obj_name):
  240. self.plot_promises.add(plot_obj_name)
  241. def plot_remove_promise(self, plot_obj_name):
  242. if plot_obj_name in self.plot_promises:
  243. self.plot_promises.remove(plot_obj_name)
  244. def has_plot_promises(self):
  245. return len(self.plot_promises) > 0
  246. def on_mouse_down(self, event):
  247. FlatCAMApp.App.log.debug("Mouse button pressed on list")
  248. def on_menu_request(self, pos):
  249. sel = len(self.view.selectedIndexes()) > 0
  250. self.app.ui.menuprojectenable.setEnabled(sel)
  251. self.app.ui.menuprojectdisable.setEnabled(sel)
  252. self.app.ui.menuprojectviewsource.setEnabled(sel)
  253. self.app.ui.menuprojectcopy.setEnabled(sel)
  254. self.app.ui.menuprojectedit.setEnabled(sel)
  255. self.app.ui.menuprojectdelete.setEnabled(sel)
  256. self.app.ui.menuprojectsave.setEnabled(sel)
  257. self.app.ui.menuprojectproperties.setEnabled(sel)
  258. if sel:
  259. self.app.ui.menuprojectgeneratecnc.setVisible(True)
  260. self.app.ui.menuprojectedit.setVisible(True)
  261. self.app.ui.menuprojectsave.setVisible(True)
  262. self.app.ui.menuprojectviewsource.setVisible(True)
  263. for obj in self.get_selected():
  264. if type(obj) != FlatCAMGeometry:
  265. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  266. if type(obj) != FlatCAMGeometry and type(obj) != FlatCAMExcellon and type(obj) != FlatCAMGerber:
  267. self.app.ui.menuprojectedit.setVisible(False)
  268. if type(obj) != FlatCAMGerber and type(obj) != FlatCAMExcellon and type(obj) != FlatCAMCNCjob:
  269. self.app.ui.menuprojectviewsource.setVisible(False)
  270. if type(obj) != FlatCAMGerber and type(obj) != FlatCAMGeometry and type(obj) != FlatCAMExcellon and \
  271. type(obj) != FlatCAMCNCjob:
  272. # meaning for Scripts and for Document type of FlatCAM object
  273. self.app.ui.menuprojectenable.setVisible(False)
  274. self.app.ui.menuprojectdisable.setVisible(False)
  275. self.app.ui.menuprojectedit.setVisible(False)
  276. self.app.ui.menuprojectproperties.setVisible(False)
  277. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  278. else:
  279. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  280. self.app.ui.menuproject.popup(self.view.mapToGlobal(pos))
  281. def index(self, row, column=0, parent=None, *args, **kwargs):
  282. if not self.hasIndex(row, column, parent):
  283. return QtCore.QModelIndex()
  284. # if not parent.isValid():
  285. # parent_item = self.root_item
  286. # else:
  287. # parent_item = parent.internalPointer()
  288. parent_item = parent.internalPointer() if parent.isValid() else self.root_item
  289. child_item = parent_item.child(row)
  290. if child_item:
  291. return self.createIndex(row, column, child_item)
  292. else:
  293. return QtCore.QModelIndex()
  294. def parent(self, index=None):
  295. if not index.isValid():
  296. return QtCore.QModelIndex()
  297. parent_item = index.internalPointer().parent_item
  298. if parent_item == self.root_item:
  299. return QtCore.QModelIndex()
  300. return self.createIndex(parent_item.row(), 0, parent_item)
  301. def rowCount(self, index=None, *args, **kwargs):
  302. if index.column() > 0:
  303. return 0
  304. if not index.isValid():
  305. parent_item = self.root_item
  306. else:
  307. parent_item = index.internalPointer()
  308. return parent_item.child_count()
  309. def columnCount(self, index=None, *args, **kwargs):
  310. if index.isValid():
  311. return index.internalPointer().column_count()
  312. else:
  313. return self.root_item.column_count()
  314. def data(self, index, role=None):
  315. if not index.isValid():
  316. return None
  317. if role in [Qt.DisplayRole, Qt.EditRole]:
  318. obj = index.internalPointer().obj
  319. if obj:
  320. return obj.options["name"]
  321. else:
  322. return index.internalPointer().data(index.column())
  323. if role == Qt.ForegroundRole:
  324. color = QColor(self.app.defaults['global_proj_item_color'])
  325. color_disabled = QColor(self.app.defaults['global_proj_item_dis_color'])
  326. obj = index.internalPointer().obj
  327. if obj:
  328. return QtGui.QBrush(color) if obj.options["plot"] else QtGui.QBrush(color_disabled)
  329. else:
  330. return index.internalPointer().data(index.column())
  331. elif role == Qt.DecorationRole:
  332. icon = index.internalPointer().icon
  333. if icon:
  334. return icon
  335. else:
  336. return QtGui.QPixmap()
  337. else:
  338. return None
  339. def setData(self, index, data, role=None):
  340. if index.isValid():
  341. obj = index.internalPointer().obj
  342. if obj:
  343. old_name = deepcopy(obj.options['name'])
  344. new_name = str(data)
  345. if old_name != new_name and new_name != '':
  346. # rename the object
  347. obj.options["name"] = deepcopy(data)
  348. self.app.object_status_changed.emit(obj, 'rename', old_name)
  349. # update the SHELL auto-completer model data
  350. try:
  351. self.app.myKeywords.remove(old_name)
  352. self.app.myKeywords.append(new_name)
  353. self.app.shell._edit.set_model_data(self.app.myKeywords)
  354. except Exception as e:
  355. log.debug(
  356. "setData() --> Could not remove the old object name from auto-completer model list. %s" %
  357. str(e))
  358. # obj.build_ui()
  359. self.app.inform.emit(_("Object renamed from <b>{old}</b> to <b>{new}</b>").format(old=old_name,
  360. new=new_name))
  361. return True
  362. def supportedDropActions(self):
  363. return Qt.MoveAction
  364. def flags(self, index):
  365. default_flags = QtCore.QAbstractItemModel.flags(self, index)
  366. if not index.isValid():
  367. return Qt.ItemIsEnabled | default_flags
  368. # Prevent groups from selection
  369. if not index.internalPointer().obj:
  370. return Qt.ItemIsEnabled
  371. else:
  372. return Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsEditable | \
  373. Qt.ItemIsDragEnabled | Qt.ItemIsDropEnabled
  374. # return QtWidgets.QAbstractItemModel.flags(self, index)
  375. def append(self, obj, active=False):
  376. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + " --> OC.append()")
  377. name = obj.options["name"]
  378. # Check promises and clear if exists
  379. if name in self.promises:
  380. self.promises.remove(name)
  381. # FlatCAMApp.App.log.debug("Promised object %s became available." % name)
  382. # FlatCAMApp.App.log.debug("%d promised objects remaining." % len(self.promises))
  383. # Prevent same name
  384. while name in self.get_names():
  385. # ## Create a new name
  386. # Ends with number?
  387. FlatCAMApp.App.log.debug("new_object(): Object name (%s) exists, changing." % name)
  388. match = re.search(r'(.*[^\d])?(\d+)$', name)
  389. if match: # Yes: Increment the number!
  390. base = match.group(1) or ''
  391. num = int(match.group(2))
  392. name = base + str(num + 1)
  393. else: # No: add a number!
  394. name += "_1"
  395. obj.options["name"] = name
  396. obj.set_ui(obj.ui_type(decimals=self.app.decimals))
  397. # Required before appending (Qt MVC)
  398. group = self.group_items[obj.kind]
  399. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  400. self.beginInsertRows(group_index, group.child_count(), group.child_count())
  401. # Append new item
  402. obj.item = TreeItem(None, self.icons[obj.kind], obj, group)
  403. # Required after appending (Qt MVC)
  404. self.endInsertRows()
  405. # Expand group
  406. if group.child_count() is 1:
  407. self.view.setExpanded(group_index, True)
  408. self.app.should_we_save = True
  409. self.app.object_status_changed.emit(obj, 'append', name)
  410. # decide if to show or hide the Notebook side of the screen
  411. if self.app.defaults["global_project_autohide"] is True:
  412. # always open the notebook on object added to collection
  413. self.app.ui.splitter.setSizes([1, 1])
  414. def get_names(self):
  415. """
  416. Gets a list of the names of all objects in the collection.
  417. :return: List of names.
  418. :rtype: list
  419. """
  420. # FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + " --> OC.get_names()")
  421. return [x.options['name'] for x in self.get_list()]
  422. def get_bounds(self):
  423. """
  424. Finds coordinates bounding all objects in the collection.
  425. :return: [xmin, ymin, xmax, ymax]
  426. :rtype: list
  427. """
  428. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + "--> OC.get_bounds()")
  429. # TODO: Move the operation out of here.
  430. xmin = Inf
  431. ymin = Inf
  432. xmax = -Inf
  433. ymax = -Inf
  434. # for obj in self.object_list:
  435. for obj in self.get_list():
  436. try:
  437. gxmin, gymin, gxmax, gymax = obj.bounds()
  438. xmin = min([xmin, gxmin])
  439. ymin = min([ymin, gymin])
  440. xmax = max([xmax, gxmax])
  441. ymax = max([ymax, gymax])
  442. except Exception as e:
  443. FlatCAMApp.App.log.warning("DEV WARNING: Tried to get bounds of empty geometry. %s" % str(e))
  444. return [xmin, ymin, xmax, ymax]
  445. def get_by_name(self, name, isCaseSensitive=None):
  446. """
  447. Fetches the FlatCAMObj with the given `name`.
  448. :param name: The name of the object.
  449. :type name: str
  450. :param isCaseSensitive: whether searching of the object is done by name where the name is case sensitive
  451. :return: The requested object or None if no such object.
  452. :rtype: FlatCAMObj or None
  453. """
  454. # FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + "--> OC.get_by_name()")
  455. if isCaseSensitive is None or isCaseSensitive is True:
  456. for obj in self.get_list():
  457. if obj.options['name'] == name:
  458. return obj
  459. else:
  460. for obj in self.get_list():
  461. if obj.options['name'].lower() == name.lower():
  462. return obj
  463. return None
  464. def delete_active(self, select_project=True):
  465. selections = self.view.selectedIndexes()
  466. if len(selections) == 0:
  467. return
  468. active = selections[0].internalPointer()
  469. group = active.parent_item
  470. # send signal with the object that is deleted
  471. # self.app.object_status_changed.emit(active.obj, 'delete')
  472. # some objects add a Tab on creation, close it here
  473. for idx in range(self.app.ui.plot_tab_area.count()):
  474. if self.app.ui.plot_tab_area.widget(idx).objectName() == active.obj.options['name']:
  475. self.app.ui.plot_tab_area.removeTab(idx)
  476. break
  477. # update the SHELL auto-completer model data
  478. name = active.obj.options['name']
  479. try:
  480. self.app.myKeywords.remove(name)
  481. self.app.shell._edit.set_model_data(self.app.myKeywords)
  482. # this is not needed any more because now the code editor is created on demand
  483. # self.app.ui.code_editor.set_model_data(self.app.myKeywords)
  484. except Exception as e:
  485. log.debug(
  486. "delete_active() --> Could not remove the old object name from auto-completer model list. %s" % str(e))
  487. self.app.object_status_changed.emit(active.obj, 'delete', name)
  488. # ############ OBJECT DELETION FROM MODEL STARTS HERE ####################
  489. self.beginRemoveRows(self.index(group.row(), 0, QtCore.QModelIndex()), active.row(), active.row())
  490. group.remove_child(active)
  491. # after deletion of object store the current list of objects into the self.app.all_objects_list
  492. self.app.all_objects_list = self.get_list()
  493. self.endRemoveRows()
  494. # ############ OBJECT DELETION FROM MODEL STOPS HERE ####################
  495. if self.app.is_legacy is False:
  496. self.app.plotcanvas.redraw()
  497. if select_project:
  498. # always go to the Project Tab after object deletion as it may be done with a shortcut key
  499. self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
  500. self.app.should_we_save = True
  501. # decide if to show or hide the Notebook side of the screen
  502. if self.app.defaults["global_project_autohide"] is True:
  503. # hide the notebook if there are no objects in the collection
  504. if not self.get_list():
  505. self.app.ui.splitter.setSizes([0, 1])
  506. def delete_all(self):
  507. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + "--> OC.delete_all()")
  508. self.app.object_status_changed.emit(None, 'delete_all', '')
  509. try:
  510. self.app.all_objects_list.clear()
  511. self.app.geo_editor.clear()
  512. self.app.exc_editor.clear()
  513. self.app.dblsidedtool.reset_fields()
  514. self.app.panelize_tool.reset_fields()
  515. self.app.cutout_tool.reset_fields()
  516. self.app.film_tool.reset_fields()
  517. self.beginResetModel()
  518. self.checked_indexes = []
  519. for group in self.root_item.child_items:
  520. group.remove_children()
  521. self.endResetModel()
  522. self.app.plotcanvas.redraw()
  523. except Exception as e:
  524. log.debug("ObjectCollection.delete_all() --> %s" % str(e))
  525. def get_active(self):
  526. """
  527. Returns the active object or None
  528. :return: FlatCAMObj or None
  529. """
  530. selections = self.view.selectedIndexes()
  531. if len(selections) == 0:
  532. return None
  533. return selections[0].internalPointer().obj
  534. def get_selected(self):
  535. """
  536. Returns list of objects selected in the view.
  537. :return: List of objects
  538. """
  539. return [sel.internalPointer().obj for sel in self.view.selectedIndexes()]
  540. def get_non_selected(self):
  541. """
  542. Returns list of objects non-selected in the view.
  543. :return: List of objects
  544. """
  545. obj_list = self.get_list()
  546. for sel in self.get_selected():
  547. obj_list.remove(sel)
  548. return obj_list
  549. def set_active(self, name):
  550. """
  551. Selects object by name from the project list. This triggers the
  552. list_selection_changed event and call on_list_selection_changed.
  553. :param name: Name of the FlatCAM Object
  554. :return: None
  555. """
  556. try:
  557. obj = self.get_by_name(name)
  558. item = obj.item
  559. group = self.group_items[obj.kind]
  560. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  561. item_index = self.index(item.row(), 0, group_index)
  562. self.view.selectionModel().select(item_index, QtCore.QItemSelectionModel.Select)
  563. except Exception as e:
  564. log.error("[ERROR] Cause: %s" % str(e))
  565. raise
  566. def set_all_active(self):
  567. """
  568. Select all objects from the project list. This triggers the
  569. list_selection_changed event and call on_list_selection_changed.
  570. :return: None
  571. """
  572. for name in self.get_names():
  573. self.set_active(name)
  574. def set_exclusive_active(self, name):
  575. """
  576. Make the object with the name in parameters the only selected object
  577. :param name: name of object to be selected and made the only active object
  578. :return: None
  579. """
  580. self.set_all_inactive()
  581. self.set_active(name)
  582. def set_inactive(self, name):
  583. """
  584. Unselect object by name from the project list. This triggers the
  585. list_selection_changed event and call on_list_selection_changed.
  586. :param name: Name of the FlatCAM Object
  587. :return: None
  588. """
  589. # log.debug("ObjectCollection.set_inactive()")
  590. obj = self.get_by_name(name)
  591. item = obj.item
  592. group = self.group_items[obj.kind]
  593. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  594. item_index = self.index(item.row(), 0, group_index)
  595. self.view.selectionModel().select(item_index, QtCore.QItemSelectionModel.Deselect)
  596. def set_all_inactive(self):
  597. """
  598. Unselect all objects from the project list. This triggers the
  599. list_selection_changed event and call on_list_selection_changed.
  600. :return: None
  601. """
  602. for name in self.get_names():
  603. self.set_inactive(name)
  604. def on_list_selection_change(self, current, previous):
  605. # FlatCAMApp.App.log.debug("on_list_selection_change()")
  606. # FlatCAMApp.App.log.debug("Current: %s, Previous %s" % (str(current), str(previous)))
  607. try:
  608. obj = current.indexes()[0].internalPointer().obj
  609. self.item_selected.emit(obj.options['name'])
  610. if obj.kind == 'gerber':
  611. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  612. color='green',
  613. name=str(obj.options['name']),
  614. tx=_("selected"))
  615. )
  616. elif obj.kind == 'excellon':
  617. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  618. color='brown',
  619. name=str(obj.options['name']),
  620. tx=_("selected"))
  621. )
  622. elif obj.kind == 'cncjob':
  623. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  624. color='blue',
  625. name=str(obj.options['name']),
  626. tx=_("selected"))
  627. )
  628. elif obj.kind == 'geometry':
  629. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  630. color='red',
  631. name=str(obj.options['name']),
  632. tx=_("selected"))
  633. )
  634. elif obj.kind == 'script':
  635. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  636. color='orange',
  637. name=str(obj.options['name']),
  638. tx=_("selected"))
  639. )
  640. elif obj.kind == 'document':
  641. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  642. color='darkCyan',
  643. name=str(obj.options['name']),
  644. tx=_("selected"))
  645. )
  646. except IndexError:
  647. self.item_selected.emit('none')
  648. # FlatCAMApp.App.log.debug("on_list_selection_change(): Index Error (Nothing selected?)")
  649. self.app.inform.emit('')
  650. try:
  651. self.app.ui.selected_scroll_area.takeWidget()
  652. except Exception as e:
  653. FlatCAMApp.App.log.debug("Nothing to remove. %s" % str(e))
  654. self.app.setup_component_editor()
  655. return
  656. if obj:
  657. obj.build_ui()
  658. def on_item_activated(self, index):
  659. """
  660. Double-click or Enter on item.
  661. :param index: Index of the item in the list.
  662. :return: None
  663. """
  664. a_idx = index.internalPointer().obj
  665. if a_idx is None:
  666. return
  667. else:
  668. try:
  669. a_idx.build_ui()
  670. except Exception as e:
  671. self.app.inform.emit('[ERROR] %s: %s' % (_("Cause of error"), str(e)))
  672. raise
  673. def get_list(self):
  674. obj_list = []
  675. for group in self.root_item.child_items:
  676. for item in group.child_items:
  677. obj_list.append(item.obj)
  678. return obj_list
  679. def update_view(self):
  680. self.dataChanged.emit(QtCore.QModelIndex(), QtCore.QModelIndex())