mirror of
https://github.com/Ultimaker/Cura.git
synced 2025-07-20 13:17:51 -06:00
115 lines
4.7 KiB
Python
115 lines
4.7 KiB
Python
from typing import List
|
|
|
|
from UM.Application import Application
|
|
from UM.Logger import Logger
|
|
from UM.Scene.SceneNode import SceneNode
|
|
from copy import deepcopy
|
|
from cura.Settings.ExtrudersModel import ExtrudersModel
|
|
|
|
|
|
## Scene nodes that are models are only seen when selecting the corresponding build plate
|
|
# Note that many other nodes can just be UM SceneNode objects.
|
|
class CuraSceneNode(SceneNode):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self._outside_buildarea = False
|
|
|
|
def setOutsideBuildArea(self, new_value):
|
|
self._outside_buildarea = new_value
|
|
|
|
def isOutsideBuildArea(self):
|
|
return self._outside_buildarea or self.callDecoration("getBuildPlateNumber") < 0
|
|
|
|
def isVisible(self):
|
|
return super().isVisible() and self.callDecoration("getBuildPlateNumber") == Application.getInstance().getBuildPlateModel().activeBuildPlate
|
|
|
|
def isSelectable(self) -> bool:
|
|
return super().isSelectable() and self.callDecoration("getBuildPlateNumber") == Application.getInstance().getBuildPlateModel().activeBuildPlate
|
|
|
|
def getPrintingExtruderPosition(self) -> int:
|
|
# took bits and pieces from extruders model, solid view
|
|
|
|
global_container_stack = Application.getInstance().getGlobalContainerStack()
|
|
per_mesh_stack = self.callDecoration("getStack")
|
|
# It's only set if you explicitly choose an extruder
|
|
extruder_id = self.callDecoration("getActiveExtruder")
|
|
|
|
machine_extruder_count = global_container_stack.getProperty("machine_extruder_count", "value")
|
|
|
|
extruder_index = 0
|
|
|
|
for extruder in Application.getInstance().getExtruderManager().getMachineExtruders(global_container_stack.getId()):
|
|
position = extruder.getMetaDataEntry("position", default = "0") # Get the position
|
|
try:
|
|
position = int(position)
|
|
except ValueError:
|
|
# Not a proper int.
|
|
position = -1
|
|
if position > machine_extruder_count:
|
|
continue
|
|
|
|
# Find out the extruder index if we know the id.
|
|
if extruder_id is not None and extruder_id == extruder.getId():
|
|
extruder_index = position
|
|
break
|
|
|
|
# Use the support extruder instead of the active extruder if this is a support_mesh
|
|
if per_mesh_stack:
|
|
if per_mesh_stack.getProperty("support_mesh", "value"):
|
|
extruder_index = int(global_container_stack.getProperty("support_extruder_nr", "value"))
|
|
|
|
return extruder_index
|
|
|
|
def getDiffuseColor(self) -> List[float]:
|
|
# took bits and pieces from extruders model, solid view
|
|
|
|
global_container_stack = Application.getInstance().getGlobalContainerStack()
|
|
machine_extruder_count = global_container_stack.getProperty("machine_extruder_count", "value")
|
|
|
|
extruder_index = self.getPrintingExtruderPosition()
|
|
|
|
material_color = ExtrudersModel.defaultColors[extruder_index]
|
|
|
|
# Collect color from the extruder we want
|
|
for extruder in Application.getInstance().getExtruderManager().getMachineExtruders(global_container_stack.getId()):
|
|
position = extruder.getMetaDataEntry("position", default = "0") # Get the position
|
|
try:
|
|
position = int(position)
|
|
except ValueError:
|
|
# Not a proper int.
|
|
position = -1
|
|
if position > machine_extruder_count:
|
|
continue
|
|
|
|
if extruder.material and position == extruder_index:
|
|
material_color = extruder.material.getMetaDataEntry("color_code", default = material_color)
|
|
break
|
|
|
|
# Colors are passed as rgb hex strings (eg "#ffffff"), and the shader needs
|
|
# an rgba list of floats (eg [1.0, 1.0, 1.0, 1.0])
|
|
return [
|
|
int(material_color[1:3], 16) / 255,
|
|
int(material_color[3:5], 16) / 255,
|
|
int(material_color[5:7], 16) / 255,
|
|
1.0
|
|
]
|
|
|
|
|
|
## Taken from SceneNode, but replaced SceneNode with CuraSceneNode
|
|
def __deepcopy__(self, memo):
|
|
copy = CuraSceneNode()
|
|
copy.setTransformation(self.getLocalTransformation())
|
|
copy.setMeshData(self._mesh_data)
|
|
copy.setVisible(deepcopy(self._visible, memo))
|
|
copy._selectable = deepcopy(self._selectable, memo)
|
|
copy._name = deepcopy(self._name, memo)
|
|
for decorator in self._decorators:
|
|
copy.addDecorator(deepcopy(decorator, memo))
|
|
|
|
for child in self._children:
|
|
copy.addChild(deepcopy(child, memo))
|
|
self.calculateBoundingBoxMesh()
|
|
return copy
|
|
|
|
def transformChanged(self) -> None:
|
|
self._transformChanged()
|