ObjectCollection.py 44 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205
  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. # ##########################################################
  9. # File modified by: Dennis Hayrullin #
  10. # File modified by: Marius Stanciu #
  11. # ##########################################################
  12. from PyQt5 import QtGui, QtCore, QtWidgets
  13. from PyQt5.QtCore import Qt, QSettings
  14. from PyQt5.QtGui import QColor
  15. # from PyQt5.QtCore import QModelIndex
  16. from appObjects.FlatCAMObj import FlatCAMObj
  17. from appObjects.FlatCAMCNCJob import CNCJobObject
  18. from appObjects.FlatCAMDocument import DocumentObject
  19. from appObjects.FlatCAMExcellon import ExcellonObject
  20. from appObjects.FlatCAMGeometry import GeometryObject
  21. from appObjects.FlatCAMGerber import GerberObject
  22. from appObjects.FlatCAMScript import ScriptObject
  23. import inspect # TODO: Remove
  24. import re
  25. import logging
  26. from copy import deepcopy
  27. from numpy import Inf
  28. import gettext
  29. import appTranslation as fcTranslate
  30. import builtins
  31. fcTranslate.apply_language('strings')
  32. if '_' not in builtins.__dict__:
  33. _ = gettext.gettext
  34. log = logging.getLogger('base')
  35. class KeySensitiveListView(QtWidgets.QTreeView):
  36. """
  37. QtGui.QListView extended to emit a signal on key press.
  38. """
  39. def __init__(self, app, parent=None):
  40. super(KeySensitiveListView, self).__init__(parent)
  41. self.setHeaderHidden(True)
  42. # self.setRootIsDecorated(False)
  43. # self.setExpandsOnDoubleClick(False)
  44. self.setEditTriggers(QtWidgets.QTreeView.NoEditTriggers) # No edit in the Project Tab Tree
  45. # Enable dragging and dropping onto the appGUI
  46. self.setAcceptDrops(True)
  47. self.filename = ""
  48. self.app = app
  49. # Enabling Drag and Drop for the items in the Project Tab
  50. # Example: https://github.com/d1vanov/PyQt5-reorderable-list-model/blob/master/reorderable_list_model.py
  51. # https://github.com/jimmykuu/PyQt-PySide-Cookbook/blob/master/tree/drop_indicator.md
  52. # self.setDragEnabled(True)
  53. # self.viewport().setAcceptDrops(True)
  54. # self.setDropIndicatorShown(True)
  55. # self.DragDropMode(QtWidgets.QAbstractItemView.InternalMove)
  56. # self.current_idx = None
  57. # self.current_group = None
  58. # self.dropped_obj = None
  59. keyPressed = QtCore.pyqtSignal(int)
  60. def keyPressEvent(self, event):
  61. # super(KeySensitiveListView, self).keyPressEvent(event)
  62. self.keyPressed.emit(event.key())
  63. def dragEnterEvent(self, event):
  64. # if event.source():
  65. # self.current_idx = self.currentIndex()
  66. # self.current_group = self.model().group_items[self.current_idx.internalPointer().obj.kind]
  67. # self.dropped_obj = self.current_idx.internalPointer().obj
  68. if event.mimeData().hasUrls:
  69. event.accept()
  70. else:
  71. event.ignore()
  72. def dragMoveEvent(self, event):
  73. self.setDropIndicatorShown(True)
  74. if event.mimeData().hasUrls:
  75. event.accept()
  76. else:
  77. event.ignore()
  78. def dropEvent(self, event):
  79. drop_indicator = self.dropIndicatorPosition()
  80. # if event.source():
  81. # new_index = self.indexAt(event.pos())
  82. # new_group = self.model().group_items[new_index.internalPointer().obj.kind]
  83. # if self.current_group == new_group:
  84. #
  85. # # delete it from the model
  86. # deleted_obj_name = self.dropped_obj.options['name']
  87. # self.model().delete_by_name(deleted_obj_name)
  88. #
  89. # # add the object to the new index
  90. # self.model().append(self.dropped_obj, to_index=new_index)
  91. #
  92. # return
  93. m = event.mimeData()
  94. if m.hasUrls:
  95. event.accept()
  96. for url in m.urls():
  97. self.filename = str(url.toLocalFile())
  98. # file drop from outside application
  99. if drop_indicator == QtWidgets.QAbstractItemView.OnItem:
  100. if self.filename == "":
  101. self.app.inform.emit(_("Cancelled."))
  102. else:
  103. if self.filename.lower().rpartition('.')[-1] in self.app.grb_list:
  104. self.app.worker_task.emit({'fcn': self.app.open_gerber,
  105. 'params': [self.filename]})
  106. else:
  107. event.ignore()
  108. if self.filename.lower().rpartition('.')[-1] in self.app.exc_list:
  109. self.app.worker_task.emit({'fcn': self.app.open_excellon,
  110. 'params': [self.filename]})
  111. else:
  112. event.ignore()
  113. if self.filename.lower().rpartition('.')[-1] in self.app.gcode_list:
  114. self.app.worker_task.emit({'fcn': self.app.open_gcode,
  115. 'params': [self.filename]})
  116. else:
  117. event.ignore()
  118. if self.filename.lower().rpartition('.')[-1] in self.app.svg_list:
  119. object_type = 'geometry'
  120. self.app.worker_task.emit({'fcn': self.app.import_svg,
  121. 'params': [self.filename, object_type, None]})
  122. if self.filename.lower().rpartition('.')[-1] in self.app.dxf_list:
  123. object_type = 'geometry'
  124. self.app.worker_task.emit({'fcn': self.app.import_dxf,
  125. 'params': [self.filename, object_type, None]})
  126. if self.filename.lower().rpartition('.')[-1] in self.app.prj_list:
  127. # self.app.open_project() is not Thread Safe
  128. self.app.open_project(self.filename)
  129. else:
  130. event.ignore()
  131. else:
  132. pass
  133. else:
  134. event.ignore()
  135. class TreeItem(KeySensitiveListView):
  136. """
  137. Item of a tree model
  138. """
  139. def __init__(self, data, icon=None, obj=None, parent_item=None):
  140. super(TreeItem, self).__init__(parent_item)
  141. self.parent_item = parent_item
  142. self.item_data = data # Columns string data
  143. self.icon = icon # Decoration
  144. self.obj = obj # FlatCAMObj
  145. self.child_items = []
  146. if parent_item:
  147. parent_item.append_child(self)
  148. def append_child(self, item):
  149. self.child_items.append(item)
  150. item.set_parent_item(self)
  151. def remove_child(self, item):
  152. child = self.child_items.pop(self.child_items.index(item))
  153. child.obj.clear(True)
  154. child.obj.delete()
  155. del child.obj
  156. del child
  157. def remove_children(self):
  158. for child in self.child_items:
  159. child.obj.clear()
  160. child.obj.delete()
  161. del child.obj
  162. del child
  163. self.child_items = []
  164. def child(self, row):
  165. return self.child_items[row]
  166. def child_count(self):
  167. return len(self.child_items)
  168. def column_count(self):
  169. return len(self.item_data)
  170. def data(self, column):
  171. return self.item_data[column]
  172. def row(self):
  173. return self.parent_item.child_items.index(self)
  174. def set_parent_item(self, parent_item):
  175. self.parent_item = parent_item
  176. def __del__(self):
  177. del self.icon
  178. class ObjectCollection(QtCore.QAbstractItemModel):
  179. """
  180. Object storage and management.
  181. """
  182. groups = [
  183. ("gerber", "Gerber"),
  184. ("excellon", "Excellon"),
  185. ("geometry", "Geometry"),
  186. ("cncjob", "CNC Job"),
  187. ("script", "Scripts"),
  188. ("document", "Document"),
  189. ]
  190. classdict = {
  191. "gerber": GerberObject,
  192. "excellon": ExcellonObject,
  193. "cncjob": CNCJobObject,
  194. "geometry": GeometryObject,
  195. "script": ScriptObject,
  196. "document": DocumentObject
  197. }
  198. icon_files = {
  199. "gerber": "assets/resources/flatcam_icon16.png",
  200. "excellon": "assets/resources/drill16.png",
  201. "cncjob": "assets/resources/cnc16.png",
  202. "geometry": "assets/resources/geometry16.png",
  203. "script": "assets/resources/script_new16.png",
  204. "document": "assets/resources/notes16_1.png"
  205. }
  206. # will emit the name of the object that was just selected
  207. item_selected = QtCore.pyqtSignal(str)
  208. update_list_signal = QtCore.pyqtSignal()
  209. root_item = None
  210. # app = None
  211. def __init__(self, app, parent=None):
  212. QtCore.QAbstractItemModel.__init__(self)
  213. self.app = app
  214. # ## Icons for the list view
  215. self.icons = {}
  216. for kind in ObjectCollection.icon_files:
  217. self.icons[kind] = QtGui.QPixmap(
  218. ObjectCollection.icon_files[kind].replace('assets/resources', self.app.resource_location))
  219. # Create root tree view item
  220. self.root_item = TreeItem(["root"])
  221. # Create group items
  222. self.group_items = {}
  223. for kind, title in ObjectCollection.groups:
  224. item = TreeItem([title], self.icons[kind])
  225. self.group_items[kind] = item
  226. self.root_item.append_child(item)
  227. # Create test sub-items
  228. # for i in self.root_item.m_child_items:
  229. # print i.data(0)
  230. # i.append_child(TreeItem(["empty"]))
  231. # ## Data # ##
  232. self.checked_indexes = []
  233. # Names of objects that are expected to become available.
  234. # For example, when the creation of a new object will run
  235. # in the background and will complete some time in the
  236. # future. This is a way to reserve the name and to let other
  237. # tasks know that they have to wait until available.
  238. self.promises = set()
  239. # same as above only for objects that are plotted
  240. self.plot_promises = set()
  241. # ## View
  242. self.view = KeySensitiveListView(self.app)
  243. self.view.setModel(self)
  244. self.view.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
  245. self.view.setSelectionMode(QtWidgets.QAbstractItemView.ExtendedSelection)
  246. if self.app.defaults["global_allow_edit_in_project_tab"] is True:
  247. self.view.setEditTriggers(QtWidgets.QTreeView.SelectedClicked) # allow Edit on Tree
  248. else:
  249. self.view.setEditTriggers(QtWidgets.QTreeView.NoEditTriggers)
  250. # self.view.setDragDropMode(QtWidgets.QAbstractItemView.InternalMove)
  251. # self.view.setDragEnabled(True)
  252. # self.view.setAcceptDrops(True)
  253. # self.view.setDropIndicatorShown(True)
  254. settings = QSettings("Open Source", "FlatCAM")
  255. if settings.contains("notebook_font_size"):
  256. fsize = settings.value('notebook_font_size', type=int)
  257. else:
  258. fsize = 12
  259. font = QtGui.QFont()
  260. font.setPixelSize(fsize)
  261. font.setFamily("Seagoe UI")
  262. self.view.setFont(font)
  263. # ## GUI Events
  264. self.view.selectionModel().selectionChanged.connect(self.on_list_selection_change)
  265. # self.view.activated.connect(self.on_item_activated)
  266. self.view.keyPressed.connect(self.app.ui.keyPressEvent)
  267. # self.view.clicked.connect(self.on_mouse_down)
  268. self.view.customContextMenuRequested.connect(self.on_menu_request)
  269. self.click_modifier = None
  270. self.update_list_signal.connect(self.on_update_list_signal)
  271. self.view.activated.connect(self.on_row_activated)
  272. self.item_selected.connect(self.on_row_selected)
  273. def promise(self, obj_name):
  274. log.debug("Object %s has been promised." % obj_name)
  275. self.promises.add(obj_name)
  276. def has_promises(self):
  277. return len(self.promises) > 0
  278. def plot_promise(self, plot_obj_name):
  279. self.plot_promises.add(plot_obj_name)
  280. def plot_remove_promise(self, plot_obj_name):
  281. if plot_obj_name in self.plot_promises:
  282. self.plot_promises.remove(plot_obj_name)
  283. def has_plot_promises(self):
  284. return len(self.plot_promises) > 0
  285. def on_mouse_down(self, event):
  286. log.debug("Mouse button pressed on list")
  287. def on_menu_request(self, pos):
  288. sel = len(self.view.selectedIndexes()) > 0
  289. self.app.ui.menuprojectenable.setEnabled(sel)
  290. self.app.ui.menuprojectdisable.setEnabled(sel)
  291. self.app.ui.menuprojectcolor.setEnabled(sel)
  292. self.app.ui.menuprojectviewsource.setEnabled(sel)
  293. self.app.ui.menuprojectcopy.setEnabled(sel)
  294. self.app.ui.menuprojectedit.setEnabled(sel)
  295. self.app.ui.menuprojectdelete.setEnabled(sel)
  296. self.app.ui.menuprojectsave.setEnabled(sel)
  297. self.app.ui.menuprojectproperties.setEnabled(sel)
  298. if sel:
  299. self.app.ui.menuprojectgeneratecnc.setVisible(True)
  300. self.app.ui.menuprojectedit.setVisible(True)
  301. self.app.ui.menuprojectsave.setVisible(True)
  302. self.app.ui.menuprojectviewsource.setVisible(True)
  303. self.app.ui.menuprojectcolor.setEnabled(False)
  304. for obj in self.get_selected():
  305. if type(obj) == GerberObject or type(obj) == ExcellonObject:
  306. self.app.ui.menuprojectcolor.setEnabled(True)
  307. if type(obj) != GeometryObject:
  308. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  309. # if type(obj) != GeometryObject and type(obj) != ExcellonObject and type(obj) != GerberObject or \
  310. # type(obj) != CNCJobObject:
  311. # self.app.ui.menuprojectedit.setVisible(False)
  312. if type(obj) != GerberObject and type(obj) != ExcellonObject and type(obj) != CNCJobObject:
  313. self.app.ui.menuprojectviewsource.setVisible(False)
  314. if type(obj) != GerberObject and type(obj) != GeometryObject and type(obj) != ExcellonObject and \
  315. type(obj) != CNCJobObject:
  316. # meaning for Scripts and for Document type of FlatCAM object
  317. self.app.ui.menuprojectenable.setVisible(False)
  318. self.app.ui.menuprojectdisable.setVisible(False)
  319. self.app.ui.menuprojectedit.setVisible(False)
  320. self.app.ui.menuprojectproperties.setVisible(False)
  321. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  322. else:
  323. self.app.ui.menuprojectgeneratecnc.setVisible(False)
  324. self.app.ui.menuproject.popup(self.view.mapToGlobal(pos))
  325. def index(self, row, column=0, parent=None, *args, **kwargs):
  326. if not self.hasIndex(row, column, parent):
  327. return QtCore.QModelIndex()
  328. # if not parent.isValid():
  329. # parent_item = self.root_item
  330. # else:
  331. # parent_item = parent.internalPointer()
  332. parent_item = parent.internalPointer() if parent.isValid() else self.root_item
  333. child_item = parent_item.child(row)
  334. if child_item:
  335. return self.createIndex(row, column, child_item)
  336. else:
  337. return QtCore.QModelIndex()
  338. def parent(self, index=None):
  339. if not index.isValid():
  340. return QtCore.QModelIndex()
  341. parent_item = index.internalPointer().parent_item
  342. if parent_item == self.root_item:
  343. return QtCore.QModelIndex()
  344. return self.createIndex(parent_item.row(), 0, parent_item)
  345. def rowCount(self, index=None, *args, **kwargs):
  346. if index.column() > 0:
  347. return 0
  348. if not index.isValid():
  349. parent_item = self.root_item
  350. else:
  351. parent_item = index.internalPointer()
  352. return parent_item.child_count()
  353. def columnCount(self, index=None, *args, **kwargs):
  354. if index.isValid():
  355. return index.internalPointer().column_count()
  356. else:
  357. return self.root_item.column_count()
  358. def data(self, index, role=None):
  359. if not index.isValid():
  360. return None
  361. if role in [Qt.DisplayRole, Qt.EditRole]:
  362. obj = index.internalPointer().obj
  363. if obj:
  364. return obj.options["name"]
  365. else:
  366. return index.internalPointer().data(index.column())
  367. if role == Qt.ForegroundRole:
  368. color = QColor(self.app.defaults['global_proj_item_color'])
  369. color_disabled = QColor(self.app.defaults['global_proj_item_dis_color'])
  370. obj = index.internalPointer().obj
  371. if obj:
  372. return QtGui.QBrush(color) if obj.options["plot"] else QtGui.QBrush(color_disabled)
  373. else:
  374. return index.internalPointer().data(index.column())
  375. elif role == Qt.DecorationRole:
  376. icon = index.internalPointer().icon
  377. if icon:
  378. return icon
  379. else:
  380. return QtGui.QPixmap()
  381. elif role == Qt.ToolTipRole:
  382. try:
  383. obj = index.internalPointer().obj
  384. except AttributeError:
  385. return None
  386. if obj:
  387. text = obj.options['name']
  388. return text
  389. else:
  390. QtWidgets.QToolTip.hideText()
  391. return None
  392. else:
  393. return None
  394. def setData(self, index, data, role=None):
  395. if index.isValid():
  396. obj = index.internalPointer().obj
  397. if obj:
  398. old_name = deepcopy(obj.options['name'])
  399. new_name = str(data)
  400. if old_name != new_name and new_name != '':
  401. # rename the object
  402. obj.options["name"] = deepcopy(data)
  403. self.app.object_status_changed.emit(obj, 'rename', old_name)
  404. # update the SHELL auto-completer model data
  405. try:
  406. self.app.myKeywords.remove(old_name)
  407. self.app.myKeywords.append(new_name)
  408. self.app.shell._edit.set_model_data(self.app.myKeywords)
  409. except Exception as e:
  410. log.debug(
  411. "setData() --> Could not remove the old object name from auto-completer model list. %s" %
  412. str(e))
  413. # obj.build_ui()
  414. self.app.inform.emit(_("Object renamed from <b>{old}</b> to <b>{new}</b>").format(old=old_name,
  415. new=new_name))
  416. self.dataChanged.emit(index, index)
  417. return True
  418. else:
  419. return False
  420. def supportedDropActions(self):
  421. return Qt.MoveAction
  422. def flags(self, index):
  423. default_flags = QtCore.QAbstractItemModel.flags(self, index)
  424. if not index.isValid():
  425. return Qt.ItemIsEnabled | default_flags
  426. # Prevent groups from selection
  427. try:
  428. if not index.internalPointer().obj:
  429. return Qt.ItemIsEnabled
  430. else:
  431. return Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsEditable | \
  432. Qt.ItemIsDragEnabled | Qt.ItemIsDropEnabled
  433. except AttributeError:
  434. return Qt.ItemIsEnabled
  435. # return QtWidgets.QAbstractItemModel.flags(self, index)
  436. def append(self, obj, active=False, to_index=None):
  437. log.debug(str(inspect.stack()[1][3]) + " --> OC.append()")
  438. name = obj.options["name"]
  439. # Check promises and clear if exists
  440. if name in self.promises:
  441. self.promises.remove(name)
  442. # log.debug("Promised object %s became available." % name)
  443. # log.debug("%d promised objects remaining." % len(self.promises))
  444. # Prevent same name
  445. while name in self.get_names():
  446. # ## Create a new name
  447. # Ends with number?
  448. log.debug("app_obj.new_object(): Object name (%s) exists, changing." % name)
  449. match = re.search(r'(.*[^\d])?(\d+)$', name)
  450. if match: # Yes: Increment the number!
  451. base = match.group(1) or ''
  452. num = int(match.group(2))
  453. name = base + str(num + 1)
  454. else: # No: add a number!
  455. name += "_1"
  456. obj.options["name"] = name
  457. obj.set_ui(obj.ui_type(app=self.app))
  458. # Required before appending (Qt MVC)
  459. group = self.group_items[obj.kind]
  460. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  461. if to_index is None:
  462. self.beginInsertRows(group_index, group.child_count(), group.child_count())
  463. # Append new item
  464. obj.item = TreeItem(None, self.icons[obj.kind], obj, group)
  465. # Required after appending (Qt MVC)
  466. self.endInsertRows()
  467. else:
  468. self.beginInsertRows(group_index, to_index.row()-1, to_index.row()-1)
  469. # Append new item
  470. obj.item = TreeItem(None, self.icons[obj.kind], obj, group)
  471. # Required after appending (Qt MVC)
  472. self.endInsertRows()
  473. # Expand group
  474. if group.child_count() == 1:
  475. self.view.setExpanded(group_index, True)
  476. self.app.should_we_save = True
  477. self.app.object_status_changed.emit(obj, 'append', name)
  478. # decide if to show or hide the Notebook side of the screen
  479. if self.app.defaults["global_project_autohide"] is True:
  480. # always open the notebook on object added to collection
  481. self.app.ui.splitter.setSizes([1, 1])
  482. def get_names(self):
  483. """
  484. Gets a list of the names of all objects in the collection.
  485. :return: List of names.
  486. :rtype: list
  487. """
  488. # log.debug(str(inspect.stack()[1][3]) + " --> OC.get_names()")
  489. return [x.options['name'] for x in self.get_list()]
  490. def get_bounds(self):
  491. """
  492. Finds coordinates bounding all objects in the collection.
  493. :return: [xmin, ymin, xmax, ymax]
  494. :rtype: list
  495. """
  496. log.debug(str(inspect.stack()[1][3]) + "--> OC.get_bounds()")
  497. # TODO: Move the operation out of here.
  498. xmin = Inf
  499. ymin = Inf
  500. xmax = -Inf
  501. ymax = -Inf
  502. # for obj in self.object_list:
  503. for obj in self.get_list():
  504. try:
  505. gxmin, gymin, gxmax, gymax = obj.bounds()
  506. xmin = min([xmin, gxmin])
  507. ymin = min([ymin, gymin])
  508. xmax = max([xmax, gxmax])
  509. ymax = max([ymax, gymax])
  510. except Exception as e:
  511. log.warning("DEV WARNING: Tried to get bounds of empty geometry. %s" % str(e))
  512. return [xmin, ymin, xmax, ymax]
  513. def get_by_name(self, name, isCaseSensitive=None):
  514. """
  515. Fetches the FlatCAMObj with the given `name`.
  516. :param name: The name of the object.
  517. :type name: str
  518. :param isCaseSensitive: whether searching of the object is done by name where the name is case sensitive
  519. :return: The requested object or None if no such object.
  520. :rtype: FlatCAMObj or None
  521. """
  522. # log.debug(str(inspect.stack()[1][3]) + "--> OC.get_by_name()")
  523. if isCaseSensitive is None or isCaseSensitive is True:
  524. for obj in self.get_list():
  525. if obj.options['name'] == name:
  526. return obj
  527. else:
  528. for obj in self.get_list():
  529. if obj.options['name'].lower() == name.lower():
  530. return obj
  531. return None
  532. def delete_active(self, select_project=True):
  533. selections = self.view.selectedIndexes()
  534. if len(selections) == 0:
  535. return
  536. active = selections[0].internalPointer()
  537. group = active.parent_item
  538. # send signal with the object that is deleted
  539. # self.app.object_status_changed.emit(active.obj, 'delete')
  540. # some objects add a Tab on creation, close it here
  541. for idx in range(self.app.ui.plot_tab_area.count()):
  542. if self.app.ui.plot_tab_area.widget(idx).objectName() == active.obj.options['name']:
  543. self.app.ui.plot_tab_area.removeTab(idx)
  544. break
  545. # update the SHELL auto-completer model data
  546. name = active.obj.options['name']
  547. try:
  548. self.app.myKeywords.remove(name)
  549. self.app.shell._edit.set_model_data(self.app.myKeywords)
  550. # this is not needed any more because now the code editor is created on demand
  551. # self.app.ui.code_editor.set_model_data(self.app.myKeywords)
  552. except Exception as e:
  553. log.debug(
  554. "delete_active() --> Could not remove the old object name from auto-completer model list. %s" % str(e))
  555. self.app.object_status_changed.emit(active.obj, 'delete', name)
  556. # ############ OBJECT DELETION FROM MODEL STARTS HERE ####################
  557. self.beginRemoveRows(self.index(group.row(), 0, QtCore.QModelIndex()), active.row(), active.row())
  558. group.remove_child(active)
  559. # after deletion of object store the current list of objects into the self.app.all_objects_list
  560. self.app.all_objects_list = self.get_list()
  561. self.endRemoveRows()
  562. # ############ OBJECT DELETION FROM MODEL STOPS HERE ####################
  563. if self.app.is_legacy is False:
  564. self.app.plotcanvas.redraw()
  565. if select_project:
  566. # always go to the Project Tab after object deletion as it may be done with a shortcut key
  567. self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
  568. self.app.should_we_save = True
  569. # decide if to show or hide the Notebook side of the screen
  570. if self.app.defaults["global_project_autohide"] is True:
  571. # hide the notebook if there are no objects in the collection
  572. if not self.get_list():
  573. self.app.ui.splitter.setSizes([0, 1])
  574. def delete_by_name(self, name, select_project=True):
  575. obj = self.get_by_name(name=name)
  576. item = obj.item
  577. group = self.group_items[obj.kind]
  578. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  579. item_index = self.index(item.row(), 0, group_index)
  580. deleted = item_index.internalPointer()
  581. group = deleted.parent_item
  582. # some objects add a Tab on creation, close it here
  583. for idx in range(self.app.ui.plot_tab_area.count()):
  584. if self.app.ui.plot_tab_area.widget(idx).objectName() == deleted.obj.options['name']:
  585. self.app.ui.plot_tab_area.removeTab(idx)
  586. break
  587. # update the SHELL auto-completer model data
  588. name = deleted.obj.options['name']
  589. try:
  590. self.app.myKeywords.remove(name)
  591. self.app.shell._edit.set_model_data(self.app.myKeywords)
  592. # this is not needed any more because now the code editor is created on demand
  593. # self.app.ui.code_editor.set_model_data(self.app.myKeywords)
  594. except Exception as e:
  595. log.debug(
  596. "delete_by_name() --> Could not remove the old object name from auto-completer model list. %s" % str(e))
  597. self.app.object_status_changed.emit(deleted.obj, 'delete', name)
  598. # ############ OBJECT DELETION FROM MODEL STARTS HERE ####################
  599. self.beginRemoveRows(self.index(group.row(), 0, QtCore.QModelIndex()), deleted.row(), deleted.row())
  600. group.remove_child(deleted)
  601. # after deletion of object store the current list of objects into the self.app.all_objects_list
  602. self.update_list_signal.emit()
  603. self.endRemoveRows()
  604. # ############ OBJECT DELETION FROM MODEL STOPS HERE ####################
  605. if self.app.is_legacy is False:
  606. self.app.plotcanvas.redraw()
  607. if select_project:
  608. # always go to the Project Tab after object deletion as it may be done with a shortcut key
  609. self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
  610. self.app.should_we_save = True
  611. # decide if to show or hide the Notebook side of the screen
  612. if self.app.defaults["global_project_autohide"] is True:
  613. # hide the notebook if there are no objects in the collection
  614. if not self.get_list():
  615. self.app.ui.splitter.setSizes([0, 1])
  616. def on_update_list_signal(self):
  617. self.app.all_objects_list = self.get_list()
  618. def delete_all(self):
  619. log.debug(str(inspect.stack()[1][3]) + "--> OC.delete_all()")
  620. self.app.object_status_changed.emit(None, 'delete_all', '')
  621. try:
  622. self.app.all_objects_list.clear()
  623. self.app.geo_editor.clear()
  624. self.app.exc_editor.clear()
  625. self.app.dblsidedtool.reset_fields()
  626. self.app.panelize_tool.reset_fields()
  627. self.app.cutout_tool.reset_fields()
  628. self.app.film_tool.reset_fields()
  629. self.beginResetModel()
  630. self.checked_indexes = []
  631. for group in self.root_item.child_items:
  632. group.remove_children()
  633. self.endResetModel()
  634. self.app.plotcanvas.redraw()
  635. except Exception as e:
  636. log.debug("ObjectCollection.delete_all() --> %s" % str(e))
  637. def get_active(self):
  638. """
  639. Returns the active object or None
  640. :return: FlatCAMObj or None
  641. """
  642. selections = self.view.selectedIndexes()
  643. if len(selections) == 0:
  644. return None
  645. return selections[0].internalPointer().obj
  646. def get_selected(self):
  647. """
  648. Returns list of objects selected in the view.
  649. :return: List of objects
  650. """
  651. return [sel.internalPointer().obj for sel in self.view.selectedIndexes()]
  652. def get_non_selected(self):
  653. """
  654. Returns list of objects non-selected in the view.
  655. :return: List of objects
  656. """
  657. obj_list = self.get_list()
  658. for sel in self.get_selected():
  659. obj_list.remove(sel)
  660. return obj_list
  661. def set_active(self, name):
  662. """
  663. Selects object by name from the project list. This triggers the
  664. list_selection_changed event and call on_list_selection_changed.
  665. :param name: Name of the FlatCAM Object
  666. :return: None
  667. """
  668. try:
  669. obj = self.get_by_name(name)
  670. item = obj.item
  671. group = self.group_items[obj.kind]
  672. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  673. item_index = self.index(item.row(), 0, group_index)
  674. self.view.selectionModel().select(item_index, QtCore.QItemSelectionModel.Select)
  675. except Exception as e:
  676. log.error("[ERROR] Cause: %s" % str(e))
  677. raise
  678. def set_all_active(self):
  679. """
  680. Select all objects from the project list. This triggers the
  681. list_selection_changed event and call on_list_selection_changed.
  682. :return: None
  683. """
  684. for name in self.get_names():
  685. self.set_active(name)
  686. def set_exclusive_active(self, name):
  687. """
  688. Make the object with the name in parameters the only selected object
  689. :param name: name of object to be selected and made the only active object
  690. :return: None
  691. """
  692. self.set_all_inactive()
  693. self.set_active(name)
  694. def set_inactive(self, name):
  695. """
  696. Unselect object by name from the project list. This triggers the
  697. list_selection_changed event and call on_list_selection_changed.
  698. :param name: Name of the FlatCAM Object
  699. :return: None
  700. """
  701. # log.debug("ObjectCollection.set_inactive()")
  702. obj = self.get_by_name(name)
  703. item = obj.item
  704. group = self.group_items[obj.kind]
  705. group_index = self.index(group.row(), 0, QtCore.QModelIndex())
  706. item_index = self.index(item.row(), 0, group_index)
  707. self.view.selectionModel().select(item_index, QtCore.QItemSelectionModel.Deselect)
  708. def set_all_inactive(self):
  709. """
  710. Unselect all objects from the project list. This triggers the
  711. list_selection_changed event and call on_list_selection_changed.
  712. :return: None
  713. """
  714. for name in self.get_names():
  715. self.set_inactive(name)
  716. def on_list_selection_change(self, current, previous):
  717. """
  718. :param current: Current selected item
  719. :param previous: Previously selected item
  720. :return:
  721. """
  722. # log.debug("on_list_selection_change()")
  723. # log.debug("Current: %s, Previous %s" % (str(current), str(previous)))
  724. try:
  725. obj = current.indexes()[0].internalPointer().obj
  726. self.item_selected.emit(obj.options['name'])
  727. if obj.kind == 'gerber':
  728. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  729. color='green',
  730. name=str(obj.options['name']),
  731. tx=_("selected"))
  732. )
  733. elif obj.kind == 'excellon':
  734. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  735. color='brown',
  736. name=str(obj.options['name']),
  737. tx=_("selected"))
  738. )
  739. elif obj.kind == 'cncjob':
  740. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  741. color='blue',
  742. name=str(obj.options['name']),
  743. tx=_("selected"))
  744. )
  745. elif obj.kind == 'geometry':
  746. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  747. color='red',
  748. name=str(obj.options['name']),
  749. tx=_("selected"))
  750. )
  751. elif obj.kind == 'script':
  752. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  753. color='orange',
  754. name=str(obj.options['name']),
  755. tx=_("selected"))
  756. )
  757. elif obj.kind == 'document':
  758. self.app.inform.emit('[selected]<span style="color:{color};">{name}</span> {tx}'.format(
  759. color='darkCyan',
  760. name=str(obj.options['name']),
  761. tx=_("selected"))
  762. )
  763. except IndexError:
  764. self.item_selected.emit('none')
  765. # 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 Exception as e:
  770. log.debug("Nothing to remove. %s" % str(e))
  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] %s: %s' % (_("Cause of error"), str(e)))
  789. raise
  790. def get_list(self):
  791. """
  792. Will return a list of all objects currently opened. Except FlatCAMScript and FlatCAMDocuments
  793. :return:
  794. """
  795. obj_list = []
  796. for group in self.root_item.child_items:
  797. for item in group.child_items:
  798. obj_list.append(item.obj)
  799. return obj_list
  800. def update_view(self):
  801. self.dataChanged.emit(QtCore.QModelIndex(), QtCore.QModelIndex())
  802. def on_row_activated(self, index):
  803. if index.isValid():
  804. if index.internalPointer().parent_item != self.root_item:
  805. self.app.ui.notebook.setCurrentWidget(self.app.ui.selected_tab)
  806. self.on_item_activated(index)
  807. def on_row_selected(self, obj_name):
  808. """
  809. This is a special string; when received it will make all Menu -> Objects entries unchecked
  810. It mean we clicked outside of the items and deselected all
  811. :param obj_name:
  812. :return:
  813. """
  814. if obj_name == 'none':
  815. for act in self.app.ui.menuobjects.actions():
  816. act.setChecked(False)
  817. return
  818. # get the name of the selected objects and add them to a list
  819. name_list = []
  820. for obj in self.get_selected():
  821. name_list.append(obj.options['name'])
  822. # set all actions as unchecked but the ones selected make them checked
  823. for act in self.app.ui.menuobjects.actions():
  824. act.setChecked(False)
  825. if act.text() in name_list:
  826. act.setChecked(True)
  827. def on_collection_updated(self, obj, state, old_name):
  828. """
  829. Create a menu from the object loaded in the collection.
  830. :param obj: object that was changed (added, deleted, renamed)
  831. :param state: what was done with the object. Can be: added, deleted, delete_all, renamed
  832. :param old_name: the old name of the object before the action that triggered this slot happened
  833. :return: None
  834. """
  835. icon_files = {
  836. "gerber": self.app.resource_location + "/flatcam_icon16.png",
  837. "excellon": self.app.resource_location + "/drill16.png",
  838. "cncjob": self.app.resource_location + "/cnc16.png",
  839. "geometry": self.app.resource_location + "/geometry16.png",
  840. "script": self.app.resource_location + "/script_new16.png",
  841. "document": self.app.resource_location + "/notes16_1.png"
  842. }
  843. if state == 'append':
  844. for act in self.app.ui.menuobjects.actions():
  845. try:
  846. act.triggered.disconnect()
  847. except TypeError:
  848. pass
  849. self.app.ui.menuobjects.clear()
  850. gerber_list = []
  851. exc_list = []
  852. cncjob_list = []
  853. geo_list = []
  854. script_list = []
  855. doc_list = []
  856. for name in self.get_names():
  857. obj_named = self.get_by_name(name)
  858. if obj_named.kind == 'gerber':
  859. gerber_list.append(name)
  860. elif obj_named.kind == 'excellon':
  861. exc_list.append(name)
  862. elif obj_named.kind == 'cncjob':
  863. cncjob_list.append(name)
  864. elif obj_named.kind == 'geometry':
  865. geo_list.append(name)
  866. elif obj_named.kind == 'script':
  867. script_list.append(name)
  868. elif obj_named.kind == 'document':
  869. doc_list.append(name)
  870. def add_act(o_name):
  871. obj_for_icon = self.get_by_name(o_name)
  872. add_action = QtWidgets.QAction(parent=self.app.ui.menuobjects)
  873. add_action.setCheckable(True)
  874. add_action.setText(o_name)
  875. add_action.setIcon(QtGui.QIcon(icon_files[obj_for_icon.kind]))
  876. add_action.triggered.connect(
  877. lambda: self.set_active(o_name) if add_action.isChecked() is True else
  878. self.set_inactive(o_name))
  879. self.app.ui.menuobjects.addAction(add_action)
  880. for name in gerber_list:
  881. add_act(name)
  882. self.app.ui.menuobjects.addSeparator()
  883. for name in exc_list:
  884. add_act(name)
  885. self.app.ui.menuobjects.addSeparator()
  886. for name in cncjob_list:
  887. add_act(name)
  888. self.app.ui.menuobjects.addSeparator()
  889. for name in geo_list:
  890. add_act(name)
  891. self.app.ui.menuobjects.addSeparator()
  892. for name in script_list:
  893. add_act(name)
  894. self.app.ui.menuobjects.addSeparator()
  895. for name in doc_list:
  896. add_act(name)
  897. self.app.ui.menuobjects.addSeparator()
  898. self.app.ui.menuobjects_selall = self.app.ui.menuobjects.addAction(
  899. QtGui.QIcon(self.app.resource_location + '/select_all.png'),
  900. _('Select All')
  901. )
  902. self.app.ui.menuobjects_unselall = self.app.ui.menuobjects.addAction(
  903. QtGui.QIcon(self.app.resource_location + '/deselect_all32.png'),
  904. _('Deselect All')
  905. )
  906. self.app.ui.menuobjects_selall.triggered.connect(lambda: self.on_objects_selection(True))
  907. self.app.ui.menuobjects_unselall.triggered.connect(lambda: self.on_objects_selection(False))
  908. elif state == 'delete':
  909. for act in self.app.ui.menuobjects.actions():
  910. if act.text() == obj.options['name']:
  911. try:
  912. act.triggered.disconnect()
  913. except TypeError:
  914. pass
  915. self.app.ui.menuobjects.removeAction(act)
  916. break
  917. elif state == 'rename':
  918. for act in self.app.ui.menuobjects.actions():
  919. if act.text() == old_name:
  920. add_action = QtWidgets.QAction(parent=self.app.ui.menuobjects)
  921. add_action.setText(obj.options['name'])
  922. add_action.setIcon(QtGui.QIcon(icon_files[obj.kind]))
  923. add_action.triggered.connect(
  924. lambda: self.set_active(obj.options['name']) if add_action.isChecked() is True else
  925. self.set_inactive(obj.options['name']))
  926. self.app.ui.menuobjects.insertAction(act, add_action)
  927. try:
  928. act.triggered.disconnect()
  929. except TypeError:
  930. pass
  931. self.app.ui.menuobjects.removeAction(act)
  932. break
  933. elif state == 'delete_all':
  934. for act in self.app.ui.menuobjects.actions():
  935. try:
  936. act.triggered.disconnect()
  937. except TypeError:
  938. pass
  939. self.app.ui.menuobjects.clear()
  940. self.app.ui.menuobjects.addSeparator()
  941. self.app.ui.menuobjects_selall = self.app.ui.menuobjects.addAction(
  942. QtGui.QIcon(self.app.resource_location + '/select_all.png'),
  943. _('Select All')
  944. )
  945. self.app.ui.menuobjects_unselall = self.app.ui.menuobjects.addAction(
  946. QtGui.QIcon(self.app.resource_location + '/deselect_all32.png'),
  947. _('Deselect All')
  948. )
  949. self.app.ui.menuobjects_selall.triggered.connect(lambda: self.on_objects_selection(True))
  950. self.app.ui.menuobjects_unselall.triggered.connect(lambda: self.on_objects_selection(False))
  951. def on_objects_selection(self, on_off):
  952. obj_list = self.get_names()
  953. if on_off is True:
  954. self.set_all_active()
  955. for act in self.app.ui.menuobjects.actions():
  956. try:
  957. act.setChecked(True)
  958. except Exception:
  959. pass
  960. if obj_list:
  961. self.app.inform[str, bool].emit('[selected] %s' % _("All objects are selected."), False)
  962. else:
  963. self.set_all_inactive()
  964. for act in self.app.ui.menuobjects.actions():
  965. try:
  966. act.setChecked(False)
  967. except Exception:
  968. pass
  969. if obj_list:
  970. self.app.inform[str, bool].emit('%s' % _("Objects selection is cleared."), False)
  971. else:
  972. self.app.inform[str, bool].emit('', False)