Cura/cura/Settings/UserProfilesModel.py

# Copyright (c) 2017 Ultimaker B.V.
# Cura is released under the terms of the LGPLv3 or higher.

from UM.Application import Application
from UM.Settings.ContainerRegistry import ContainerRegistry

from cura.QualityManager import QualityManager
from cura.Settings.ProfilesModel import ProfilesModel
from cura.Settings.ExtruderManager import ExtruderManager

##  QML Model for listing the current list of valid quality changes profiles.
#
class UserProfilesModel(ProfilesModel):
    def __init__(self, parent = None):
        super().__init__(parent)

        #Need to connect to the metaDataChanged signal of the active materials.
        self.__current_extruders = []
        self.__current_materials = []

        Application.getInstance().getExtruderManager().extrudersChanged.connect(self.__onExtrudersChanged)
        self.__onExtrudersChanged()
        self.__current_materials = [extruder.material for extruder in self.__current_extruders]
        for material in self.__current_materials:
            material.metaDataChanged.connect(self._onContainerChanged)

        self._empty_quality = ContainerRegistry.getInstance().findContainers(id = "empty_quality")[0]

    ##  Fetch the list of containers to display.
    #
    #   See UM.Settings.Models.InstanceContainersModel._fetchInstanceContainers().
    def _fetchInstanceContainers(self):
        global_container_stack = Application.getInstance().getGlobalContainerStack()
        if not global_container_stack:
            return {}, {}

        # Fetch the list of quality changes.
        quality_manager = QualityManager.getInstance()
        machine_definition = quality_manager.getParentMachineDefinition(global_container_stack.definition)
        quality_changes_list = quality_manager.findAllQualityChangesForMachine(machine_definition)

        extruder_manager = ExtruderManager.getInstance()
        active_extruder = extruder_manager.getActiveExtruderStack()
        extruder_stacks = self._getOrderedExtruderStacksList()

        # Fetch the list of usable qualities across all extruders.
        # The actual list of quality profiles come from the first extruder in the extruder list.
        quality_list = quality_manager.findAllUsableQualitiesForMachineAndExtruders(global_container_stack, extruder_stacks)

        # Filter the quality_change by the list of available quality_types
        quality_type_set = set([x.getMetaDataEntry("quality_type") for x in quality_list])
        quality_type_set.add(self._empty_quality.getMetaDataEntry("quality_type"))

        filtered_quality_changes = {qc.getId():qc for qc in quality_changes_list if
                                    qc.getMetaDataEntry("quality_type") in quality_type_set and
                                    qc.getMetaDataEntry("extruder") is not None and
                                    (qc.getMetaDataEntry("extruder") == active_extruder.definition.getMetaDataEntry("quality_definition") or
                                     qc.getMetaDataEntry("extruder") == active_extruder.definition.getId())}

        return filtered_quality_changes, {} #Only return true profiles for now, no metadata. The quality manager is not able to get only metadata yet.

    ##  Called when a container changed on an extruder stack.
    #
    #   If it's the material we need to connect to the metaDataChanged signal of
    #   that.
    def __onContainerChanged(self, new_container):
        #Careful not to update when a quality or quality changes profile changed!
        #If you then update you're going to have an infinite recursion because the update may change the container.
        if new_container.getMetaDataEntry("type") == "material":
            for material in self.__current_materials:
                material.metaDataChanged.disconnect(self._onContainerChanged)
            self.__current_materials = [extruder.material for extruder in self.__current_extruders]
            for material in self.__current_materials:
                material.metaDataChanged.connect(self._onContainerChanged)

    ##  Called when the current set of extruders change.
    #
    #   This makes sure that we are listening to the signal for when the
    #   materials change.
    def __onExtrudersChanged(self):
        for extruder in self.__current_extruders:
            extruder.containersChanged.disconnect(self.__onContainerChanged)
        self.__current_extruders = Application.getInstance().getExtruderManager().getExtruderStacks()
        for extruder in self.__current_extruders:
            extruder.containersChanged.connect(self.__onContainerChanged)