TechbloxModdingAPI/TechbloxModdingAPI/Blocks/RotationEngine.cs

using RobocraftX.Common;
using RobocraftX.UECS;
using Svelto.ECS;
using Svelto.ECS.EntityStructs;
using Unity.Mathematics;
using UnityEngine;
using TechbloxModdingAPI.Engines;
using TechbloxModdingAPI.Utility;

namespace TechbloxModdingAPI.Blocks
{
    /// <summary>
    /// Engine which executes block movement actions
    /// </summary>
    public class RotationEngine : IApiEngine
    {
        public string Name { get; } = "TechbloxModdingAPIRotationGameEngine";

        public EntitiesDB entitiesDB { set; private get; }

		public bool isRemovable => false;

		public bool IsInGame = false;

        public void Dispose()
        {
            IsInGame = false;
        }

        public void Ready()
        {
            IsInGame = true;
        }

        // implementations for Rotation static class

        internal float3 RotateBlock(EGID blockID, BlockEngine.BlockInitData data, Vector3 vector)
        {
            if (!entitiesDB.Exists<RotationEntityStruct>(blockID))
            {
                if (data.Group == null) return float3.zero;
                var init = new EntityInitializer(blockID, data.Group, data.Reference);
                init.GetOrCreate<RotationEntityStruct>().rotation = Quaternion.Euler(vector);
                init.GetOrCreate<GridRotationStruct>().rotation = Quaternion.Euler(vector);
                init.GetOrCreate<LocalTransformEntityStruct>().rotation = Quaternion.Euler(vector);
                return vector;
            }
            ref RotationEntityStruct rotStruct = ref this.entitiesDB.QueryEntity<RotationEntityStruct>(blockID);
            ref GridRotationStruct gridStruct = ref this.entitiesDB.QueryEntity<GridRotationStruct>(blockID);
            ref LocalTransformEntityStruct transStruct = ref this.entitiesDB.QueryEntity<LocalTransformEntityStruct>(blockID);
            ref UECSPhysicsEntityStruct phyStruct = ref this.entitiesDB.QueryEntity<UECSPhysicsEntityStruct>(blockID);
            // main (persistent) position
            Quaternion newRotation = rotStruct.rotation;
            newRotation.eulerAngles = vector;
            rotStruct.rotation = newRotation;
            // placement grid rotation
            Quaternion newGridRotation = gridStruct.rotation;
            newGridRotation.eulerAngles = vector;
            gridStruct.rotation = newGridRotation;
            // rendered position
            Quaternion newTransRotation = rotStruct.rotation;
            newTransRotation.eulerAngles = vector;
            transStruct.rotation = newTransRotation;
            // collision position
            FullGameFields._physicsWorld.EntityManager.SetComponentData(phyStruct.uecsEntity, new Unity.Transforms.Rotation
            {
                Value = rotStruct.rotation
            });
            entitiesDB.QueryEntity<GridConnectionsEntityStruct>(blockID).isProcessed = false;
            return ((Quaternion)rotStruct.rotation).eulerAngles;

        }

        internal float3 GetRotation(EGID blockID, BlockEngine.BlockInitData data)
        {
            if (!entitiesDB.Exists<RotationEntityStruct>(blockID))
            {
                if (data.Group == null) return float3.zero;
                var init = new EntityInitializer(blockID, data.Group, data.Reference);
                return init.Has<RotationEntityStruct>()
                    ? (float3) ((Quaternion) init.Get<RotationEntityStruct>().rotation).eulerAngles
                    : float3.zero;
            }

            ref RotationEntityStruct rotStruct = ref entitiesDB.QueryEntity<RotationEntityStruct>(blockID);
            return ((Quaternion) rotStruct.rotation).eulerAngles;
        }
    }
}