From c8899addfd2deed3d84be2de234681db64987722 Mon Sep 17 00:00:00 2001 From: John Wiegley Date: Mon, 30 Apr 2007 06:26:38 +0000 Subject: Rearranged the sources a bit. --- src/py_balance.cc | 202 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 202 insertions(+) create mode 100644 src/py_balance.cc (limited to 'src/py_balance.cc') diff --git a/src/py_balance.cc b/src/py_balance.cc new file mode 100644 index 00000000..372bf1e9 --- /dev/null +++ b/src/py_balance.cc @@ -0,0 +1,202 @@ +using namespace boost::python; +using namespace ledger; + +unsigned int balance_len(balance_t& bal) +{ + return bal.amounts.size(); +} + +amount_t balance_getitem(balance_t& bal, int i) +{ + std::size_t len = bal.amounts.size(); + + if (abs(i) >= len) { + PyErr_SetString(PyExc_IndexError, "Index out of range"); + throw_error_already_set(); + } + + int x = i < 0 ? len + i : i; + amounts_map::iterator elem = bal.amounts.begin(); + while (--x >= 0) + elem++; + + return (*elem).second; +} + +unsigned int balance_pair_len(balance_pair_t& bal_pair) +{ + return balance_len(bal_pair.quantity); +} + +amount_t balance_pair_getitem(balance_pair_t& bal_pair, int i) +{ + return balance_getitem(bal_pair.quantity, i); +} + +void export_balance() +{ + class_< balance_t > ("Balance") + .def(init()) + .def(init()) + .def(init()) + .def(init()) + .def(init()) + + .def(self += self) + .def(self += other()) + .def(self += long()) + .def(self + self) + .def(self + other()) + .def(self + long()) + .def(self -= self) + .def(self -= other()) + .def(self -= long()) + .def(self - self) + .def(self - other()) + .def(self - long()) + .def(self *= self) + .def(self *= other()) + .def(self *= long()) + .def(self * self) + .def(self * other()) + .def(self * long()) + .def(self /= self) + .def(self /= other()) + .def(self /= long()) + .def(self / self) + .def(self / other()) + .def(self / long()) + .def(- self) + + .def(self < self) + .def(self < other()) + .def(self < long()) + .def(self <= self) + .def(self <= other()) + .def(self <= long()) + .def(self > self) + .def(self > other()) + .def(self > long()) + .def(self >= self) + .def(self >= other()) + .def(self >= long()) + .def(self == self) + .def(self == other()) + .def(self == long()) + .def(self != self) + .def(self != other()) + .def(self != long()) + .def(! self) + + .def(self_ns::str(self)) + + .def("__abs__", &balance_t::abs) + .def("__len__", balance_len) + .def("__getitem__", balance_getitem) + + .def("valid", &balance_t::valid) + + .def("realzero", &balance_t::realzero) + .def("amount", &balance_t::amount) + .def("value", &balance_t::value) + .def("price", &balance_t::price) + .def("date", &balance_t::date) + .def("strip_annotations", &balance_t::strip_annotations) + .def("write", &balance_t::write) + .def("round", &balance_t::round) + .def("negate", &balance_t::negate) + .def("negated", &balance_t::negated) + ; + + class_< balance_pair_t > ("BalancePair") + .def(init()) + .def(init()) + .def(init()) + .def(init()) + .def(init()) + .def(init()) + + .def(self += self) + .def(self += other()) + .def(self += other()) + .def(self += long()) + .def(self + self) + .def(self + other()) + .def(self + other()) + .def(self + long()) + .def(self -= self) + .def(self -= other()) + .def(self -= other()) + .def(self -= long()) + .def(self - self) + .def(self - other()) + .def(self - other()) + .def(self - long()) + .def(self *= self) + .def(self *= other()) + .def(self *= other()) + .def(self *= long()) + .def(self * self) + .def(self * other()) + .def(self * other()) + .def(self * long()) + .def(self /= self) + .def(self /= other()) + .def(self /= other()) + .def(self /= long()) + .def(self / self) + .def(self / other()) + .def(self / other()) + .def(self / long()) + .def(- self) + + .def(self < self) + .def(self < other()) + .def(self < other()) + .def(self < long()) + .def(self <= self) + .def(self <= other()) + .def(self <= other()) + .def(self <= long()) + .def(self > self) + .def(self > other()) + .def(self > other()) + .def(self > long()) + .def(self >= self) + .def(self >= other()) + .def(self >= other()) + .def(self >= long()) + .def(self == self) + .def(self == other()) + .def(self == other()) + .def(self == long()) + .def(self != self) + .def(self != other()) + .def(self != other()) + .def(self != long()) + .def(! self) + + .def(self_ns::str(self)) + + .def("__abs__", &balance_pair_t::abs) + .def("__len__", balance_pair_len) + .def("__getitem__", balance_pair_getitem) + + .def("valid", &balance_pair_t::valid) + + .def("realzero", &balance_pair_t::realzero) + .def("amount", &balance_pair_t::amount) + .def("value", &balance_pair_t::value) + .def("price", &balance_pair_t::price) + .def("date", &balance_pair_t::date) + .def("strip_annotations", &balance_pair_t::strip_annotations) + .def("write", &balance_pair_t::write) + .def("round", &balance_pair_t::round) + .def("negate", &balance_pair_t::negate) + .def("negated", &balance_pair_t::negated) + + .add_property("cost", + make_getter(&balance_pair_t::cost, + return_value_policy())) + ; +} -- cgit v1.2.3