Files
zeling_v2/Packages/com.kybernetik.animancer/Runtime/Interfaces/IAnimationClipCollection.cs
2026-05-08 11:04:00 +08:00

266 lines
11 KiB
C#

// Animancer // https://kybernetik.com.au/animancer // Copyright 2018-2026 Kybernetik //
using System.Collections;
using System.Collections.Generic;
using UnityEngine;
using UnityEngine.Playables;
namespace Animancer
{
/// <summary>
/// A variant of <see cref="IAnimationClipSource"/> which uses a <see cref="ICollection{T}"/>
/// instead of a <see cref="List{T}"/> so that it can take a <see cref="HashSet{T}"/>
/// to efficiently avoid adding duplicates. <see cref="AnimancerUtilities"/> contains
/// various extension methods for this purpose.
/// </summary>
/// <remarks>
/// <see cref="IAnimationClipSource"/> still needs to be the main point of entry
/// for the Animation Window, so this interface is only used internally.
/// </remarks>
/// https://kybernetik.com.au/animancer/api/Animancer/IAnimationClipCollection
///
public interface IAnimationClipCollection
{
/************************************************************************************************************************/
/// <summary>Adds all the animations associated with this object to the `clips`.</summary>
void GatherAnimationClips(ICollection<AnimationClip> clips);
/************************************************************************************************************************/
}
/************************************************************************************************************************/
/// https://kybernetik.com.au/animancer/api/Animancer/AnimancerUtilities
public static partial class AnimancerUtilities
{
/************************************************************************************************************************/
/// <summary>[Animancer Extension]
/// Adds the `clip` to the `clips` if it wasn't there already.
/// </summary>
public static void Gather(this ICollection<AnimationClip> clips, AnimationClip clip)
{
if (clip != null && !clips.Contains(clip))
clips.Add(clip);
}
/************************************************************************************************************************/
/// <summary>[Animancer Extension]
/// Calls <see cref="Gather(ICollection{AnimationClip}, AnimationClip)"/> for each of the `newClips`.
/// </summary>
public static void Gather(this ICollection<AnimationClip> clips, IList<AnimationClip> gatherFrom)
{
if (gatherFrom == null)
return;
for (int i = gatherFrom.Count - 1; i >= 0; i--)
clips.Gather(gatherFrom[i]);
}
/************************************************************************************************************************/
/// <summary>[Animancer Extension]
/// Calls <see cref="Gather(ICollection{AnimationClip}, AnimationClip)"/> for each of the `newClips`.
/// </summary>
public static void Gather(this ICollection<AnimationClip> clips, IEnumerable<AnimationClip> gatherFrom)
{
if (gatherFrom == null)
return;
foreach (var clip in gatherFrom)
clips.Gather(clip);
}
/************************************************************************************************************************/
/// <summary>[Animancer Extension]
/// Calls <see cref="Gather(ICollection{AnimationClip}, AnimationClip)"/> for each clip in the `asset`.
/// </summary>
public static void GatherFromAsset(this ICollection<AnimationClip> clips, PlayableAsset asset)
{
if (asset == null)
return;
// We want to get the tracks out of a TimelineAsset without actually referencing that class directly
// because it comes from an optional package and Animancer does not need to depend on that package.
var method = asset.GetType().GetMethod("GetRootTracks");
if (method != null &&
typeof(IEnumerable).IsAssignableFrom(method.ReturnType) &&
method.GetParameters().Length == 0)
{
var rootTracks = method.Invoke(asset, null);
GatherFromTracks(clips, rootTracks as IEnumerable);
}
}
/************************************************************************************************************************/
/// <summary>Gathers all the animations in the `tracks`.</summary>
private static void GatherFromTracks(ICollection<AnimationClip> clips, IEnumerable tracks)
{
if (tracks.IsNullOrDestroyed())
return;
foreach (var track in tracks)
{
if (track.IsNullOrDestroyed())
continue;
var trackType = track.GetType();
var getClips = trackType.GetMethod("GetClips");
if (getClips != null &&
typeof(IEnumerable).IsAssignableFrom(getClips.ReturnType) &&
getClips.GetParameters().Length == 0)
{
if (getClips.Invoke(track, null) is IEnumerable trackClips)
{
foreach (var clip in trackClips)
{
var animationClip = clip.GetType().GetProperty("animationClip");
if (animationClip != null &&
animationClip.PropertyType == typeof(AnimationClip))
{
var getClip = animationClip.GetGetMethod();
clips.Gather(getClip.Invoke(clip, null) as AnimationClip);
}
}
}
}
var getChildTracks = trackType.GetMethod("GetChildTracks");
if (getChildTracks != null &&
typeof(IEnumerable).IsAssignableFrom(getChildTracks.ReturnType) &&
getChildTracks.GetParameters().Length == 0)
{
var childTracks = getChildTracks.Invoke(track, null);
GatherFromTracks(clips, childTracks as IEnumerable);
}
}
}
/************************************************************************************************************************/
/// <summary>[Animancer Extension]
/// Calls <see cref="Gather(ICollection{AnimationClip}, AnimationClip)"/>
/// for each clip gathered by <see cref="IAnimationClipSource.GetAnimationClips"/>.
/// </summary>
public static void GatherFromSource(this ICollection<AnimationClip> clips, IAnimationClipSource source)
{
if (source.IsNullOrDestroyed())
return;
var list = ListPool.Acquire<AnimationClip>();
source.GetAnimationClips(list);
clips.Gather(list);
ListPool.Release(list);
}
/************************************************************************************************************************/
/// <summary>[Animancer Extension]
/// Calls <see cref="GatherFromSource(ICollection{AnimationClip}, object)"/> for each item in the `source`.
/// </summary>
public static void GatherFromSource(this ICollection<AnimationClip> clips, IEnumerable source)
{
if (source.IsNullOrDestroyed() ||
AnimationGathererRecursionGuard.DontGatherFrom.Contains(source.GetType()))
return;
using var _ = AnimationGathererRecursionGuard.Begin();
if (AnimationGathererRecursionGuard.HasCheckedObject(source))
return;
foreach (var item in source)
clips.GatherFromSource(item);
}
/************************************************************************************************************************/
/// <summary>[Animancer Extension]
/// Calls <see cref="Gather(ICollection{AnimationClip}, AnimationClip)"/>
/// for each clip in the `source`, supporting both
/// <see cref="IAnimationClipSource"/> and <see cref="IAnimationClipCollection"/>.
/// </summary>
public static bool GatherFromSource(this ICollection<AnimationClip> clips, object source)
{
if (source.IsNullOrDestroyed() ||
AnimationGathererRecursionGuard.DontGatherFrom.Contains(source.GetType()))
return false;
using var _ = AnimationGathererRecursionGuard.Begin();
if (AnimationGathererRecursionGuard.HasCheckedObject(source))
return false;
if (TryGetWrappedObject(source, out AnimationClip clip))
{
clips.Gather(clip);
return true;
}
if (TryGetWrappedObject(source, out IAnimationClipCollection collectionSource))
{
collectionSource.GatherAnimationClips(clips);
return true;
}
if (TryGetWrappedObject(source, out IAnimationClipSource listSource))
{
clips.GatherFromSource(listSource);
return true;
}
if (TryGetWrappedObject(source, out IEnumerable enumerable))
{
clips.GatherFromSource(enumerable);
return true;
}
return false;
}
/************************************************************************************************************************/
/// <summary>
/// Attempts to get the <see cref="AnimationClip.frameRate"/> from the `clipSource`
/// and returns true if successful. If it has multiple animations with different rates,
/// this method returns false.
/// </summary>
public static bool TryGetFrameRate(object clipSource, out float frameRate)
{
using (SetPool<AnimationClip>.Instance.Acquire(out var clips))
{
clips.GatherFromSource(clipSource);
if (clips.Count == 0)
{
frameRate = float.NaN;
return false;
}
frameRate = float.NaN;
foreach (var clip in clips)
{
if (float.IsNaN(frameRate))
{
frameRate = clip.frameRate;
}
else if (frameRate != clip.frameRate)
{
frameRate = float.NaN;
return false;
}
}
return frameRate > 0;
}
}
/************************************************************************************************************************/
}
}