public static void UnifyChannelsWithSameBoneIndex(this EditableBoneWeight editableBoneWeight)
{
int channelCount = editableBoneWeight.GetChannelCount();
for (int i = 0; i < channelCount; ++i)
{
BoneWeightData data = editableBoneWeight.GetBoneWeightData(i);
if (!editableBoneWeight.IsChannelEnabled(i))
{
continue;
}
bool weightChanged = false;
for (int j = i + 1; j < channelCount; ++j)
{
BoneWeightData otherData = editableBoneWeight.GetBoneWeightData(j);
if (otherData.boneIndex == data.boneIndex)
{
weightChanged = true;
data.weight += otherData.weight;
editableBoneWeight.EnableChannel(j, false);
}
}
if (weightChanged)
{
editableBoneWeight.SetBoneWeightData(i, data);
editableBoneWeight.CompensateOtherChannels(i);
}
}
}
private void ChannelsGUI()
{
for (int channelIndex = 0; channelIndex < 4; ++channelIndex)
{
bool channelEnabled;
bool isChannelEnabledMixed;
bool isBoneIndexMixed;
bool isWeightMixed;
BoneWeightData boneWeightData;
spriteMeshData.GetMultiEditChannelData(selection, channelIndex, out channelEnabled, out boneWeightData, out isChannelEnabledMixed, out isBoneIndexMixed, out isWeightMixed);
BoneWeightData newBoneWeightData = new BoneWeightData();
newBoneWeightData = boneWeightData;
bool newChannelEnabled = channelEnabled;
EditorGUI.BeginChangeCheck();
WeightChannelDrawer(ref newChannelEnabled, ref newBoneWeightData, isChannelEnabledMixed, isBoneIndexMixed, isWeightMixed);
if (EditorGUI.EndChangeCheck())
{
spriteMeshData.SetMultiEditChannelData(selection, channelIndex, channelEnabled, newChannelEnabled, boneWeightData, newBoneWeightData);
}
}
}
public static void RemoveBone(this EditableBoneWeight editableBoneWeight, int boneIndex)
{
int channelCount = editableBoneWeight.GetChannelCount();
for (int i = 0; i < channelCount; ++i)
{
BoneWeightData data = editableBoneWeight.GetBoneWeightData(i);
if (data.boneIndex > boneIndex)
{
data.boneIndex -= 1;
}
else if (data.boneIndex == boneIndex)
{
data.boneIndex = 0;
data.weight = 0f;
editableBoneWeight.EnableChannel(i, false);
}
editableBoneWeight.SetBoneWeightData(i, data);
}
editableBoneWeight.NormalizeChannels();
editableBoneWeight.SortChannels();
}
public void SetBoneWeightData(int channelIndex, BoneWeightData boneWeightData)
{
if (channelIndex < 0 || channelIndex >= GetChannelCount())
{
throw new IndexOutOfRangeException("channel index out of range");
}
m_Channels[channelIndex].boneWeightData = boneWeightData;
}
public static void GetMultiEditChannelData(this SpriteMeshData spriteMeshData, ISelection selection, int channelIndex,
out bool channelEnabled, out BoneWeightData boneWeightData,
out bool isChannelEnabledMixed, out bool isBoneIndexMixed, out bool isWeightMixed)
{
if (selection == null)
{
throw new ArgumentNullException("selection is null");
}
bool first = true;
channelEnabled = false;
boneWeightData = new BoneWeightData();
isChannelEnabledMixed = false;
isBoneIndexMixed = false;
isWeightMixed = false;
foreach (int i in selection)
{
EditableBoneWeight editableBoneWeight = spriteMeshData.vertices[i].editableBoneWeight;
BoneWeightData otherBoneWeightData = editableBoneWeight.GetBoneWeightData(channelIndex);
if (first)
{
channelEnabled = editableBoneWeight.IsChannelEnabled(channelIndex);
boneWeightData.boneIndex = otherBoneWeightData.boneIndex;
boneWeightData.weight = otherBoneWeightData.weight;
first = false;
}
else
{
if (channelEnabled != editableBoneWeight.IsChannelEnabled(channelIndex))
{
isChannelEnabledMixed = true;
channelEnabled = false;
}
if (boneWeightData.boneIndex != otherBoneWeightData.boneIndex)
{
isBoneIndexMixed = true;
boneWeightData.boneIndex = -1;
}
if (boneWeightData.weight != otherBoneWeightData.weight)
{
isWeightMixed = true;
boneWeightData.weight = 0f;
}
}
}
}
public static void ValidateChannels(this EditableBoneWeight editableBoneWeight)
{
for (int channelIndex = 0; channelIndex < editableBoneWeight.GetChannelCount(); ++channelIndex)
{
BoneWeightData data = editableBoneWeight.GetBoneWeightData(channelIndex);
if (!editableBoneWeight.IsChannelEnabled(channelIndex))
{
data.weight = 0f;
}
data.weight = Mathf.Clamp01(data.weight);
editableBoneWeight.SetBoneWeightData(channelIndex, data);
}
}
private static EditableBoneWeight Lerp(EditableBoneWeight first, EditableBoneWeight second, float t)
{
EditableBoneWeight result = new EditableBoneWeight();
foreach (BoneWeightChannel channel in first)
{
if (!channel.enabled)
{
continue;
}
BoneWeightData data = channel.boneWeightData;
data.weight *= (1f - t);
if (data.weight > 0f)
{
result.AddChannel(data, true);
}
}
foreach (BoneWeightChannel channel in second)
{
if (!channel.enabled)
{
continue;
}
BoneWeightData data = channel.boneWeightData;
data.weight *= t;
if (data.weight > 0f)
{
result.AddChannel(data, true);
}
}
result.UnifyChannelsWithSameBoneIndex();
if (result.GetWeightSum() > 1f)
{
result.NormalizeChannels();
}
result.FilterChannels(0f);
result.ClampChannels(4, true);
return(result);
}
public static float GetWeightSum(this EditableBoneWeight editableBoneWeight)
{
float sum = 0f;
for (int i = 0; i < editableBoneWeight.GetChannelCount(); ++i)
{
if (editableBoneWeight.IsChannelEnabled(i))
{
BoneWeightData data = editableBoneWeight.GetBoneWeightData(i);
sum += data.weight;
}
}
return(sum);
}
public static int GetChannelFromBoneIndex(this EditableBoneWeight editableBoneWeight, int boneIndex)
{
for (int i = 0; i < editableBoneWeight.GetChannelCount(); ++i)
{
if (editableBoneWeight.IsChannelEnabled(i))
{
BoneWeightData data = editableBoneWeight.GetBoneWeightData(i);
if (data.boneIndex == boneIndex)
{
return(i);
}
}
}
return(-1);
}
public static void CompensateOtherChannels(this EditableBoneWeight editableBoneWeight, int masterChannelIndex)
{
ValidateChannels(editableBoneWeight);
int validChannelCount = 0;
float sum = 0f;
for (int i = 0; i < editableBoneWeight.GetChannelCount(); ++i)
{
if (i != masterChannelIndex && editableBoneWeight.IsChannelEnabled(i))
{
BoneWeightData data = editableBoneWeight.GetBoneWeightData(i);
sum += data.weight;
++validChannelCount;
}
}
if (validChannelCount == 0)
{
return;
}
BoneWeightData channelData = editableBoneWeight.GetBoneWeightData(masterChannelIndex);
float targetSum = 1f - channelData.weight;
for (int i = 0; i < editableBoneWeight.GetChannelCount(); ++i)
{
if (i != masterChannelIndex && editableBoneWeight.IsChannelEnabled(i))
{
BoneWeightData data = editableBoneWeight.GetBoneWeightData(i);
if (sum == 0f)
{
data.weight = targetSum / validChannelCount;
}
else
{
data.weight = data.weight * targetSum / sum;
}
editableBoneWeight.SetBoneWeightData(i, data);
}
}
}
private void SetWeight(float value, bool createNewChannel = true)
{
if (boneIndex == -1)
{
return;
}
Debug.Assert(selection != null);
for (int i = 0; i < spriteMeshData.vertices.Count; ++i)
{
if (selection.Count == 0 && emptySelectionEditsAll ||
selection.Count > 0 && selection.IsSelected(i))
{
EditableBoneWeight editableBoneWeight = spriteMeshData.vertices[i].editableBoneWeight;
int channel = editableBoneWeight.GetChannelFromBoneIndex(boneIndex);
if (channel == -1)
{
if (createNewChannel)
{
editableBoneWeight.AddChannel(new BoneWeightData(boneIndex, 0f), true);
channel = editableBoneWeight.GetChannelFromBoneIndex(boneIndex);
}
else
{
continue;
}
}
BoneWeightData data = editableBoneWeight.GetBoneWeightData(channel);
data.weight += value;
editableBoneWeight.SetBoneWeightData(channel, data);
if (editableBoneWeight.GetWeightSum() > 1f)
{
editableBoneWeight.CompensateOtherChannels(channel);
}
editableBoneWeight.FilterChannels(0f);
}
}
}
public static void FilterChannels(this EditableBoneWeight editableBoneWeight, float weightTolerance)
{
int channelCount = editableBoneWeight.GetChannelCount();
for (int i = 0; i < channelCount; ++i)
{
BoneWeightData data = editableBoneWeight.GetBoneWeightData(i);
if (data.weight <= weightTolerance)
{
data.boneIndex = 0;
data.weight = 0f;
editableBoneWeight.SetBoneWeightData(i, data);
editableBoneWeight.EnableChannel(i, false);
}
}
}
public static void SmoothWeights(BoneWeight[] boneWeightIn, IList <int> indices, int boneCount, BoneWeight[] boneWeightOut)
{
Debug.Assert(boneWeightIn != null);
Debug.Assert(boneWeightOut != null);
Debug.Assert(boneWeightIn != boneWeightOut);
Debug.Assert(boneWeightIn.Length == boneWeightOut.Length);
PrepareTempBuffers(boneWeightIn.Length, boneCount);
EditableBoneWeight editableBoneWeight = new EditableBoneWeight();
for (int i = 0; i < boneWeightIn.Length; ++i)
{
editableBoneWeight.SetFromBoneWeight(boneWeightIn[i]);
for (int channelIndex = 0; channelIndex < editableBoneWeight.GetChannelCount(); ++channelIndex)
{
if (editableBoneWeight.IsChannelEnabled(channelIndex))
{
BoneWeightData boneWeightData = editableBoneWeight.GetBoneWeightData(channelIndex);
m_DataInTemp[i, boneWeightData.boneIndex] = boneWeightData.weight;
}
}
}
SmoothPerVertexData(indices, m_DataInTemp, m_DataOutTemp);
for (int i = 0; i < boneWeightIn.Length; ++i)
{
editableBoneWeight.Clear();
for (int boneIndex = 0; boneIndex < boneCount; ++boneIndex)
{
float weight = m_DataOutTemp[i, boneIndex];
int boneIndex2 = weight > 0f ? boneIndex : 0;
editableBoneWeight.AddChannel(new BoneWeightData(boneIndex2, weight), weight > 0);
}
editableBoneWeight.ClampChannels(4);
editableBoneWeight.NormalizeChannels();
boneWeightOut[i] = editableBoneWeight.ToBoneWeight(false);
}
}
private void WeightChannelDrawer(
ref bool isChannelEnabled, ref BoneWeightData boneWeightData,
bool isChannelEnabledMixed = false, bool isBoneIndexMixed = false, bool isWeightMixed = false)
{
EditorGUILayout.BeginHorizontal();
EditorGUIUtility.fieldWidth = 1f;
EditorGUIUtility.labelWidth = 1f;
EditorGUI.showMixedValue = isChannelEnabledMixed;
isChannelEnabled = EditorGUILayout.Toggle(GUIContent.none, isChannelEnabled);
EditorGUIUtility.fieldWidth = 30f;
EditorGUIUtility.labelWidth = 30f;
using (new EditorGUI.DisabledScope(!isChannelEnabled && !isChannelEnabledMixed))
{
int tempBoneIndex = GUI.enabled ? boneWeightData.boneIndex : -1;
EditorGUI.BeginChangeCheck();
EditorGUI.showMixedValue = GUI.enabled && isBoneIndexMixed;
tempBoneIndex = EditorGUILayout.Popup(tempBoneIndex, MeshModuleUtility.GetBoneNameList(spriteMeshData));
if (EditorGUI.EndChangeCheck())
{
boneWeightData.boneIndex = tempBoneIndex;
}
EditorGUIUtility.fieldWidth = 45f;
EditorGUI.showMixedValue = isWeightMixed;
boneWeightData.weight = EditorGUILayout.Slider(GUIContent.none, boneWeightData.weight, 0f, 1f);
}
EditorGUILayout.EndHorizontal();
EditorGUI.showMixedValue = false;
EditorGUIUtility.labelWidth = -1;
EditorGUIUtility.fieldWidth = -1;
}
public static void SetMultiEditChannelData(this SpriteMeshData spriteMeshData, ISelection selection, int channelIndex,
bool referenceChannelEnabled, bool newChannelEnabled, BoneWeightData referenceData, BoneWeightData newData)
{
if (selection == null)
{
throw new ArgumentNullException("selection is null");
}
bool channelEnabledChanged = referenceChannelEnabled != newChannelEnabled;
bool boneIndexChanged = referenceData.boneIndex != newData.boneIndex;
bool weightChanged = referenceData.weight != newData.weight;
foreach (int i in selection)
{
EditableBoneWeight editableBoneWeight = spriteMeshData.vertices[i].editableBoneWeight;
BoneWeightData data = editableBoneWeight.GetBoneWeightData(channelIndex);
if (channelEnabledChanged)
{
editableBoneWeight.EnableChannel(channelIndex, newChannelEnabled);
}
if (boneIndexChanged)
{
data.boneIndex = newData.boneIndex;
}
if (weightChanged)
{
data.weight = newData.weight;
}
editableBoneWeight.SetBoneWeightData(channelIndex, data);
if (channelEnabledChanged || weightChanged)
{
editableBoneWeight.CompensateOtherChannels(channelIndex);
}
}
}
public static void NormalizeChannels(this EditableBoneWeight editableBoneWeight)
{
ValidateChannels(editableBoneWeight);
float sum = editableBoneWeight.GetWeightSum();
if (sum == 0f || sum == 1f)
{
return;
}
for (int i = 0; i < editableBoneWeight.GetChannelCount(); ++i)
{
if (editableBoneWeight.IsChannelEnabled(i))
{
BoneWeightData data = editableBoneWeight.GetBoneWeightData(i);
data.weight = data.weight / sum;
editableBoneWeight.SetBoneWeightData(i, data);
}
}
}
public BoneWeightChannel(BoneWeightData data, bool enabled)
{
m_Enabled = enabled;
boneWeightData = data;
}
private void CreateVertex(Vector2 position, int edgeIndex)
{
position = MeshModuleUtility.ClampPositionToRect(position, spriteMeshData.frame);
undoObject.RegisterCompleteObjectUndo("Create Vertex");
BoneWeight boneWeight = new BoneWeight();
Vector3Int indices;
Vector3 barycentricCoords;
if (spriteMeshData.FindTriangle(position, out indices, out barycentricCoords))
{
EditableBoneWeight bw1 = spriteMeshData.vertices[indices.x].editableBoneWeight;
EditableBoneWeight bw2 = spriteMeshData.vertices[indices.y].editableBoneWeight;
EditableBoneWeight bw3 = spriteMeshData.vertices[indices.z].editableBoneWeight;
EditableBoneWeight result = new EditableBoneWeight();
foreach (BoneWeightChannel channel in bw1)
{
if (!channel.enabled)
{
continue;
}
BoneWeightData data = channel.boneWeightData;
data.weight *= barycentricCoords.x;
if (data.weight > 0f)
{
result.AddChannel(data, true);
}
}
foreach (BoneWeightChannel channel in bw2)
{
if (!channel.enabled)
{
continue;
}
BoneWeightData data = channel.boneWeightData;
data.weight *= barycentricCoords.y;
if (data.weight > 0f)
{
result.AddChannel(data, true);
}
}
foreach (BoneWeightChannel channel in bw3)
{
if (!channel.enabled)
{
continue;
}
BoneWeightData data = channel.boneWeightData;
data.weight *= barycentricCoords.z;
if (data.weight > 0f)
{
result.AddChannel(data, true);
}
}
result.UnifyChannelsWithSameBoneIndex();
result.FilterChannels(0f);
result.ClampChannels(4, true);
boneWeight = result.ToBoneWeight(true);
}
else if (edgeIndex != -1)
{
Edge edge = spriteMeshData.edges[edgeIndex];
Vector2 pos1 = spriteMeshData.vertices[edge.index1].position;
Vector2 pos2 = spriteMeshData.vertices[edge.index2].position;
Vector2 dir1 = (position - pos1);
Vector2 dir2 = (pos2 - pos1);
float t = Vector2.Dot(dir1, dir2.normalized) / dir2.magnitude;
t = Mathf.Clamp01(t);
BoneWeight bw1 = spriteMeshData.vertices[edge.index1].editableBoneWeight.ToBoneWeight(true);
BoneWeight bw2 = spriteMeshData.vertices[edge.index2].editableBoneWeight.ToBoneWeight(true);
boneWeight = EditableBoneWeightUtility.Lerp(bw1, bw2, t);
}
spriteMeshData.CreateVertex(position, edgeIndex);
spriteMeshData.vertices[spriteMeshData.vertices.Count - 1].editableBoneWeight.SetFromBoneWeight(boneWeight);
spriteMeshData.Triangulate(triangulator);
}
public void AddChannel(BoneWeightData boneWeightData, bool enabled)
{
m_Channels.Add(new BoneWeightChannel(boneWeightData, enabled));
}
