"""
Copyright (C) 2023 Adobe.
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
"""

# file: shader/shader.py
# brief: Shader preset class object definition
# author Adobe - 3D & Immersive
# copyright 2023 Adobe Inc. All rights reserved.


from ..common import CLASS_SHADER_INPUTS


class ShaderPreset_Output():
    def __init__(self, output):
        self.id = output["id"]
        self.label = output["label"]
        self.enabled = output["enabled"]
        self.optional = output["optional"]
        self.colorspace = output["colorspace"]
        self.format = output["format"]
        self.bitdepth = output["bitdepth"]
        self.normal = output["normal"] if "normal" in output else False

    def to_json(self):
        _obj = {
            "id": self.id,
            "label": self.label,
            "enabled": self.enabled,
            "optional": self.optional,
            "colorspace": self.colorspace,
            "format": self.format,
            "bitdepth": self.bitdepth
        }
        if self.normal:
            _obj["normal"] = True
        return _obj


class ShaderPreset_Input():
    def __init__(self, input):
        self.id = input["id"]
        self.label = input["label"]
        self.type = input["type"]
        self.default = input["default"]
        self.min = input["min"] if "min" in input else None
        self.max = input["max"] if "max" in input else None

    def to_json(self):
        _obj = {
            "id": self.id,
            "label": self.label,
            "type": self.type,
            "default": self.default
        }
        if self.min is not None:
            _obj["min"] = self.min
        if self.max is not None:
            _obj["max"] = self.max
        return _obj


class ShaderPreset():
    def __init__(self, filename, data):
        self.filename = filename
        self.name = filename.replace(".json", "")
        self.label = data["label"]
        self.inputs = {}
        self.outputs = {}
        self.inputs_class_name = CLASS_SHADER_INPUTS.format(filename.replace(".json", "").replace(" ", "_"))

        for _key, _input in data["inputs"].items():
            _item = ShaderPreset_Input(_input)
            self.inputs[_key] = _item

        for _key, _output in data["outputs"].items():
            _item = ShaderPreset_Output(_output)
            self.outputs[_key] = _item

    def to_json(self):
        _obj = {
            "label": self.label,
            "inputs": {},
            "outputs": {},
            "inputs_class_name": self.inputs_class_name
        }

        for _key, _input in self.inputs.items():
            _obj["inputs"][_key] = _input.to_json()

        for _key, _output in self.outputs.items():
            _obj["outputs"][_key] = _output.to_json()

        return _obj