using System;
using Unity.Entities;
using UnityEngine;

[Serializable]
public struct CharAnimState : IInterpolatedData<CharAnimState>, IComponentData
{
    public Vector3 position;
    public float rotation;
    public float aimYaw;
    public float aimPitch;
    public float moveYaw;                                       // Global rotation 0->360 deg

    public CharacterPredictedState.StateData.LocoState charLocoState;
    public int charLocoTick;
    public CharacterPredictedState.StateData.Action charAction;
    public int charActionTick;
    public int damageTick;
    public float damageDirection;
    public int sprinting;
    public float sprintWeight;
    
    // Custom properties for Animation states
    public CharacterPredictedState.StateData.LocoState previousCharLocoState;
    public int groundMoveExitTick; 
    public float moveAngleLocal;                                // Movement rotation realtive to character forward -180->180 deg clockwise
    public float shootPoseWeight;
    public Vector2 locomotionVector;
    public float locomotionPhase;        
    public float banking;
    public float landAnticWeight;
    public float turnStartAngle;
    public short turnDirection;                                 // -1 TurnLeft, 0 Idle, 1 TurnRight
    public float squashTime;
    public float squashWeight;
    public float inAirTime;
    public float jumpTime;
    public float simpleTime;
    public Vector2 footIkOffset;
    public Vector3 footIkNormalLeft;
    public Vector3 footIkNormaRight;
    
    public void Serialize(ref NetworkWriter writer, IEntityReferenceSerializer refSerializer)
    {
        writer.SetFieldSection(NetworkWriter.FieldSectionType.OnlyNotPredicting);
        writer.WriteVector3Q("position", position, 2);
        writer.WriteFloatQ("rotation", rotation, 0);
        writer.WriteFloatQ("aimYaw", aimYaw, 0);
        writer.WriteFloatQ("aimPitch", aimPitch, 0);
        writer.WriteFloatQ("moveYaw", moveYaw, 0);

        writer.WriteInt32("charLocoState", (int)charLocoState);
        writer.WriteInt32("charLocoTick", charLocoTick);
        writer.WriteInt32("characterAction", (int)charAction);
        writer.WriteInt32("characterActionTick", charActionTick);
        writer.WriteBoolean("sprinting", sprinting == 1);
        writer.WriteFloatQ("sprintWeight", sprintWeight, 2);
        writer.WriteInt32("damageTick", damageTick);
        writer.WriteFloatQ("damageDirection", damageDirection,1);
        
        writer.WriteFloatQ("moveAngleLocal", moveAngleLocal, 0);
        writer.WriteFloatQ("shootPoseWeight", shootPoseWeight);
        writer.WriteVector2Q("locomotionVector", locomotionVector);
        writer.WriteFloatQ("locomotionPhase", locomotionPhase);
        writer.WriteFloatQ("banking", banking);
        writer.WriteFloatQ("landAnticWeight", landAnticWeight, 2);
        writer.WriteFloatQ("turnStartAngle", turnStartAngle,0);
        writer.WriteInt16("turnDirection", turnDirection);
        writer.WriteFloatQ("squashTime", squashTime, 2);
        writer.WriteFloatQ("squashWeight", squashWeight, 2);
        writer.WriteFloatQ("inAirTime", inAirTime, 2);
        writer.WriteFloatQ("jumpTime", jumpTime, 2);
        writer.WriteFloatQ("simpleTime", simpleTime, 2);
        writer.WriteVector2Q("footIkOffset", footIkOffset, 2);
        writer.WriteVector3Q("footIkNormalLeft", footIkNormalLeft, 2);
        writer.WriteVector3Q("footIkNormaRight", footIkNormaRight, 2);
        writer.ClearFieldSection();
    }

    public void Deserialize(ref NetworkReader reader, IEntityReferenceSerializer refSerializer, int tick)
    {
        position = reader.ReadVector3Q();
        rotation = reader.ReadFloatQ();
        aimYaw = reader.ReadFloatQ();
        aimPitch = reader.ReadFloatQ();
        moveYaw = reader.ReadFloatQ();

        charLocoState = (CharacterPredictedState.StateData.LocoState)reader.ReadInt32();
        charLocoTick = reader.ReadInt32();
        charAction = (CharacterPredictedState.StateData.Action)reader.ReadInt32();
        charActionTick = reader.ReadInt32();
        sprinting = reader.ReadBoolean() ? 1 : 0;
        sprintWeight = reader.ReadFloatQ();
        
        damageTick = reader.ReadInt32();
        damageDirection = reader.ReadFloatQ();

        moveAngleLocal = reader.ReadFloatQ();
        shootPoseWeight = reader.ReadFloatQ();
        locomotionVector = reader.ReadVector2Q();
        locomotionPhase = reader.ReadFloatQ();
        banking = reader.ReadFloatQ();
        landAnticWeight = reader.ReadFloatQ();
        turnStartAngle = reader.ReadFloatQ();
        turnDirection = reader.ReadInt16();
        squashTime = reader.ReadFloatQ();
        squashWeight = reader.ReadFloatQ();
        inAirTime = reader.ReadFloatQ();
        jumpTime = reader.ReadFloatQ();
        simpleTime = reader.ReadFloatQ();
        footIkOffset = reader.ReadVector2Q();
        footIkNormalLeft = reader.ReadVector3Q();
        footIkNormaRight = reader.ReadVector3Q();
    }

