ObjectCollection.py 29 KB

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