PythonStdMapUtils.h

Go to the documentation of this file.

// PythonStdMapUtils.h:
// --------------------
//

#ifndef __PYIPSDKBASE_PYTHONSTDMAPUTILS_H__
#define __PYIPSDKBASE_PYTHONSTDMAPUTILS_H__

#include <boost/python/suite/indexing/map_indexing_suite.hpp>

namespace ipsdk {
namespace python {


template <typename Container>
std::string
mapToString(const Container& dataColl)
{
    std::stringstream ss;
    ss << "{";
    bool first = true;
    for (typename Container::const_iterator
        i = dataColl.begin(), e = dataColl.end(); i != e; ++i) {
        if (first) {
            first = false;
        }
        else {
            ss << ", ";
        }
        ss << "(" << i->first << ", " << i->second << ")";
    }
    ss << "}";
    return ss.str();
}


// export map
template<class Container>
void
export_map(const char* py_name) {

    boost::python::class_<Container>(py_name)
        .def(boost::python::map_indexing_suite<Container>())
        .def("__str__", &mapToString<Container>);
}

// export map
template<class Container>
void
export_map_no_str(const char* py_name) {

    boost::python::class_<Container>(py_name)
        .def(boost::python::map_indexing_suite<Container>());
}


} // end of namespace python
} // end of namespace ipsdk

#endif // __PYIPSDKBASE_PYTHONSTDMAPUTILS_H__