    public void Interpolate(ref CharAnimState prevState, ref CharAnimState nextState, float f)
    {
        position = Vector3.Lerp(prevState.position, nextState.position, f);
        rotation = Mathf.LerpAngle(prevState.rotation, nextState.rotation, f);
        aimYaw = Mathf.LerpAngle(prevState.aimYaw, nextState.aimYaw, f);
        aimPitch = Mathf.LerpAngle(prevState.aimPitch, nextState.aimPitch, f);
        moveYaw = Mathf.LerpAngle(prevState.moveYaw, nextState.moveYaw, f);

        charLocoState = prevState.charLocoState;
        charLocoTick = prevState.charLocoTick;
        charAction = prevState.charAction;
        charActionTick = prevState.charActionTick;
        sprinting = prevState.sprinting;
        sprintWeight =  Mathf.Lerp(prevState.sprintWeight, nextState.sprintWeight, f);
        
        damageTick = prevState.damageTick;
        damageDirection = prevState.damageDirection;

        moveAngleLocal = Mathf.LerpAngle(prevState.moveAngleLocal, nextState.moveAngleLocal, f);
        shootPoseWeight = Mathf.Lerp(prevState.shootPoseWeight, nextState.shootPoseWeight, f);
        locomotionVector = Vector2.Lerp(prevState.locomotionVector, nextState.locomotionVector, f);
        locomotionPhase = Mathf.Lerp(prevState.locomotionPhase, nextState.locomotionPhase, f);
        banking = Mathf.Lerp(prevState.banking, nextState.banking, f);
        landAnticWeight = Mathf.Lerp(prevState.landAnticWeight, nextState.landAnticWeight, f);
        turnStartAngle = prevState.turnStartAngle;
        turnDirection = prevState.turnDirection;
        squashTime = Mathf.Lerp(prevState.squashTime, nextState.squashTime, f);
        squashWeight = Mathf.Lerp(prevState.squashWeight, nextState.squashWeight, f);
        inAirTime = Mathf.Lerp(prevState.inAirTime, nextState.inAirTime, f);
        jumpTime = Mathf.Lerp(prevState.jumpTime, nextState.jumpTime, f);
        simpleTime = Mathf.Lerp(prevState.simpleTime, nextState.simpleTime, f);
        footIkOffset = Vector2.Lerp(prevState.footIkOffset, nextState.footIkOffset, f);
        footIkNormalLeft = Vector3.Lerp(prevState.footIkNormalLeft, nextState.footIkNormalLeft, f);
        footIkNormaRight = Vector3.Lerp(prevState.footIkNormaRight, nextState.footIkNormaRight, f);
    }

    public override string ToString()
    {
        System.Text.StringBuilder strBuilder = new System.Text.StringBuilder();
        strBuilder.AppendLine("position" + position);
        strBuilder.AppendLine("rotation" + rotation);
        strBuilder.AppendLine("aimYaw" + aimYaw);
        strBuilder.AppendLine("aimPitch" + aimPitch);
        strBuilder.AppendLine("moveYaw" + moveYaw);
        strBuilder.AppendLine("charLocoState" + charLocoState);
        strBuilder.AppendLine("charLocoTick" + charLocoTick);
        strBuilder.AppendLine("characterAction" + charAction);
        strBuilder.AppendLine("characterActionTick" + charActionTick);
        strBuilder.AppendLine("sprinting" + sprinting);
        strBuilder.AppendLine("shootPoseWeight" + shootPoseWeight);
        strBuilder.AppendLine("locomotionVector" + locomotionVector);
        strBuilder.AppendLine("locomotionPhase" + locomotionPhase);
        strBuilder.AppendLine("banking" + banking);
        strBuilder.AppendLine("landAnticWeight" + landAnticWeight);
        strBuilder.AppendLine("turnStartAngle" + turnStartAngle);
        strBuilder.AppendLine("turnDirection" + turnDirection);
        strBuilder.AppendLine("footIkOffset" + footIkOffset);

        return strBuilder.ToString();
    }
}

class CharacterAnimState : ComponentDataWrapper<CharAnimState>
{
}