ObjectCollection.py 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894
  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. # from PyQt5.QtCore import QModelIndex
  9. from FlatCAMObj import *
  10. import inspect # TODO: Remove
  11. import FlatCAMApp
  12. from PyQt5 import QtGui, QtCore, QtWidgets
  13. from PyQt5.QtCore import Qt
  14. import webbrowser
  15. class KeySensitiveListView(QtWidgets.QTreeView):
  16. """
  17. QtGui.QListView extended to emit a signal on key press.
  18. """
  19. def __init__(self, app, parent=None):
  20. super(KeySensitiveListView, self).__init__(parent)
  21. self.setHeaderHidden(True)
  22. self.setEditTriggers(QtWidgets.QTreeView.SelectedClicked)
  23. # self.setRootIsDecorated(False)
  24. # self.setExpandsOnDoubleClick(False)
  25. # Enable dragging and dropping onto the GUI
  26. self.setAcceptDrops(True)
  27. self.filename = ""
  28. self.app = app
  29. keyPressed = QtCore.pyqtSignal(int)
  30. def keyPressEvent(self, event):
  31. super(KeySensitiveListView, self).keyPressEvent(event)
  32. self.keyPressed.emit(event.key())
  33. def dragEnterEvent(self, event):
  34. if event.mimeData().hasUrls:
  35. event.accept()
  36. else:
  37. event.ignore()
  38. def dragMoveEvent(self, event):
  39. if event.mimeData().hasUrls:
  40. event.accept()
  41. else:
  42. event.ignore()
  43. def dropEvent(self, event):
  44. if event.mimeData().hasUrls:
  45. event.setDropAction(QtCore.Qt.CopyAction)
  46. event.accept()
  47. for url in event.mimeData().urls():
  48. self.filename = str(url.toLocalFile())
  49. if self.filename == "":
  50. self.app.inform.emit("Open cancelled.")
  51. else:
  52. if self.filename.lower().rpartition('.')[-1] in self.app.grb_list:
  53. self.app.worker_task.emit({'fcn': self.app.open_gerber,
  54. 'params': [self.filename]})
  55. else:
  56. event.ignore()
  57. if self.filename.lower().rpartition('.')[-1] in self.app.exc_list:
  58. self.app.worker_task.emit({'fcn': self.app.open_excellon,
  59. 'params': [self.filename]})
  60. else:
  61. event.ignore()
  62. if self.filename.lower().rpartition('.')[-1] in self.app.gcode_list:
  63. self.app.worker_task.emit({'fcn': self.app.open_gcode,
  64. 'params': [self.filename]})
  65. else:
  66. event.ignore()
  67. if self.filename.lower().rpartition('.')[-1] in self.app.svg_list:
  68. object_type = 'geometry'
  69. self.app.worker_task.emit({'fcn': self.app.import_svg,
  70. 'params': [self.filename, object_type, None]})
  71. if self.filename.lower().rpartition('.')[-1] in self.app.dxf_list:
  72. object_type = 'geometry'
  73. self.app.worker_task.emit({'fcn': self.app.import_dxf,
  74. 'params': [self.filename, object_type, None]})
  75. if self.filename.lower().rpartition('.')[-1] in self.app.prj_list:
  76. # self.app.open_project() is not Thread Safe
  77. self.app.open_project(self.filename)
  78. else:
  79. event.ignore()
  80. else:
  81. event.ignore()
  82. class TreeItem:
  83. """
  84. Item of a tree model
  85. """
  86. def __init__(self, data, icon=None, obj=None, parent_item=None):
  87. self.parent_item = parent_item
  88. self.item_data = data # Columns string data
  89. self.icon = icon # Decoration
  90. self.obj = obj # FlatCAMObj
  91. self.child_items = []
  92. if parent_item:
  93. parent_item.append_child(self)
  94. def append_child(self, item):
  95. self.child_items.append(item)
  96. item.set_parent_item(self)
  97. def remove_child(self, item):
  98. child = self.child_items.pop(self.child_items.index(item))
  99. child.obj.clear(True)
  100. child.obj.delete()
  101. del child.obj
  102. del child
  103. def remove_children(self):
  104. for child in self.child_items:
  105. child.obj.clear()
  106. child.obj.delete()
  107. del child.obj
  108. del child
  109. self.child_items = []
  110. def child(self, row):
  111. return self.child_items[row]
  112. def child_count(self):
  113. return len(self.child_items)
  114. def column_count(self):
  115. return len(self.item_data)
  116. def data(self, column):
  117. return self.item_data[column]
  118. def row(self):
  119. return self.parent_item.child_items.index(self)
  120. def set_parent_item(self, parent_item):
  121. self.parent_item = parent_item
  122. def __del__(self):
  123. del self.icon
  124. class ObjectCollection(QtCore.QAbstractItemModel):
  125. """
  126. Object storage and management.
  127. """
  128. groups = [
  129. ("gerber", "Gerber"),
  130. ("excellon", "Excellon"),
  131. ("geometry", "Geometry"),
  132. ("cncjob", "CNC Job")
  133. ]
  134. classdict = {
  135. "gerber": FlatCAMGerber,
  136. "excellon": FlatCAMExcellon,
  137. "cncjob": FlatCAMCNCjob,
  138. "geometry": FlatCAMGeometry
  139. }
  140. icon_files = {
  141. "gerber": "share/flatcam_icon16.png",
  142. "excellon": "share/drill16.png",
  143. "cncjob": "share/cnc16.png",
  144. "geometry": "share/geometry16.png"
  145. }
  146. root_item = None
  147. # app = None
  148. def __init__(self, app, parent=None):
  149. QtCore.QAbstractItemModel.__init__(self)
  150. ### Icons for the list view
  151. self.icons = {}
  152. for kind in ObjectCollection.icon_files:
  153. self.icons[kind] = QtGui.QPixmap(ObjectCollection.icon_files[kind])
  154. # Create root tree view item
  155. self.root_item = TreeItem(["root"])
  156. # Create group items
  157. self.group_items = {}
  158. for kind, title in ObjectCollection.groups:
  159. item = TreeItem([title], self.icons[kind])
  160. self.group_items[kind] = item
  161. self.root_item.append_child(item)
  162. # Create test sub-items
  163. # for i in self.root_item.m_child_items:
  164. # print i.data(0)
  165. # i.append_child(TreeItem(["empty"]))
  166. ### Data ###
  167. self.checked_indexes = []
  168. # Names of objects that are expected to become available.
  169. # For example, when the creation of a new object will run
  170. # in the background and will complete some time in the
  171. # future. This is a way to reserve the name and to let other
  172. # tasks know that they have to wait until available.
  173. self.promises = set()
  174. ### View
  175. self.view = KeySensitiveListView(app)
  176. self.view.setSelectionMode(QtWidgets.QAbstractItemView.ExtendedSelection)
  177. self.view.setModel(self)
  178. self.view.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
  179. font = QtGui.QFont()
  180. font.setPixelSize(12)
  181. font.setFamily("Seagoe UI")
  182. self.view.setFont(font)
  183. ## GUI Events
  184. self.view.selectionModel().selectionChanged.connect(self.on_list_selection_change)
  185. self.view.activated.connect(self.on_item_activated)
  186. self.view.keyPressed.connect(self.on_key)
  187. self.view.clicked.connect(self.on_mouse_down)
  188. self.view.customContextMenuRequested.connect(self.on_menu_request)
  189. self.click_modifier = None
  190. def promise(self, obj_name):
  191. FlatCAMApp.App.log.debug("Object %s has been promised." % obj_name)
  192. self.promises.add(obj_name)
  193. def has_promises(self):
  194. return len(self.promises) > 0
  195. def on_key(self, key):
  196. modifiers = QtWidgets.QApplication.keyboardModifiers()
  197. active = self.get_active()
  198. selected = self.get_selected()
  199. if modifiers == QtCore.Qt.ControlModifier:
  200. if key == QtCore.Qt.Key_A:
  201. self.app.on_selectall()
  202. if key == QtCore.Qt.Key_C:
  203. self.app.on_copy_object()
  204. if key == QtCore.Qt.Key_E:
  205. self.app.on_fileopenexcellon()
  206. if key == QtCore.Qt.Key_G:
  207. self.app.on_fileopengerber()
  208. if key == QtCore.Qt.Key_N:
  209. self.app.on_file_new_click()
  210. if key == QtCore.Qt.Key_M:
  211. self.app.measurement_tool.run()
  212. if key == QtCore.Qt.Key_O:
  213. self.app.on_file_openproject()
  214. if key == QtCore.Qt.Key_S:
  215. self.app.on_file_saveproject()
  216. return
  217. elif modifiers == QtCore.Qt.ShiftModifier:
  218. # Copy Object Name
  219. # Copy Object Name
  220. if key == QtCore.Qt.Key_C:
  221. self.app.on_copy_name()
  222. # Toggle axis
  223. if key == QtCore.Qt.Key_G:
  224. if self.toggle_axis is False:
  225. self.app.plotcanvas.v_line.set_data(color=(0.70, 0.3, 0.3, 1.0))
  226. self.app.plotcanvas.h_line.set_data(color=(0.70, 0.3, 0.3, 1.0))
  227. self.app.plotcanvas.redraw()
  228. self.app.toggle_axis = True
  229. else:
  230. self.app.plotcanvas.v_line.set_data(color=(0.0, 0.0, 0.0, 0.0))
  231. self.app.plotcanvas.h_line.set_data(color=(0.0, 0.0, 0.0, 0.0))
  232. self.appplotcanvas.redraw()
  233. self.app.toggle_axis = False
  234. # Open Preferences Window
  235. if key == QtCore.Qt.Key_P:
  236. self.app.on_preferences()
  237. return
  238. # Rotate Object by 90 degree CCW
  239. if key == QtCore.Qt.Key_R:
  240. self.app.on_rotate(silent=True, preset=-90)
  241. return
  242. # Run a Script
  243. if key == QtCore.Qt.Key_S:
  244. self.app.on_filerunscript()
  245. return
  246. # Toggle Workspace
  247. if key == QtCore.Qt.Key_W:
  248. self.app.on_workspace_menu()
  249. return
  250. # Skew on X axis
  251. if key == QtCore.Qt.Key_X:
  252. self.app.on_skewx()
  253. return
  254. # Skew on Y axis
  255. if key == QtCore.Qt.Key_Y:
  256. self.app.on_skewy()
  257. return
  258. elif modifiers == QtCore.Qt.AltModifier:
  259. # 2-Sided PCB Tool
  260. if key == QtCore.Qt.Key_D:
  261. self.app.dblsidedtool.run()
  262. return
  263. # Non-Copper Clear Tool
  264. if key == QtCore.Qt.Key_N:
  265. self.app.ncclear_tool.run()
  266. return
  267. # Transformation Tool
  268. if key == QtCore.Qt.Key_R:
  269. self.app.transform_tool.run()
  270. return
  271. # Cutout Tool
  272. if key == QtCore.Qt.Key_U:
  273. self.app.cutout_tool.run()
  274. return
  275. else:
  276. # Open Manual
  277. if key == QtCore.Qt.Key_F1:
  278. webbrowser.open(self.app.manual_url)
  279. # Open Video Help
  280. if key == QtCore.Qt.Key_F2:
  281. webbrowser.open(self.app.video_url)
  282. # Zoom Fit
  283. if key == QtCore.Qt.Key_1:
  284. self.app.on_zoom_fit(None)
  285. # Zoom In
  286. if key == QtCore.Qt.Key_2:
  287. self.app.plotcanvas.zoom(1 / self.app.defaults['zoom_ratio'], self.app.mouse)
  288. # Zoom Out
  289. if key == QtCore.Qt.Key_3:
  290. self.app.plotcanvas.zoom(self.app.defaults['zoom_ratio'], self.app.mouse)
  291. # Delete
  292. if key == QtCore.Qt.Key_Delete and active:
  293. # Delete via the application to
  294. # ensure cleanup of the GUI
  295. active.app.on_delete()
  296. # Space = Toggle Active/Inactive
  297. if key == QtCore.Qt.Key_Space:
  298. for select in selected:
  299. select.ui.plot_cb.toggle()
  300. self.app.delete_selection_shape()
  301. # Copy Object Name
  302. if key == QtCore.Qt.Key_E:
  303. self.app.object2editor()
  304. # Grid toggle
  305. if key == QtCore.Qt.Key_G:
  306. self.app.geo_editor.grid_snap_btn.trigger()
  307. # Jump to coords
  308. if key == QtCore.Qt.Key_J:
  309. self.app.on_jump_to()
  310. # New Excellon
  311. if key == QtCore.Qt.Key_L:
  312. self.app.new_excellon_object()
  313. # Move tool toggle
  314. if key == QtCore.Qt.Key_M:
  315. self.app.move_tool.toggle()
  316. # New Geometry
  317. if key == QtCore.Qt.Key_N:
  318. self.app.on_new_geometry()
  319. # Set Origin
  320. if key == QtCore.Qt.Key_O:
  321. self.app.on_set_origin()
  322. return
  323. # Set Origin
  324. if key == QtCore.Qt.Key_P:
  325. self.app.properties_tool.run()
  326. return
  327. # Change Units
  328. if key == QtCore.Qt.Key_Q:
  329. if self.app.options["units"] == 'MM':
  330. self.app.general_options_form.general_app_group.units_radio.set_value("IN")
  331. else:
  332. self.app.general_options_form.general_app_group.units_radio.set_value("MM")
  333. self.app.on_toggle_units()
  334. # Rotate Object by 90 degree CW
  335. if key == QtCore.Qt.Key_R:
  336. self.app.on_rotate(silent=True, preset=90)
  337. # Shell toggle
  338. if key == QtCore.Qt.Key_S:
  339. self.app.on_toggle_shell()
  340. # Transform Tool
  341. if key == QtCore.Qt.Key_T:
  342. self.app.transform_tool.run()
  343. # Zoom Fit
  344. if key == QtCore.Qt.Key_V:
  345. self.app.on_zoom_fit(None)
  346. # Mirror on X the selected object(s)
  347. if key == QtCore.Qt.Key_X:
  348. self.app.on_flipx()
  349. # Mirror on Y the selected object(s)
  350. if key == QtCore.Qt.Key_Y:
  351. self.app.on_flipy()
  352. # Show shortcut list
  353. if key == QtCore.Qt.Key_Ampersand:
  354. self.app.on_shortcut_list()
  355. if key == QtCore.Qt.Key_QuoteLeft:
  356. self.app.on_shortcut_list()
  357. return
  358. def on_mouse_down(self, event):
  359. FlatCAMApp.App.log.debug("Mouse button pressed on list")
  360. def on_menu_request(self, pos):
  361. sel = len(self.view.selectedIndexes()) > 0
  362. self.app.ui.menuprojectenable.setEnabled(sel)
  363. self.app.ui.menuprojectdisable.setEnabled(sel)
  364. self.app.ui.menuprojectcopy.setEnabled(sel)
  365. self.app.ui.menuprojectedit.setEnabled(sel)
  366. self.app.ui.menuprojectdelete.setEnabled(sel)
  367. self.app.ui.menuprojectproperties.setEnabled(sel)
  368. if sel:
  369. self.app.ui.menuprojectgeneratecnc.setVisible(True)
  370. self.app.ui.menuprojectedit.setVisible(True)
  371. for obj in self.get_selected():
  372. if type(obj) != FlatCAMGeometry:
  373. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  374. if type(obj) != FlatCAMGeometry and type(obj) != FlatCAMExcellon:
  375. self.app.ui.menuprojectedit.setVisible(False)
  376. else:
  377. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  378. self.app.ui.menuproject.popup(self.view.mapToGlobal(pos))
  379. def index(self, row, column=0, parent=None, *args, **kwargs):
  380. if not self.hasIndex(row, column, parent):
  381. return QtCore.QModelIndex()
  382. if not parent.isValid():
  383. parent_item = self.root_item
  384. else:
  385. parent_item = parent.internalPointer()
  386. child_item = parent_item.child(row)
  387. if child_item:
  388. return self.createIndex(row, column, child_item)
  389. else:
  390. return QtCore.QModelIndex()
  391. def parent(self, index=None):
  392. if not index.isValid():
  393. return QtCore.QModelIndex()
  394. parent_item = index.internalPointer().parent_item
  395. if parent_item == self.root_item:
  396. return QtCore.QModelIndex()
  397. return self.createIndex(parent_item.row(), 0, parent_item)
  398. def rowCount(self, index=None, *args, **kwargs):
  399. if index.column() > 0:
  400. return 0
  401. if not index.isValid():
  402. parent_item = self.root_item
  403. else:
  404. parent_item = index.internalPointer()
  405. return parent_item.child_count()
  406. def columnCount(self, index=None, *args, **kwargs):
  407. if index.isValid():
  408. return index.internalPointer().column_count()
  409. else:
  410. return self.root_item.column_count()
  411. def data(self, index, role=None):
  412. if not index.isValid():
  413. return None
  414. if role in [Qt.DisplayRole, Qt.EditRole]:
  415. obj = index.internalPointer().obj
  416. if obj:
  417. return obj.options["name"]
  418. else:
  419. return index.internalPointer().data(index.column())
  420. if role == Qt.ForegroundRole:
  421. obj = index.internalPointer().obj
  422. if obj:
  423. return QtGui.QBrush(QtCore.Qt.black) if obj.options["plot"] else QtGui.QBrush(QtCore.Qt.darkGray)
  424. else:
  425. return index.internalPointer().data(index.column())
  426. elif role == Qt.DecorationRole:
  427. icon = index.internalPointer().icon
  428. if icon:
  429. return icon
  430. else:
  431. return QtGui.QPixmap()
  432. else:
  433. return None
  434. def setData(self, index, data, role=None):
  435. if index.isValid():
  436. obj = index.internalPointer().obj
  437. if obj:
  438. old_name = obj.options['name']
  439. # rename the object
  440. obj.options["name"] = str(data)
  441. new_name = obj.options['name']
  442. # update the SHELL auto-completer model data
  443. try:
  444. self.app.myKeywords.remove(old_name)
  445. self.app.myKeywords.append(new_name)
  446. self.app.shell._edit.set_model_data(self.app.myKeywords)
  447. except:
  448. log.debug(
  449. "setData() --> Could not remove the old object name from auto-completer model list")
  450. obj.build_ui()
  451. self.app.inform.emit("Object renamed from %s to %s" % (old_name, new_name))
  452. return True
  453. def flags(self, index):
  454. if not index.isValid():
  455. return 0
  456. # Prevent groups from selection
  457. if not index.internalPointer().obj:
  458. return Qt.ItemIsEnabled
  459. else:
  460. return Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsEditable
  461. return QtWidgets.QAbstractItemModel.flags(self, index)
  462. # def data(self, index, role=Qt.Qt.DisplayRole):
  463. # if not index.isValid() or not 0 <= index.row() < self.rowCount():
  464. # return QtCore.QVariant()
  465. # row = index.row()
  466. # if role == Qt.Qt.DisplayRole:
  467. # return self.object_list[row].options["name"]
  468. # if role == Qt.Qt.DecorationRole:
  469. # return self.icons[self.object_list[row].kind]
  470. # # if role == Qt.Qt.CheckStateRole:
  471. # # if row in self.checked_indexes:
  472. # # return Qt.Qt.Checked
  473. # # else:
  474. # # return Qt.Qt.Unchecked
  475. def print_list(self):
  476. for obj in self.get_list():
  477. print(obj)
  478. def append(self, obj, active=False):
  479. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + " --> OC.append()")
  480. name = obj.options["name"]
  481. # Check promises and clear if exists
  482. if name in self.promises:
  483. self.promises.remove(name)
  484. FlatCAMApp.App.log.debug("Promised object %s became available." % name)
  485. FlatCAMApp.App.log.debug("%d promised objects remaining." % len(self.promises))
  486. # Prevent same name
  487. while name in self.get_names():
  488. ## Create a new name
  489. # Ends with number?
  490. FlatCAMApp.App.log.debug("new_object(): Object name (%s) exists, changing." % name)
  491. match = re.search(r'(.*[^\d])?(\d+)$', name)
  492. if match: # Yes: Increment the number!
  493. base = match.group(1) or ''
  494. num = int(match.group(2))
  495. name = base + str(num + 1)
  496. else: # No: add a number!
  497. name += "_1"
  498. obj.options["name"] = name
  499. obj.set_ui(obj.ui_type())
  500. # Required before appending (Qt MVC)
  501. group = self.group_items[obj.kind]
  502. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  503. self.beginInsertRows(group_index, group.child_count(), group.child_count())
  504. # Append new item
  505. obj.item = TreeItem(None, self.icons[obj.kind], obj, group)
  506. # Required after appending (Qt MVC)
  507. self.endInsertRows()
  508. # Expand group
  509. if group.child_count() is 1:
  510. self.view.setExpanded(group_index, True)
  511. def get_names(self):
  512. """
  513. Gets a list of the names of all objects in the collection.
  514. :return: List of names.
  515. :rtype: list
  516. """
  517. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + " --> OC.get_names()")
  518. return [x.options['name'] for x in self.get_list()]
  519. def get_bounds(self):
  520. """
  521. Finds coordinates bounding all objects in the collection.
  522. :return: [xmin, ymin, xmax, ymax]
  523. :rtype: list
  524. """
  525. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + "--> OC.get_bounds()")
  526. # TODO: Move the operation out of here.
  527. xmin = Inf
  528. ymin = Inf
  529. xmax = -Inf
  530. ymax = -Inf
  531. # for obj in self.object_list:
  532. for obj in self.get_list():
  533. try:
  534. gxmin, gymin, gxmax, gymax = obj.bounds()
  535. xmin = min([xmin, gxmin])
  536. ymin = min([ymin, gymin])
  537. xmax = max([xmax, gxmax])
  538. ymax = max([ymax, gymax])
  539. except:
  540. FlatCAMApp.App.log.warning("DEV WARNING: Tried to get bounds of empty geometry.")
  541. return [xmin, ymin, xmax, ymax]
  542. def get_by_name(self, name, isCaseSensitive=None):
  543. """
  544. Fetches the FlatCAMObj with the given `name`.
  545. :param name: The name of the object.
  546. :type name: str
  547. :return: The requested object or None if no such object.
  548. :rtype: FlatCAMObj or None
  549. """
  550. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + "--> OC.get_by_name()")
  551. if isCaseSensitive is None or isCaseSensitive is True:
  552. for obj in self.get_list():
  553. if obj.options['name'] == name:
  554. return obj
  555. else:
  556. for obj in self.get_list():
  557. if obj.options['name'].lower() == name.lower():
  558. return obj
  559. return None
  560. def delete_active(self):
  561. selections = self.view.selectedIndexes()
  562. if len(selections) == 0:
  563. return
  564. active = selections[0].internalPointer()
  565. group = active.parent_item
  566. # update the SHELL auto-completer model data
  567. name = active.obj.options['name']
  568. try:
  569. self.app.myKeywords.remove(name)
  570. self.app.shell._edit.set_model_data(self.app.myKeywords)
  571. except:
  572. log.debug(
  573. "delete_active() --> Could not remove the old object name from auto-completer model list")
  574. self.beginRemoveRows(self.index(group.row(), 0, QtCore.QModelIndex()), active.row(), active.row())
  575. group.remove_child(active)
  576. # after deletion of object store the current list of objects into the self.app.all_objects_list
  577. self.app.all_objects_list = self.get_list()
  578. self.endRemoveRows()
  579. # always go to the Project Tab after object deletion as it may be done with a shortcut key
  580. self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
  581. def delete_all(self):
  582. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + "--> OC.delete_all()")
  583. self.beginResetModel()
  584. self.checked_indexes = []
  585. for group in self.root_item.child_items:
  586. group.remove_children()
  587. self.endResetModel()
  588. self.app.plotcanvas.redraw()
  589. self.app.all_objects_list.clear()
  590. self.app.geo_editor.clear()
  591. self.app.exc_editor.clear()
  592. self.app.dblsidedtool.reset_fields()
  593. self.app.panelize_tool.reset_fields()
  594. self.app.cutout_tool.reset_fields()
  595. self.app.film_tool.reset_fields()
  596. def get_active(self):
  597. """
  598. Returns the active object or None
  599. :return: FlatCAMObj or None
  600. """
  601. selections = self.view.selectedIndexes()
  602. if len(selections) == 0:
  603. return None
  604. return selections[0].internalPointer().obj
  605. def get_selected(self):
  606. """
  607. Returns list of objects selected in the view.
  608. :return: List of objects
  609. """
  610. return [sel.internalPointer().obj for sel in self.view.selectedIndexes()]
  611. def get_non_selected(self):
  612. """
  613. Returns list of objects non-selected in the view.
  614. :return: List of objects
  615. """
  616. l = self.get_list()
  617. for sel in self.get_selected():
  618. l.remove(sel)
  619. return l
  620. def set_active(self, name):
  621. """
  622. Selects object by name from the project list. This triggers the
  623. list_selection_changed event and call on_list_selection_changed.
  624. :param name: Name of the FlatCAM Object
  625. :return: None
  626. """
  627. try:
  628. obj = self.get_by_name(name)
  629. item = obj.item
  630. group = self.group_items[obj.kind]
  631. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  632. item_index = self.index(item.row(), 0, group_index)
  633. self.view.selectionModel().select(item_index, QtCore.QItemSelectionModel.Select)
  634. except Exception as e:
  635. log.error("[ERROR] Cause: %s" % str(e))
  636. raise
  637. def set_inactive(self, name):
  638. """
  639. Unselect object by name from the project list. This triggers the
  640. list_selection_changed event and call on_list_selection_changed.
  641. :param name: Name of the FlatCAM Object
  642. :return: None
  643. """
  644. obj = self.get_by_name(name)
  645. item = obj.item
  646. group = self.group_items[obj.kind]
  647. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  648. item_index = self.index(item.row(), 0, group_index)
  649. self.view.selectionModel().select(item_index, QtCore.QItemSelectionModel.Deselect)
  650. def set_all_inactive(self):
  651. """
  652. Unselect all objects from the project list. This triggers the
  653. list_selection_changed event and call on_list_selection_changed.
  654. :return: None
  655. """
  656. for name in self.get_names():
  657. self.set_inactive(name)
  658. def on_list_selection_change(self, current, previous):
  659. FlatCAMApp.App.log.debug("on_list_selection_change()")
  660. FlatCAMApp.App.log.debug("Current: %s, Previous %s" % (str(current), str(previous)))
  661. try:
  662. obj = current.indexes()[0].internalPointer().obj
  663. except IndexError:
  664. FlatCAMApp.App.log.debug("on_list_selection_change(): Index Error (Nothing selected?)")
  665. try:
  666. self.app.ui.selected_scroll_area.takeWidget()
  667. except:
  668. FlatCAMApp.App.log.debug("Nothing to remove")
  669. self.app.setup_component_editor()
  670. return
  671. if obj:
  672. obj.build_ui()
  673. def on_item_activated(self, index):
  674. """
  675. Double-click or Enter on item.
  676. :param index: Index of the item in the list.
  677. :return: None
  678. """
  679. a_idx = index.internalPointer().obj
  680. if a_idx is None:
  681. return
  682. else:
  683. try:
  684. a_idx.build_ui()
  685. except Exception as e:
  686. self.app.inform.emit("[ERROR] Cause of error: %s" % str(e))
  687. raise
  688. def get_list(self):
  689. obj_list = []
  690. for group in self.root_item.child_items:
  691. for item in group.child_items:
  692. obj_list.append(item.obj)
  693. return obj_list
  694. def update_view(self):
  695. self.dataChanged.emit(QtCore.QModelIndex(), QtCore.QModelIndex())