//+------------------------------------------------------------------+
//|                                                EA31337 framework |
//|                       Copyright 2016-2020, 31337 Investments Ltd |
//|                                       https://github.com/EA31337 |
//+------------------------------------------------------------------+

/*
 * This file is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

// Prevents processing this includes file for the second time.
#ifndef TICKER_MQH
#define TICKER_MQH

// Forward declaration.
class Chart;

// Includes.
#include "Chart.mqh"
#include "Log.mqh"
#include "SymbolInfo.mqh"
//#include "Market.mqh"

// Define an assert macros.
#define PROCESS_METHOD(method, no) ((method & (1<<no)) == 1<<no)

/**
 * Class to provide methods handling ticks.
 */
class Ticker {

  // Structs.
  struct TTick {
    datetime dt;
    double bid, ask;
    double vol;
  };

  protected:
    ulong total_added, total_ignored, total_processed, total_saved;
    // Struct variables.
    MqlTick data[];
    // Class variables.
    SymbolInfo *symbol;
    Ref<Log> logger;

  public:

    // Public variables.
    int index;

    /**
     * Class constructor.
     */
    Ticker(SymbolInfo *_symbol = NULL, Log *_logger = NULL, int size = 1000) :
      symbol(Object::IsValid(_symbol) ? _symbol : new SymbolInfo),
      logger(Object::IsValid(_logger) ? _logger : new Log),
      total_added(0),
      total_ignored(0),
      total_processed(0),
      total_saved(0),
      index(-1) {
        ArrayResize(data, size, size);
    }

    /**
     * Class deconstructor.
     */
    ~Ticker() {
      Object::Delete(symbol);
    }

    Log* Logger() { return logger.Ptr(); }

    /* Getters */

    /**
     * Get number of added ticks.
     */
    ulong GetTotalAdded() {
      return total_added;
    }

    /**
     * Get number of ignored ticks.
     */
    ulong GetTotalIgnored() {
      return total_ignored;
    }

    /**
     * Get number of parsed ticks.
     */
    ulong GetTotalProcessed() {
      return total_processed;
    }


    /**
     * Get number of saved ticks.
     */
    ulong GetTotalSaved() {
      return total_saved;
    }

    /* Other methods */

    /**
     * Processes tick.
     *
     * @param
     * _method Ignore method (0-15).
     * _tf Timeframe to use.
     * @return
     * Returns true when tick should be parsed, otherwise ignored.
     */
    bool Process(Chart *_chart, uint _method) {
      total_processed++;
      if (_method == 0 || total_processed == 1) {
        return true;
      }
      double _last_bid = symbol.GetLastBid();
      double _bid = symbol.GetBid();
      bool _res = _last_bid != _bid;
      if (PROCESS_METHOD(_method, 0)) _res &= (_chart.GetOpen() == _bid); // 1
      if (PROCESS_METHOD(_method, 1)) _res &= (_chart.iTime() == TimeCurrent()); // 2
      if (PROCESS_METHOD(_method, 2)) _res &= (_bid >= _chart.GetHigh()) || (_bid <= _chart.GetLow()); // 4
      if (!_res) {
        total_ignored++;
      }
      return _res;
    }

    /**
     * Append a new tick to an array.
     */
    bool Add(const MqlTick &_tick) {
      if (index++ >= ArraySize(data) - 1) {
        if (ArrayResize(data, index + 100, 1000) < 0) {
          Logger().Error(StringFormat("Cannot resize array (index: %d)!", index), __FUNCTION__);
          return false;
        }
      }
      data[index] = _tick;
      total_added++;
      return true;
    }
    bool Add() {
      MqlTick _tick = this.symbol.GetTick();
      return Add(_tick);
    }

    /**
     * Empties the tick array.
     */
    void Reset() {
      total_added = 0;
      index = 0;
    }

    /**
     * Save ticks into CSV file.
     */
    bool SaveToCSV(string filename = NULL, bool verbose = true) {
      ResetLastError();
      datetime _dt = index > 0 ? data[index].time : TimeCurrent();
      filename = filename != NULL
        ? filename
        : StringFormat("%s_%s_ticks.csv",
          symbol.GetSymbol(), DateTime::TimeToStr(_dt, TIME_DATE));
      int _handle = FileOpen(filename, FILE_WRITE|FILE_CSV, ",");
      if (_handle != INVALID_HANDLE) {
        total_saved = 0;
        FileWrite(_handle, "Datatime", "Bid", "Ask", "Volume");
        for (int i = 0; i < index; i++) {
          if (data[i].time > 0) {
            FileWrite(_handle,
                DateTime::TimeToStr(data[i].time, TIME_DATE|TIME_MINUTES|TIME_SECONDS),
                data[i].bid,
                data[i].ask,
                data[i].volume);
            total_saved++;
          }
        }
        FileClose(_handle);
        if (verbose) {
          Logger().Info(StringFormat("%s: %d ticks written to '%s' file.", __FUNCTION__, total_saved, filename));
        }
        return true;
      }
      else {
        if (verbose) {
          Logger().Error(StringFormat("%s: Cannot open file for writting, error: %s", __FUNCTION__, GetLastError()));
        }
        return false;
      }
    }

  /**
   * Returns textual representation of the Market class.
   */
  string ToString() {
    return StringFormat(
      "Processed: %d; Ignored: %d; Added: %d; Saved: %d;",
      GetTotalProcessed(),
      GetTotalIgnored(),
      GetTotalAdded(),
      GetTotalSaved()
    );
  }

};

#endif // TICKER_MQH