using GameCult.Aetheria.State.Verse;
using Unity.Collections;
using Unity.Jobs;
using Unity.Mathematics;

public struct AetheriaDaemonRenderNativeView
{
    public long FrameId;
    public long Generation;
    public int Count;
    public NativeArray<float> PositionX;
    public NativeArray<float> PositionY;
    public NativeArray<float> PositionZ;
    public NativeArray<float> RotationRadians;
    public NativeArray<float> VelocityX;
    public NativeArray<float> VelocityY;
    public NativeArray<float> VelocityZ;
    public NativeArray<float> PhysicsBodyRadius;
    public NativeArray<float> PhysicsBodyMass;
    public NativeArray<float> PhysicsBodyInverseMass;
    public NativeArray<float> RenderScale;
    public NativeArray<byte> RenderVisibility;
    public NativeArray<int> RenderLod;
    public NativeArray<uint> RenderGroupId;

    public bool IsCreated =>
        Count > 0 &&
        PositionX.IsCreated &&
        PositionY.IsCreated &&
        PositionZ.IsCreated;

    public bool HasRotation => RotationRadians.IsCreated;
    public bool HasVelocity => VelocityX.IsCreated && VelocityY.IsCreated && VelocityZ.IsCreated;
    public bool HasPhysicsRadius => PhysicsBodyRadius.IsCreated;
    public bool HasPhysicsMass => PhysicsBodyMass.IsCreated;
    public bool HasPhysicsInverseMass => PhysicsBodyInverseMass.IsCreated;
    public bool HasRenderScale => RenderScale.IsCreated;
    public bool HasRenderVisibility => RenderVisibility.IsCreated;
    public bool HasRenderLod => RenderLod.IsCreated;
    public bool HasRenderGroupId => RenderGroupId.IsCreated;

    public JobHandle ScheduleBuildRenderMatrices(
        NativeArray<float4x4> matrices,
        float uniformScale = 1.0f,
        int innerLoopBatchCount = 64,
        JobHandle dependsOn = default)
    {
        if (!IsCreated)
        {
            return dependsOn;
        }

        var count = math.min(Count, matrices.IsCreated ? matrices.Length : 0);
        if (count <= 0)
        {
            return dependsOn;
        }

        var job = new AetheriaDaemonRenderMatrixJob
        {
            PositionX = PositionX,
            PositionY = PositionY,
            PositionZ = PositionZ,
            RotationRadians = RotationRadians,
            PhysicsBodyRadius = PhysicsBodyRadius,
            RenderScale = RenderScale,
            Matrices = matrices,
            HasRotation = HasRotation,
            HasPhysicsRadius = HasPhysicsRadius,
            HasRenderScale = HasRenderScale,
            UniformScale = uniformScale
        };

        return job.Schedule(count, math.max(1, innerLoopBatchCount), dependsOn);
    }

