CuraSceneController.py 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109
  1. from UM.Logger import Logger
  2. from PyQt5.QtCore import Qt, pyqtSlot, QObject
  3. from PyQt5.QtWidgets import QApplication
  4. from cura.ObjectsModel import ObjectsModel
  5. from cura.BuildPlateModel import BuildPlateModel
  6. from UM.Application import Application
  7. from UM.Scene.Iterator.DepthFirstIterator import DepthFirstIterator
  8. from UM.Scene.SceneNode import SceneNode
  9. from UM.Scene.Selection import Selection
  10. class CuraSceneController(QObject):
  11. def __init__(self, objects_model: ObjectsModel, build_plate_model: BuildPlateModel):
  12. super().__init__()
  13. self._objects_model = objects_model
  14. self._build_plate_model = build_plate_model
  15. self._active_build_plate = -1
  16. self._last_selected_index = 0
  17. self._max_build_plate = 1 # default
  18. Application.getInstance().getController().getScene().sceneChanged.connect(self.updateMaxBuildPlate) # it may be a bit inefficient when changing a lot simultaneously
  19. def updateMaxBuildPlate(self, *args):
  20. if args:
  21. source = args[0]
  22. else:
  23. source = None
  24. if not issubclass(type(source), SceneNode):
  25. return
  26. max_build_plate = self._calcMaxBuildPlate()
  27. changed = False
  28. if max_build_plate != self._max_build_plate:
  29. self._max_build_plate = max_build_plate
  30. changed = True
  31. if changed:
  32. self._build_plate_model.setMaxBuildPlate(self._max_build_plate)
  33. build_plates = [{"name": "Build Plate %d" % (i + 1), "buildPlateNumber": i} for i in range(self._max_build_plate + 1)]
  34. self._build_plate_model.setItems(build_plates)
  35. if self._active_build_plate > self._max_build_plate:
  36. build_plate_number = 0
  37. if self._last_selected_index >= 0: # go to the buildplate of the item you last selected
  38. item = self._objects_model.getItem(self._last_selected_index)
  39. if "node" in item:
  40. node = item["node"]
  41. build_plate_number = node.callDecoration("getBuildPlateNumber")
  42. self.setActiveBuildPlate(build_plate_number)
  43. # self.buildPlateItemsChanged.emit() # TODO: necessary after setItems?
  44. def _calcMaxBuildPlate(self):
  45. max_build_plate = 0
  46. for node in DepthFirstIterator(Application.getInstance().getController().getScene().getRoot()):
  47. if node.callDecoration("isSliceable"):
  48. build_plate_number = node.callDecoration("getBuildPlateNumber")
  49. max_build_plate = max(build_plate_number, max_build_plate)
  50. return max_build_plate
  51. ## Either select or deselect an item
  52. @pyqtSlot(int)
  53. def changeSelection(self, index):
  54. modifiers = QApplication.keyboardModifiers()
  55. ctrl_is_active = modifiers & Qt.ControlModifier
  56. shift_is_active = modifiers & Qt.ShiftModifier
  57. if ctrl_is_active:
  58. item = self._objects_model.getItem(index)
  59. node = item["node"]
  60. if Selection.isSelected(node):
  61. Selection.remove(node)
  62. else:
  63. Selection.add(node)
  64. elif shift_is_active:
  65. polarity = 1 if index + 1 > self._last_selected_index else -1
  66. for i in range(self._last_selected_index, index + polarity, polarity):
  67. item = self._objects_model.getItem(i)
  68. node = item["node"]
  69. Selection.add(node)
  70. else:
  71. # Single select
  72. item = self._objects_model.getItem(index)
  73. node = item["node"]
  74. build_plate_number = node.callDecoration("getBuildPlateNumber")
  75. if build_plate_number is not None and build_plate_number != -1:
  76. self.setActiveBuildPlate(build_plate_number)
  77. Selection.clear()
  78. Selection.add(node)
  79. self._last_selected_index = index
  80. @pyqtSlot(int)
  81. def setActiveBuildPlate(self, nr):
  82. if nr == self._active_build_plate:
  83. return
  84. Logger.log("d", "Select build plate: %s" % nr)
  85. self._active_build_plate = nr
  86. Selection.clear()
  87. self._build_plate_model.setActiveBuildPlate(nr)
  88. self._objects_model.setActiveBuildPlate(nr)
  89. @staticmethod
  90. def createCuraSceneController():
  91. objects_model = Application.getInstance().getObjectsModel()
  92. build_plate_model = Application.getInstance().getBuildPlateModel()
  93. return CuraSceneController(objects_model = objects_model, build_plate_model = build_plate_model)