ObjectCollection.py 29 KB

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