hikyuu2/hikyuu_pywrap/trade_sys/_Environment.cpp

76 lines
2.2 KiB
C++

/*
* _Environment.cpp
*
* Created on: 2013-3-2
* Author: fasiondog
*/
/*
* _TradeRecord.cpp
*
* Created on: 2013-2-25
* Author: fasiondog
*/
#include <boost/python.hpp>
#include <hikyuu/trade_sys/environment/EnvironmentBase.h>
#include <hikyuu/trade_sys/environment/build_in.h>
#include "../_Parameter.h"
#include "../pickle_support.h"
using namespace boost::python;
using namespace hku;
class EnvironmentWrap: public EnvironmentBase, public wrapper<EnvironmentBase>{
public:
EnvironmentWrap() : EnvironmentBase() {}
EnvironmentWrap(const string& name): EnvironmentBase(name) {}
void _reset() {
if (override func = get_override("_reset")) {
func();
} else {
EnvironmentBase::_reset();
}
}
void default_reset() {
this->EnvironmentBase::_reset();
}
EnvironmentPtr _clone() {
return this->get_override("_clone")();
}
void _calculate() {
this->get_override("_calculate")();
}
};
string (EnvironmentBase::*ev_get_name)() const = &EnvironmentBase::name;
void (EnvironmentBase::*ev_set_name)(const string&) = &EnvironmentBase::name;
void export_Environment() {
class_<EnvironmentWrap, boost::noncopyable>("EnvironmentBase", init<>())
.def(init<const string&>())
.def(self_ns::str(self))
.add_property("name", ev_get_name, ev_set_name)
.def("getParam", &EnvironmentBase::getParam<boost::any>)
.def("setParam", &EnvironmentBase::setParam<object>)
.def("setQuery", &EnvironmentBase::setQuery)
.def("getQuery", &EnvironmentBase::getQuery)
.def("isValid", &EnvironmentBase::isValid)
.def("_addValid", &EnvironmentBase::_addValid)
.def("reset", &EnvironmentBase::reset)
.def("clone", &EnvironmentBase::clone)
.def("_reset", &EnvironmentBase::_reset, &EnvironmentWrap::default_reset)
.def("_clone", pure_virtual(&EnvironmentBase::_clone))
.def("_calculate", pure_virtual(&EnvironmentBase::_calculate))
;
register_ptr_to_python<EnvironmentPtr>();
def("EV_TwoLine", EV_TwoLine, (arg("fast"), arg("slow"), arg("market") = "SH"));
}