Luis Gonzalez
108 lines
3.6 KiB
C#
108 lines
3.6 KiB
C#
#if RUKHANKA_DEBUG_INFO
|
|
|
|
using Rukhanka.DebugDrawer;
|
|
using Unity.Burst;
|
|
using Unity.Burst.Intrinsics;
|
|
using Unity.Collections;
|
|
using Unity.Collections.LowLevel.Unsafe;
|
|
using Unity.Entities;
|
|
using Unity.Mathematics;
|
|
using Unity.Transforms;
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
namespace Rukhanka
|
|
{
|
|
partial class BoneVisualizationSystem
|
|
{
|
|
[BurstCompile]
|
|
partial struct RenderBonesCPUAnimatorsJob: IJobEntity
|
|
{
|
|
[ReadOnly]
|
|
public NativeList<BoneTransform> bonePoses;
|
|
|
|
public uint colorTriangles;
|
|
public uint colorLines;
|
|
|
|
public Drawer drawer;
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
public void Execute(Entity e, in RigDefinitionComponent rd, in BoneVisualizationComponent bvc)
|
|
{
|
|
var bt = RuntimeAnimationData.GetAnimationDataForRigRO(bonePoses, rd);
|
|
|
|
var len = bt.Length;
|
|
|
|
for (int l = rd.rigBlob.Value.rootBoneIndex; l < len; ++l)
|
|
{
|
|
ref var rb = ref rd.rigBlob.Value.bones[l];
|
|
|
|
if (rb.parentBoneIndex < 0)
|
|
continue;
|
|
|
|
var bonePose0 = bt[l];
|
|
var bonePose1 = bt[rb.parentBoneIndex];
|
|
|
|
if (math.any(math.abs(bonePose0.pos - bonePose1.pos)))
|
|
{
|
|
drawer.DrawBoneMesh(bonePose0.pos, bonePose1.pos, colorTriangles, colorLines);
|
|
if (bvc.tripodSize > 0)
|
|
{
|
|
var fwd = math.rotate(bonePose0.rot, math.forward()) * bvc.tripodSize;
|
|
var left = math.rotate(bonePose0.rot, math.up()) * bvc.tripodSize;
|
|
var up = math.rotate(bonePose0.rot, math.left()) * bvc.tripodSize;
|
|
drawer.DrawLine(bonePose0.pos, bonePose0.pos + fwd, 0x0000ffff);
|
|
drawer.DrawLine(bonePose0.pos, bonePose0.pos + up, 0x00ff00ff);
|
|
drawer.DrawLine(bonePose0.pos, bonePose0.pos + left, 0xff0000ff);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
//------------------------------------------------------------------------------//
|
|
|
|
[BurstCompile]
|
|
struct PrepareGPURigsJob: IJobChunk
|
|
{
|
|
[ReadOnly]
|
|
public ComponentLookup<LocalTransform> localTransformLookup;
|
|
[ReadOnly]
|
|
public ComponentLookup<BoneVisualizationComponent> boneVisualizationLookup;
|
|
[ReadOnly]
|
|
public ComponentTypeHandle<GPURigFrameOffsetsComponent> gpuFrameOffsetsTypeHandle;
|
|
[ReadOnly]
|
|
public EntityTypeHandle entityTypeHandle;
|
|
|
|
[NativeDisableParallelForRestriction]
|
|
public NativeArray<GPUBoneRendererRigInfo> frameRigData;
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
public void Execute(in ArchetypeChunk chunk, int unfilteredChunkIndex, bool useEnabledMask, in v128 chunkEnabledMask)
|
|
{
|
|
var entityFrameOffsets = chunk.GetNativeArray(ref gpuFrameOffsetsTypeHandle);
|
|
var chunkFrameOffsets = chunk.GetChunkComponentData(ref gpuFrameOffsetsTypeHandle);
|
|
var entities = chunk.GetNativeArray(entityTypeHandle);
|
|
|
|
var cee = new ChunkEntityEnumerator(useEnabledMask, chunkEnabledMask, chunk.Count);
|
|
while (cee.NextEntityIndex(out var i))
|
|
{
|
|
var e = entities[i];
|
|
var gpuFrameOffsets = entityFrameOffsets[i];
|
|
gpuFrameOffsets.AddOffsets(chunkFrameOffsets);
|
|
|
|
var rigInfo = new GPUBoneRendererRigInfo();
|
|
if (boneVisualizationLookup.HasComponent(e))
|
|
{
|
|
localTransformLookup.TryGetComponent(e, out rigInfo.rigWorldPose);
|
|
}
|
|
frameRigData[gpuFrameOffsets.rigIndex] = rigInfo;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
#endif
|