12345qiupeng
/
gtsam
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add helper functions to better handle gtsam.Symbol on python

release/4.3a0
Ellon Mendes 2015-11-23 20:56:51 +01:00
parent 49d02c798f
commit 8ae3dda6a6
1 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
python/handwritten/inference/Symbol.cpp
View File

@ -20,6 +20,8 @@
#define NO_IMPORT_ARRAY #define NO_IMPORT_ARRAY
#include <numpy_eigen/NumpyEigenConverter.hpp> #include <numpy_eigen/NumpyEigenConverter.hpp>
#include <sstream> // for stringstream
#include "gtsam/inference/Symbol.h" #include "gtsam/inference/Symbol.h"
using namespace boost::python; using namespace boost::python;
@ -43,6 +45,20 @@ std::string selfToString(const Symbol & self)
return (std::string)self; return (std::string)self;
} }
// Helper function to convert a Symbol to int using int() cast in python
size_t selfToKey(const Symbol & self)
{
return self.key();
}
// Helper function to recover symbol's unsigned char as string
std::string chrFromSelf(const Symbol & self)
{
std::stringstream ss;
ss << self.chr();
return ss.str();
}
void exportSymbol(){ void exportSymbol(){
class_<Symbol, boost::shared_ptr<Symbol> >("Symbol") class_<Symbol, boost::shared_ptr<Symbol> >("Symbol")
@ -53,7 +69,6 @@ class_<Symbol, boost::shared_ptr<Symbol> >("Symbol")
.def("print", &Symbol::print, print_overloads(args("s"))) .def("print", &Symbol::print, print_overloads(args("s")))
.def("equals", &Symbol::equals, equals_overloads(args("q","tol"))) .def("equals", &Symbol::equals, equals_overloads(args("q","tol")))
.def("key", &Symbol::key) .def("key", &Symbol::key)
.def("chr", &Symbol::chr)
.def("index", &Symbol::index) .def("index", &Symbol::index)
.def(self < self) .def(self < self)
.def(self == self) .def(self == self)
@ -61,6 +76,8 @@ class_<Symbol, boost::shared_ptr<Symbol> >("Symbol")
.def(self != self) .def(self != self)
.def(self != other<Key>()) .def(self != other<Key>())
.def("__repr__", &selfToString) .def("__repr__", &selfToString)
.def("__int__", &selfToKey)
.def("chr", &chrFromSelf)
; ;
} }