/*
 * _OrderBroker.cpp
 *
 *  Created on: 2017年6月28日
 *      Author: fasiondog
 */

#include <boost/python.hpp>
#include <hikyuu/trade_manage/OrderBrokerBase.h>
#include "../_Parameter.h"
#include "../pickle_support.h"

using namespace boost::python;
using namespace hku;

class OrderBrokerWrap : public OrderBrokerBase, public wrapper<OrderBrokerBase> {
public:
    OrderBrokerWrap() : OrderBrokerBase() {}
    OrderBrokerWrap(const string& name) : OrderBrokerBase(name) {}

    Datetime _buy(Datetime datetime, const string& market, const string& code, price_t price,
                  double num) {
        return this->get_override("_buy")(datetime, market, code, price, num);
    }

    Datetime _sell(Datetime datetime, const string& market, const string& code, price_t price,
                   double num) {
        return this->get_override("_sell")(datetime, market, code, price, num);
    }
};

string (OrderBrokerBase::*ob_get_name)() const = &OrderBrokerBase::name;
void (OrderBrokerBase::*ob_set_name)(const string&) = &OrderBrokerBase::name;

void export_OrderBroker() {
    class_<OrderBrokerWrap, boost::noncopyable>("OrderBrokerBase", init<>())
      .def(init<const string&>())
      .def(self_ns::str(self))
      .add_property("name", ob_get_name, ob_set_name)
      .def("buy", &OrderBrokerBase::buy)
      .def("sell", &OrderBrokerBase::sell)
      .def("_buy", pure_virtual(&OrderBrokerBase::_buy))
      .def("_sell", pure_virtual(&OrderBrokerBase::_sell));

    register_ptr_to_python<OrderBrokerPtr>();
}