57 lines
1.8 KiB
C++
57 lines
1.8 KiB
C++
/* ----------------------------------------------------------------------------
|
|
|
|
* GTSAM Copyright 2010, Georgia Tech Research Corporation,
|
|
* Atlanta, Georgia 30332-0415
|
|
* All Rights Reserved
|
|
* Authors: Frank Dellaert, et al. (see THANKS for the full author list)
|
|
|
|
* See LICENSE for the license information
|
|
|
|
* -------------------------------------------------------------------------- */
|
|
|
|
/**
|
|
* @brief exports NonlinearFactorGraph class to python
|
|
* @author Andrew Melim
|
|
* @author Ellon Paiva Mendes (LAAS-CNRS)
|
|
**/
|
|
|
|
#include <boost/python.hpp>
|
|
|
|
#define NO_IMPORT_ARRAY
|
|
#include <numpy_eigen/NumpyEigenConverter.hpp>
|
|
|
|
#include "gtsam/nonlinear/NonlinearFactorGraph.h"
|
|
#include "gtsam/nonlinear/NonlinearFactor.h"
|
|
|
|
using namespace boost::python;
|
|
using namespace gtsam;
|
|
|
|
BOOST_PYTHON_MEMBER_FUNCTION_OVERLOADS(print_overloads, NonlinearFactorGraph::print, 0, 1);
|
|
|
|
boost::shared_ptr<NonlinearFactor> getNonlinearFactor(
|
|
const NonlinearFactorGraph& graph, size_t idx) {
|
|
auto p = boost::dynamic_pointer_cast<NonlinearFactor>(graph.at(idx));
|
|
if (!p) throw std::runtime_error("No NonlinearFactor at requested index");
|
|
return p;
|
|
};
|
|
|
|
void exportNonlinearFactorGraph(){
|
|
|
|
typedef NonlinearFactorGraph::sharedFactor sharedFactor;
|
|
|
|
void (NonlinearFactorGraph::*push_back1)(const sharedFactor&) = &NonlinearFactorGraph::push_back;
|
|
void (NonlinearFactorGraph::*add1)(const sharedFactor&) = &NonlinearFactorGraph::add;
|
|
|
|
class_<NonlinearFactorGraph>("NonlinearFactorGraph", init<>())
|
|
.def("size",&NonlinearFactorGraph::size)
|
|
.def("push_back", push_back1)
|
|
.def("add", add1)
|
|
.def("resize", &NonlinearFactorGraph::resize)
|
|
.def("empty", &NonlinearFactorGraph::empty)
|
|
.def("print", &NonlinearFactorGraph::print, print_overloads(args("s")))
|
|
;
|
|
|
|
def("getNonlinearFactor", getNonlinearFactor);
|
|
|
|
}
|