forked from unity3d-jp/UnityChanSpringBone
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathAutoSpringBoneSetup.cs
More file actions
executable file
·207 lines (187 loc) · 8.15 KB
/
AutoSpringBoneSetup.cs
File metadata and controls
executable file
·207 lines (187 loc) · 8.15 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
using Unity.Animations.SpringBones.GameObjectExtensions;
using System.Collections.Generic;
using System.Linq;
using UnityEditor;
using UnityEngine;
#if UNITY_2020_2_OR_NEWER
using Localization = UnityEditor.L10n;
#else
using Localization = UnityEditor.Localization.Editor.Localization;
#endif
namespace Unity.Animations.SpringBones
{
public class AutoSpringBoneSetup
{
public class AutoSetupParameters
{
public bool createPivots;
public AutoSetupParameters()
{
createPivots = true;
}
}
public static string FindCharacterBoneListPath(GameObject rootObject)
{
const string BoneListFilePattern = "*_BoneList.txt";
var modelDirectory = GetModelDirectoryFromGameObject(rootObject);
if (modelDirectory.Length > 0)
{
var filePaths = DirectoryUtil.GetFiles(modelDirectory, BoneListFilePattern);
if (filePaths.Length > 0)
{
return filePaths[0];
}
}
return "";
}
public static IEnumerable<GameObject> GetBonesDesignatedForDynamics(GameObject rootObject)
{
// First try to find bone list from character's model data folder
var boneListPath = FindCharacterBoneListPath(rootObject);
if (boneListPath.Length > 0)
{
var boneListText = FileUtil.ReadAllText(boneListPath);
var boneNames = GetDynamicsBoneNamesFromBoneList(boneListText).ToArray();
var dynamicsBones = new List<GameObject>(boneNames.Count());
var allNodes = rootObject.GetComponentsInChildren<Transform>(true);
foreach (var boneName in boneNames)
{
var transform = allNodes.FirstOrDefault(item => item.name == boneName);
if (transform != null)
{
dynamicsBones.Add(transform.gameObject);
}
else
{
Debug.LogWarning("ダイナミクス骨が見つかりません: " + boneName);
}
}
return dynamicsBones;
}
return new List<GameObject>(0);
}
public static IEnumerable<string> GetDynamicsBoneNamesFromBoneList(string boneListText)
{
const int DynamicsBoneColumn = 3;
char[] EntrySeparators = { ',' };
var sourceLines = boneListText.Split("\r\n".ToCharArray());
var boneNames = new List<string>(sourceLines.Length);
foreach (var sourceLine in sourceLines)
{
var line = sourceLine.Trim();
if (line.Length > 0 && !line.StartsWith("//"))
{
var entries = line.Split(EntrySeparators, System.StringSplitOptions.None);
if (entries.Length > DynamicsBoneColumn)
{
var boneName = entries[DynamicsBoneColumn].Trim();
if (boneName.Length > 0)
{
boneNames.Add(boneName);
}
}
}
}
return boneNames;
}
public static SpringManager PerformSetup
(
GameObject rootObject,
IEnumerable<GameObject> newSpringBones,
AutoSetupParameters parameters
)
{
SpringBoneSetup.DestroySpringManagersAndBones(rootObject);
SpringColliderSetup.DestroySpringColliders(rootObject);
var springBones = new List<SpringBone>();
foreach (var boneParent in newSpringBones.Where(boneParent => boneParent.transform.childCount > 0))
{
springBones.Add(boneParent.AddComponent<SpringBone>());
}
var manager = rootObject.AddComponent<SpringManager>();
SpringBoneSetup.FindAndAssignSpringBones(manager, true);
if (parameters.createPivots)
{
foreach (var springBone in springBones)
{
SpringBoneSetup.CreateSpringPivotNode(springBone);
}
}
return manager;
}
public static void UpdateSpringManagerFromBoneList(SpringManager springManager)
{
var rootObject = springManager.gameObject;
var designatedSpringBones = GetBonesDesignatedForDynamics(rootObject);
var currentSpringBones = rootObject.GetComponentsInChildren<SpringBone>(true);
var skinBones = GameObjectUtil.GetAllBones(rootObject);
var springBonesToDestroy = currentSpringBones
.Where(bone => !designatedSpringBones.Contains(bone.gameObject));
foreach (var boneToDestroy in springBonesToDestroy)
{
var pivot = boneToDestroy.pivotNode;
if (pivot != null
&& SpringBoneSetup.IsPivotProbablySafeToDestroy(pivot, skinBones))
{
SpringBoneSetup.DestroyUnityObject(pivot.gameObject);
}
SpringBoneSetup.DestroyUnityObject(boneToDestroy);
}
var objectsToAddBonesTo = designatedSpringBones
.Where(bone => bone.GetComponent<SpringBone>() == null);
foreach (var newBoneOwner in objectsToAddBonesTo)
{
var newSpringBone = newBoneOwner.AddComponent<SpringBone>();
SpringBoneSetup.CreateSpringPivotNode(newSpringBone);
}
SpringBoneSetup.FindAndAssignSpringBones(springManager, true);
}
// private
private static bool IsProbablyFacialObject(Transform transform)
{
string[] boneNameTokensToFind = { "head", "neck" };
var transformToCheck = transform;
while (transformToCheck != null)
{
var lowerName = transformToCheck.name.ToLowerInvariant();
if (boneNameTokensToFind.Any(item => lowerName.Contains(item)))
{
return true;
}
transformToCheck = transformToCheck.parent;
}
return false;
}
private static string GetModelDirectoryFromGameObject(GameObject rootObject)
{
// Find a mesh?
var meshPaths = rootObject.GetComponentsInChildren<SkinnedMeshRenderer>(true)
.Where(renderer => renderer.sharedMesh != null
&& !IsProbablyFacialObject(renderer.transform))
.Select(renderer => AssetDatabase.GetAssetPath(renderer.sharedMesh))
.Where(path => path.Length > 0);
if (meshPaths.Any())
{
var modelDirectory = System.IO.Path.GetDirectoryName(meshPaths.First());
return PathUtil.AssetPathToSystemPath(modelDirectory);
}
// Find a texture and go up from there
var renderers = rootObject.GetComponentsInChildren<Renderer>(true)
.Where(renderer => !IsProbablyFacialObject(renderer.transform));
foreach (var renderer in renderers)
{
var materials = renderer.sharedMaterials;
var texturePaths = materials
.Where(material => material.mainTexture != null)
.Select(material => AssetDatabase.GetAssetPath(material.mainTexture));
if (texturePaths.Any())
{
var textureDirectory = System.IO.Path.GetDirectoryName(texturePaths.First());
var modelDirectory = System.IO.Path.GetDirectoryName(textureDirectory);
return PathUtil.AssetPathToSystemPath(modelDirectory);
}
}
return "";
}
}
}