44 lines
1.1 KiB
C#
44 lines
1.1 KiB
C#
using System.Collections.Generic;
|
|
using System.Linq;
|
|
|
|
namespace Unity.VisualScripting
|
|
{
|
|
[UnitOrder(304)]
|
|
public abstract class Average<T> : MultiInputUnit<T>
|
|
{
|
|
/// <summary>
|
|
/// The average.
|
|
/// </summary>
|
|
[DoNotSerialize]
|
|
[PortLabelHidden]
|
|
public ValueOutput average { get; private set; }
|
|
|
|
protected override void Definition()
|
|
{
|
|
base.Definition();
|
|
|
|
average = ValueOutput(nameof(average), Operation).Predictable();
|
|
|
|
foreach (var multiInput in multiInputs)
|
|
{
|
|
Requirement(multiInput, average);
|
|
}
|
|
}
|
|
|
|
public abstract T Operation(T a, T b);
|
|
public abstract T Operation(IEnumerable<T> values);
|
|
|
|
public T Operation(Flow flow)
|
|
{
|
|
if (inputCount == 2)
|
|
{
|
|
return Operation(flow.GetValue<T>(multiInputs[0]), flow.GetValue<T>(multiInputs[1]));
|
|
}
|
|
else
|
|
{
|
|
return Operation(multiInputs.Select(flow.GetValue<T>));
|
|
}
|
|
}
|
|
}
|
|
}
|