using System;
using UnityEngine;

namespace UnityEditor.TerrainTools
{
    internal class DefaultBrushUIGroup : BaseBrushUIGroup
    {
        internal class FeatureDefaults
        {
            public float Size { get; set; }
            public float Rotation { get; set; }
            public float Strength { get; set; }
            public float Spacing { get; set; }
            public float Scatter { get; set; }
        }

        [Flags]
        public enum Feature
        {
            Size = 1 << 0,
            Rotation = 1 << 1,
            Strength = 1 << 2,
            Spacing = 1 << 3,
            Scatter = 1 << 4,
            Smoothing = 1 << 5,

            All = Size | Rotation | Strength | Spacing | Scatter | Smoothing,

            NoScatter = All & ~Scatter,
            NoSpacing = All & ~Spacing,
            NoSmoothing = All & ~Smoothing,
        }


        public DefaultBrushUIGroup(string name, Func<TerrainToolsAnalytics.IBrushParameter[]> analyticsCall = null, Feature feature = Feature.All, FeatureDefaults defaults = null) : base(name, analyticsCall)
        {
            //Scatter must be first.
            if ((feature & Feature.Scatter) != 0)
            {
                m_HasBrushScatter = true;
                AddScatterController(new BrushScatterVariator(name, this, this, defaults?.Scatter?? brushScatter));
            }

            if ((feature & Feature.Size) != 0)
            {
                m_HasBrushSize = true;
                AddSizeController(new BrushSizeVariator(name, this, this, defaults?.Size?? brushSize));
            }
            if ((feature & Feature.Rotation) != 0)
            {
                m_HasBrushRotation = true;
                AddRotationController(new BrushRotationVariator(name, this, this, false, defaults?.Rotation?? brushRotation));
            }
            if ((feature & Feature.Strength) != 0)
            {
                m_HasBrushStrength = true;
                AddStrengthController(new BrushStrengthVariator(name, this, this, defaults?.Strength?? brushStrength));
            }
            if ((feature & Feature.Spacing) != 0)
            {
                m_HasBrushSpacing = true;
                AddSpacingController(new BrushSpacingVariator(name, this, this, defaults?.Spacing?? brushSpacing));
            }

            if ((feature & Feature.Smoothing) != 0)
            {
                AddSmoothingController(new DefaultBrushSmoother(name));
            }

            AddModifierKeyController(new DefaultBrushModifierKeys());
        }
    }
}