/*
 * _KData.cpp
 *
 *  Created on: 2012-9-28
 *      Author: fasiondog
 */

#include <hikyuu/serialization/KData_serialization.h>
#include <hikyuu/indicator/crt/KDATA.h>
#include "pybind_utils.h"

using namespace hku;
namespace py = pybind11;

const KRecord& (KData::*KData_getKRecord1)(size_t pos) const = &KData::getKRecord;
const KRecord& (KData::*KData_getKRecord2)(Datetime datetime) const = &KData::getKRecord;

void export_KData(py::module& m) {
    py::class_<KData>(
      m, "KData",
      "通过 Stock.getKData 获取的K线数据，由 KRecord 组成的数组，可象 list 一样进行遍历")
      .def(py::init<>())
      .def("__str__", &KData::toString)
      .def("__repr__", &KData::toString)

      .def_property_readonly("start_pos", &KData::startPos,
                             "获取在原始K线记录中对应的起始位置，如果KData为空返回0")
      .def_property_readonly(
        "end_pos", &KData::endPos,
        "获取在原始K线记录中对应范围的下一条记录的位置，如果为空返回0,其他等于lastPos + 1")
      .def_property_readonly(
        "last_pos", &KData::lastPos,
        "获取在原始K线记录中对应的最后一条记录的位置，如果为空返回0,其他等于endPos - 1")

      .def_property_readonly("open", &KData::open,
                             "返回包含开盘价的 Indicator 实例，相当于 OPEN(k)")
      .def_property_readonly("close", &KData::close,
                             "返回包含收盘价的 Indicator 实例，相当于 CLOSE(k)")
      .def_property_readonly("high", &KData::high,
                             "返回包含最高价的 Indicator 实例，相当于 HIGH(k)")
      .def_property_readonly("low", &KData::low, "返回包含最低价的 Indicator 实例，相当于 LOW(k)")
      .def_property_readonly("amo", &KData::amo, "返回包含成交金额的 Indicator 实例，相当于 AMO(k)")
      .def_property_readonly("vol", &KData::vol, "返回包含成交量的 Indicator 实例，相当于 VOL(k)")

      .def("get_datetime_list", &KData::getDatetimeList, R"(get_datetime_list(self)

        返回交易日期列表

        :rtype: DatetimeList)")

      .def("get", KData_getKRecord1, py::return_value_policy::copy, R"(get(self, pos)

        获取指定索引位置的K线记录

        :param int pos: 位置索引
        :rtype: KRecord)")

      .def("get_by_datetime", KData_getKRecord2, py::return_value_policy::copy,
           R"(get_by_datetime(self, datetime)

        获取指定时间的K线记录。

        :param Datetime datetime: 指定的日期
        :rtype: KRecord)")

      .def("_getPos", &KData::getPos)  // python中需要将Null的情况改写为None

      .def("_getPosInStock", &KData::getPosInStock)

      .def("empty", &KData::empty, R"(empty(self)

        判断是否为空

        :rtype: bool)")

      .def("get_query", &KData::getQuery, py::return_value_policy::copy, R"(get_query(self)

        获取关联的查询条件

        :rtype: KQuery)")

      .def("get_stock", &KData::getStock, py::return_value_policy::copy, R"(get_stock(self)

        获取关联的Stock

        :rtype: Stock)")

      .def("get_kdata", &KData::getKData, R"(get_kdata(self, start_date, end_date)
      
        通过当前 KData 获取一个保持数据类型、复权类型不变的新的 KData（注意，不是原 KData 的子集）

        :param Datetime start: 新的起始日期
        :param Datetime end: 新的结束日期
        :rtype: KData)")

      .def("tocsv", &KData::tocsv, R"(tocsv(self, filename)

        将数据保存至CSV文件

        :param str filename: 指定保存的文件名称)")

      .def("__len__", &KData::size)

      .def(py::self == py::self)
      .def(py::self != py::self)

        DEF_PICKLE(KData);
}