/*
 * _Stock.cpp
 *
 *  Created on: 2011-12-4
 *      Author: fasiondog
 */

#include <boost/python.hpp>
#include <hikyuu/serialization/Stock_serialization.h>
#include <hikyuu/KData.h>
#include "_Parameter.h"
#include "pickle_support.h"

using namespace boost::python;
using namespace hku;

BOOST_PYTHON_MEMBER_FUNCTION_OVERLOADS(getIndex_overloads, getIndex, 1, 2)

StockWeightList (Stock::*getWeight1)() const = &Stock::getWeight;
StockWeightList (Stock::*getWeight2)(const Datetime&, const Datetime&) const = &Stock::getWeight;

DatetimeList (Stock::*getDatetimeList1)(size_t, size_t,
                                        KQuery::KType) const = &Stock::getDatetimeList;
DatetimeList (Stock::*getDatetimeList2)(const KQuery&) const = &Stock::getDatetimeList;

void export_Stock() {
    class_<Stock>("Stock", "证券对象", init<>())
      .def(init<const string&, const string&, const string&>())
      .def(init<const Stock&>())

      .def("__str__", &Stock::toString)
      .def("__repr__", &Stock::toString)

      .add_property("id", &Stock::id, "内部id")
      .add_property("market",
                    make_function(&Stock::market, return_value_policy<copy_const_reference>()),
                    "所属市场简称，市场简称是市场的唯一标识")
      .add_property("code",
                    make_function(&Stock::code, return_value_policy<copy_const_reference>()),
                    "证券代码")
      .add_property("market_code",
                    make_function(&Stock::market_code, return_value_policy<copy_const_reference>()),
                    "市场简称+证券代码，如: sh000001")
      .add_property("name",
                    make_function(&Stock::name, return_value_policy<copy_const_reference>()),
                    "证券名称")
      .add_property("type", &Stock::type, "证券类型，参见：constant")
      .add_property("valid", &Stock::valid, "该证券当前是否有效")
      .add_property("start_datetime", &Stock::startDatetime, "证券起始日期")
      .add_property("last_datetime", &Stock::lastDatetime, "证券最后日期")
      .add_property("tick", &Stock::tick, "最小跳动量")
      .add_property("tick_value", &Stock::tickValue, "最小跳动量价值")
      .add_property("unit", &Stock::unit, "每单位价值 = tickValue / tick")
      .add_property("precision", &Stock::precision, "价格精度")
      .add_property("atom", &Stock::atom, "最小交易数量，同min_tradeNumber")
      .add_property("min_trade_number", &Stock::minTradeNumber, "最小交易数量")
      .add_property("max_trade_number", &Stock::maxTradeNumber, "最大交易数量")

      .def("is_null", &Stock::isNull, R"(is_null(self)

    是否为Null

    :rtype: bool)")

      .def("is_buffer", &Stock::isBuffer, R"(指定类型的K线数据是否被缓存)")

      .def("get_kdata", &Stock::getKData, R"(get_kdata(self, query)

    获取K线数据

    :param Query query: 查询条件
    :return: 满足查询条件的K线数据
    :rtype: KData)")

      .def("get_timeLine_list", &Stock::getTimeLineList, R"(get_timeLine_list(self, query)

    获取分时线

    :param Query query: 查询条件（查询条件中的K线类型、复权类型参数此时无用）
    :rtype: TimeLineList)")

      .def("get_trans_list", &Stock::getTransList, R"(get_trans_list(self, query)

    获取历史分笔数据

    :param Query query: 查询条件（查询条件中的K线类型、复权类型参数此时无用）
    :rtype: TransList)")

      .def("get_count", &Stock::getCount, arg("ktype") = KQuery::DAY,
           R"(get_count(self, [ktype=Query.DAY])

    获取不同类型K线数据量

    :param Query.KType ktype: K线数据类别
    :return: K线记录数
    :rtype: int)")

      .def("get_market_value", &Stock::getMarketValue, R"(get_market_value(self, date, ktype)

    获取指定时刻的市值，即小于等于指定时刻的最后一条记录的收盘价

    :param Datetime date: 指定时刻
    :param Query.KType ktype: K线数据类别
    :return: 指定时刻的市值
    :rtype: float)")

      .def("get_krecord", &Stock::getKRecord, (arg("pos"), arg("ktype") = KQuery::DAY),
           R"(get_krecord(self, pos[, ktype=Query.DAY])

    获取指定索引的K线数据记录，未作越界检查

    :param int pos: 指定的索引位置
    :param Query.KType ktype: K线数据类别
    :return: K线记录
    :rtype: KRecord)")

      .def("get_krecord_by_datetime", &Stock::getKRecordByDate,
           (arg("date"), arg("ktype") = KQuery::DAY),
           R"(get_krecord_by_datetime(self, datetime[, ktype=Query.DAY])

    根据数据类型（日线/周线等），获取指定时刻的KRecord

    :param Datetime datetime: 指定日期时刻
    :param Query.KType ktype: K线数据类别
    :return: K线记录
    :rtype: KRecord)")

      .def("get_krecord_list", &Stock::getKRecordList, R"(get_krecord_list(self, start, end, ktype)

    获取K线记录 [start, end)，一般不直接使用.

    :param int start: 起始位置
    :param int end: 结束位置
    :param Query.KType ktype: K线类别
    :return: K线记录列表
    :rtype: KRecordList)")

      .def("get_datetime_list", getDatetimeList1)
      .def("get_datetime_list", getDatetimeList2, R"(get_datetime_list(self, query)

    获取日期列表

    :param Query query: 查询条件
    :rtype: DatetimeList

get_date_list(self, start, end, ktype)

    获取日期列表

    :param int start: 起始位置
    :param ind end: 结束位置
    :param Query.KType ktype: K线类型
    :rtype: DatetimeList )")

      .def("get_finance_info", &Stock::getFinanceInfo, R"(get_finance_info(self)

    获取当前财务信息
  
    :rtype: Parameter)")

      .def("get_history_finance_info", &Stock::getHistoryFinanceInfo,
           R"(get_history_finance_info(self, date)

    获取历史财务信息, 字段含义参见：https://hikyuu.org/finance_fields.html

    :param Datetime date: 指定日期必须是0331、0630、0930、1231，如 Datetime(201109300000)
    :rtype: PriceList)")

      .def("realtime_update", &Stock::realtimeUpdate, R"(realtime_update(self, krecord)

    （临时函数）只用于更新缓存中的日线数据

    :param KRecord krecord: 新增的实时K线记录)")

      .def("get_weight", getWeight1)
      .def("get_weight", getWeight2, R"(get_weight(self, [start, end])

    获取指定时间段[start,end)内的权息信息。未指定起始、结束时刻时，获取全部权息记录。

    :param Datetime start: 起始时刻
    :param Datetime end: 结束时刻
    :rtype: StockWeightList)")

      .def("load_kdata_to_buffer", &Stock::loadKDataToBuffer, R"(load_kdata_to_buffer(self, ktype)

    将指定类别的K线数据加载至内存缓存

    :param Query.KType ktype: K线类型)")

      .def("release_kdata_buffer", &Stock::releaseKDataBuffer, R"(release_kdata_buffer(self, ktype)

    释放指定类别的内存K线数据

    :param Query.KType ktype: K线类型)")

      .def(self == self)
      .def(self != self)

#if HKU_PYTHON_SUPPORT_PICKLE
      .def_pickle(normal_pickle_suite<Stock>())
#endif
      ;
}