mirror of
https://github.com/NotAKidoS/NAK_CVR_Mods.git
synced 2025-09-02 22:39:22 +00:00
rewrite
This commit is contained in:
NotAKidoS
parent
16e888c64d
commit
cbd65aa813
8 changed files with 635 additions and 539 deletions
450
DesktopVRIK/DesktopVRIKCalibrator.cs
Normal file
450
DesktopVRIK/DesktopVRIKCalibrator.cs
Normal file
|
|
@ -0,0 +1,450 @@
|
|||
using ABI.CCK.Components;
|
||||
using ABI_RC.Core;
|
||||
using ABI_RC.Core.Base;
|
||||
using ABI_RC.Core.Player;
|
||||
using ABI_RC.Systems.IK;
|
||||
using ABI_RC.Systems.IK.SubSystems;
|
||||
using HarmonyLib;
|
||||
using RootMotion.FinalIK;
|
||||
using UnityEngine;
|
||||
using UnityEngine.Events;
|
||||
|
||||
namespace NAK.Melons.DesktopVRIK;
|
||||
|
||||
public class DesktopVRIKCalibrator
|
||||
{
|
||||
public DesktopVRIKCalibrator()
|
||||
{
|
||||
// Get base game scripts.
|
||||
ikSystem = IKSystem.Instance;
|
||||
playerSetup = PlayerSetup.Instance;
|
||||
|
||||
// Get traverse to private shit in iksystem.
|
||||
_vrikTraverse = Traverse.Create(ikSystem).Field("_vrik");
|
||||
_avatarTraverse = Traverse.Create(ikSystem).Field("_avatar");
|
||||
_animatorManagerTraverse = Traverse.Create(ikSystem).Field("_animatorManager");
|
||||
_poseHandlerTraverse = Traverse.Create(ikSystem).Field("_poseHandler");
|
||||
_avatarRootHeightTraverse = Traverse.Create(ikSystem).Field("_avatarRootHeight");
|
||||
|
||||
// Get traverse to private shit in playersetup.
|
||||
_lookIKTraverse = Traverse.Create(playerSetup).Field("lookIK");
|
||||
}
|
||||
|
||||
//Settings
|
||||
public bool Setting_UseVRIKToes = true;
|
||||
public bool Setting_FindUnmappedToes = true;
|
||||
|
||||
//DesktopVRIK
|
||||
public CVRAvatar avatar;
|
||||
public Animator animator;
|
||||
public Transform avatarTransform;
|
||||
public VRIK vrik;
|
||||
public LookAtIK lookAtIK;
|
||||
public HumanPoseHandler humanPoseHandler;
|
||||
public HumanPose initialHumanPose;
|
||||
public bool fixTransformsRequired;
|
||||
public float initialFootDistance;
|
||||
public float initialStepThreshold;
|
||||
|
||||
//Traverse
|
||||
private IKSystem ikSystem;
|
||||
private PlayerSetup playerSetup;
|
||||
private Traverse _vrikTraverse;
|
||||
private Traverse _lookIKTraverse;
|
||||
private Traverse _avatarTraverse;
|
||||
private Traverse _animatorManagerTraverse;
|
||||
private Traverse _poseHandlerTraverse;
|
||||
private Traverse _avatarRootHeightTraverse;
|
||||
|
||||
//public void RecalibrateDesktopVRIK()
|
||||
//{
|
||||
// if (avatar != null)
|
||||
// {
|
||||
// //calibrate VRIK
|
||||
// CalibrateDesktopVRIK();
|
||||
// }
|
||||
// else
|
||||
// {
|
||||
// //we never calibrated
|
||||
// SetupDesktopVRIK();
|
||||
// }
|
||||
//}
|
||||
|
||||
public void SetupDesktopVRIK()
|
||||
{
|
||||
//store avatar root transform & center it
|
||||
avatar = playerSetup._avatar.GetComponent<CVRAvatar>();
|
||||
animator = avatar.GetComponent<Animator>();
|
||||
avatarTransform = avatar.transform;
|
||||
avatarTransform.localPosition = Vector3.zero;
|
||||
lookAtIK = _lookIKTraverse.GetValue<LookAtIK>();
|
||||
|
||||
//prepare for VRIK
|
||||
PrepareIKSystem();
|
||||
CalibrateDesktopVRIK();
|
||||
|
||||
//add presolver update listener
|
||||
vrik.onPreSolverUpdate.AddListener(new UnityAction(DesktopVRIK.Instance.OnPreSolverUpdate));
|
||||
}
|
||||
|
||||
private void CalibrateDesktopVRIK()
|
||||
{
|
||||
//calibrate VRIK
|
||||
PrepareAvatarVRIK();
|
||||
SetAvatarIKPose(true);
|
||||
CalculateInitialIKScaling();
|
||||
CalibrateHeadIK();
|
||||
ForceInitiateVRIKSolver();
|
||||
SetAvatarIKPose(false);
|
||||
}
|
||||
|
||||
private void PrepareIKSystem()
|
||||
{
|
||||
// Get the animator manager and human pose handler
|
||||
var animatorManager = _animatorManagerTraverse.GetValue<CVRAnimatorManager>();
|
||||
humanPoseHandler = _poseHandlerTraverse.GetValue<HumanPoseHandler>();
|
||||
|
||||
// Store the avatar component
|
||||
_avatarTraverse.SetValue(avatar);
|
||||
|
||||
// Set the animator for the IK system
|
||||
ikSystem.animator = animator;
|
||||
if (ikSystem.animator != null)
|
||||
{
|
||||
animatorManager.SetAnimator(ikSystem.animator, ikSystem.animator.runtimeAnimatorController);
|
||||
}
|
||||
|
||||
// Set the avatar height float
|
||||
float avatarHeight = ikSystem.vrPlaySpace.transform.InverseTransformPoint(avatarTransform.position).y;
|
||||
_avatarRootHeightTraverse.SetValue(avatarHeight);
|
||||
|
||||
// Create a new human pose handler and dispose the old one
|
||||
if (humanPoseHandler != null)
|
||||
{
|
||||
humanPoseHandler.Dispose();
|
||||
_poseHandlerTraverse.SetValue(null);
|
||||
}
|
||||
humanPoseHandler = new HumanPoseHandler(ikSystem.animator.avatar, avatarTransform);
|
||||
_poseHandlerTraverse.SetValue(humanPoseHandler);
|
||||
|
||||
// Find valid human bones
|
||||
IKSystem.BoneExists.Clear();
|
||||
foreach (HumanBodyBones bone in Enum.GetValues(typeof(HumanBodyBones)))
|
||||
{
|
||||
if (bone != HumanBodyBones.LastBone)
|
||||
{
|
||||
IKSystem.BoneExists.Add(bone, ikSystem.animator.GetBoneTransform(bone) != null);
|
||||
}
|
||||
}
|
||||
|
||||
// Prepare BodySystem for calibration
|
||||
BodySystem.TrackingLeftArmEnabled = false;
|
||||
BodySystem.TrackingRightArmEnabled = false;
|
||||
BodySystem.TrackingLeftLegEnabled = false;
|
||||
BodySystem.TrackingRightLegEnabled = false;
|
||||
BodySystem.TrackingPositionWeight = 0f;
|
||||
}
|
||||
|
||||
private void PrepareAvatarVRIK()
|
||||
{
|
||||
//add and configure VRIK
|
||||
vrik = avatar.gameObject.AddComponentIfMissing<VRIK>();
|
||||
vrik.AutoDetectReferences();
|
||||
ConfigureVRIKReferences();
|
||||
_vrikTraverse.SetValue(vrik);
|
||||
|
||||
//in testing, not really needed
|
||||
//only required if Setting_FindUnmappedToes
|
||||
//and non-human mapped toes are found
|
||||
vrik.fixTransforms = fixTransformsRequired;
|
||||
|
||||
//default solver settings
|
||||
vrik.solver.locomotion.weight = 0f;
|
||||
vrik.solver.locomotion.angleThreshold = 30f;
|
||||
vrik.solver.locomotion.maxLegStretch = 0.75f;
|
||||
vrik.solver.spine.minHeadHeight = 0f;
|
||||
vrik.solver.IKPositionWeight = 1f;
|
||||
//disable to not bleed into anims
|
||||
vrik.solver.spine.chestClampWeight = 0f;
|
||||
vrik.solver.spine.maintainPelvisPosition = 0f;
|
||||
//for body leaning
|
||||
vrik.solver.spine.neckStiffness = 0.0001f; //cannot be 0
|
||||
vrik.solver.spine.bodyPosStiffness = 1f;
|
||||
vrik.solver.spine.bodyRotStiffness = 0.2f;
|
||||
//disable so avatar doesnt try and walk away
|
||||
//fixes nameplate spazzing on remote
|
||||
vrik.solver.locomotion.velocityFactor = 0f;
|
||||
vrik.solver.locomotion.maxVelocity = 0f;
|
||||
//disable so PAM & BID dont make body shake
|
||||
vrik.solver.spine.rotateChestByHands = 0f;
|
||||
//enable so knees on fucked models work better
|
||||
vrik.solver.leftLeg.useAnimatedBendNormal = true;
|
||||
vrik.solver.rightLeg.useAnimatedBendNormal = true;
|
||||
//enable to prioritize LookAtIK
|
||||
vrik.solver.spine.headClampWeight = 0.2f;
|
||||
//disable to not go on tippytoes
|
||||
vrik.solver.spine.positionWeight = 0f;
|
||||
vrik.solver.spine.rotationWeight = 1f;
|
||||
|
||||
//vrik.solver.spine.maintainPelvisPosition = 1f;
|
||||
//vrik.solver.locomotion.weight = 0f;
|
||||
//vrik.solver.spine.positionWeight = 0f;
|
||||
//vrik.solver.spine.pelvisPositionWeight = 0f;
|
||||
//vrik.solver.leftArm.positionWeight = 0f;
|
||||
//vrik.solver.leftArm.rotationWeight = 0f;
|
||||
//vrik.solver.rightArm.positionWeight = 0f;
|
||||
//vrik.solver.rightArm.rotationWeight = 0f;
|
||||
//vrik.solver.leftLeg.positionWeight = 0f;
|
||||
//vrik.solver.leftLeg.rotationWeight = 0f;
|
||||
//vrik.solver.rightLeg.positionWeight = 0f;
|
||||
//vrik.solver.rightLeg.rotationWeight = 0f;
|
||||
//vrik.solver.IKPositionWeight = 0f;
|
||||
|
||||
//THESE ARE CONFIGURABLE IN GAME IK SETTINGS
|
||||
//vrik.solver.leftLeg.target = null;
|
||||
//vrik.solver.leftLeg.bendGoal = null;
|
||||
//vrik.solver.leftLeg.positionWeight = 0f;
|
||||
//vrik.solver.leftLeg.bendGoalWeight = 0f;
|
||||
//vrik.solver.rightLeg.target = null;
|
||||
//vrik.solver.rightLeg.bendGoal = null;
|
||||
//vrik.solver.rightLeg.positionWeight = 0f;
|
||||
//vrik.solver.rightLeg.bendGoalWeight = 0f;
|
||||
//vrik.solver.spine.pelvisTarget = null;
|
||||
//vrik.solver.spine.chestGoal = null;
|
||||
//vrik.solver.spine.positionWeight = 0f;
|
||||
//vrik.solver.spine.rotationWeight = 0f;
|
||||
//vrik.solver.spine.pelvisPositionWeight = 0f;
|
||||
//vrik.solver.spine.pelvisRotationWeight = 0f;
|
||||
//vrik.solver.spine.chestGoalWeight = 0f;
|
||||
}
|
||||
|
||||
private void CalculateInitialIKScaling()
|
||||
{
|
||||
// Get distance between feets and thighs (fixes some weird armatures)
|
||||
float footDistance = Vector3.Distance(vrik.references.leftFoot.position, vrik.references.rightFoot.position);
|
||||
float thighDistance = Vector3.Distance(vrik.references.leftThigh.position, vrik.references.rightThigh.position);
|
||||
float greatestDistance = Mathf.Min(footDistance, thighDistance);
|
||||
initialFootDistance = greatestDistance * 0.5f;
|
||||
initialStepThreshold = greatestDistance * 0.4f;
|
||||
|
||||
//set initial values now, as avatars without scaling dont apply it
|
||||
vrik.solver.locomotion.footDistance = initialFootDistance;
|
||||
vrik.solver.locomotion.stepThreshold = initialStepThreshold;
|
||||
}
|
||||
|
||||
private void CalibrateHeadIK()
|
||||
{
|
||||
// Lazy HeadIKTarget calibration
|
||||
if (vrik.solver.spine.headTarget == null)
|
||||
{
|
||||
vrik.solver.spine.headTarget = new GameObject("Head IK Target").transform;
|
||||
}
|
||||
vrik.solver.spine.headTarget.parent = vrik.references.head;
|
||||
vrik.solver.spine.headTarget.localPosition = Vector3.zero;
|
||||
vrik.solver.spine.headTarget.localRotation = Quaternion.identity;
|
||||
}
|
||||
|
||||
private void SetAvatarIKPose(bool enforceTPose)
|
||||
{
|
||||
int ikposeLayerIndex = animator.GetLayerIndex("IKPose");
|
||||
int locoLayerIndex = animator.GetLayerIndex("Locomotion/Emotes");
|
||||
|
||||
// Use custom IKPose if found.
|
||||
if (ikposeLayerIndex != -1 && locoLayerIndex != -1)
|
||||
{
|
||||
animator.SetLayerWeight(ikposeLayerIndex, enforceTPose ? 1f : 0f);
|
||||
animator.SetLayerWeight(locoLayerIndex, enforceTPose ? 0f : 1f);
|
||||
animator.Update(0f);
|
||||
return;
|
||||
}
|
||||
|
||||
// Otherwise use DesktopVRIK IKPose & revert afterwards.
|
||||
if (enforceTPose)
|
||||
{
|
||||
humanPoseHandler.GetHumanPose(ref initialHumanPose);
|
||||
humanPoseHandler.GetHumanPose(ref ikSystem.humanPose);
|
||||
for (int i = 0; i < IKPoseMuscles.Length; i++)
|
||||
{
|
||||
IKSystem.Instance.ApplyMuscleValue((MuscleIndex)i, IKPoseMuscles[i], ref ikSystem.humanPose.muscles);
|
||||
}
|
||||
humanPoseHandler.SetHumanPose(ref ikSystem.humanPose);
|
||||
}
|
||||
else
|
||||
{
|
||||
humanPoseHandler.SetHumanPose(ref initialHumanPose);
|
||||
}
|
||||
}
|
||||
|
||||
private void ForceInitiateVRIKSolver()
|
||||
{
|
||||
//force immediate calibration before animator decides to fuck us
|
||||
vrik.solver.SetToReferences(vrik.references);
|
||||
vrik.solver.Initiate(vrik.transform);
|
||||
}
|
||||
|
||||
private void ConfigureVRIKReferences()
|
||||
{
|
||||
fixTransformsRequired = false;
|
||||
|
||||
Transform leftShoulderBone = vrik.references.leftShoulder;
|
||||
Transform rightShoulderBone = vrik.references.rightShoulder;
|
||||
Transform assumedChest = leftShoulderBone?.parent;
|
||||
|
||||
// Repair chest & spine bone references (valve models were messed up)
|
||||
if (assumedChest != null && rightShoulderBone.parent == assumedChest &&
|
||||
vrik.references.chest != assumedChest)
|
||||
{
|
||||
vrik.references.chest = assumedChest;
|
||||
vrik.references.spine = assumedChest.parent;
|
||||
}
|
||||
|
||||
if (!Setting_UseVRIKToes)
|
||||
{
|
||||
vrik.references.leftToes = null;
|
||||
vrik.references.rightToes = null;
|
||||
}
|
||||
else if (Setting_FindUnmappedToes)
|
||||
{
|
||||
Transform leftToes = vrik.references.leftToes;
|
||||
Transform rightToes = vrik.references.rightToes;
|
||||
if (leftToes == null && rightToes == null)
|
||||
{
|
||||
leftToes = FindUnmappedToe(vrik.references.leftFoot);
|
||||
rightToes = FindUnmappedToe(vrik.references.rightFoot);
|
||||
if (leftToes != null && rightToes != null)
|
||||
{
|
||||
fixTransformsRequired = true;
|
||||
vrik.references.leftToes = leftToes;
|
||||
vrik.references.rightToes = rightToes;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Fix error when there is no finger bones
|
||||
// Making up bullshit cause VRIK is evil otherwise
|
||||
if (vrik.references.leftHand.childCount == 0)
|
||||
{
|
||||
vrik.solver.leftArm.wristToPalmAxis = Vector3.up;
|
||||
vrik.solver.leftArm.palmToThumbAxis = -Vector3.forward;
|
||||
}
|
||||
if (vrik.references.rightHand.childCount == 0)
|
||||
{
|
||||
vrik.solver.rightArm.wristToPalmAxis = Vector3.up;
|
||||
vrik.solver.rightArm.palmToThumbAxis = Vector3.forward;
|
||||
}
|
||||
}
|
||||
|
||||
private Transform FindUnmappedToe(Transform foot)
|
||||
{
|
||||
foreach (Transform bone in foot)
|
||||
{
|
||||
if (bone.name.ToLowerInvariant().Contains("toe") ||
|
||||
bone.name.ToLowerInvariant().EndsWith("_end"))
|
||||
{
|
||||
return bone;
|
||||
}
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
|
||||
private static readonly float[] IKPoseMuscles = new float[]
|
||||
{
|
||||
0.00133321f,
|
||||
8.195831E-06f,
|
||||
8.537738E-07f,
|
||||
-0.002669832f,
|
||||
-7.651234E-06f,
|
||||
-0.001659694f,
|
||||
0f,
|
||||
0f,
|
||||
0f,
|
||||
0.04213953f,
|
||||
0.0003007996f,
|
||||
-0.008032114f,
|
||||
-0.03059979f,
|
||||
-0.0003182998f,
|
||||
0.009640567f,
|
||||
0f,
|
||||
0f,
|
||||
0f,
|
||||
0f,
|
||||
0f,
|
||||
0f,
|
||||
0.5768794f,
|
||||
0.01061097f,
|
||||
-0.1127839f,
|
||||
0.9705755f,
|
||||
0.07972051f,
|
||||
-0.0268422f,
|
||||
0.007237188f,
|
||||
0f,
|
||||
0.5768792f,
|
||||
0.01056608f,
|
||||
-0.1127519f,
|
||||
0.9705756f,
|
||||
0.07971933f,
|
||||
-0.02682396f,
|
||||
0.007229362f,
|
||||
0f,
|
||||
-5.651802E-06f,
|
||||
-3.034899E-07f,
|
||||
0.4100508f,
|
||||
0.3610304f,
|
||||
-0.0838329f,
|
||||
0.9262537f,
|
||||
0.1353517f,
|
||||
-0.03578902f,
|
||||
0.06005657f,
|
||||
-4.95989E-06f,
|
||||
-1.43007E-06f,
|
||||
0.4096187f,
|
||||
0.363263f,
|
||||
-0.08205152f,
|
||||
0.9250782f,
|
||||
0.1345718f,
|
||||
-0.03572125f,
|
||||
0.06055461f,
|
||||
-1.079177f,
|
||||
0.2095419f,
|
||||
0.6140652f,
|
||||
0.6365265f,
|
||||
0.6683931f,
|
||||
-0.4764312f,
|
||||
0.8099416f,
|
||||
0.8099371f,
|
||||
0.6658203f,
|
||||
-0.7327053f,
|
||||
0.8113618f,
|
||||
0.8114051f,
|
||||
0.6643661f,
|
||||
-0.40341f,
|
||||
0.8111364f,
|
||||
0.8111367f,
|
||||
0.6170399f,
|
||||
-0.2524227f,
|
||||
0.8138723f,
|
||||
0.8110135f,
|
||||
-1.079171f,
|
||||
0.2095456f,
|
||||
0.6140658f,
|
||||
0.6365255f,
|
||||
0.6683878f,
|
||||
-0.4764301f,
|
||||
0.8099402f,
|
||||
0.8099376f,
|
||||
0.6658241f,
|
||||
-0.7327023f,
|
||||
0.8113653f,
|
||||
0.8113793f,
|
||||
0.664364f,
|
||||
-0.4034042f,
|
||||
0.811136f,
|
||||
0.8111364f,
|
||||
0.6170469f,
|
||||
-0.2524345f,
|
||||
0.8138595f,
|
||||
0.8110138f
|
||||
};
|
||||
}
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue