ObjectCollection.py 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906
  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. # Eanble all plots
  260. if key == Qt.Key_1:
  261. self.app.enable_all_plots()
  262. # Disable all plots
  263. if key == Qt.Key_2:
  264. self.app.disable_all_plots()
  265. # Disable all other plots
  266. if key == Qt.Key_3:
  267. self.app.disable_other_plots()
  268. # 2-Sided PCB Tool
  269. if key == QtCore.Qt.Key_D:
  270. self.app.dblsidedtool.run()
  271. return
  272. # Non-Copper Clear Tool
  273. if key == QtCore.Qt.Key_N:
  274. self.app.ncclear_tool.run()
  275. return
  276. # Transformation Tool
  277. if key == QtCore.Qt.Key_R:
  278. self.app.transform_tool.run()
  279. return
  280. # Cutout Tool
  281. if key == QtCore.Qt.Key_U:
  282. self.app.cutout_tool.run()
  283. return
  284. else:
  285. # Open Manual
  286. if key == QtCore.Qt.Key_F1:
  287. webbrowser.open(self.app.manual_url)
  288. # Open Video Help
  289. if key == QtCore.Qt.Key_F2:
  290. webbrowser.open(self.app.video_url)
  291. # Zoom Fit
  292. if key == QtCore.Qt.Key_1:
  293. self.app.on_zoom_fit(None)
  294. # Zoom In
  295. if key == QtCore.Qt.Key_2:
  296. self.app.plotcanvas.zoom(1 / self.app.defaults['zoom_ratio'], self.app.mouse)
  297. # Zoom Out
  298. if key == QtCore.Qt.Key_3:
  299. self.app.plotcanvas.zoom(self.app.defaults['zoom_ratio'], self.app.mouse)
  300. # Delete
  301. if key == QtCore.Qt.Key_Delete and active:
  302. # Delete via the application to
  303. # ensure cleanup of the GUI
  304. active.app.on_delete()
  305. # Space = Toggle Active/Inactive
  306. if key == QtCore.Qt.Key_Space:
  307. for select in selected:
  308. select.ui.plot_cb.toggle()
  309. self.app.delete_selection_shape()
  310. # Copy Object Name
  311. if key == QtCore.Qt.Key_E:
  312. self.app.object2editor()
  313. # Grid toggle
  314. if key == QtCore.Qt.Key_G:
  315. self.app.geo_editor.grid_snap_btn.trigger()
  316. # Jump to coords
  317. if key == QtCore.Qt.Key_J:
  318. self.app.on_jump_to()
  319. # New Excellon
  320. if key == QtCore.Qt.Key_L:
  321. self.app.new_excellon_object()
  322. # Move tool toggle
  323. if key == QtCore.Qt.Key_M:
  324. self.app.move_tool.toggle()
  325. # New Geometry
  326. if key == QtCore.Qt.Key_N:
  327. self.app.on_new_geometry()
  328. # Set Origin
  329. if key == QtCore.Qt.Key_O:
  330. self.app.on_set_origin()
  331. return
  332. # Set Origin
  333. if key == QtCore.Qt.Key_P:
  334. self.app.properties_tool.run()
  335. return
  336. # Change Units
  337. if key == QtCore.Qt.Key_Q:
  338. if self.app.options["units"] == 'MM':
  339. self.app.general_options_form.general_app_group.units_radio.set_value("IN")
  340. else:
  341. self.app.general_options_form.general_app_group.units_radio.set_value("MM")
  342. self.app.on_toggle_units()
  343. # Rotate Object by 90 degree CW
  344. if key == QtCore.Qt.Key_R:
  345. self.app.on_rotate(silent=True, preset=90)
  346. # Shell toggle
  347. if key == QtCore.Qt.Key_S:
  348. self.app.on_toggle_shell()
  349. # Transform Tool
  350. if key == QtCore.Qt.Key_T:
  351. self.app.transform_tool.run()
  352. # Zoom Fit
  353. if key == QtCore.Qt.Key_V:
  354. self.app.on_zoom_fit(None)
  355. # Mirror on X the selected object(s)
  356. if key == QtCore.Qt.Key_X:
  357. self.app.on_flipx()
  358. # Mirror on Y the selected object(s)
  359. if key == QtCore.Qt.Key_Y:
  360. self.app.on_flipy()
  361. # Show shortcut list
  362. if key == QtCore.Qt.Key_Ampersand:
  363. self.app.on_shortcut_list()
  364. if key == QtCore.Qt.Key_QuoteLeft:
  365. self.app.on_shortcut_list()
  366. return
  367. def on_mouse_down(self, event):
  368. FlatCAMApp.App.log.debug("Mouse button pressed on list")
  369. def on_menu_request(self, pos):
  370. sel = len(self.view.selectedIndexes()) > 0
  371. self.app.ui.menuprojectenable.setEnabled(sel)
  372. self.app.ui.menuprojectdisable.setEnabled(sel)
  373. self.app.ui.menuprojectcopy.setEnabled(sel)
  374. self.app.ui.menuprojectedit.setEnabled(sel)
  375. self.app.ui.menuprojectdelete.setEnabled(sel)
  376. self.app.ui.menuprojectproperties.setEnabled(sel)
  377. if sel:
  378. self.app.ui.menuprojectgeneratecnc.setVisible(True)
  379. self.app.ui.menuprojectedit.setVisible(True)
  380. for obj in self.get_selected():
  381. if type(obj) != FlatCAMGeometry:
  382. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  383. if type(obj) != FlatCAMGeometry and type(obj) != FlatCAMExcellon:
  384. self.app.ui.menuprojectedit.setVisible(False)
  385. else:
  386. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  387. self.app.ui.menuproject.popup(self.view.mapToGlobal(pos))
  388. def index(self, row, column=0, parent=None, *args, **kwargs):
  389. if not self.hasIndex(row, column, parent):
  390. return QtCore.QModelIndex()
  391. if not parent.isValid():
  392. parent_item = self.root_item
  393. else:
  394. parent_item = parent.internalPointer()
  395. child_item = parent_item.child(row)
  396. if child_item:
  397. return self.createIndex(row, column, child_item)
  398. else:
  399. return QtCore.QModelIndex()
  400. def parent(self, index=None):
  401. if not index.isValid():
  402. return QtCore.QModelIndex()
  403. parent_item = index.internalPointer().parent_item
  404. if parent_item == self.root_item:
  405. return QtCore.QModelIndex()
  406. return self.createIndex(parent_item.row(), 0, parent_item)
  407. def rowCount(self, index=None, *args, **kwargs):
  408. if index.column() > 0:
  409. return 0
  410. if not index.isValid():
  411. parent_item = self.root_item
  412. else:
  413. parent_item = index.internalPointer()
  414. return parent_item.child_count()
  415. def columnCount(self, index=None, *args, **kwargs):
  416. if index.isValid():
  417. return index.internalPointer().column_count()
  418. else:
  419. return self.root_item.column_count()
  420. def data(self, index, role=None):
  421. if not index.isValid():
  422. return None
  423. if role in [Qt.DisplayRole, Qt.EditRole]:
  424. obj = index.internalPointer().obj
  425. if obj:
  426. return obj.options["name"]
  427. else:
  428. return index.internalPointer().data(index.column())
  429. if role == Qt.ForegroundRole:
  430. obj = index.internalPointer().obj
  431. if obj:
  432. return QtGui.QBrush(QtCore.Qt.black) if obj.options["plot"] else QtGui.QBrush(QtCore.Qt.darkGray)
  433. else:
  434. return index.internalPointer().data(index.column())
  435. elif role == Qt.DecorationRole:
  436. icon = index.internalPointer().icon
  437. if icon:
  438. return icon
  439. else:
  440. return QtGui.QPixmap()
  441. else:
  442. return None
  443. def setData(self, index, data, role=None):
  444. if index.isValid():
  445. obj = index.internalPointer().obj
  446. if obj:
  447. old_name = obj.options['name']
  448. # rename the object
  449. obj.options["name"] = str(data)
  450. new_name = obj.options['name']
  451. # update the SHELL auto-completer model data
  452. try:
  453. self.app.myKeywords.remove(old_name)
  454. self.app.myKeywords.append(new_name)
  455. self.app.shell._edit.set_model_data(self.app.myKeywords)
  456. except:
  457. log.debug(
  458. "setData() --> Could not remove the old object name from auto-completer model list")
  459. obj.build_ui()
  460. self.app.inform.emit("Object renamed from %s to %s" % (old_name, new_name))
  461. return True
  462. def flags(self, index):
  463. if not index.isValid():
  464. return 0
  465. # Prevent groups from selection
  466. if not index.internalPointer().obj:
  467. return Qt.ItemIsEnabled
  468. else:
  469. return Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsEditable
  470. return QtWidgets.QAbstractItemModel.flags(self, index)
  471. # def data(self, index, role=Qt.Qt.DisplayRole):
  472. # if not index.isValid() or not 0 <= index.row() < self.rowCount():
  473. # return QtCore.QVariant()
  474. # row = index.row()
  475. # if role == Qt.Qt.DisplayRole:
  476. # return self.object_list[row].options["name"]
  477. # if role == Qt.Qt.DecorationRole:
  478. # return self.icons[self.object_list[row].kind]
  479. # # if role == Qt.Qt.CheckStateRole:
  480. # # if row in self.checked_indexes:
  481. # # return Qt.Qt.Checked
  482. # # else:
  483. # # return Qt.Qt.Unchecked
  484. def print_list(self):
  485. for obj in self.get_list():
  486. print(obj)
  487. def append(self, obj, active=False):
  488. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + " --> OC.append()")
  489. name = obj.options["name"]
  490. # Check promises and clear if exists
  491. if name in self.promises:
  492. self.promises.remove(name)
  493. FlatCAMApp.App.log.debug("Promised object %s became available." % name)
  494. FlatCAMApp.App.log.debug("%d promised objects remaining." % len(self.promises))
  495. # Prevent same name
  496. while name in self.get_names():
  497. ## Create a new name
  498. # Ends with number?
  499. FlatCAMApp.App.log.debug("new_object(): Object name (%s) exists, changing." % name)
  500. match = re.search(r'(.*[^\d])?(\d+)$', name)
  501. if match: # Yes: Increment the number!
  502. base = match.group(1) or ''
  503. num = int(match.group(2))
  504. name = base + str(num + 1)
  505. else: # No: add a number!
  506. name += "_1"
  507. obj.options["name"] = name
  508. obj.set_ui(obj.ui_type())
  509. # Required before appending (Qt MVC)
  510. group = self.group_items[obj.kind]
  511. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  512. self.beginInsertRows(group_index, group.child_count(), group.child_count())
  513. # Append new item
  514. obj.item = TreeItem(None, self.icons[obj.kind], obj, group)
  515. # Required after appending (Qt MVC)
  516. self.endInsertRows()
  517. # Expand group
  518. if group.child_count() is 1:
  519. self.view.setExpanded(group_index, True)
  520. def get_names(self):
  521. """
  522. Gets a list of the names of all objects in the collection.
  523. :return: List of names.
  524. :rtype: list
  525. """
  526. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + " --> OC.get_names()")
  527. return [x.options['name'] for x in self.get_list()]
  528. def get_bounds(self):
  529. """
  530. Finds coordinates bounding all objects in the collection.
  531. :return: [xmin, ymin, xmax, ymax]
  532. :rtype: list
  533. """
  534. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + "--> OC.get_bounds()")
  535. # TODO: Move the operation out of here.
  536. xmin = Inf
  537. ymin = Inf
  538. xmax = -Inf
  539. ymax = -Inf
  540. # for obj in self.object_list:
  541. for obj in self.get_list():
  542. try:
  543. gxmin, gymin, gxmax, gymax = obj.bounds()
  544. xmin = min([xmin, gxmin])
  545. ymin = min([ymin, gymin])
  546. xmax = max([xmax, gxmax])
  547. ymax = max([ymax, gymax])
  548. except:
  549. FlatCAMApp.App.log.warning("DEV WARNING: Tried to get bounds of empty geometry.")
  550. return [xmin, ymin, xmax, ymax]
  551. def get_by_name(self, name, isCaseSensitive=None):
  552. """
  553. Fetches the FlatCAMObj with the given `name`.
  554. :param name: The name of the object.
  555. :type name: str
  556. :return: The requested object or None if no such object.
  557. :rtype: FlatCAMObj or None
  558. """
  559. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + "--> OC.get_by_name()")
  560. if isCaseSensitive is None or isCaseSensitive is True:
  561. for obj in self.get_list():
  562. if obj.options['name'] == name:
  563. return obj
  564. else:
  565. for obj in self.get_list():
  566. if obj.options['name'].lower() == name.lower():
  567. return obj
  568. return None
  569. def delete_active(self):
  570. selections = self.view.selectedIndexes()
  571. if len(selections) == 0:
  572. return
  573. active = selections[0].internalPointer()
  574. group = active.parent_item
  575. # update the SHELL auto-completer model data
  576. name = active.obj.options['name']
  577. try:
  578. self.app.myKeywords.remove(name)
  579. self.app.shell._edit.set_model_data(self.app.myKeywords)
  580. except:
  581. log.debug(
  582. "delete_active() --> Could not remove the old object name from auto-completer model list")
  583. self.beginRemoveRows(self.index(group.row(), 0, QtCore.QModelIndex()), active.row(), active.row())
  584. group.remove_child(active)
  585. # after deletion of object store the current list of objects into the self.app.all_objects_list
  586. self.app.all_objects_list = self.get_list()
  587. self.endRemoveRows()
  588. # always go to the Project Tab after object deletion as it may be done with a shortcut key
  589. self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
  590. def delete_all(self):
  591. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + "--> OC.delete_all()")
  592. self.beginResetModel()
  593. self.checked_indexes = []
  594. for group in self.root_item.child_items:
  595. group.remove_children()
  596. self.endResetModel()
  597. self.app.plotcanvas.redraw()
  598. self.app.all_objects_list.clear()
  599. self.app.geo_editor.clear()
  600. self.app.exc_editor.clear()
  601. self.app.dblsidedtool.reset_fields()
  602. self.app.panelize_tool.reset_fields()
  603. self.app.cutout_tool.reset_fields()
  604. self.app.film_tool.reset_fields()
  605. def get_active(self):
  606. """
  607. Returns the active object or None
  608. :return: FlatCAMObj or None
  609. """
  610. selections = self.view.selectedIndexes()
  611. if len(selections) == 0:
  612. return None
  613. return selections[0].internalPointer().obj
  614. def get_selected(self):
  615. """
  616. Returns list of objects selected in the view.
  617. :return: List of objects
  618. """
  619. return [sel.internalPointer().obj for sel in self.view.selectedIndexes()]
  620. def get_non_selected(self):
  621. """
  622. Returns list of objects non-selected in the view.
  623. :return: List of objects
  624. """
  625. l = self.get_list()
  626. for sel in self.get_selected():
  627. l.remove(sel)
  628. return l
  629. def set_active(self, name):
  630. """
  631. Selects object by name from the project list. This triggers the
  632. list_selection_changed event and call on_list_selection_changed.
  633. :param name: Name of the FlatCAM Object
  634. :return: None
  635. """
  636. try:
  637. obj = self.get_by_name(name)
  638. item = obj.item
  639. group = self.group_items[obj.kind]
  640. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  641. item_index = self.index(item.row(), 0, group_index)
  642. self.view.selectionModel().select(item_index, QtCore.QItemSelectionModel.Select)
  643. except Exception as e:
  644. log.error("[ERROR] Cause: %s" % str(e))
  645. raise
  646. def set_inactive(self, name):
  647. """
  648. Unselect object by name from the project list. This triggers the
  649. list_selection_changed event and call on_list_selection_changed.
  650. :param name: Name of the FlatCAM Object
  651. :return: None
  652. """
  653. obj = self.get_by_name(name)
  654. item = obj.item
  655. group = self.group_items[obj.kind]
  656. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  657. item_index = self.index(item.row(), 0, group_index)
  658. self.view.selectionModel().select(item_index, QtCore.QItemSelectionModel.Deselect)
  659. def set_all_inactive(self):
  660. """
  661. Unselect all objects from the project list. This triggers the
  662. list_selection_changed event and call on_list_selection_changed.
  663. :return: None
  664. """
  665. for name in self.get_names():
  666. self.set_inactive(name)
  667. def on_list_selection_change(self, current, previous):
  668. FlatCAMApp.App.log.debug("on_list_selection_change()")
  669. FlatCAMApp.App.log.debug("Current: %s, Previous %s" % (str(current), str(previous)))
  670. try:
  671. obj = current.indexes()[0].internalPointer().obj
  672. except IndexError:
  673. FlatCAMApp.App.log.debug("on_list_selection_change(): Index Error (Nothing selected?)")
  674. try:
  675. self.app.ui.selected_scroll_area.takeWidget()
  676. except:
  677. FlatCAMApp.App.log.debug("Nothing to remove")
  678. self.app.setup_component_editor()
  679. return
  680. if obj:
  681. obj.build_ui()
  682. def on_item_activated(self, index):
  683. """
  684. Double-click or Enter on item.
  685. :param index: Index of the item in the list.
  686. :return: None
  687. """
  688. a_idx = index.internalPointer().obj
  689. if a_idx is None:
  690. return
  691. else:
  692. try:
  693. a_idx.build_ui()
  694. except Exception as e:
  695. self.app.inform.emit("[ERROR] Cause of error: %s" % str(e))
  696. raise
  697. def get_list(self):
  698. obj_list = []
  699. for group in self.root_item.child_items:
  700. for item in group.child_items:
  701. obj_list.append(item.obj)
  702. return obj_list
  703. def update_view(self):
  704. self.dataChanged.emit(QtCore.QModelIndex(), QtCore.QModelIndex())