mirror of
https://github.com/hanetzer/sdraw_mods_cvr.git
synced 2025-09-03 10:29:22 +00:00
246 lines
10 KiB
C#
246 lines
10 KiB
C#
using ABI_RC.Core.InteractionSystem;
|
|
using ABI_RC.Core.Player;
|
|
using ABI_RC.Systems.MovementSystem;
|
|
using RootMotion.Dynamics;
|
|
using RootMotion.FinalIK;
|
|
using System.Collections.Generic;
|
|
using UnityEngine;
|
|
|
|
namespace ml_prm
|
|
{
|
|
class RagdollController : MonoBehaviour
|
|
{
|
|
VRIK m_vrIK = null;
|
|
float m_vrIkWeight = 1f;
|
|
|
|
bool m_enabled = false;
|
|
|
|
readonly List<Rigidbody> m_rigidBodies = null;
|
|
readonly List<Collider> m_colliders = null;
|
|
Transform m_puppetRoot = null;
|
|
Transform m_puppet = null;
|
|
BipedRagdollReferences m_puppetReferences;
|
|
BipedRagdollReferences m_avatarReferences;
|
|
readonly List<System.Tuple<Transform, Transform>> m_boneLinks = null;
|
|
|
|
bool m_avatarReady = false;
|
|
Vector3 m_lastPosition = Vector3.zero;
|
|
Vector3 m_velocity = Vector3.zero;
|
|
|
|
internal RagdollController()
|
|
{
|
|
m_rigidBodies = new List<Rigidbody>();
|
|
m_colliders = new List<Collider>();
|
|
m_boneLinks = new List<System.Tuple<Transform, Transform>>();
|
|
}
|
|
|
|
// Unity events
|
|
void Start()
|
|
{
|
|
m_puppetRoot = new GameObject("[PlayerAvatarPuppet]").transform;
|
|
m_puppetRoot.parent = PlayerSetup.Instance.transform;
|
|
m_puppetRoot.localPosition = Vector3.zero;
|
|
m_puppetRoot.localRotation = Quaternion.identity;
|
|
|
|
Settings.SwitchChange += this.SwitchRagdoll;
|
|
}
|
|
|
|
void OnDestroy()
|
|
{
|
|
Settings.SwitchChange -= this.SwitchRagdoll;
|
|
}
|
|
|
|
void Update()
|
|
{
|
|
Vector3 l_pos = PlayerSetup.Instance.transform.position;
|
|
m_velocity = (m_velocity + (l_pos - m_lastPosition) / Time.deltaTime) * 0.5f;
|
|
m_lastPosition = l_pos;
|
|
|
|
if(Settings.Hotkey && Input.GetKeyDown(KeyCode.R) && !ViewManager.Instance.isGameMenuOpen())
|
|
SwitchRagdoll();
|
|
}
|
|
|
|
void LateUpdate()
|
|
{
|
|
if(m_enabled && m_avatarReady)
|
|
{
|
|
foreach(var l_link in m_boneLinks)
|
|
l_link.Item1.CopyGlobal(l_link.Item2);
|
|
}
|
|
}
|
|
|
|
// Game events
|
|
internal void OnAvatarClear()
|
|
{
|
|
if(m_enabled)
|
|
MovementSystem.Instance.SetImmobilized(false);
|
|
|
|
if(m_puppet != null)
|
|
Object.Destroy(m_puppet.gameObject);
|
|
m_puppet = null;
|
|
|
|
m_vrIK = null;
|
|
m_enabled = false;
|
|
m_avatarReady = false;
|
|
m_rigidBodies.Clear();
|
|
m_colliders.Clear();
|
|
m_avatarReferences = new BipedRagdollReferences();
|
|
m_puppetReferences = new BipedRagdollReferences();
|
|
m_boneLinks.Clear();
|
|
}
|
|
|
|
internal void OnAvatarSetup()
|
|
{
|
|
if(PlayerSetup.Instance._animator.isHuman)
|
|
{
|
|
m_avatarReferences = BipedRagdollReferences.FromAvatar(PlayerSetup.Instance._animator);
|
|
|
|
m_puppet = new GameObject("Root").transform;
|
|
m_puppet.parent = m_puppetRoot;
|
|
m_puppet.localPosition = Vector3.zero;
|
|
m_puppet.localRotation = Quaternion.identity;
|
|
|
|
m_puppetReferences.root = m_puppet;
|
|
m_puppetReferences.hips = CloneTransform(m_avatarReferences.hips, m_puppetReferences.root, "Hips");
|
|
m_puppetReferences.spine = CloneTransform(m_avatarReferences.spine, m_puppetReferences.hips, "Spine");
|
|
m_puppetReferences.chest = CloneTransform(m_avatarReferences.chest, m_puppetReferences.spine, "Chest");
|
|
m_puppetReferences.head = CloneTransform(m_avatarReferences.head, m_puppetReferences.chest, "Head");
|
|
|
|
m_puppetReferences.leftUpperArm = CloneTransform(m_avatarReferences.leftUpperArm, m_puppetReferences.chest, "LeftUpperArm");
|
|
m_puppetReferences.leftLowerArm = CloneTransform(m_avatarReferences.leftLowerArm, m_puppetReferences.leftUpperArm, "LeftLowerArm");
|
|
m_puppetReferences.leftHand = CloneTransform(m_avatarReferences.leftHand, m_puppetReferences.leftLowerArm, "LeftHand");
|
|
|
|
m_puppetReferences.rightUpperArm = CloneTransform(m_avatarReferences.rightUpperArm, m_puppetReferences.chest, "RightUpperArm");
|
|
m_puppetReferences.rightLowerArm = CloneTransform(m_avatarReferences.rightLowerArm, m_puppetReferences.rightUpperArm, "RightLowerArm");
|
|
m_puppetReferences.rightHand = CloneTransform(m_avatarReferences.rightHand, m_puppetReferences.rightLowerArm, "RightHand");
|
|
|
|
m_puppetReferences.leftUpperLeg = CloneTransform(m_avatarReferences.leftUpperLeg, m_puppetReferences.hips, "LeftUpperLeg");
|
|
m_puppetReferences.leftLowerLeg = CloneTransform(m_avatarReferences.leftLowerLeg, m_puppetReferences.leftUpperLeg, "LeftLowerLeg");
|
|
m_puppetReferences.leftFoot = CloneTransform(m_avatarReferences.leftFoot, m_puppetReferences.leftLowerLeg, "LeftFoot");
|
|
|
|
m_puppetReferences.rightUpperLeg = CloneTransform(m_avatarReferences.rightUpperLeg, m_puppetReferences.hips, "RightUpperLeg");
|
|
m_puppetReferences.rightLowerLeg = CloneTransform(m_avatarReferences.rightLowerLeg, m_puppetReferences.rightUpperLeg, "RightLowerLeg");
|
|
m_puppetReferences.rightFoot = CloneTransform(m_avatarReferences.rightFoot, m_puppetReferences.rightLowerLeg, "RightFoot");
|
|
|
|
BipedRagdollCreator.Options l_options = BipedRagdollCreator.AutodetectOptions(m_puppetReferences);
|
|
l_options.joints = RagdollCreator.JointType.Character;
|
|
BipedRagdollCreator.Create(m_puppetReferences, l_options);
|
|
|
|
Transform[] l_puppetTransforms = m_puppetReferences.GetRagdollTransforms();
|
|
Transform[] l_avatarTransforms = m_avatarReferences.GetRagdollTransforms();
|
|
for(int i = 0; i < l_puppetTransforms.Length; i++)
|
|
{
|
|
if(l_puppetTransforms[i] != null)
|
|
{
|
|
Rigidbody l_body = l_puppetTransforms[i].GetComponent<Rigidbody>();
|
|
if(l_body != null)
|
|
{
|
|
m_rigidBodies.Add(l_body);
|
|
l_body.isKinematic = true;
|
|
l_body.angularDrag = 0.5f;
|
|
l_body.drag = 1.0f;
|
|
l_body.collisionDetectionMode = CollisionDetectionMode.ContinuousDynamic;
|
|
}
|
|
|
|
CharacterJoint l_joint = l_puppetTransforms[i].GetComponent<CharacterJoint>();
|
|
if(l_joint != null)
|
|
{
|
|
l_joint.enablePreprocessing = false;
|
|
l_joint.enableProjection = true;
|
|
}
|
|
|
|
Collider l_collider = l_puppetTransforms[i].GetComponent<Collider>();
|
|
if(l_collider != null)
|
|
{
|
|
Physics.IgnoreCollision(MovementSystem.Instance.proxyCollider, l_collider, true);
|
|
l_collider.enabled = false;
|
|
m_colliders.Add(l_collider);
|
|
}
|
|
|
|
if(l_avatarTransforms[i] != null)
|
|
m_boneLinks.Add(System.Tuple.Create(l_puppetTransforms[i], l_avatarTransforms[i]));
|
|
}
|
|
}
|
|
|
|
m_vrIK = PlayerSetup.Instance._avatar.GetComponent<VRIK>();
|
|
if(m_vrIK != null)
|
|
{
|
|
m_vrIK.onPreSolverUpdate.AddListener(this.OnIKPreUpdate);
|
|
m_vrIK.onPostSolverUpdate.AddListener(this.OnIKPostUpdate);
|
|
}
|
|
|
|
m_avatarReady = true;
|
|
}
|
|
}
|
|
|
|
internal void OnSeatSitDown(CVRSeat p_seat)
|
|
{
|
|
if(m_enabled && m_avatarReady && !p_seat.occupied)
|
|
SwitchRagdoll();
|
|
}
|
|
|
|
// IK updates
|
|
void OnIKPreUpdate()
|
|
{
|
|
if(m_enabled)
|
|
{
|
|
m_vrIkWeight = m_vrIK.solver.IKPositionWeight;
|
|
m_vrIK.solver.IKPositionWeight = 0f;
|
|
}
|
|
}
|
|
void OnIKPostUpdate()
|
|
{
|
|
if(m_enabled)
|
|
m_vrIK.solver.IKPositionWeight = m_vrIkWeight;
|
|
}
|
|
|
|
// Arbitrary
|
|
public void SwitchRagdoll()
|
|
{
|
|
if(m_avatarReady && (MovementSystem.Instance.lastSeat == null))
|
|
{
|
|
m_enabled = !m_enabled;
|
|
|
|
MovementSystem.Instance.SetImmobilized(m_enabled);
|
|
|
|
if(m_enabled)
|
|
{
|
|
foreach(var l_link in m_boneLinks)
|
|
l_link.Item2.CopyGlobal(l_link.Item1);
|
|
|
|
foreach(Rigidbody l_body in m_rigidBodies)
|
|
l_body.isKinematic = false;
|
|
|
|
Vector3 l_velocity = m_velocity * Settings.Multiplier;
|
|
foreach(Rigidbody l_body in m_rigidBodies)
|
|
{
|
|
l_body.velocity = l_velocity;
|
|
l_body.angularVelocity = Vector3.zero;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
foreach(Rigidbody l_body in m_rigidBodies)
|
|
l_body.isKinematic = true;
|
|
|
|
if(!Settings.RestorePosition && (m_puppetReferences.hips != null))
|
|
{
|
|
Vector3 l_pos = m_puppetReferences.hips.position;
|
|
PlayerSetup.Instance.transform.position = l_pos;
|
|
}
|
|
}
|
|
|
|
foreach(Collider l_collider in m_colliders)
|
|
l_collider.enabled = m_enabled;
|
|
}
|
|
}
|
|
|
|
static Transform CloneTransform(Transform p_source, Transform p_parent, string p_name)
|
|
{
|
|
Transform l_target = new GameObject(p_name).transform;
|
|
l_target.parent = p_parent;
|
|
p_source.CopyGlobal(l_target);
|
|
return l_target;
|
|
}
|
|
}
|
|
}
|