1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133
|
//
// Copyright (c) 2003-2008 greg Landrum and Rational Discovery LLC
//
// @@ All Rights Reserved @@
// This file is part of the RDKit.
// The contents are covered by the terms of the BSD license
// which is included in the file license.txt, found at the root
// of the RDKit source tree.
//
#ifndef _RD_WRAP_H_
#define _RD_WRAP_H_
//
// Generic Wrapper utility functionality
//
#include <boost/python.hpp>
#include <boost/python/stl_iterator.hpp>
// code for windows DLL handling taken from
// http://www.boost.org/more/separate_compilation.html
#include <boost/config.hpp>
#ifdef BOOST_HAS_DECLSPEC // defined in config system
// we need to import/export our code only if the user has specifically
// asked for it by defining either BOOST_ALL_DYN_LINK if they want all boost
// libraries to be dynamically linked, or RDKIT_WRAP_DYN_LINK
// if they want just this one to be dynamically liked:
#if defined(BOOST_ALL_DYN_LINK) || defined(RDKIT_WRAP_DYN_LINK)
// export if this is our own source, otherwise import:
#ifdef RDKIT_WRAP_SOURCE
# define RDKIT_WRAP_DECL __declspec(dllexport)
#else
# define RDKIT_WRAP_DECL __declspec(dllimport)
#endif // RDKIT_WRAP_SOURCE
#endif // DYN_LINK
#endif // BOOST_HAS_DECLSPEC
//
// if RDKIT_WRAP_DECL isn't defined yet define it now:
#ifndef RDKIT_WRAP_DECL
#define RDKIT_WRAP_DECL
#endif
#include <boost/python/suite/indexing/vector_indexing_suite.hpp>
#include <boost/cstdint.hpp>
#include "list_indexing_suite.hpp"
#include <vector>
#include <list>
#include <iostream>
#include "Exceptions.h"
namespace python = boost::python;
RDKIT_WRAP_DECL void
throw_index_error(int key); //!< construct and throw an \c IndexError
RDKIT_WRAP_DECL void
throw_value_error(const std::string err); //!< construct and throw a \c ValueError
RDKIT_WRAP_DECL void
translate_index_error(IndexErrorException const&e);
RDKIT_WRAP_DECL void
translate_value_error(ValueErrorException const&e);
//! \brief Registers a templated converter for returning \c vectors of a
//! particular type.
//! This should be used instead of calling \c vector_to_python<T>()
//! directly because this will catch the appropriate exception if
//! the specified converter has already been registered.
template <typename T>
void RegisterVectorConverter(bool noproxy=false){
std::string name = "_vect";
name += typeid(T).name();
if(noproxy){
python::class_< std::vector<T> >(name.c_str())
.def(python::vector_indexing_suite< std::vector<T>, 1>());
} else {
python::class_< std::vector<T> >(name.c_str())
.def(python::vector_indexing_suite< std::vector<T> >());
}
}
//! \brief Registers a templated converter for returning \c lists of a
//! particular type.
//! This should be used instead of calling \c list_to_python<T>()
//! directly because this will catch the appropriate exception if
//! the specified converter has already been registered.
template <typename T>
void RegisterListConverter(bool noproxy=false){
std::string name = "_list";
name += typeid(T).name();
if(noproxy){
python::class_< std::list<T> >(name.c_str())
.def(python::list_indexing_suite< std::list<T>, 1 >());
} else {
python::class_< std::list<T> >(name.c_str())
.def(python::list_indexing_suite< std::list<T> >());
}
}
template <typename T>
std::vector<T> *pythonObjectToVect(const python::object &obj,T maxV){
std::vector<T> *res=0;
if(obj){
res=new std::vector<T>;
python::stl_input_iterator<T> beg(obj),end;
while(beg!=end){
T v=*beg;
if(v>=maxV){
throw_value_error("list element larger than allowed value");
}
res->push_back(v);
++beg;
}
}
return res;
}
template <typename T>
std::vector<T> *pythonObjectToVect(const python::object &obj){
std::vector<T> *res=0;
if(obj){
res=new std::vector<T>;
unsigned int nFrom=python::extract<unsigned int>(obj.attr("__len__")());
for(unsigned int i=0;i<nFrom;++i){
T v=python::extract<T>(obj[i]);
res->push_back(v);
}
}
return res;
}
#endif
|