using System.Collections.Generic; using UnityEngine; using Kinect = Windows.Kinect; public class BodySourceView : MonoBehaviour { public Transform[] joints; public GameObject BodySourceManager; public Transform cam; public PlayerReplay playerReplay; public ModeController modeController; public GameObject body; public bool wristLeftLate; public bool wristRightLate; private Dictionary _Bodies = new Dictionary(); private BodySourceManager _BodyManager; private Dictionary _BoneMap = new Dictionary() { { Kinect.JointType.FootLeft, Kinect.JointType.AnkleLeft }, { Kinect.JointType.AnkleLeft, Kinect.JointType.KneeLeft }, { Kinect.JointType.KneeLeft, Kinect.JointType.HipLeft }, { Kinect.JointType.HipLeft, Kinect.JointType.SpineBase }, { Kinect.JointType.FootRight, Kinect.JointType.AnkleRight }, { Kinect.JointType.AnkleRight, Kinect.JointType.KneeRight }, { Kinect.JointType.KneeRight, Kinect.JointType.HipRight }, { Kinect.JointType.HipRight, Kinect.JointType.SpineBase }, { Kinect.JointType.HandTipLeft, Kinect.JointType.HandLeft }, { Kinect.JointType.ThumbLeft, Kinect.JointType.HandLeft }, { Kinect.JointType.HandLeft, Kinect.JointType.WristLeft }, { Kinect.JointType.WristLeft, Kinect.JointType.ElbowLeft }, { Kinect.JointType.ElbowLeft, Kinect.JointType.ShoulderLeft }, { Kinect.JointType.ShoulderLeft, Kinect.JointType.SpineShoulder }, { Kinect.JointType.HandTipRight, Kinect.JointType.HandRight }, { Kinect.JointType.ThumbRight, Kinect.JointType.HandRight }, { Kinect.JointType.HandRight, Kinect.JointType.WristRight }, { Kinect.JointType.WristRight, Kinect.JointType.ElbowRight }, { Kinect.JointType.ElbowRight, Kinect.JointType.ShoulderRight }, { Kinect.JointType.ShoulderRight, Kinect.JointType.SpineShoulder }, { Kinect.JointType.SpineBase, Kinect.JointType.SpineMid }, { Kinect.JointType.SpineMid, Kinect.JointType.SpineShoulder }, { Kinect.JointType.SpineShoulder, Kinect.JointType.Neck }, { Kinect.JointType.Neck, Kinect.JointType.Head }, }; void Update() { if (BodySourceManager == null) { return; } _BodyManager = BodySourceManager.GetComponent(); if (_BodyManager == null) { return; } Kinect.Body[] data = _BodyManager.GetData(); if (data == null) { return; } List trackedIds = new List(); foreach (var body in data) { if (body == null) { continue; } if (body.IsTracked) { trackedIds.Add(body.TrackingId); } } List knownIds = new List(_Bodies.Keys); // First delete untracked bodies foreach (ulong trackingId in knownIds) { if (!trackedIds.Contains(trackingId)) { Destroy(_Bodies[trackingId]); _Bodies.Remove(trackingId); } } foreach (var body in data) { if (body == null) { continue; } if (body.IsTracked) { if (!_Bodies.ContainsKey(body.TrackingId)) { _Bodies[body.TrackingId] = CreateBodyObject(body.TrackingId); } RefreshBodyObject(body, _Bodies[body.TrackingId]); } } } private GameObject CreateBodyObject(ulong id) { body = new GameObject("Body:" + id); body.transform.parent = gameObject.transform; for (Kinect.JointType jt = Kinect.JointType.SpineBase; jt <= Kinect.JointType.ThumbRight; jt++) { GameObject jointObj = GameObject.CreatePrimitive(PrimitiveType.Cube); // Give WristLeft and WristRight rigidbody if (jt == Kinect.JointType.WristLeft) { jointObj.tag = "WristLeft"; Rigidbody rb = jointObj.AddComponent(); rb.useGravity = false; rb.isKinematic = true; } else if (jt == Kinect.JointType.WristRight) { jointObj.tag = "WristRight"; Rigidbody rb = jointObj.AddComponent(); rb.useGravity = false; rb.isKinematic = true; } LineRenderer lr = jointObj.AddComponent(); //lr.SetVertexCount(2); lr.positionCount = 2; //lr.SetWidth(0.05f, 0.05f); lr.startWidth = 0.05f; lr.endWidth = 0.05f; Material whiteDiffuseMat = new Material(Shader.Find("Sprites/Default")); whiteDiffuseMat.color = new Color(1, 1, 1, 0.5f); lr.material = whiteDiffuseMat; jointObj.transform.localScale = new Vector3(0.05f, 0.05f, 0.05f); jointObj.name = jt.ToString(); jointObj.transform.parent = body.transform; } return body; } private void RefreshBodyObject(Kinect.Body body, GameObject bodyObject) { Vector3[] initJoints = new Vector3[25]; for (Kinect.JointType jt = Kinect.JointType.SpineBase; jt <= Kinect.JointType.ThumbRight; jt++) { Kinect.Joint sourceJoint = body.Joints[jt]; initJoints[(int)jt] = GetVector3FromJoint(sourceJoint); } Vector3[] rescaledJoints = RescaleJoints(initJoints); for (Kinect.JointType jt = Kinect.JointType.SpineBase; jt <= Kinect.JointType.ThumbRight; jt++) { Kinect.Joint sourceJoint = body.Joints[jt]; Kinect.Joint? targetJoint = null; if (_BoneMap.ContainsKey(jt)) { targetJoint = body.Joints[_BoneMap[jt]]; } Transform jointObj = bodyObject.transform.Find(jt.ToString()); //jointObj.localPosition = GetVector3FromJoint(sourceJoint); jointObj.localPosition = rescaledJoints[(int)jt]; if (jt == Kinect.JointType.Head) { // Make head joint invisible jointObj.GetComponent().enabled = false; } if (jt == Kinect.JointType.Neck) { if (modeController.perspective == ModeController.Perspective.FirstPersonPerspective) { cam.position = new Vector3(jointObj.position.x, jointObj.position.y + 0.05f, jointObj.position.z - 0.05f); jointObj.GetComponent().enabled = false; } else { cam.position = new Vector3(jointObj.position.x, jointObj.position.y + 0.1f, jointObj.position.z + 0.5f); } } // Make these joints invisible if (jt == Kinect.JointType.ThumbLeft || jt == Kinect.JointType.ThumbRight || jt == Kinect.JointType.HandLeft || jt == Kinect.JointType.HandRight || jt == Kinect.JointType.HandTipLeft || jt == Kinect.JointType.HandTipRight) { jointObj.GetComponent().enabled = false; } LineRenderer lr = jointObj.GetComponent(); if (targetJoint.HasValue && jt != Kinect.JointType.Neck && jt != Kinect.JointType.ThumbLeft && jt != Kinect.JointType.ThumbRight && jt != Kinect.JointType.HandLeft && jt != Kinect.JointType.HandRight && jt != Kinect.JointType.HandTipLeft && jt != Kinect.JointType.HandTipRight) // Do not make LineRenderer for these joints { lr.SetPosition(0, jointObj.localPosition); //lr.SetPosition(1, GetVector3FromJoint(targetJoint.Value)); lr.SetPosition(1, rescaledJoints[(int)_BoneMap[jt]]); // Coloring the line renderer //lr.SetColors(GetColorForState(sourceJoint.TrackingState), GetColorForState(targetJoint.Value.TrackingState)); //lr.startColor = GetColorForState(sourceJoint.TrackingState); //lr.endColor = GetColorForState(targetJoint.Value.TrackingState); } else { lr.enabled = false; } if (modeController.perspective == ModeController.Perspective.FirstPersonPerspective && jt == Kinect.JointType.SpineShoulder) { lr.enabled = false; } // Record joints to PlayerReplay.cs playerReplay.joints[(int)jt] = jointObj; } } private static Color GetColorForState(Kinect.TrackingState state) { switch (state) { case Kinect.TrackingState.Tracked: return Color.green; case Kinect.TrackingState.Inferred: return Color.red; default: return Color.black; } } private Vector3 GetVector3FromJoint(Kinect.Joint joint) { //return new Vector3(joint.Position.X * -10, joint.Position.Y * 10, joint.Position.Z * 10); return new Vector3(joint.Position.X * -1, joint.Position.Y, joint.Position.Z); } private Quaternion ConvertKinectOrientationToUnity(Kinect.Vector4 orientation) { Quaternion orientationUnity; orientationUnity.x = -orientation.X; orientationUnity.y = orientation.Y; orientationUnity.z = orientation.Z; orientationUnity.w = orientation.W; return orientationUnity; } private Vector3[] RescaleJoints(Vector3[] initJoints) { Vector3[] rescaledJoints = new Vector3[25]; rescaledJoints[(int)Kinect.JointType.SpineBase] = Vector3.zero; // Lower part rescaledJoints[(int)Kinect.JointType.HipLeft] = GetTargetJointWithDesiredLength( initJoints[(int)Kinect.JointType.SpineBase], initJoints[(int)Kinect.JointType.HipLeft], 0.08f) + rescaledJoints[(int)Kinect.JointType.SpineBase]; rescaledJoints[(int)Kinect.JointType.HipRight] = GetTargetJointWithDesiredLength( initJoints[(int)Kinect.JointType.SpineBase], initJoints[(int)Kinect.JointType.HipRight], 0.08f) + rescaledJoints[(int)Kinect.JointType.SpineBase]; rescaledJoints[(int)Kinect.JointType.KneeLeft] = GetTargetJointWithDesiredLength( initJoints[(int)Kinect.JointType.HipLeft], initJoints[(int)Kinect.JointType.KneeLeft], 0.38f) + rescaledJoints[(int)Kinect.JointType.HipLeft]; rescaledJoints[(int)Kinect.JointType.KneeRight] = GetTargetJointWithDesiredLength( initJoints[(int)Kinect.JointType.HipRight], initJoints[(int)Kinect.JointType.KneeRight], 0.38f) + rescaledJoints[(int)Kinect.JointType.HipRight]; rescaledJoints[(int)Kinect.JointType.AnkleLeft] = GetTargetJointWithDesiredLength( initJoints[(int)Kinect.JointType.KneeLeft], initJoints[(int)Kinect.JointType.AnkleLeft], 0.35f) + rescaledJoints[(int)Kinect.JointType.KneeLeft]; rescaledJoints[(int)Kinect.JointType.AnkleRight] = GetTargetJointWithDesiredLength( initJoints[(int)Kinect.JointType.KneeRight], initJoints[(int)Kinect.JointType.AnkleRight], 0.35f) + rescaledJoints[(int)Kinect.JointType.KneeRight]; rescaledJoints[(int)Kinect.JointType.FootLeft] = GetTargetJointWithDesiredLength( initJoints[(int)Kinect.JointType.AnkleLeft], initJoints[(int)Kinect.JointType.FootLeft], 0.1f) + rescaledJoints[(int)Kinect.JointType.AnkleLeft] + new Vector3(0, 0.03f, 0); //initJoints[(int)Kinect.JointType.AnkleLeft], initJoints[(int)Kinect.JointType.FootLeft], 0.1f) + rescaledJoints[(int)Kinect.JointType.AnkleLeft]; rescaledJoints[(int)Kinect.JointType.FootRight] = GetTargetJointWithDesiredLength( initJoints[(int)Kinect.JointType.AnkleRight], initJoints[(int)Kinect.JointType.FootRight], 0.1f) + rescaledJoints[(int)Kinect.JointType.AnkleRight] + new Vector3(0, 0.03f, 0); //initJoints[(int)Kinect.JointType.AnkleRight], initJoints[(int)Kinect.JointType.FootRight], 0.1f) + rescaledJoints[(int)Kinect.JointType.AnkleRight]; // Upper part rescaledJoints[(int)Kinect.JointType.SpineMid] = GetTargetJointWithDesiredLength( initJoints[(int)Kinect.JointType.SpineBase], initJoints[(int)Kinect.JointType.SpineMid], 0.2f) + rescaledJoints[(int)Kinect.JointType.SpineBase]; rescaledJoints[(int)Kinect.JointType.SpineShoulder] = GetTargetJointWithDesiredLength( initJoints[(int)Kinect.JointType.SpineMid], initJoints[(int)Kinect.JointType.SpineShoulder], 0.2f) + rescaledJoints[(int)Kinect.JointType.SpineMid]; rescaledJoints[(int)Kinect.JointType.ShoulderLeft] = GetTargetJointWithDesiredLength( initJoints[(int)Kinect.JointType.SpineShoulder], initJoints[(int)Kinect.JointType.ShoulderLeft], 0.12f) + rescaledJoints[(int)Kinect.JointType.SpineShoulder]; rescaledJoints[(int)Kinect.JointType.ShoulderRight] = GetTargetJointWithDesiredLength( initJoints[(int)Kinect.JointType.SpineShoulder], initJoints[(int)Kinect.JointType.ShoulderRight], 0.12f) + rescaledJoints[(int)Kinect.JointType.SpineShoulder]; rescaledJoints[(int)Kinect.JointType.ElbowLeft] = GetTargetJointWithDesiredLength( initJoints[(int)Kinect.JointType.ShoulderLeft], initJoints[(int)Kinect.JointType.ElbowLeft], 0.28f) + rescaledJoints[(int)Kinect.JointType.ShoulderLeft]; rescaledJoints[(int)Kinect.JointType.ElbowRight] = GetTargetJointWithDesiredLength( initJoints[(int)Kinect.JointType.ShoulderRight], initJoints[(int)Kinect.JointType.ElbowRight], 0.28f) + rescaledJoints[(int)Kinect.JointType.ShoulderRight]; rescaledJoints[(int)Kinect.JointType.WristLeft] = GetTargetJointWithDesiredLength( initJoints[(int)Kinect.JointType.ElbowLeft], initJoints[(int)Kinect.JointType.WristLeft], 0.28f) + rescaledJoints[(int)Kinect.JointType.ElbowLeft]; rescaledJoints[(int)Kinect.JointType.WristRight] = GetTargetJointWithDesiredLength( initJoints[(int)Kinect.JointType.ElbowRight], initJoints[(int)Kinect.JointType.WristRight], 0.28f) + rescaledJoints[(int)Kinect.JointType.ElbowRight]; rescaledJoints[(int)Kinect.JointType.Neck] = GetTargetJointWithDesiredLength( initJoints[(int)Kinect.JointType.SpineShoulder], initJoints[(int)Kinect.JointType.Neck], 0.1f) + rescaledJoints[(int)Kinect.JointType.SpineShoulder]; return rescaledJoints; } private Vector3 GetTargetJointWithDesiredLength(Vector3 source, Vector3 target, float desiredLength) { Vector3 sourceToTarget = target - source; float length = sourceToTarget.magnitude; return sourceToTarget * desiredLength / length; } }