using Unity.Collections;
using Unity.Jobs;
using LitMotion;
using LitMotion.Adapters;

[assembly: RegisterGenericJobType(typeof(MotionUpdateJob<FixedString32Bytes, StringOptions, FixedString32BytesMotionAdapter>))]
[assembly: RegisterGenericJobType(typeof(MotionUpdateJob<FixedString64Bytes, StringOptions, FixedString64BytesMotionAdapter>))]
[assembly: RegisterGenericJobType(typeof(MotionUpdateJob<FixedString128Bytes, StringOptions, FixedString128BytesMotionAdapter>))]
[assembly: RegisterGenericJobType(typeof(MotionUpdateJob<FixedString512Bytes, StringOptions, FixedString512BytesMotionAdapter>))]
[assembly: RegisterGenericJobType(typeof(MotionUpdateJob<FixedString4096Bytes, StringOptions, FixedString4096BytesMotionAdapter>))]

namespace LitMotion.Adapters
{
    public readonly struct FixedString32BytesMotionAdapter : IMotionAdapter<FixedString32Bytes, StringOptions>
    {
        public FixedString32Bytes Evaluate(ref FixedString32Bytes startValue, ref FixedString32Bytes endValue, ref StringOptions options, in MotionEvaluationContext context)
        {
            var start = startValue;
            var end = endValue;
            var customScrambleChars = options.CustomScrambleChars;
            ref var randomState = ref options.RandomState;
            if (randomState.state == 0) randomState = ref SharedRandom.Random;
            FixedStringHelper.Interpolate(ref start, ref end, context.Progress, options.ScrambleMode, options.RichTextEnabled, ref randomState, ref customScrambleChars, out var result);
            return result;
        }
    }

    public readonly struct FixedString64BytesMotionAdapter : IMotionAdapter<FixedString64Bytes, StringOptions>
    {
        public FixedString64Bytes Evaluate(ref FixedString64Bytes startValue, ref FixedString64Bytes endValue, ref StringOptions options, in MotionEvaluationContext context)
        {
            var start = startValue;
            var end = endValue;
            var customScrambleChars = options.CustomScrambleChars;
            ref var randomState = ref options.RandomState;
            if (randomState.state == 0) randomState = ref SharedRandom.Random;
            FixedStringHelper.Interpolate(ref start, ref end, context.Progress, options.ScrambleMode, options.RichTextEnabled, ref randomState, ref customScrambleChars, out var result);
            return result;
        }
    }

    public readonly struct FixedString128BytesMotionAdapter : IMotionAdapter<FixedString128Bytes, StringOptions>
    {
        public FixedString128Bytes Evaluate(ref FixedString128Bytes startValue, ref FixedString128Bytes endValue, ref StringOptions options, in MotionEvaluationContext context)
        {
            var start = startValue;
            var end = endValue;
            var customScrambleChars = options.CustomScrambleChars;
            ref var randomState = ref options.RandomState;
            if (randomState.state == 0) randomState = ref SharedRandom.Random;
            FixedStringHelper.Interpolate(ref start, ref end, context.Progress, options.ScrambleMode, options.RichTextEnabled, ref randomState, ref customScrambleChars, out var result);
            return result;
        }
    }

    public readonly struct FixedString512BytesMotionAdapter : IMotionAdapter<FixedString512Bytes, StringOptions>
    {
        public FixedString512Bytes Evaluate(ref FixedString512Bytes startValue, ref FixedString512Bytes endValue, ref StringOptions options, in MotionEvaluationContext context)
        {
            var start = startValue;
            var end = endValue;
            var customScrambleChars = options.CustomScrambleChars;
            ref var randomState = ref options.RandomState;
            if (randomState.state == 0) randomState = ref SharedRandom.Random;
            FixedStringHelper.Interpolate(ref start, ref end, context.Progress, options.ScrambleMode, options.RichTextEnabled, ref randomState, ref customScrambleChars, out var result);
            return result;
        }
    }

    public readonly struct FixedString4096BytesMotionAdapter : IMotionAdapter<FixedString4096Bytes, StringOptions>
    {
        public FixedString4096Bytes Evaluate(ref FixedString4096Bytes startValue, ref FixedString4096Bytes endValue, ref StringOptions options, in MotionEvaluationContext context)
        {
            var start = startValue;
            var end = endValue;
            var customScrambleChars = options.CustomScrambleChars;
            ref var randomState = ref options.RandomState;
            if (randomState.state == 0) randomState = ref SharedRandom.Random;
            FixedStringHelper.Interpolate(ref start, ref end, context.Progress, options.ScrambleMode, options.RichTextEnabled, ref randomState, ref customScrambleChars, out var result);
            return result;
        }
    }
}