ObjectCollection.py 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926
  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. self.app = app
  181. ### View
  182. self.view = KeySensitiveListView(app)
  183. self.view.setModel(self)
  184. self.view.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
  185. self.view.setSelectionMode(QtWidgets.QAbstractItemView.ExtendedSelection)
  186. # self.view.setDragDropMode(QtWidgets.QAbstractItemView.InternalMove)
  187. # self.view.setDragEnabled(True)
  188. # self.view.setAcceptDrops(True)
  189. # self.view.setDropIndicatorShown(True)
  190. font = QtGui.QFont()
  191. font.setPixelSize(12)
  192. font.setFamily("Seagoe UI")
  193. self.view.setFont(font)
  194. ## GUI Events
  195. self.view.selectionModel().selectionChanged.connect(self.on_list_selection_change)
  196. self.view.activated.connect(self.on_item_activated)
  197. # self.view.keyPressed.connect(self.on_key)
  198. self.view.keyPressed.connect(self.app.ui.keyPressEvent)
  199. self.view.clicked.connect(self.on_mouse_down)
  200. self.view.customContextMenuRequested.connect(self.on_menu_request)
  201. self.click_modifier = None
  202. def promise(self, obj_name):
  203. FlatCAMApp.App.log.debug("Object %s has been promised." % obj_name)
  204. self.promises.add(obj_name)
  205. def has_promises(self):
  206. return len(self.promises) > 0
  207. # def on_key(self, key):
  208. # modifiers = QtWidgets.QApplication.keyboardModifiers()
  209. # active = self.get_active()
  210. # selected = self.get_selected()
  211. #
  212. # if modifiers == QtCore.Qt.ControlModifier:
  213. # if key == QtCore.Qt.Key_A:
  214. # self.app.on_selectall()
  215. #
  216. # if key == QtCore.Qt.Key_C:
  217. # self.app.on_copy_object()
  218. #
  219. # if key == QtCore.Qt.Key_E:
  220. # self.app.on_fileopenexcellon()
  221. #
  222. # if key == QtCore.Qt.Key_G:
  223. # self.app.on_fileopengerber()
  224. #
  225. # if key == QtCore.Qt.Key_N:
  226. # self.app.on_file_new_click()
  227. #
  228. # if key == QtCore.Qt.Key_M:
  229. # self.app.measurement_tool.run()
  230. # if key == QtCore.Qt.Key_O:
  231. # self.app.on_file_openproject()
  232. #
  233. # if key == QtCore.Qt.Key_S:
  234. # self.app.on_file_saveproject()
  235. #
  236. # # Toggle Plot Area
  237. # if key == QtCore.Qt.Key_F10:
  238. # self.app.on_toggle_plotarea()
  239. #
  240. # return
  241. # elif modifiers == QtCore.Qt.ShiftModifier:
  242. #
  243. # # Copy Object Name
  244. # # Copy Object Name
  245. # if key == QtCore.Qt.Key_C:
  246. # self.app.on_copy_name()
  247. #
  248. # # Toggle axis
  249. # if key == QtCore.Qt.Key_G:
  250. # if self.toggle_axis is False:
  251. # self.app.plotcanvas.v_line.set_data(color=(0.70, 0.3, 0.3, 1.0))
  252. # self.app.plotcanvas.h_line.set_data(color=(0.70, 0.3, 0.3, 1.0))
  253. # self.app.plotcanvas.redraw()
  254. # self.app.toggle_axis = True
  255. # else:
  256. # self.app.plotcanvas.v_line.set_data(color=(0.0, 0.0, 0.0, 0.0))
  257. #
  258. # self.app.plotcanvas.h_line.set_data(color=(0.0, 0.0, 0.0, 0.0))
  259. # self.appplotcanvas.redraw()
  260. # self.app.toggle_axis = False
  261. #
  262. # # Open Preferences Window
  263. # if key == QtCore.Qt.Key_P:
  264. # self.app.on_preferences()
  265. # return
  266. #
  267. # # Rotate Object by 90 degree CCW
  268. # if key == QtCore.Qt.Key_R:
  269. # self.app.on_rotate(silent=True, preset=-90)
  270. # return
  271. #
  272. # # Run a Script
  273. # if key == QtCore.Qt.Key_S:
  274. # self.app.on_filerunscript()
  275. # return
  276. #
  277. # # Toggle Workspace
  278. # if key == QtCore.Qt.Key_W:
  279. # self.app.on_workspace_menu()
  280. # return
  281. #
  282. # # Skew on X axis
  283. # if key == QtCore.Qt.Key_X:
  284. # self.app.on_skewx()
  285. # return
  286. #
  287. # # Skew on Y axis
  288. # if key == QtCore.Qt.Key_Y:
  289. # self.app.on_skewy()
  290. # return
  291. #
  292. # elif modifiers == QtCore.Qt.AltModifier:
  293. # # Eanble all plots
  294. # if key == Qt.Key_1:
  295. # self.app.enable_all_plots()
  296. #
  297. # # Disable all plots
  298. # if key == Qt.Key_2:
  299. # self.app.disable_all_plots()
  300. #
  301. # # Disable all other plots
  302. # if key == Qt.Key_3:
  303. # self.app.disable_other_plots()
  304. #
  305. # # 2-Sided PCB Tool
  306. # if key == QtCore.Qt.Key_D:
  307. # self.app.dblsidedtool.run()
  308. # return
  309. #
  310. # # Non-Copper Clear Tool
  311. # if key == QtCore.Qt.Key_N:
  312. # self.app.ncclear_tool.run()
  313. # return
  314. #
  315. # # Transformation Tool
  316. # if key == QtCore.Qt.Key_R:
  317. # self.app.transform_tool.run()
  318. # return
  319. #
  320. # # Cutout Tool
  321. # if key == QtCore.Qt.Key_U:
  322. # self.app.cutout_tool.run()
  323. # return
  324. #
  325. # else:
  326. # # Open Manual
  327. # if key == QtCore.Qt.Key_F1:
  328. # webbrowser.open(self.app.manual_url)
  329. #
  330. # # Open Video Help
  331. # if key == QtCore.Qt.Key_F2:
  332. # webbrowser.open(self.app.video_url)
  333. #
  334. # # Switch to Project Tab
  335. # if key == QtCore.Qt.Key_1:
  336. # self.app.on_select_tab('project')
  337. #
  338. # # Switch to Selected Tab
  339. # if key == QtCore.Qt.Key_2:
  340. # self.app.on_select_tab('selected')
  341. #
  342. # # Switch to Tool Tab
  343. # if key == QtCore.Qt.Key_3:
  344. # self.app.on_select_tab('tool')
  345. #
  346. # # Delete
  347. # if key == QtCore.Qt.Key_Delete and active:
  348. # # Delete via the application to
  349. # # ensure cleanup of the GUI
  350. # active.app.on_delete()
  351. #
  352. # # Space = Toggle Active/Inactive
  353. # if key == QtCore.Qt.Key_Space:
  354. # for select in selected:
  355. # select.ui.plot_cb.toggle()
  356. # self.app.delete_selection_shape()
  357. #
  358. # # Copy Object Name
  359. # if key == QtCore.Qt.Key_E:
  360. # self.app.object2editor()
  361. #
  362. # # Grid toggle
  363. # if key == QtCore.Qt.Key_G:
  364. # self.app.ui.grid_snap_btn.trigger()
  365. #
  366. # # Jump to coords
  367. # if key == QtCore.Qt.Key_J:
  368. # self.app.on_jump_to()
  369. #
  370. # # New Excellon
  371. # if key == QtCore.Qt.Key_L:
  372. # self.app.new_excellon_object()
  373. #
  374. # # Move tool toggle
  375. # if key == QtCore.Qt.Key_M:
  376. # self.app.move_tool.toggle()
  377. #
  378. # # New Geometry
  379. # if key == QtCore.Qt.Key_N:
  380. # self.app.on_new_geometry()
  381. #
  382. # # Set Origin
  383. # if key == QtCore.Qt.Key_O:
  384. # self.app.on_set_origin()
  385. # return
  386. #
  387. # # Set Origin
  388. # if key == QtCore.Qt.Key_P:
  389. # self.app.properties_tool.run()
  390. # return
  391. #
  392. # # Change Units
  393. # if key == QtCore.Qt.Key_Q:
  394. # if self.app.options["units"] == 'MM':
  395. # self.app.general_options_form.general_app_group.units_radio.set_value("IN")
  396. # else:
  397. # self.app.general_options_form.general_app_group.units_radio.set_value("MM")
  398. # self.app.on_toggle_units()
  399. #
  400. # # Rotate Object by 90 degree CW
  401. # if key == QtCore.Qt.Key_R:
  402. # self.app.on_rotate(silent=True, preset=90)
  403. #
  404. # # Shell toggle
  405. # if key == QtCore.Qt.Key_S:
  406. # self.app.on_toggle_shell()
  407. #
  408. # # Transform Tool
  409. # if key == QtCore.Qt.Key_T:
  410. # self.app.transform_tool.run()
  411. #
  412. # # Zoom Fit
  413. # if key == QtCore.Qt.Key_V:
  414. # self.app.on_zoom_fit(None)
  415. #
  416. # # Mirror on X the selected object(s)
  417. # if key == QtCore.Qt.Key_X:
  418. # self.app.on_flipx()
  419. #
  420. # # Mirror on Y the selected object(s)
  421. # if key == QtCore.Qt.Key_Y:
  422. # self.app.on_flipy()
  423. #
  424. # # Zoom In
  425. # if key == QtCore.Qt.Key_Equal:
  426. # self.app.plotcanvas.zoom(1 / self.app.defaults['zoom_ratio'], self.app.mouse)
  427. #
  428. # # Zoom Out
  429. # if key == QtCore.Qt.Key_Minus:
  430. # self.app.plotcanvas.zoom(self.app.defaults['zoom_ratio'], self.app.mouse)
  431. #
  432. # # Show shortcut list
  433. # if key == QtCore.Qt.Key_Ampersand:
  434. # self.app.on_shortcut_list()
  435. #
  436. # if key == QtCore.Qt.Key_QuoteLeft:
  437. # self.app.on_shortcut_list()
  438. # return
  439. def on_mouse_down(self, event):
  440. FlatCAMApp.App.log.debug("Mouse button pressed on list")
  441. def on_menu_request(self, pos):
  442. sel = len(self.view.selectedIndexes()) > 0
  443. self.app.ui.menuprojectenable.setEnabled(sel)
  444. self.app.ui.menuprojectdisable.setEnabled(sel)
  445. self.app.ui.menuprojectcopy.setEnabled(sel)
  446. self.app.ui.menuprojectedit.setEnabled(sel)
  447. self.app.ui.menuprojectdelete.setEnabled(sel)
  448. self.app.ui.menuprojectproperties.setEnabled(sel)
  449. if sel:
  450. self.app.ui.menuprojectgeneratecnc.setVisible(True)
  451. self.app.ui.menuprojectedit.setVisible(True)
  452. for obj in self.get_selected():
  453. if type(obj) != FlatCAMGeometry:
  454. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  455. if type(obj) != FlatCAMGeometry and type(obj) != FlatCAMExcellon:
  456. self.app.ui.menuprojectedit.setVisible(False)
  457. else:
  458. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  459. self.app.ui.menuproject.popup(self.view.mapToGlobal(pos))
  460. def index(self, row, column=0, parent=None, *args, **kwargs):
  461. if not self.hasIndex(row, column, parent):
  462. return QtCore.QModelIndex()
  463. # if not parent.isValid():
  464. # parent_item = self.root_item
  465. # else:
  466. # parent_item = parent.internalPointer()
  467. parent_item = parent.internalPointer() if parent.isValid() else self.root_item
  468. child_item = parent_item.child(row)
  469. if child_item:
  470. return self.createIndex(row, column, child_item)
  471. else:
  472. return QtCore.QModelIndex()
  473. def parent(self, index=None):
  474. if not index.isValid():
  475. return QtCore.QModelIndex()
  476. parent_item = index.internalPointer().parent_item
  477. if parent_item == self.root_item:
  478. return QtCore.QModelIndex()
  479. return self.createIndex(parent_item.row(), 0, parent_item)
  480. def rowCount(self, index=None, *args, **kwargs):
  481. if index.column() > 0:
  482. return 0
  483. if not index.isValid():
  484. parent_item = self.root_item
  485. else:
  486. parent_item = index.internalPointer()
  487. return parent_item.child_count()
  488. def columnCount(self, index=None, *args, **kwargs):
  489. if index.isValid():
  490. return index.internalPointer().column_count()
  491. else:
  492. return self.root_item.column_count()
  493. def data(self, index, role=None):
  494. if not index.isValid():
  495. return None
  496. if role in [Qt.DisplayRole, Qt.EditRole]:
  497. obj = index.internalPointer().obj
  498. if obj:
  499. return obj.options["name"]
  500. else:
  501. return index.internalPointer().data(index.column())
  502. if role == Qt.ForegroundRole:
  503. obj = index.internalPointer().obj
  504. if obj:
  505. return QtGui.QBrush(QtCore.Qt.black) if obj.options["plot"] else QtGui.QBrush(QtCore.Qt.darkGray)
  506. else:
  507. return index.internalPointer().data(index.column())
  508. elif role == Qt.DecorationRole:
  509. icon = index.internalPointer().icon
  510. if icon:
  511. return icon
  512. else:
  513. return QtGui.QPixmap()
  514. else:
  515. return None
  516. def setData(self, index, data, role=None):
  517. if index.isValid():
  518. obj = index.internalPointer().obj
  519. if obj:
  520. old_name = obj.options['name']
  521. # rename the object
  522. obj.options["name"] = str(data)
  523. new_name = obj.options['name']
  524. # update the SHELL auto-completer model data
  525. try:
  526. self.app.myKeywords.remove(old_name)
  527. self.app.myKeywords.append(new_name)
  528. self.app.shell._edit.set_model_data(self.app.myKeywords)
  529. except:
  530. log.debug(
  531. "setData() --> Could not remove the old object name from auto-completer model list")
  532. obj.build_ui()
  533. self.app.inform.emit("Object renamed from %s to %s" % (old_name, new_name))
  534. return True
  535. def supportedDropActions(self):
  536. return Qt.MoveAction
  537. def flags(self, index):
  538. default_flags = QtCore.QAbstractItemModel.flags(self, index)
  539. if not index.isValid():
  540. return Qt.ItemIsEnabled | default_flags
  541. # Prevent groups from selection
  542. if not index.internalPointer().obj:
  543. return Qt.ItemIsEnabled
  544. else:
  545. return Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsEditable | \
  546. Qt.ItemIsDragEnabled | Qt.ItemIsDropEnabled
  547. # return QtWidgets.QAbstractItemModel.flags(self, index)
  548. def append(self, obj, active=False):
  549. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + " --> OC.append()")
  550. name = obj.options["name"]
  551. # Check promises and clear if exists
  552. if name in self.promises:
  553. self.promises.remove(name)
  554. # FlatCAMApp.App.log.debug("Promised object %s became available." % name)
  555. # FlatCAMApp.App.log.debug("%d promised objects remaining." % len(self.promises))
  556. # Prevent same name
  557. while name in self.get_names():
  558. ## Create a new name
  559. # Ends with number?
  560. FlatCAMApp.App.log.debug("new_object(): Object name (%s) exists, changing." % name)
  561. match = re.search(r'(.*[^\d])?(\d+)$', name)
  562. if match: # Yes: Increment the number!
  563. base = match.group(1) or ''
  564. num = int(match.group(2))
  565. name = base + str(num + 1)
  566. else: # No: add a number!
  567. name += "_1"
  568. obj.options["name"] = name
  569. obj.set_ui(obj.ui_type())
  570. # Required before appending (Qt MVC)
  571. group = self.group_items[obj.kind]
  572. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  573. self.beginInsertRows(group_index, group.child_count(), group.child_count())
  574. # Append new item
  575. obj.item = TreeItem(None, self.icons[obj.kind], obj, group)
  576. # Required after appending (Qt MVC)
  577. self.endInsertRows()
  578. # Expand group
  579. if group.child_count() is 1:
  580. self.view.setExpanded(group_index, True)
  581. def get_names(self):
  582. """
  583. Gets a list of the names of all objects in the collection.
  584. :return: List of names.
  585. :rtype: list
  586. """
  587. # FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + " --> OC.get_names()")
  588. return [x.options['name'] for x in self.get_list()]
  589. def get_bounds(self):
  590. """
  591. Finds coordinates bounding all objects in the collection.
  592. :return: [xmin, ymin, xmax, ymax]
  593. :rtype: list
  594. """
  595. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + "--> OC.get_bounds()")
  596. # TODO: Move the operation out of here.
  597. xmin = Inf
  598. ymin = Inf
  599. xmax = -Inf
  600. ymax = -Inf
  601. # for obj in self.object_list:
  602. for obj in self.get_list():
  603. try:
  604. gxmin, gymin, gxmax, gymax = obj.bounds()
  605. xmin = min([xmin, gxmin])
  606. ymin = min([ymin, gymin])
  607. xmax = max([xmax, gxmax])
  608. ymax = max([ymax, gymax])
  609. except:
  610. FlatCAMApp.App.log.warning("DEV WARNING: Tried to get bounds of empty geometry.")
  611. return [xmin, ymin, xmax, ymax]
  612. def get_by_name(self, name, isCaseSensitive=None):
  613. """
  614. Fetches the FlatCAMObj with the given `name`.
  615. :param name: The name of the object.
  616. :type name: str
  617. :return: The requested object or None if no such object.
  618. :rtype: FlatCAMObj or None
  619. """
  620. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + "--> OC.get_by_name()")
  621. if isCaseSensitive is None or isCaseSensitive is True:
  622. for obj in self.get_list():
  623. if obj.options['name'] == name:
  624. return obj
  625. else:
  626. for obj in self.get_list():
  627. if obj.options['name'].lower() == name.lower():
  628. return obj
  629. return None
  630. def delete_active(self):
  631. selections = self.view.selectedIndexes()
  632. if len(selections) == 0:
  633. return
  634. active = selections[0].internalPointer()
  635. group = active.parent_item
  636. # update the SHELL auto-completer model data
  637. name = active.obj.options['name']
  638. try:
  639. self.app.myKeywords.remove(name)
  640. self.app.shell._edit.set_model_data(self.app.myKeywords)
  641. except:
  642. log.debug(
  643. "delete_active() --> Could not remove the old object name from auto-completer model list")
  644. self.beginRemoveRows(self.index(group.row(), 0, QtCore.QModelIndex()), active.row(), active.row())
  645. group.remove_child(active)
  646. # after deletion of object store the current list of objects into the self.app.all_objects_list
  647. self.app.all_objects_list = self.get_list()
  648. self.endRemoveRows()
  649. # always go to the Project Tab after object deletion as it may be done with a shortcut key
  650. self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
  651. def delete_all(self):
  652. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + "--> OC.delete_all()")
  653. self.beginResetModel()
  654. self.checked_indexes = []
  655. for group in self.root_item.child_items:
  656. group.remove_children()
  657. self.endResetModel()
  658. self.app.plotcanvas.redraw()
  659. self.app.all_objects_list.clear()
  660. self.app.geo_editor.clear()
  661. self.app.exc_editor.clear()
  662. self.app.dblsidedtool.reset_fields()
  663. self.app.panelize_tool.reset_fields()
  664. self.app.cutout_tool.reset_fields()
  665. self.app.film_tool.reset_fields()
  666. def get_active(self):
  667. """
  668. Returns the active object or None
  669. :return: FlatCAMObj or None
  670. """
  671. selections = self.view.selectedIndexes()
  672. if len(selections) == 0:
  673. return None
  674. return selections[0].internalPointer().obj
  675. def get_selected(self):
  676. """
  677. Returns list of objects selected in the view.
  678. :return: List of objects
  679. """
  680. return [sel.internalPointer().obj for sel in self.view.selectedIndexes()]
  681. def get_non_selected(self):
  682. """
  683. Returns list of objects non-selected in the view.
  684. :return: List of objects
  685. """
  686. l = self.get_list()
  687. for sel in self.get_selected():
  688. l.remove(sel)
  689. return l
  690. def set_active(self, name):
  691. """
  692. Selects object by name from the project list. This triggers the
  693. list_selection_changed event and call on_list_selection_changed.
  694. :param name: Name of the FlatCAM Object
  695. :return: None
  696. """
  697. try:
  698. obj = self.get_by_name(name)
  699. item = obj.item
  700. group = self.group_items[obj.kind]
  701. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  702. item_index = self.index(item.row(), 0, group_index)
  703. self.view.selectionModel().select(item_index, QtCore.QItemSelectionModel.Select)
  704. except Exception as e:
  705. log.error("[ERROR] Cause: %s" % str(e))
  706. raise
  707. def set_inactive(self, name):
  708. """
  709. Unselect object by name from the project list. This triggers the
  710. list_selection_changed event and call on_list_selection_changed.
  711. :param name: Name of the FlatCAM Object
  712. :return: None
  713. """
  714. obj = self.get_by_name(name)
  715. item = obj.item
  716. group = self.group_items[obj.kind]
  717. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  718. item_index = self.index(item.row(), 0, group_index)
  719. self.view.selectionModel().select(item_index, QtCore.QItemSelectionModel.Deselect)
  720. def set_all_inactive(self):
  721. """
  722. Unselect all objects from the project list. This triggers the
  723. list_selection_changed event and call on_list_selection_changed.
  724. :return: None
  725. """
  726. for name in self.get_names():
  727. self.set_inactive(name)
  728. def on_list_selection_change(self, current, previous):
  729. # FlatCAMApp.App.log.debug("on_list_selection_change()")
  730. # FlatCAMApp.App.log.debug("Current: %s, Previous %s" % (str(current), str(previous)))
  731. try:
  732. obj = current.indexes()[0].internalPointer().obj
  733. except IndexError:
  734. FlatCAMApp.App.log.debug("on_list_selection_change(): Index Error (Nothing selected?)")
  735. try:
  736. self.app.ui.selected_scroll_area.takeWidget()
  737. except:
  738. FlatCAMApp.App.log.debug("Nothing to remove")
  739. self.app.setup_component_editor()
  740. return
  741. if obj:
  742. obj.build_ui()
  743. def on_item_activated(self, index):
  744. """
  745. Double-click or Enter on item.
  746. :param index: Index of the item in the list.
  747. :return: None
  748. """
  749. a_idx = index.internalPointer().obj
  750. if a_idx is None:
  751. return
  752. else:
  753. try:
  754. a_idx.build_ui()
  755. except Exception as e:
  756. self.app.inform.emit("[ERROR] Cause of error: %s" % str(e))
  757. raise
  758. def get_list(self):
  759. obj_list = []
  760. for group in self.root_item.child_items:
  761. for item in group.child_items:
  762. obj_list.append(item.obj)
  763. return obj_list
  764. def update_view(self):
  765. self.dataChanged.emit(QtCore.QModelIndex(), QtCore.QModelIndex())