hikyuu2/hikyuu_pywrap/trade_manage/_TradeCost.cpp

/*
 * _TradeCose.cpp
 *
 *  Created on: 2013-2-14
 *      Author: fasiondog
 */

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

using namespace boost::python;
using namespace hku;

class TradeCostWrap: public TradeCostBase, public wrapper<TradeCostBase>{
public:
    TradeCostWrap(const string& name) : TradeCostBase(name) { }

    CostRecord getBuyCost(const Datetime& datetime, const Stock& stock,
            price_t price, size_t num) const {
        return this->get_override("getBuyCost")(datetime, stock, price, num);
    }

    CostRecord getSellCost(const Datetime& datetime, const Stock& stock,
            price_t price, size_t num) const {
        return this->get_override("getSellCost")(datetime, stock, price, num);
    }

    TradeCostPtr _clone() {
        return this->get_override("_clone")();
    }

    CostRecord getBorrowCashCost(const Datetime& datetime,
            price_t cash) const {
        if (override getBorrowCashCost = get_override("getBorrowCashCost")) {
            return getBorrowCashCost(datetime, cash);
        }
        return TradeCostBase::getBorrowCashCost(datetime, cash);
    }

    CostRecord default_getBorrowCashCost(const Datetime& datetime,
            price_t cash) const {
        return this->TradeCostBase::getBorrowCashCost(datetime, cash);
    }

    CostRecord getReturnCashCost(const Datetime& borrow_datetime,
            const Datetime& return_datetime, price_t cash) const {
        if (override getReturnCashCost = get_override("getReturnCashCost")) {
            return getReturnCashCost(borrow_datetime, return_datetime, cash);
        }
        return TradeCostBase::getReturnCashCost(borrow_datetime, return_datetime, cash);
    }

    CostRecord default_getReturnCashCost(const Datetime& borrow_datetime,
            const Datetime& return_datetime, price_t cash) const {
        return this->TradeCostBase::getReturnCashCost(borrow_datetime, return_datetime, cash);
    }

    CostRecord getBorrowStockCost(const Datetime& datetime,
            const Stock& stock, price_t price, size_t num) const {
        if (override getBorrowStockCost = get_override("getBorrowStockCost")) {
            return getBorrowStockCost(datetime, stock, price, num);
        }
        return TradeCostBase::getBorrowStockCost(datetime, stock, price, num);
    }

    CostRecord default_getBorrowStockCost(const Datetime& datetime,
            const Stock& stock, price_t price, size_t num) const {
        return this->TradeCostBase::getBorrowStockCost(datetime, stock, price, num);
    }

    CostRecord getReturnStockCost(const Datetime& borrow_datetime,
            const Datetime& return_datetime,
            const Stock& stock, price_t price, size_t num)  const {
        if (override getReturnStockCost = get_override("getReturnStockCost")) {
            return getReturnStockCost(borrow_datetime, return_datetime,
                    stock, price, num);
        }
        return TradeCostBase::getReturnStockCost(borrow_datetime,
                return_datetime, stock, price, num);
    }

    CostRecord default_getReturnStockCost(const Datetime& borrow_datetime,
            const Datetime& return_datetime,
            const Stock& stock, price_t price, size_t num) const {
        return this->TradeCostBase::getReturnStockCost(borrow_datetime,
                return_datetime, stock, price, num);
    }
};


void export_TradeCost() {
    class_<TradeCostWrap, boost::noncopyable>("TradeCostBase", init<const string&>())
            .def(self_ns::str(self))
            .add_property("name", make_function(&TradeCostBase::name, return_value_policy<copy_const_reference>()))
            //.add_property("params", make_function(&TradeCostBase::getParameter, return_internal_reference<>()))
            .def("getParam", &TradeCostBase::getParam<boost::any>)
            .def("setParam", &TradeCostBase::setParam<object>)
            .def("clone", &TradeCostBase::clone)
            .def("getBuyCost", pure_virtual(&TradeCostBase::getBuyCost))
            .def("getSellCost", pure_virtual(&TradeCostBase::getSellCost))
            .def("getBorrowCashCost", &TradeCostBase::getBorrowCashCost, &TradeCostWrap::default_getBorrowCashCost)
            .def("getReturnCashCost", &TradeCostBase::getReturnCashCost, &TradeCostWrap::default_getReturnCashCost)
            //.def("getBorrowStockCost", &TradeCostBase::getBorrowStockCost, &TradeCostWrap::default_getBorrowStockCost)
            //.def("getReturnStockCost", &TradeCostBase::getReturnStockCost, &TradeCostWrap::default_getReturnStockCost)
            .def("_clone", pure_virtual(&TradeCostBase::_clone))
#if HKU_PYTHON_SUPPORT_PICKLE
            .def_pickle(name_init_pickle_suite<TradeCostBase>())
#endif
            ;
    register_ptr_to_python<TradeCostPtr>();
}