using System.Collections.Generic;
using sc.splinemesher.pro.runtime;
using Unity.Mathematics;
using UnityEngine;
using UnityEngine.Splines;

public class SetScaleData : MonoBehaviour
{
    public SplineCurveMesher splineMesher;
    
    public float frequency = 0.1f;
    public float amplitude = 1f;

    public PathIndexUnit pathIndexUnit;
    
    private void OnValidate()
    {
        if(!splineMesher) return;
        
        int splineCount = splineMesher.SplineContainer.Splines.Count;

        //One SplineData per spline
        List<SplineData> scaleData = new List<SplineData>(splineCount);
        for (int i = 0; i < splineCount; i++)
        {
            Spline spline = splineMesher.SplineContainer.Splines[i];
            float length = spline.GetLength();

            //One point every 2 units for this example
            int pointCount = Mathf.CeilToInt(length / 2f);

            //Holds the various data points
            SplineData data = new SplineData
            {
                PathIndexUnit = pathIndexUnit
            };

            if (pathIndexUnit == PathIndexUnit.Knot)
            {
                //One point per knot
                pointCount = spline.Count;
            }
            
            for (int j = 0; j < pointCount; j++)
            {
                //Normalized progress along spline
                float t = j / (float)(pointCount-1);
                
                float distance = t * length;

                float index = 0;

                switch (pathIndexUnit)
                {
                    case PathIndexUnit.Distance: index = distance; break;
                    case PathIndexUnit.Normalized: index = t; break;
                    case PathIndexUnit.Knot: index = spline.ConvertIndexUnit(t, PathIndexUnit.Knot); break;
                }
                float scale = Mathf.Sin(distance * frequency) * amplitude;
                scale = scale * 0.5f + 0.5f;
                
                //Each data point has an 'index' to set its position along the spline
                //And a 'value'. In this case a float3 for a 3D scale.
                data.Add(index, new float3(scale));
            }
            
            scaleData.Add(data);
        }
        
        splineMesher.scaleData = scaleData;
        splineMesher.Rebuild();
    }
}