Cura/resources/qml/Preferences/Materials/MaterialsList.qml

// Copyright (c) 2018 Ultimaker B.V.
// Uranium is released under the terms of the LGPLv3 or higher.

import QtQuick 2.7
import QtQuick.Controls 1.4
import QtQuick.Controls.Styles 1.4
import QtQuick.Layouts 1.3
import QtQuick.Dialogs 1.2

import UM 1.2 as UM
import Cura 1.0 as Cura

Item
{
    id: materialList
    width: materialScrollView.width - 17
    height: childrenRect.height

    // Children
    UM.I18nCatalog { id: catalog; name: "cura"; }
    Cura.MaterialBrandsModel { id: materialsModel }
    Cura.FavoriteMaterialsModel { id: favoriteMaterialsModel }
    Cura.GenericMaterialsModel { id: genericMaterialsModel }

    property var currentType: null
    property var currentBrand: null
    property var expandedBrands: UM.Preferences.getValue("cura/expanded_brands").split(";")
    property var expandedTypes: UM.Preferences.getValue("cura/expanded_types").split(";")

    // Store information about which parts of the tree are expanded
    function persistExpandedCategories()
    {
        UM.Preferences.setValue("cura/expanded_brands", materialList.expandedBrands.join(";"))
        UM.Preferences.setValue("cura/expanded_types", materialList.expandedTypes.join(";"))
    }

    // Expand the list of materials in order to select the current material
    function expandActiveMaterial(search_root_id)
    {
        for (var material_idx = 0; material_idx < genericMaterialsModel.rowCount(); material_idx++)
        {
            var material = genericMaterialsModel.getItem(material_idx)
            if (material.root_material_id == search_root_id)
            {
                if (materialList.expandedBrands.indexOf("Generic") == -1)
                {
                    materialList.expandedBrands.push("Generic");
                    materialList.currentBrand = "Generic"
                }
                base.currentItem = material
                persistExpandedCategories()
                return true
            }
        }
        for (var brand_idx = 0; brand_idx < materialsModel.rowCount(); brand_idx++)
        {
            var brand = materialsModel.getItem(brand_idx)
            var types_model = brand.material_types
            for (var type_idx = 0; type_idx < types_model.rowCount(); type_idx++)
            {
                var type = types_model.getItem(type_idx)
                var colors_model = type.colors
                for (var material_idx = 0; material_idx < colors_model.rowCount(); material_idx++)
                {
                    var material = colors_model.getItem(material_idx)
                    if (material.root_material_id == search_root_id)
                    {
                        if (materialList.expandedBrands.indexOf(brand.name) == -1)
                        {
                            materialList.expandedBrands.push(brand.name);
                            materialList.currentBrand = brand.name
                        }
                        if (materialList.expandedTypes.indexOf(brand.name + "_" + type.name) == -1)
                        {
                            materialList.expandedTypes.push(brand.name + "_" + type.name)
                            materialList.currentType = brand.name + "_" + type.name
                        }
                        base.currentItem = material
                        persistExpandedCategories()
                        return true
                    }
                }
            }
            return false
        }
    }

    Connections
    {
        target: materialsModel
        onItemsChanged:
        {
            var correctlyExpanded = materialList.expandActiveMaterial(base.newRootMaterialIdToSwitchTo)
            if (base.toActivateNewMaterial)
            {
                var position = Cura.ExtruderManager.activeExtruderIndex
                Cura.MachineManager.setMaterial(position, base.currentItem.container_node)
            }
            base.newRootMaterialIdToSwitchTo = ""
            base.toActivateNewMaterial = false
        }
    }
    
    Column
    {
        Rectangle
        {
            property var expanded: true

            id: favorites_section
            height: childrenRect.height
            width: materialList.width
            Rectangle
            {
                id: favorites_header_background
                color: UM.Theme.getColor("favorites_header_bar")
                anchors.fill: favorites_header
            }
            Row
            {
                id: favorites_header
                Label
                {
                    id: favorites_name
                    text: "Favorites"
                    height: UM.Theme.getSize("favorites_row").height
                    width: materialList.width - UM.Theme.getSize("favorites_button").width
                    verticalAlignment: Text.AlignVCenter
                    leftPadding: 4
                }
                Button
                {
                    text: ""
                    implicitWidth: UM.Theme.getSize("favorites_button").width
                    implicitHeight: UM.Theme.getSize("favorites_button").height
                    UM.RecolorImage {
                        anchors
                        {
                            verticalCenter: parent.verticalCenter
                            horizontalCenter: parent.horizontalCenter
                        }
                        width: UM.Theme.getSize("standard_arrow").width
                        height: UM.Theme.getSize("standard_arrow").height
                        sourceSize.width: width
                        sourceSize.height: height
                        color: "black"
                        source: favorites_section.expanded ? UM.Theme.getIcon("arrow_bottom") : UM.Theme.getIcon("arrow_left")
                    }
                    style: ButtonStyle
                    {
                        background: Rectangle
                        {
                            anchors.fill: parent
                            color: "transparent"
                        }
                    }
                }
            }
            MouseArea
            {
                anchors.fill: favorites_header
                onPressed:
                {
                    favorites_section.expanded = !favorites_section.expanded
                }
            }
            Column
            {
                anchors.top: favorites_header.bottom
                anchors.left: parent.left
                width: materialList.width
                height: favorites_section.expanded ? childrenRect.height : 0
                visible: favorites_section.expanded
                Repeater
                {
                    model: favoriteMaterialsModel
                    delegate: MaterialsSlot
                    {
                        material: model
                    }
                }
            }
        }

        MaterialsBrandSection
        {
            id: genericSection
            sectionName: "Generic"
            elements: genericMaterialsModel
            hasMaterialTypes: false
        }
        Repeater
        {
            id: brand_list
            model: materialsModel
            delegate: MaterialsBrandSection
            {
                id: brandSection
                sectionName: model.name
                elements: model.material_types
                hasMaterialTypes: true
            }
        }
    }
}