"""Warianty progu dla EmaEnsemble - tylko do badania, nie do produkcji.

Chodzi o histereze: wejscie i wyjscie na tym samym progu (0.5) daje pile,
bo cena bez konca oscyluje wokol srodkowych EMA. Ponizsze warianty roznia sie
STRUKTURA reguly, nie wartoscia parametru: "powyzej wszystkich EMA" albo
"ponizej wszystkich" to punkty wyznaczone przez sam zbior, nie przez dostrajanie.
"""
from __future__ import annotations

import numpy as np
import talib.abstract as ta
from pandas import DataFrame

from freqtrade.strategy import IStrategy

EMA_SET = (400, 600, 800, 1000, 1200)


class _Base(IStrategy):
    INTERFACE_VERSION = 3
    timeframe = "1h"
    can_short = False
    process_only_new_candles = True
    startup_candle_count = 1300
    minimal_roi = {"0": 10}
    stoploss = -0.15
    trailing_stop = True

    enter_at = 0.5   # wejscie gdy udzial przekroczy ten poziom
    exit_at = 0.5    # wyjscie gdy spadnie ponizej

    def populate_indicators(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
        above = np.zeros(len(dataframe))
        for p in EMA_SET:
            above += (dataframe["close"] > ta.EMA(dataframe, timeperiod=p)).astype(float).to_numpy()
        dataframe["ema_score"] = above / len(EMA_SET)
        return dataframe

    def populate_entry_trend(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
        s = dataframe["ema_score"]
        dataframe.loc[(s >= self.enter_at) & (s.shift(1) < self.enter_at)
                      & (dataframe["volume"] > 0), "enter_long"] = 1
        return dataframe

    def populate_exit_trend(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
        s = dataframe["ema_score"]
        dataframe.loc[(s <= self.exit_at) & (s.shift(1) > self.exit_at), "exit_long"] = 1
        return dataframe


class EnsAllAll(_Base):
    """Wejscie powyzej WSZYSTKICH EMA, wyjscie ponizej wszystkich."""
    enter_at, exit_at = 1.0, 0.0


class EnsAllHalf(_Base):
    """Wejscie powyzej wszystkich, wyjscie gdy wiekszosc juz nie trzyma."""
    enter_at, exit_at = 1.0, 0.4


class EnsHalfAll(_Base):
    """Wejscie gdy wiekszosc przebita, wyjscie dopiero ponizej wszystkich."""
    enter_at, exit_at = 0.6, 0.0


class EnsHalfZero(_Base):
    """Wejscie na wiekszosci, wyjscie gdy zostanie tylko najdluzsza EMA."""
    enter_at, exit_at = 0.6, 0.2