# cython: language_level=3
#  Drakkar-Software OctoBot-Trading
#  Copyright (c) Drakkar-Software, All rights reserved.
#
#  This library is free software; you can redistribute it and/or
#  modify it under the terms of the GNU Lesser General Public
#  License as published by the Free Software Foundation; either
#  version 3.0 of the License, or (at your option) any later version.
#
#  This library is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
#  Lesser General License for more details.
#
#  You should have received a copy of the GNU Lesser General Public
#  License along with this library.
from octobot_trading.util.initializable cimport Initializable

cimport numpy as np
np.import_array()

cdef class CandlesManager(Initializable):
    cdef object logger

    cdef public bint candles_initialized

    cdef public np.float64_t[::1] close_candles
    cdef public np.float64_t[::1] open_candles
    cdef public np.float64_t[::1] high_candles
    cdef public np.float64_t[::1] low_candles
    cdef public np.float64_t[::1] time_candles
    cdef public np.float64_t[::1] volume_candles

    cdef public int close_candles_index
    cdef public int open_candles_index
    cdef public int high_candles_index
    cdef public int low_candles_index
    cdef public int time_candles_index
    cdef public int volume_candles_index

    cdef public bint reached_max

    cpdef np.ndarray get_symbol_close_candles(self, int limit=*)
    cpdef np.ndarray get_symbol_open_candles(self, int limit=*)
    cpdef np.ndarray get_symbol_high_candles(self, int limit=*)
    cpdef np.ndarray get_symbol_low_candles(self, int limit=*)
    cpdef np.ndarray get_symbol_time_candles(self, int limit=*)
    cpdef np.ndarray get_symbol_volume_candles(self, int limit=*)

    cpdef dict get_symbol_prices(self, object limit=*)
    cpdef void add_old_and_new_candles(self, list candles_data)
    cpdef void add_new_candle(self, list new_candle_data)
    cpdef void replace_all_candles(self, list all_candles_data)

    # private
    cdef void _set_all_candles(self, object new_candles_data)
    cdef void _change_current_candle(self)
    cdef bint _should_add_new_candle(self, new_open_time)
    cdef void _check_max_candles(self)
    cdef object _inc_candle_index(self)
    cdef void _reset_candles(self)
    cdef np.ndarray _extract_limited_data(self, np.float64_t[::1] data, int limit=*, int max_limit=*)
