File size: 1,111 Bytes
18a519f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
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>));
            }
        }
    }
}