mirror of
https://github.com/mihakralj/QuanTAlib.git
synced 2026-08-16 01:28:05 +00:00
54 lines
1.2 KiB
C#
54 lines
1.2 KiB
C#
namespace QuanTAlib;
|
|
|
|
public class SMA
|
|
{
|
|
private CircularBuffer buffer = null!;
|
|
private int period;
|
|
private double sum;
|
|
public TValue Value { get; private set; }
|
|
public bool IsHot { get; private set; }
|
|
|
|
public SMA(int period)
|
|
{
|
|
Init(period);
|
|
}
|
|
|
|
public void Init(int period)
|
|
{
|
|
this.period = period;
|
|
this.buffer = new CircularBuffer(period);
|
|
this.sum = 0;
|
|
this.IsHot = false;
|
|
this.Value = default;
|
|
}
|
|
|
|
public TValue Update(TValue input, bool IsNew = true)
|
|
{
|
|
if (buffer.Count == 0 || isNew)
|
|
{
|
|
if (buffer.Count == period)
|
|
{
|
|
sum -= buffer[0];
|
|
}
|
|
buffer.Add(input);
|
|
sum += input.Value;
|
|
}
|
|
else
|
|
{
|
|
sum -= buffer[buffer.Count - 1];
|
|
sum += input.Value;
|
|
buffer[buffer.Count - 1] = input;
|
|
}
|
|
|
|
double sma = sum / buffer.Count;
|
|
Value = new TValue(input.Time, sma, isNew, IsHot);
|
|
return Value;
|
|
}
|
|
|
|
double sma = buffer.Count > 0 ? sum / buffer.Count : double.NaN;
|
|
IsHot = buffer.Count >= period;
|
|
Value = new TValue(input.Time, sma, IsNew, IsHot);
|
|
return Value;
|
|
}
|
|
}
|