KSP-Conformal-Decals/Source/ConformalDecals/MaterialModifiers/MaterialPropertyCollection.cs

using System;
using System.Collections.Generic;
using UnityEngine;

namespace ConformalDecals.MaterialModifiers {
    public class MaterialPropertyCollection {
        public TextureMaterialProperty MainTextureProperty { get; }

        public Material ParsedMaterial { get; }

        public bool UseBaseNormal { get; }

        private readonly List<MaterialProperty>        _materialProperties;
        private readonly List<TextureMaterialProperty> _textureMaterialProperties;

        public String BaseNormalSrc { get; }
        public String BaseNormalDest { get; }

        private const string _normalTextureName = "_BumpMap";

        public MaterialPropertyCollection(ConfigNode node, PartModule module) {

            // Initialize fields
            _materialProperties = new List<MaterialProperty>();
            _textureMaterialProperties = new List<TextureMaterialProperty>();

            // Get shader
            var shaderString = node.GetValue("shader") ?? throw new FormatException("Missing shader name in material");

            var shaderRef = Shabby.Shabby.FindShader(shaderString);

            // note to self: null coalescing does not work on UnityEngine classes
            if (shaderRef == null) {
                throw new FormatException($"Shader not found: '{shaderString}'");
            }

            ParsedMaterial = new Material(shaderRef);

            // Get useBaseNormal value
            var useBaseNormalString = node.GetValue("useBaseNormal");

            if (useBaseNormalString != null) {
                if (bool.TryParse(useBaseNormalString, out var useBaseNormalRef)) {
                    UseBaseNormal = useBaseNormalRef;
                }
                else {
                    throw new FormatException($"Improperly formatted bool value for 'useBaseNormal' : {useBaseNormalString}");
                }
            }
            else {
                UseBaseNormal = false;
            }

            // Get basenormal source and destination property names
            BaseNormalSrc = node.GetValue("baseNormalSource") ?? _normalTextureName;
            BaseNormalDest = node.GetValue("baseNormalDestination") ?? _normalTextureName;

            // Parse all materialProperties
            foreach (ConfigNode propertyNode in node.nodes) {
                try {
                    MaterialProperty property;
                    switch (propertyNode.name) {
                        case "FLOAT":
                            property = new FloatMaterialProperty(propertyNode);
                            break;

                        case "COLOR":
                            property = new ColorMaterialProperty(propertyNode);
                            break;

                        case "TEXTURE":
                            property = new TextureMaterialProperty(propertyNode);
                            var textureModifier = (TextureMaterialProperty) property;
                            if (textureModifier.IsMain) {
                                if (MainTextureProperty == null) {
                                    MainTextureProperty = textureModifier;
                                }
                                else {
                                    // multiple textures have been marked as main!
                                    // non-fatal issue, ignore this one and keep using current main texture
                                    module.LogWarning(
                                        $"Material texture property {textureModifier.TextureUrl} is marked as main, but material already has a main texture! \n" +
                                        $"Defaulting to {MainTextureProperty.TextureUrl}");
                                }
                            }

                            _textureMaterialProperties.Add(textureModifier);
                            break;

                        default:
                            throw new FormatException($"Invalid property type '{propertyNode.name}' in material");
                    }

                    _materialProperties.Add(property);
                    property.Modify(ParsedMaterial);
                }

                catch (Exception e) {
                    // Catch exception from parsing current material property
                    // And print it to the log as an Error
                    module.LogError(e.Message);
                }
            }
        }

        public void UpdateMaterial(Vector2 scale) {
            foreach (var textureProperty in _textureMaterialProperties) {
                textureProperty.UpdateScale(ParsedMaterial, scale);
            }
        }
    }
}