NanobindTraitInclude.cpp

Go to the documentation of this file.

/***************************************
        Auteur : Thibaut Oprinsen
        Mail : thibaut.oprinsen@lapp.in2p3.fr
        Licence : CeCILL-C
****************************************/
 
#include "NanobindTraitInclude.h"

NanobindTraitInclude::NanobindTraitInclude()
: PAbstractNanobindTraitBackend()
{}

NanobindTraitInclude::~NanobindTraitInclude(){}
 
 
// Get the header file to include for a given type
PString get_nanobind_include(const PString& type) {
        static const std::map<PString, PString> type_to_header = {
                {"std::array", "<nanobind/stl/array.h>"},
                {"std::chrono", "<nanobind/stl/chrono.h>"},
                {"std::complex", "<nanobind/stl/complex.h>"},
                {"std::filesystem","<nanobind/stl/filesystem.h>"},
                {"std::function", "<nanobind/stl/function.h>"},
                {"std::list", "<nanobind/stl/list.h>"},
                {"std::map", "<nanobind/stl/map.h>"},
                {"std::optional", "<nanobind/stl/optional.h>"},
                {"std::pair", "<nanobind/stl/pair.h>"},
                {"std::set", "<nanobind/stl/set.h>"},
                {"std::string", "<nanobind/stl/string.h>"},
                {"std::string_view", "<nanobind/stl/string_view.h>"},
                {"std::wstring", "<nanobind/stl/wstring.h>"},
                {"std::tuple", "<nanobind/stl/tuple.h>"},
                {"std::shared_ptr", "<nanobind/stl/shared_ptr.h>"},
                {"std::unique_ptr", "<nanobind/stl/unique_ptr.h>"},
                {"std::unordered_set", "<nanobind/stl/unordered_set.h>"},
                {"std::unordered_map", "<nanobind/stl/unordered_map.h>"},
                {"std::variant", "<nanobind/stl/variant.h>"},
                {"std::vector", "<nanobind/stl/vector.h>"},
                {"nb::ndarray", "<nanobind/ndarray.h>"},
                {"Eigen::Matrix", "<nanobind/eigen/dense.h>"},
                {"Eigen::Array", "<nanobind/eigen/dense.h>"},
                {"Eigen::Ref", "<nanobind/eigen/dense.h>"},
                {"Eigen::Map", "<nanobind/eigen/dense.h>"},
                {"Eigen::SparseMatrix","<nanobind/eigen/sparse.h>"}
        };
 
        typedef std::map<PString, PString>::const_iterator MapIt;
        for(MapIt it = type_to_header.begin(); it != type_to_header.end(); ++it) {
                // Search if the type starts with the key + '<' (for templates)
                PString key = it->first + "<";
                if(type.substr(0, key.size()) == key) {
                        return it->second;
                }
                // Or if the type is exactly the key (for non-templates)
                if(type == it->first) {
                        return it->second;
                }
        }
 
        return "";
}


void NanobindTraitInclude::headerExtraInclude(std::set<std::string> & setInclude, const PClassConfig & classConfig, const GeneratorMode & mode) const{
        const PVecClassAttribute & listAttr(classConfig.getListAttribute());
        for(PVecClassAttribute::const_iterator it(listAttr.begin()); it != listAttr.end(); ++it){
                PString header = get_nanobind_include(it->getType());
                if(header != ""){
                        setInclude.insert("#include " + header);
                }
        }
}