mirror of
https://github.com/Ultimaker/Cura.git
synced 2025-07-10 00:07:51 -06:00
131 lines
4.8 KiB
Python
131 lines
4.8 KiB
Python
# Copyright (c) 2017 Ultimaker B.V.
|
|
# Cura is released under the terms of the AGPLv3 or higher.
|
|
|
|
from UM.Settings.DefinitionContainer import DefinitionContainer
|
|
from UM.Settings.InstanceContainer import InstanceContainer
|
|
from UM.Settings.ContainerRegistry import ContainerRegistry
|
|
|
|
from .GlobalStack import GlobalStack
|
|
from .ExtruderStack import ExtruderStack
|
|
from .CuraContainerStack import CuraContainerStack
|
|
|
|
class CuraStackBuilder:
|
|
@classmethod
|
|
def createMachine(cls, name: str, definition_id: str) -> GlobalStack:
|
|
cls.__registry = ContainerRegistry.getInstance()
|
|
definitions = cls.__registry.findDefinitionContainers(id = definition_id)
|
|
if not definitions:
|
|
Logger.log("w", "Definition {definition} was not found!", definition = definition_id)
|
|
return None
|
|
|
|
machine_definition = definitions[0]
|
|
name = cls.__registry.createUniqueName("machine", "", name, machine_definition.name)
|
|
|
|
new_global_stack = cls.createGlobalStack(
|
|
new_stack_id = name,
|
|
definition = machine_definition,
|
|
quality = "default",
|
|
material = "default",
|
|
variant = "default",
|
|
)
|
|
|
|
for extruder_definition in cls.__registry.findDefinitionContainers(machine = machine_definition.id):
|
|
position = extruder_definition.getMetaDataEntry("position", None)
|
|
if not position:
|
|
Logger.log("w", "Extruder definition %s specifies no position metadata entry.", extruder_definition.id)
|
|
|
|
new_extruder_id = cls.__registry.uniqueName(extruder_definition.id)
|
|
new_extruder = cls.createExtruderStack(
|
|
new_extruder_id = new_extruder_id,
|
|
definition = extruder_definition,
|
|
machine_definition = machine_definition,
|
|
quality = "default",
|
|
material = "default",
|
|
variant = "default",
|
|
next_stack = new_global_stack
|
|
)
|
|
|
|
return new_global_stack
|
|
|
|
|
|
@classmethod
|
|
def createExtruderStack(cls, new_stack_id: str, definition: DefinitionContainer, machine_definition: DefinitionContainer, **kwargs) -> ExtruderStack:
|
|
cls.__registry = ContainerRegistry.getInstance()
|
|
|
|
stack = ExtruderStack(new_stack_id)
|
|
|
|
user_container = InstanceContainer(new_stack_id + "_user")
|
|
user_container.addMetaDataEntry("type", "user")
|
|
user_container.addMetaDataEntry("machine", new_stack_id)
|
|
|
|
stack.setUserChanges(user_container)
|
|
|
|
if "quality_changes" in kwargs:
|
|
stack.setQualityChangesById(kwargs["quality_changes"])
|
|
|
|
if "quality" in kwargs:
|
|
stack.setQualityById(kwargs["quality"])
|
|
|
|
if "material" in kwargs:
|
|
stack.setMaterialById(kwargs["material"])
|
|
|
|
if "variant" in kwargs:
|
|
stack.setVariantById(kwargs["variant"])
|
|
|
|
if "definition_changes" in kwargs:
|
|
stack.setDefinitionChangesById(kwargs["definition_changes"])
|
|
|
|
if "definition" in kwargs:
|
|
stack.setDefinitionById(kwargs["definition"])
|
|
|
|
if "next_stack" in kwargs:
|
|
stack.setNextStack(kwargs["next_stack"])
|
|
|
|
# Only add the created containers to the registry after we have set all the other
|
|
# properties. This makes the create operation more transactional, since any problems
|
|
# setting properties will not result in incomplete containers being added.
|
|
registry.addContainer(stack)
|
|
registry.addContainer(user_container)
|
|
|
|
return stack
|
|
|
|
@staticmethod
|
|
def createGlobalStack(new_stack_id: str, definition: DefinitionContainer, **kwargs) -> GlobalStack:
|
|
registry = ContainerRegistry.getInstance()
|
|
|
|
stack = GlobalStack(new_stack_id)
|
|
|
|
stack.setDefinition(definition)
|
|
|
|
user_container = InstanceContainer(new_stack_id + "_user")
|
|
user_container.addMetaDataEntry("type", "user")
|
|
user_container.addMetaDataEntry("machine", new_stack_id)
|
|
user_container.setDefinition(definition)
|
|
|
|
stack.setUserChanges(user_container)
|
|
|
|
if "quality_changes" in kwargs:
|
|
stack.setQualityChangesById(kwargs["quality_changes"])
|
|
|
|
if "quality" in kwargs:
|
|
stack.setQualityById(kwargs["quality"])
|
|
|
|
if "material" in kwargs:
|
|
stack.setMaterialById(kwargs["material"])
|
|
|
|
if "variant" in kwargs:
|
|
stack.setVariantById(kwargs["variant"])
|
|
|
|
if "definition_changes" in kwargs:
|
|
stack.setDefinitionChangesById(kwargs["definition_changes"])
|
|
|
|
registry.addContainer(stack)
|
|
registry.addContainer(user_container)
|
|
|
|
return stack
|
|
|
|
# Convenience variable
|
|
# It should get set before any private functions are called so the privates do not need to
|
|
# re-get the container registry.
|
|
__registry = None # type: ContainerRegistry
|
|
|