# -*- coding: utf-8 -*-
from pandas import Series
from pandas_ta._typing import DictLike, Int, IntFloat
from pandas_ta.utils import v_offset, v_series



def hwma(
    close: Series,
    na: IntFloat = None, nb: IntFloat = None, nc: IntFloat = None,
    offset: Int = None, **kwargs: DictLike
) -> Series:
    """Holt-Winter Moving Average

    This indicator uses a three parameter Holt-Winter Moving Average for
    smoothing.

    Sources:
        * [rengel8](https://github.com/rengel8) based on a publication for
          MetaTrader 5.
        * [mql5](https://www.mql5.com/en/code/20856)

    Parameters:
        close (Series): ```close``` Series
        na (float): Smoothed series parameter (from 0 to 1). Default: 0.2
        nb (float): Trend parameter (from 0 to 1). Default: 0.1
        nc (float): Seasonality parameter (from 0 to 1). Default: 0.1
        offset (int): Post shift. Default: ```0```

    Other Parameters:
        fillna (value): ```pd.DataFrame.fillna(value)```

    Returns:
        Series: 1 column
    """
    # Validate
    close = v_series(close, 1)
    na = float(na) if isinstance(na, float) and 0 < na < 1 else 0.2
    nb = float(nb) if isinstance(nb, float) and 0 < nb < 1 else 0.1
    nc = float(nc) if isinstance(nc, float) and 0 < nc < 1 else 0.1
    offset = v_offset(offset)

    if close is None:
        return

    # Calculate
    last_a = last_v = 0
    last_f = close.iloc[0]

    result = []
    m = close.size
    for i in range(m):
        F = (1.0 - na) * (last_f + last_v + 0.5 * last_a) + na * close.iloc[i]
        V = (1.0 - nb) * (last_v + last_a) + nb * (F - last_f)
        A = (1.0 - nc) * last_a + nc * (V - last_v)
        result.append((F + V + 0.5 * A))
        last_a, last_f, last_v = A, F, V  # update values

    hwma = Series(result, index=close.index)

    # Offset
    if offset != 0:
        hwma = hwma.shift(offset)

    # Fill
    if "fillna" in kwargs:
        hwma.fillna(kwargs["fillna"], inplace=True)

    # Name and Category
    hwma.name = f"HWMA_{na}_{nb}_{nc}"
    hwma.category = "overlap"

    return hwma