    public static bool TryCreate(
        AetheriaRuntimeDaemonSoaViewIndex index,
        AetheriaDaemonSoaMemoryMap map,
        out AetheriaDaemonRenderNativeView view)
    {
        view = default;
        if (index == null || map == null)
        {
            return false;
        }

        if (!map.TryCreateFirstNativeArrayOfKind(
                index,
                AetheriaRuntimeDaemonSoaColumnKinds.PositionX,
                out view.PositionX) ||
            !map.TryCreateFirstNativeArrayOfKind(
                index,
                AetheriaRuntimeDaemonSoaColumnKinds.PositionY,
                out view.PositionY) ||
            !map.TryCreateFirstNativeArrayOfKind(
                index,
                AetheriaRuntimeDaemonSoaColumnKinds.PositionZ,
                out view.PositionZ))
        {
            view = default;
            return false;
        }

        view.Count = view.PositionX.Length;
        if (view.PositionY.Length != view.Count || view.PositionZ.Length != view.Count)
        {
            view = default;
            return false;
        }

        TryAssignOptionalColumn(index, map, AetheriaRuntimeDaemonSoaColumnKinds.RotationRadians, view.Count, out view.RotationRadians);
        TryAssignOptionalColumn(index, map, AetheriaRuntimeDaemonSoaColumnKinds.VelocityX, view.Count, out view.VelocityX);
        TryAssignOptionalColumn(index, map, AetheriaRuntimeDaemonSoaColumnKinds.VelocityY, view.Count, out view.VelocityY);
        TryAssignOptionalColumn(index, map, AetheriaRuntimeDaemonSoaColumnKinds.VelocityZ, view.Count, out view.VelocityZ);
        TryAssignOptionalColumn(index, map, AetheriaRuntimeDaemonSoaColumnKinds.PhysicsBodyRadius, view.Count, out view.PhysicsBodyRadius);
        TryAssignOptionalColumn(index, map, AetheriaRuntimeDaemonSoaColumnKinds.PhysicsBodyMass, view.Count, out view.PhysicsBodyMass);
        TryAssignOptionalColumn(index, map, AetheriaRuntimeDaemonSoaColumnKinds.PhysicsBodyInverseMass, view.Count, out view.PhysicsBodyInverseMass);
        TryAssignOptionalColumn(index, map, AetheriaRuntimeDaemonSoaColumnKinds.RenderScale, view.Count, out view.RenderScale);
        TryAssignOptionalByteColumn(index, map, AetheriaRuntimeDaemonSoaColumnKinds.RenderVisibility, view.Count, out view.RenderVisibility);
        TryAssignOptionalIntColumn(index, map, AetheriaRuntimeDaemonSoaColumnKinds.RenderLod, view.Count, out view.RenderLod);
        TryAssignOptionalUIntColumn(index, map, AetheriaRuntimeDaemonSoaColumnKinds.RenderGroupId, view.Count, out view.RenderGroupId);

        if ((view.VelocityX.IsCreated || view.VelocityY.IsCreated || view.VelocityZ.IsCreated) &&
            !view.HasVelocity)
        {
            view.VelocityX = default;
            view.VelocityY = default;
            view.VelocityZ = default;
        }

        view.FrameId = index.View.FrameId;
        view.Generation = index.View.Generation;
        return true;
    }

    private static bool TryAssignOptionalColumn(
        AetheriaRuntimeDaemonSoaViewIndex index,
        AetheriaDaemonSoaMemoryMap map,
        string kind,
        int expectedCount,
        out NativeArray<float> array)
    {
        array = default;
        if (!index.TryGetFirstColumnOfKind(kind, out _))
        {
            return false;
        }

        if (!map.TryCreateFirstNativeArrayOfKind(index, kind, out array))
        {
            array = default;
            return false;
        }

        if (array.Length == expectedCount)
        {
            return true;
        }

        array = default;
        return false;
    }

    private static bool TryAssignOptionalUIntColumn(
        AetheriaRuntimeDaemonSoaViewIndex index,
        AetheriaDaemonSoaMemoryMap map,
        string kind,
        int expectedCount,
        out NativeArray<uint> array)
    {
        array = default;
        if (!index.TryGetFirstColumnOfKind(kind, out _))
        {
            return false;
        }

        if (!map.TryCreateFirstNativeArrayOfKind(index, kind, out array))
        {
            array = default;
            return false;
        }

        if (array.Length == expectedCount)
        {
            return true;
        }

        array = default;
        return false;
    }

    private static bool TryAssignOptionalIntColumn(
        AetheriaRuntimeDaemonSoaViewIndex index,
        AetheriaDaemonSoaMemoryMap map,
        string kind,
        int expectedCount,
        out NativeArray<int> array)
    {
        array = default;
        if (!index.TryGetFirstColumnOfKind(kind, out _))
        {
            return false;
        }

        if (!map.TryCreateFirstNativeArrayOfKind(index, kind, out array))
        {
            array = default;
            return false;
        }

        if (array.Length == expectedCount)
        {
            return true;
        }

        array = default;
        return false;
    }

    private static bool TryAssignOptionalByteColumn(
        AetheriaRuntimeDaemonSoaViewIndex index,
        AetheriaDaemonSoaMemoryMap map,
        string kind,
        int expectedCount,
        out NativeArray<byte> array)
    {
        array = default;
        if (!index.TryGetFirstColumnOfKind(kind, out _))
        {
            return false;
        }

        if (!map.TryCreateFirstNativeArrayOfKind(index, kind, out array))
        {
            array = default;
            return false;
        }

        if (array.Length == expectedCount)
        {
            return true;
        }

        array = default;
        return false;
    }
}