#if RUKHANKA_DEBUG_INFO

#if !RUKHANKA_NO_DEBUG_DRAWER
using Rukhanka.DebugDrawer;
#endif

using Rukhanka.Toolbox;
using Unity.Collections.LowLevel.Unsafe;
using Unity.Entities;
using Unity.Jobs;
using Unity.Mathematics;
using Unity.Transforms;
using UnityEngine;

/////////////////////////////////////////////////////////////////////////////////

namespace Rukhanka
{

[WorldSystemFilter(WorldSystemFilterFlags.Default)]
[UpdateInGroup(typeof(PresentationSystemGroup))]
[UpdateAfter(typeof(RukhankaDeformationSystemGroup))]
public partial class BoneVisualizationSystem: SystemBase
{
	EntityQuery cpuBoneVisualizeQuery, gpuBoneVisualizeQuery;
	
	struct GPUBoneRendererRigInfo
	{
		public LocalTransform rigWorldPose;
	}
	
	FrameFencedGPUBufferPool<GPUBoneRendererRigInfo> frameRigInfoCB;
	internal GPUBoneRenderer gpuBoneRenderer;
	
/////////////////////////////////////////////////////////////////////////////////

	protected override void OnCreate()
	{
		gpuBoneRenderer = new ();
		frameRigInfoCB = new (0xff, GraphicsBuffer.Target.Structured, GraphicsBuffer.UsageFlags.LockBufferForWrite);
			
		cpuBoneVisualizeQuery = SystemAPI.QueryBuilder()
			.WithAll<RigDefinitionComponent, BoneVisualizationComponent>()
			.WithNone<GPUAnimationEngineTag>()
			.Build();
		
		gpuBoneVisualizeQuery = SystemAPI.QueryBuilder()
			.WithAll<RigDefinitionComponent, BoneVisualizationComponent, GPUAnimationEngineTag>()
			.Build();
		
		RequireForUpdate(cpuBoneVisualizeQuery);
	}
	
/////////////////////////////////////////////////////////////////////////////////

	protected override void OnUpdate()
	{
#if !RUKHANKA_NO_DEBUG_DRAWER
		Dependency = RenderBonesForCPUAnimators(Dependency);
#endif
		Dependency = RenderBonesForGPUAnimators(Dependency);
	}
	
/////////////////////////////////////////////////////////////////////////////////
	
	protected override void OnDestroy()
	{
		frameRigInfoCB.Dispose();
	}
	
/////////////////////////////////////////////////////////////////////////////////
	
	JobHandle RenderBonesForCPUAnimators(JobHandle dependsOn)
	{
		if (!SystemAPI.TryGetSingletonRW<Drawer>(out var dd))
			return dependsOn;
		
		if (!SystemAPI.TryGetSingleton<RuntimeAnimationData>(out var runtimeData))
			return dependsOn;
		
		var hasDCC = SystemAPI.TryGetSingleton<DebugConfigurationComponent>(out var dcc);
		var defaultColor = new float4(0, 1, 1, 1);
		var boneColorLines = hasDCC ? dcc.cpuRigColor : defaultColor;
		var boneColorTriangles = new float4(defaultColor.x, defaultColor.y, defaultColor.z, defaultColor.w * 0.3f);
		var boneColorLinesUINT = ColorTools.ToUint(boneColorLines);
		var boneColorTriUINT = ColorTools.ToUint(boneColorTriangles);
		
		var renderBonesJob = new RenderBonesCPUAnimatorsJob()
		{
			bonePoses = runtimeData.worldSpaceBonesBuffer,
			drawer = dd.ValueRW,
			colorLines = boneColorLinesUINT,
			colorTriangles = boneColorTriUINT
		};
		
		var rv = renderBonesJob.ScheduleParallel(cpuBoneVisualizeQuery, dependsOn);
		return rv;
	}
	
/////////////////////////////////////////////////////////////////////////////////
	
	JobHandle RenderBonesForGPUAnimators(JobHandle dependsOn)
	{
		if (!SystemAPI.TryGetSingleton<GPURuntimeAnimationData>(out var rad))
			return dependsOn;
		
		var rigCount = (int)rad.frameAnimatedRigsCounter;
		if (rigCount == 0 || gpuBoneVisualizeQuery.CalculateEntityCount() == 0)
			return dependsOn;
		
		frameRigInfoCB.Grow(rigCount);
		frameRigInfoCB.BeginFrame();
		var frameRigData = frameRigInfoCB.LockBufferForWrite(0, rigCount);
		
		var prepareBoneDataJob = new PrepareGPURigsJob()
		{
			frameRigData = frameRigData,
			localTransformLookup = SystemAPI.GetComponentLookup<LocalTransform>(true),
			boneVisualizationLookup = SystemAPI.GetComponentLookup<BoneVisualizationComponent>(true),
			gpuFrameOffsetsTypeHandle = SystemAPI.GetComponentTypeHandle<GPURigFrameOffsetsComponent>(true),
			entityTypeHandle = SystemAPI.GetEntityTypeHandle()
		};
		
		var rv = prepareBoneDataJob.ScheduleParallel(gpuBoneVisualizeQuery, dependsOn);
		rv.Complete();
		
		frameRigInfoCB.UnlockBufferAfterWrite(rigCount);
		
		var hasDCC = SystemAPI.TryGetSingleton<DebugConfigurationComponent>(out var dcc);
		var defaultColor = new float4(0, 1, 0, 1);
		var boneColor = hasDCC ? dcc.gpuRigColor : defaultColor;
		gpuBoneRenderer.RenderBones(frameRigInfoCB, boneColor);
		
		frameRigInfoCB.EndFrame();
		return rv;
	}
}
}

#endif