Luis Gonzalez
96 lines
3.6 KiB
C#
96 lines
3.6 KiB
C#
#if RUKHANKA_DEBUG_INFO
|
|
|
|
using Rukhanka.DebugDrawer;
|
|
using Unity.Mathematics;
|
|
using UnityEngine;
|
|
using UnityEngine.Rendering;
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
namespace Rukhanka
|
|
{
|
|
public class GPUBoneRenderer
|
|
{
|
|
Material boneDrawerMat;
|
|
Mesh boneMesh;
|
|
|
|
readonly int ShaderID_animatedBoneWorkload = Shader.PropertyToID("animatedBoneWorkload");
|
|
readonly int ShaderID_animationJobs = Shader.PropertyToID("animationJobs");
|
|
readonly int ShaderID_rigDefinitions = Shader.PropertyToID("rigDefinitions");
|
|
readonly int ShaderID_rigBones = Shader.PropertyToID("rigBones");
|
|
readonly int ShaderID_rigSpaceBoneTransforms = Shader.PropertyToID("rigSpaceBoneTransforms");
|
|
readonly int ShaderID_rigVisualizationData = Shader.PropertyToID("rigVisualizationData");
|
|
|
|
GraphicsBuffer framePerBoneAnimationWorkloadGB;
|
|
GraphicsBuffer frameRigAnimationJobsGB;
|
|
GraphicsBuffer rigDefinitionGB;
|
|
GraphicsBuffer rigBonesGB;
|
|
GraphicsBuffer rigSpaceAnimatedBonesGB;
|
|
int frameAnimatedBonesCount;
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
public GPUBoneRenderer()
|
|
{
|
|
boneDrawerMat = CoreUtils.CreateEngineMaterial(Shader.Find($"BoneRendererGPU"));
|
|
boneDrawerMat.enableInstancing = true;
|
|
boneMesh = DrawerManagedSingleton.CreateBoneMesh();
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
public void SetGPUBuffersForFrame
|
|
(
|
|
GraphicsBuffer framePerBoneAnimationWorkloadGB,
|
|
GraphicsBuffer frameRigAnimationJobsGB,
|
|
GraphicsBuffer rigDefinitionGB,
|
|
GraphicsBuffer rigBonesGB,
|
|
GraphicsBuffer rigSpaceAnimatedBonesGB,
|
|
int frameAnimatedBonesCount
|
|
)
|
|
{
|
|
this.framePerBoneAnimationWorkloadGB = framePerBoneAnimationWorkloadGB;
|
|
this.frameRigAnimationJobsGB = frameRigAnimationJobsGB;
|
|
this.rigDefinitionGB = rigDefinitionGB;
|
|
this.rigBonesGB = rigBonesGB;
|
|
this.rigSpaceAnimatedBonesGB = rigSpaceAnimatedBonesGB;
|
|
this.frameAnimatedBonesCount = frameAnimatedBonesCount;
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
public void RenderBones(GraphicsBuffer rigVisualizationDataGB, float4 boneColor)
|
|
{
|
|
var rp = new RenderParams();
|
|
rp.camera = null;
|
|
rp.layer = 0;
|
|
rp.lightProbeProxyVolume = null;
|
|
rp.lightProbeUsage = LightProbeUsage.Off;
|
|
rp.motionVectorMode = MotionVectorGenerationMode.ForceNoMotion;
|
|
rp.receiveShadows = false;
|
|
rp.reflectionProbeUsage = ReflectionProbeUsage.Off;
|
|
rp.rendererPriority = 0;
|
|
rp.renderingLayerMask = 0xffffffff;
|
|
rp.shadowCastingMode = ShadowCastingMode.Off;
|
|
rp.worldBounds = new Bounds(Vector3.zero, Vector3.one * 100000);
|
|
|
|
rp.matProps = new MaterialPropertyBlock();
|
|
rp.matProps.SetBuffer(ShaderID_animatedBoneWorkload, framePerBoneAnimationWorkloadGB);
|
|
rp.matProps.SetBuffer(ShaderID_animationJobs, frameRigAnimationJobsGB);
|
|
rp.matProps.SetBuffer(ShaderID_rigDefinitions, rigDefinitionGB);
|
|
rp.matProps.SetBuffer(ShaderID_rigBones, rigBonesGB);
|
|
rp.matProps.SetBuffer(ShaderID_rigSpaceBoneTransforms, rigSpaceAnimatedBonesGB);
|
|
rp.matProps.SetBuffer(ShaderID_rigVisualizationData, rigVisualizationDataGB);
|
|
rp.material = boneDrawerMat;
|
|
|
|
rp.matProps.SetVector("boneColor", boneColor);
|
|
Graphics.RenderMeshPrimitives(rp, boneMesh, 1, frameAnimatedBonesCount);
|
|
boneColor.w = 0.3f;
|
|
rp.matProps.SetVector("boneColor", boneColor);
|
|
Graphics.RenderMeshPrimitives(rp, boneMesh, 0, frameAnimatedBonesCount);
|
|
}
|
|
}
|
|
}
|
|
|
|
#endif
|