using System.Runtime.CompilerServices;
namespace QuanTAlib;

/// <summary>
/// EMA: Exponential Moving Average
/// </summary>
/// <remarks>
/// EMA needs very short history buffer and calculates the EMA value using just the
/// previous EMA value. The weight of the new datapoint (alpha) is alpha = 2 / (period + 1)
///
/// Key characteristics:
/// - Uses no buffer, relying only on the previous EMA value.
/// - The weight of new data points is calculated as alpha = 2 / (period + 1).
/// - Provides a balance between responsiveness and smoothing. No overshooting. Significant lag
///
/// Calculation method:
/// This implementation can use SMA for the first Period bars as a seeding value for EMA when useSma is true.
///
/// Sources:
/// - https://stockcharts.com/school/doku.php?id=chart_school:technical_indicators:moving_averages
/// - https://www.investopedia.com/ask/answers/122314/what-exponential-moving-average-ema-formula-and-how-ema-calculated.asp
/// - https://blog.fugue88.ws/archives/2017-01/The-correct-way-to-start-an-Exponential-Moving-Average-EMA
/// </remarks>
public class Ema : AbstractBase
{
    private readonly int _period;
    private readonly double _k;
    private readonly bool _useSma;
    private readonly double _epsilon = 1e-10;
    private CircularBuffer _sma;
    private double _lastEma, _p_lastEma;
    private double _e, _p_e;
    private bool _isInit, _p_isInit;

    /// <summary>
    /// Initializes a new instance of the Ema class with a specified period.
    /// </summary>
    /// <param name="period">The period for EMA calculation.</param>
    /// <param name="useSma">Whether to use SMA for initial values. Default is true.</param>
    /// <exception cref="ArgumentOutOfRangeException">Thrown when period is less than 1.</exception>
    public Ema(int period, bool useSma = true)
    {
        if (period < 1)
        {
            throw new System.ArgumentOutOfRangeException(nameof(period), "Period must be greater than or equal to 1.");
        }
        _period = period;
        _k = 2.0 / (_period + 1);
        _useSma = useSma;
        _sma = new(_period);
        Name = "Ema";
        WarmupPeriod = (int)System.Math.Ceiling(System.Math.Log(0.05) / System.Math.Log(1 - _k)); //95th percentile
        Init();
    }

    /// <summary>
    /// Initializes a new instance of the Ema class with a specified alpha value.
    /// </summary>
    /// <param name="alpha">The smoothing factor for EMA calculation.</param>
    public Ema(double alpha)
    {
        _k = alpha;
        _useSma = false;
        _sma = new(1);
        Name = "Ema";
        _period = 1;
        WarmupPeriod = (int)System.Math.Ceiling(System.Math.Log(0.05) / System.Math.Log(1 - _k)); //95th percentile
        Init();
    }

    /// <summary>
    /// Initializes a new instance of the Ema class with a specified source and period.
    /// </summary>
    /// <param name="source">The source object for event subscription.</param>
    /// <param name="period">The period for EMA calculation.</param>
    /// <param name="useSma">Whether to use SMA for initial values. Default is true.</param>
    public Ema(object source, int period, bool useSma = true) : this(period, useSma)
    {
        var pubEvent = source.GetType().GetEvent("Pub");
        pubEvent?.AddEventHandler(source, new ValueSignal(Sub));
    }

    [MethodImpl(MethodImplOptions.AggressiveInlining)]
    public override void Init()
    {
        base.Init();
        _e = 1.0;
        _lastEma = 0;
        _isInit = false;
        _p_isInit = false;
        _sma = new(_period);
    }

    [MethodImpl(MethodImplOptions.AggressiveInlining)]
    protected override void ManageState(bool isNew)
    {
        if (isNew)
        {
            _p_lastEma = _lastEma;
            _p_isInit = _isInit;
            _p_e = _e;
            _index++;
        }
        else
        {
            _lastEma = _p_lastEma;
            _isInit = _p_isInit;
            _e = _p_e;
        }
    }

    [MethodImpl(MethodImplOptions.AggressiveInlining)]
    private double CalculateEma(double input, double lastEma)
    {
        return (_k * (input - lastEma)) + lastEma;
    }

    [MethodImpl(MethodImplOptions.AggressiveInlining)]
    private double CompensateEma(double ema)
    {
        return (_useSma || _e <= _epsilon) ? ema : ema / (1 - _e);
    }

    protected override double Calculation()
    {
        ManageState(Input.IsNew);

        double ema;
        if (!_isInit && _useSma)
        {
            _sma.Add(Input.Value, Input.IsNew);
            ema = _sma.Average();
            if (_index >= _period)
            {
                _isInit = true;
            }
        }
        else
        {
            // Compensator for early EMA values
            _e = (_e > _epsilon) ? (1 - _k) * _e : 0;
            ema = CalculateEma(Input.Value, _lastEma);
            ema = CompensateEma(ema);
        }

        _lastEma = ema;
        IsHot = _index >= WarmupPeriod;
        return ema;
    }
}