t1 / TFDContents / Assets / KinectDemos / VariousDemos / Scripts / PoseModelHelper.cs @ 3
이력 | 보기 | 이력해설 | 다운로드 (6.4 KB)
1 | 3 | KTH | using UnityEngine; |
---|---|---|---|
2 | //using Windows.Kinect; |
||
3 | |||
4 | using System; |
||
5 | using System.Collections; |
||
6 | using System.Collections.Generic; |
||
7 | using System.Runtime.InteropServices; |
||
8 | using System.IO; |
||
9 | using System.Text; |
||
10 | |||
11 | |||
12 | /// <summary> |
||
13 | /// Pose model helper matches joints to model transforms. |
||
14 | /// </summary> |
||
15 | [RequireComponent(typeof(Animator))] |
||
16 | public class PoseModelHelper : MonoBehaviour |
||
17 | { |
||
18 | // Variable to hold all them bones. It will initialize the same size as initialRotations. |
||
19 | private Transform[] bones; |
||
20 | |||
21 | |||
22 | /// <summary> |
||
23 | /// Gets the number of bone transforms (array length). |
||
24 | /// </summary> |
||
25 | /// <returns>The number of bone transforms.</returns> |
||
26 | public int GetBoneTransformCount() |
||
27 | { |
||
28 | return bones != null ? bones.Length : 0; |
||
29 | } |
||
30 | |||
31 | /// <summary> |
||
32 | /// Gets the bone transform by index. |
||
33 | /// </summary> |
||
34 | /// <returns>The bone transform.</returns> |
||
35 | /// <param name="index">Index</param> |
||
36 | public Transform GetBoneTransform(int index) |
||
37 | { |
||
38 | if(index >= 0 && index < bones.Length) |
||
39 | { |
||
40 | return bones[index]; |
||
41 | } |
||
42 | |||
43 | return null; |
||
44 | } |
||
45 | |||
46 | /// <summary> |
||
47 | /// Gets the bone index by joint type. |
||
48 | /// </summary> |
||
49 | /// <returns>The bone index.</returns> |
||
50 | /// <param name="joint">Joint type</param> |
||
51 | /// <param name="bMirrored">If set to <c>true</c> gets the mirrored joint index.</param> |
||
52 | public int GetBoneIndexByJoint(KinectInterop.JointType joint, bool bMirrored) |
||
53 | { |
||
54 | int boneIndex = -1; |
||
55 | |||
56 | if(jointMap2boneIndex.ContainsKey(joint)) |
||
57 | { |
||
58 | boneIndex = !bMirrored ? jointMap2boneIndex[joint] : mirrorJointMap2boneIndex[joint]; |
||
59 | } |
||
60 | |||
61 | return boneIndex; |
||
62 | } |
||
63 | |||
64 | |||
65 | // transform caching gives performance boost since Unity calls GetComponent<Transform>() each time you call transform |
||
66 | private Transform _transformCache; |
||
67 | public new Transform transform |
||
68 | { |
||
69 | get |
||
70 | { |
||
71 | if (!_transformCache) |
||
72 | { |
||
73 | _transformCache = base.transform; |
||
74 | } |
||
75 | |||
76 | return _transformCache; |
||
77 | } |
||
78 | } |
||
79 | |||
80 | |||
81 | public void Awake() |
||
82 | { |
||
83 | // check for double start |
||
84 | if(bones != null) |
||
85 | return; |
||
86 | if(!gameObject.activeInHierarchy) |
||
87 | return; |
||
88 | |||
89 | // inits the bones array |
||
90 | bones = new Transform[31]; |
||
91 | |||
92 | // Map bones to the points the Kinect tracks |
||
93 | MapBones(); |
||
94 | } |
||
95 | |||
96 | // If the bones to be mapped have been declared, map that bone to the model. |
||
97 | private void MapBones() |
||
98 | { |
||
99 | // get bone transforms from the animator component |
||
100 | Animator animatorComponent = GetComponent<Animator>(); |
||
101 | |||
102 | for (int boneIndex = 0; boneIndex < bones.Length; boneIndex++) |
||
103 | { |
||
104 | if (!boneIndex2MecanimMap.ContainsKey(boneIndex)) |
||
105 | continue; |
||
106 | |||
107 | bones[boneIndex] = animatorComponent ? animatorComponent.GetBoneTransform(boneIndex2MecanimMap[boneIndex]) : null; |
||
108 | } |
||
109 | } |
||
110 | |||
111 | // dictionaries to speed up bones' processing |
||
112 | // the author of the terrific idea for kinect-joints to mecanim-bones mapping |
||
113 | // along with its initial implementation, including following dictionary is |
||
114 | // Mikhail Korchun (korchoon@gmail.com). Big thanks to this guy! |
||
115 | private readonly Dictionary<int, HumanBodyBones> boneIndex2MecanimMap = new Dictionary<int, HumanBodyBones> |
||
116 | { |
||
117 | {0, HumanBodyBones.Hips}, |
||
118 | {1, HumanBodyBones.Spine}, |
||
119 | {2, HumanBodyBones.Chest}, |
||
120 | {3, HumanBodyBones.Neck}, |
||
121 | {4, HumanBodyBones.Head}, |
||
122 | |||
123 | {5, HumanBodyBones.LeftUpperArm}, |
||
124 | {6, HumanBodyBones.LeftLowerArm}, |
||
125 | {7, HumanBodyBones.LeftHand}, |
||
126 | {8, HumanBodyBones.LeftIndexProximal}, |
||
127 | {9, HumanBodyBones.LeftIndexDistal}, |
||
128 | {10, HumanBodyBones.LeftThumbProximal}, |
||
129 | |||
130 | {11, HumanBodyBones.RightUpperArm}, |
||
131 | {12, HumanBodyBones.RightLowerArm}, |
||
132 | {13, HumanBodyBones.RightHand}, |
||
133 | {14, HumanBodyBones.RightIndexProximal}, |
||
134 | {15, HumanBodyBones.RightIndexDistal}, |
||
135 | {16, HumanBodyBones.RightThumbProximal}, |
||
136 | |||
137 | {17, HumanBodyBones.LeftUpperLeg}, |
||
138 | {18, HumanBodyBones.LeftLowerLeg}, |
||
139 | {19, HumanBodyBones.LeftFoot}, |
||
140 | {20, HumanBodyBones.LeftToes}, |
||
141 | |||
142 | {21, HumanBodyBones.RightUpperLeg}, |
||
143 | {22, HumanBodyBones.RightLowerLeg}, |
||
144 | {23, HumanBodyBones.RightFoot}, |
||
145 | {24, HumanBodyBones.RightToes}, |
||
146 | }; |
||
147 | |||
148 | private readonly Dictionary<KinectInterop.JointType, int> jointMap2boneIndex = new Dictionary<KinectInterop.JointType, int> |
||
149 | { |
||
150 | {KinectInterop.JointType.SpineBase, 0}, |
||
151 | {KinectInterop.JointType.SpineMid, 1}, |
||
152 | {KinectInterop.JointType.SpineShoulder, 2}, |
||
153 | {KinectInterop.JointType.Neck, 3}, |
||
154 | {KinectInterop.JointType.Head, 4}, |
||
155 | |||
156 | {KinectInterop.JointType.ShoulderLeft, 5}, |
||
157 | {KinectInterop.JointType.ElbowLeft, 6}, |
||
158 | {KinectInterop.JointType.WristLeft, 7}, |
||
159 | {KinectInterop.JointType.HandLeft, 8}, |
||
160 | |||
161 | {KinectInterop.JointType.HandTipLeft, 9}, |
||
162 | {KinectInterop.JointType.ThumbLeft, 10}, |
||
163 | |||
164 | {KinectInterop.JointType.ShoulderRight, 11}, |
||
165 | {KinectInterop.JointType.ElbowRight, 12}, |
||
166 | {KinectInterop.JointType.WristRight, 13}, |
||
167 | {KinectInterop.JointType.HandRight, 14}, |
||
168 | |||
169 | {KinectInterop.JointType.HandTipRight, 15}, |
||
170 | {KinectInterop.JointType.ThumbRight, 16}, |
||
171 | |||
172 | {KinectInterop.JointType.HipLeft, 17}, |
||
173 | {KinectInterop.JointType.KneeLeft, 18}, |
||
174 | {KinectInterop.JointType.AnkleLeft, 19}, |
||
175 | {KinectInterop.JointType.FootLeft, 20}, |
||
176 | |||
177 | {KinectInterop.JointType.HipRight, 21}, |
||
178 | {KinectInterop.JointType.KneeRight, 22}, |
||
179 | {KinectInterop.JointType.AnkleRight, 23}, |
||
180 | {KinectInterop.JointType.FootRight, 24}, |
||
181 | }; |
||
182 | |||
183 | private readonly Dictionary<KinectInterop.JointType, int> mirrorJointMap2boneIndex = new Dictionary<KinectInterop.JointType, int> |
||
184 | { |
||
185 | {KinectInterop.JointType.SpineBase, 0}, |
||
186 | {KinectInterop.JointType.SpineMid, 1}, |
||
187 | {KinectInterop.JointType.SpineShoulder, 2}, |
||
188 | {KinectInterop.JointType.Neck, 3}, |
||
189 | {KinectInterop.JointType.Head, 4}, |
||
190 | |||
191 | {KinectInterop.JointType.ShoulderRight, 5}, |
||
192 | {KinectInterop.JointType.ElbowRight, 6}, |
||
193 | {KinectInterop.JointType.WristRight, 7}, |
||
194 | {KinectInterop.JointType.HandRight, 8}, |
||
195 | |||
196 | {KinectInterop.JointType.HandTipRight, 9}, |
||
197 | {KinectInterop.JointType.ThumbRight, 10}, |
||
198 | |||
199 | {KinectInterop.JointType.ShoulderLeft, 11}, |
||
200 | {KinectInterop.JointType.ElbowLeft, 12}, |
||
201 | {KinectInterop.JointType.WristLeft, 13}, |
||
202 | {KinectInterop.JointType.HandLeft, 14}, |
||
203 | |||
204 | {KinectInterop.JointType.HandTipLeft, 15}, |
||
205 | {KinectInterop.JointType.ThumbLeft, 16}, |
||
206 | |||
207 | {KinectInterop.JointType.HipRight, 17}, |
||
208 | {KinectInterop.JointType.KneeRight, 18}, |
||
209 | {KinectInterop.JointType.AnkleRight, 19}, |
||
210 | {KinectInterop.JointType.FootRight, 20}, |
||
211 | |||
212 | {KinectInterop.JointType.HipLeft, 21}, |
||
213 | {KinectInterop.JointType.KneeLeft, 22}, |
||
214 | {KinectInterop.JointType.AnkleLeft, 23}, |
||
215 | {KinectInterop.JointType.FootLeft, 24}, |
||
216 | }; |
||
217 | |||
218 | |||
219 | } |