ObjectCollection.py 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964
  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.menuprojectviewsource.setEnabled(sel)
  446. self.app.ui.menuprojectcopy.setEnabled(sel)
  447. self.app.ui.menuprojectedit.setEnabled(sel)
  448. self.app.ui.menuprojectdelete.setEnabled(sel)
  449. self.app.ui.menuprojectsave.setEnabled(sel)
  450. self.app.ui.menuprojectproperties.setEnabled(sel)
  451. if sel:
  452. self.app.ui.menuprojectgeneratecnc.setVisible(True)
  453. self.app.ui.menuprojectedit.setVisible(True)
  454. self.app.ui.menuprojectsave.setVisible(True)
  455. self.app.ui.menuprojectviewsource.setVisible(True)
  456. for obj in self.get_selected():
  457. if type(obj) != FlatCAMGeometry:
  458. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  459. if type(obj) != FlatCAMGeometry and type(obj) != FlatCAMExcellon:
  460. self.app.ui.menuprojectedit.setVisible(False)
  461. if type(obj) != FlatCAMGeometry and type(obj) != FlatCAMExcellon and type(obj) != FlatCAMCNCjob:
  462. self.app.ui.menuprojectsave.setVisible(False)
  463. if type(obj) != FlatCAMGerber and type(obj) != FlatCAMExcellon:
  464. self.app.ui.menuprojectviewsource.setVisible(False)
  465. else:
  466. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  467. self.app.ui.menuproject.popup(self.view.mapToGlobal(pos))
  468. def index(self, row, column=0, parent=None, *args, **kwargs):
  469. if not self.hasIndex(row, column, parent):
  470. return QtCore.QModelIndex()
  471. # if not parent.isValid():
  472. # parent_item = self.root_item
  473. # else:
  474. # parent_item = parent.internalPointer()
  475. parent_item = parent.internalPointer() if parent.isValid() else self.root_item
  476. child_item = parent_item.child(row)
  477. if child_item:
  478. return self.createIndex(row, column, child_item)
  479. else:
  480. return QtCore.QModelIndex()
  481. def parent(self, index=None):
  482. if not index.isValid():
  483. return QtCore.QModelIndex()
  484. parent_item = index.internalPointer().parent_item
  485. if parent_item == self.root_item:
  486. return QtCore.QModelIndex()
  487. return self.createIndex(parent_item.row(), 0, parent_item)
  488. def rowCount(self, index=None, *args, **kwargs):
  489. if index.column() > 0:
  490. return 0
  491. if not index.isValid():
  492. parent_item = self.root_item
  493. else:
  494. parent_item = index.internalPointer()
  495. return parent_item.child_count()
  496. def columnCount(self, index=None, *args, **kwargs):
  497. if index.isValid():
  498. return index.internalPointer().column_count()
  499. else:
  500. return self.root_item.column_count()
  501. def data(self, index, role=None):
  502. if not index.isValid():
  503. return None
  504. if role in [Qt.DisplayRole, Qt.EditRole]:
  505. obj = index.internalPointer().obj
  506. if obj:
  507. return obj.options["name"]
  508. else:
  509. return index.internalPointer().data(index.column())
  510. if role == Qt.ForegroundRole:
  511. obj = index.internalPointer().obj
  512. if obj:
  513. return QtGui.QBrush(QtCore.Qt.black) if obj.options["plot"] else QtGui.QBrush(QtCore.Qt.darkGray)
  514. else:
  515. return index.internalPointer().data(index.column())
  516. elif role == Qt.DecorationRole:
  517. icon = index.internalPointer().icon
  518. if icon:
  519. return icon
  520. else:
  521. return QtGui.QPixmap()
  522. else:
  523. return None
  524. def setData(self, index, data, role=None):
  525. if index.isValid():
  526. obj = index.internalPointer().obj
  527. if obj:
  528. old_name = obj.options['name']
  529. # rename the object
  530. obj.options["name"] = str(data)
  531. new_name = obj.options['name']
  532. # update the SHELL auto-completer model data
  533. try:
  534. self.app.myKeywords.remove(old_name)
  535. self.app.myKeywords.append(new_name)
  536. self.app.shell._edit.set_model_data(self.app.myKeywords)
  537. except:
  538. log.debug(
  539. "setData() --> Could not remove the old object name from auto-completer model list")
  540. obj.build_ui()
  541. self.app.inform.emit("Object renamed from %s to %s" % (old_name, new_name))
  542. return True
  543. def supportedDropActions(self):
  544. return Qt.MoveAction
  545. def flags(self, index):
  546. default_flags = QtCore.QAbstractItemModel.flags(self, index)
  547. if not index.isValid():
  548. return Qt.ItemIsEnabled | default_flags
  549. # Prevent groups from selection
  550. if not index.internalPointer().obj:
  551. return Qt.ItemIsEnabled
  552. else:
  553. return Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsEditable | \
  554. Qt.ItemIsDragEnabled | Qt.ItemIsDropEnabled
  555. # return QtWidgets.QAbstractItemModel.flags(self, index)
  556. def append(self, obj, active=False):
  557. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + " --> OC.append()")
  558. name = obj.options["name"]
  559. # Check promises and clear if exists
  560. if name in self.promises:
  561. self.promises.remove(name)
  562. # FlatCAMApp.App.log.debug("Promised object %s became available." % name)
  563. # FlatCAMApp.App.log.debug("%d promised objects remaining." % len(self.promises))
  564. # Prevent same name
  565. while name in self.get_names():
  566. ## Create a new name
  567. # Ends with number?
  568. FlatCAMApp.App.log.debug("new_object(): Object name (%s) exists, changing." % name)
  569. match = re.search(r'(.*[^\d])?(\d+)$', name)
  570. if match: # Yes: Increment the number!
  571. base = match.group(1) or ''
  572. num = int(match.group(2))
  573. name = base + str(num + 1)
  574. else: # No: add a number!
  575. name += "_1"
  576. obj.options["name"] = name
  577. obj.set_ui(obj.ui_type())
  578. # Required before appending (Qt MVC)
  579. group = self.group_items[obj.kind]
  580. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  581. self.beginInsertRows(group_index, group.child_count(), group.child_count())
  582. # Append new item
  583. obj.item = TreeItem(None, self.icons[obj.kind], obj, group)
  584. # Required after appending (Qt MVC)
  585. self.endInsertRows()
  586. # Expand group
  587. if group.child_count() is 1:
  588. self.view.setExpanded(group_index, True)
  589. self.app.should_we_save = True
  590. # decide if to show or hide the Notebook side of the screen
  591. if self.app.defaults["global_project_autohide"] is True:
  592. # always open the notebook on object added to collection
  593. self.app.ui.splitter.setSizes([1, 1])
  594. def get_names(self):
  595. """
  596. Gets a list of the names of all objects in the collection.
  597. :return: List of names.
  598. :rtype: list
  599. """
  600. # FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + " --> OC.get_names()")
  601. return [x.options['name'] for x in self.get_list()]
  602. def get_bounds(self):
  603. """
  604. Finds coordinates bounding all objects in the collection.
  605. :return: [xmin, ymin, xmax, ymax]
  606. :rtype: list
  607. """
  608. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + "--> OC.get_bounds()")
  609. # TODO: Move the operation out of here.
  610. xmin = Inf
  611. ymin = Inf
  612. xmax = -Inf
  613. ymax = -Inf
  614. # for obj in self.object_list:
  615. for obj in self.get_list():
  616. try:
  617. gxmin, gymin, gxmax, gymax = obj.bounds()
  618. xmin = min([xmin, gxmin])
  619. ymin = min([ymin, gymin])
  620. xmax = max([xmax, gxmax])
  621. ymax = max([ymax, gymax])
  622. except:
  623. FlatCAMApp.App.log.warning("DEV WARNING: Tried to get bounds of empty geometry.")
  624. return [xmin, ymin, xmax, ymax]
  625. def get_by_name(self, name, isCaseSensitive=None):
  626. """
  627. Fetches the FlatCAMObj with the given `name`.
  628. :param name: The name of the object.
  629. :type name: str
  630. :return: The requested object or None if no such object.
  631. :rtype: FlatCAMObj or None
  632. """
  633. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + "--> OC.get_by_name()")
  634. if isCaseSensitive is None or isCaseSensitive is True:
  635. for obj in self.get_list():
  636. if obj.options['name'] == name:
  637. return obj
  638. else:
  639. for obj in self.get_list():
  640. if obj.options['name'].lower() == name.lower():
  641. return obj
  642. return None
  643. def delete_active(self):
  644. selections = self.view.selectedIndexes()
  645. if len(selections) == 0:
  646. return
  647. active = selections[0].internalPointer()
  648. group = active.parent_item
  649. # update the SHELL auto-completer model data
  650. name = active.obj.options['name']
  651. try:
  652. self.app.myKeywords.remove(name)
  653. self.app.shell._edit.set_model_data(self.app.myKeywords)
  654. except:
  655. log.debug(
  656. "delete_active() --> Could not remove the old object name from auto-completer model list")
  657. self.beginRemoveRows(self.index(group.row(), 0, QtCore.QModelIndex()), active.row(), active.row())
  658. group.remove_child(active)
  659. # after deletion of object store the current list of objects into the self.app.all_objects_list
  660. self.app.all_objects_list = self.get_list()
  661. self.endRemoveRows()
  662. # always go to the Project Tab after object deletion as it may be done with a shortcut key
  663. self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
  664. self.app.should_we_save = True
  665. # decide if to show or hide the Notebook side of the screen
  666. if self.app.defaults["global_project_autohide"] is True:
  667. # hide the notebook if there are no objects in the collection
  668. if not self.get_list():
  669. self.app.ui.splitter.setSizes([0, 1])
  670. def delete_all(self):
  671. FlatCAMApp.App.log.debug(str(inspect.stack()[1][3]) + "--> OC.delete_all()")
  672. self.beginResetModel()
  673. self.checked_indexes = []
  674. for group in self.root_item.child_items:
  675. group.remove_children()
  676. self.endResetModel()
  677. self.app.plotcanvas.redraw()
  678. self.app.all_objects_list.clear()
  679. self.app.geo_editor.clear()
  680. self.app.exc_editor.clear()
  681. self.app.dblsidedtool.reset_fields()
  682. self.app.panelize_tool.reset_fields()
  683. self.app.cutout_tool.reset_fields()
  684. self.app.film_tool.reset_fields()
  685. def get_active(self):
  686. """
  687. Returns the active object or None
  688. :return: FlatCAMObj or None
  689. """
  690. selections = self.view.selectedIndexes()
  691. if len(selections) == 0:
  692. return None
  693. return selections[0].internalPointer().obj
  694. def get_selected(self):
  695. """
  696. Returns list of objects selected in the view.
  697. :return: List of objects
  698. """
  699. return [sel.internalPointer().obj for sel in self.view.selectedIndexes()]
  700. def get_non_selected(self):
  701. """
  702. Returns list of objects non-selected in the view.
  703. :return: List of objects
  704. """
  705. l = self.get_list()
  706. for sel in self.get_selected():
  707. l.remove(sel)
  708. return l
  709. def set_active(self, name):
  710. """
  711. Selects object by name from the project list. This triggers the
  712. list_selection_changed event and call on_list_selection_changed.
  713. :param name: Name of the FlatCAM Object
  714. :return: None
  715. """
  716. try:
  717. obj = self.get_by_name(name)
  718. item = obj.item
  719. group = self.group_items[obj.kind]
  720. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  721. item_index = self.index(item.row(), 0, group_index)
  722. self.view.selectionModel().select(item_index, QtCore.QItemSelectionModel.Select)
  723. except Exception as e:
  724. log.error("[ERROR] Cause: %s" % str(e))
  725. raise
  726. def set_inactive(self, name):
  727. """
  728. Unselect object by name from the project list. This triggers the
  729. list_selection_changed event and call on_list_selection_changed.
  730. :param name: Name of the FlatCAM Object
  731. :return: None
  732. """
  733. obj = self.get_by_name(name)
  734. item = obj.item
  735. group = self.group_items[obj.kind]
  736. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  737. item_index = self.index(item.row(), 0, group_index)
  738. self.view.selectionModel().select(item_index, QtCore.QItemSelectionModel.Deselect)
  739. def set_all_inactive(self):
  740. """
  741. Unselect all objects from the project list. This triggers the
  742. list_selection_changed event and call on_list_selection_changed.
  743. :return: None
  744. """
  745. for name in self.get_names():
  746. self.set_inactive(name)
  747. def on_list_selection_change(self, current, previous):
  748. # FlatCAMApp.App.log.debug("on_list_selection_change()")
  749. # FlatCAMApp.App.log.debug("Current: %s, Previous %s" % (str(current), str(previous)))
  750. try:
  751. obj = current.indexes()[0].internalPointer().obj
  752. if obj.kind == 'gerber':
  753. self.app.inform.emit('[selected]<span style="color:%s;">%s</span> selected' %
  754. ('green', str(obj.options['name'])))
  755. elif obj.kind == 'excellon':
  756. self.app.inform.emit('[selected]<span style="color:%s;">%s</span> selected' %
  757. ('brown', str(obj.options['name'])))
  758. elif obj.kind == 'cncjob':
  759. self.app.inform.emit('[selected]<span style="color:%s;">%s</span> selected' %
  760. ('blue', str(obj.options['name'])))
  761. elif obj.kind == 'geometry':
  762. self.app.inform.emit('[selected]<span style="color:%s;">%s</span> selected' %
  763. ('red', str(obj.options['name'])))
  764. except IndexError:
  765. FlatCAMApp.App.log.debug("on_list_selection_change(): Index Error (Nothing selected?)")
  766. self.app.inform.emit('')
  767. try:
  768. self.app.ui.selected_scroll_area.takeWidget()
  769. except:
  770. FlatCAMApp.App.log.debug("Nothing to remove")
  771. self.app.setup_component_editor()
  772. return
  773. if obj:
  774. obj.build_ui()
  775. def on_item_activated(self, index):
  776. """
  777. Double-click or Enter on item.
  778. :param index: Index of the item in the list.
  779. :return: None
  780. """
  781. a_idx = index.internalPointer().obj
  782. if a_idx is None:
  783. return
  784. else:
  785. try:
  786. a_idx.build_ui()
  787. except Exception as e:
  788. self.app.inform.emit("[ERROR] Cause of error: %s" % str(e))
  789. raise
  790. def get_list(self):
  791. obj_list = []
  792. for group in self.root_item.child_items:
  793. for item in group.child_items:
  794. obj_list.append(item.obj)
  795. return obj_list
  796. def update_view(self):
  797. self.dataChanged.emit(QtCore.QModelIndex(), QtCore.QModelIndex())