NanobindTraitInclude.cpp File Reference

#include "NanobindTraitInclude.h"

Include dependency graph for NanobindTraitInclude.cpp:

Go to the source code of this file.

Functions
PString	get_nanobind_include (const PString &type)

Function Documentation

get_nanobind_include()

PString get_nanobind_include

(

const PString &

type

)

Definition at line 19 of file NanobindTraitInclude.cpp.

                                                  {
        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 "";
}

Referenced by NanobindTraitInclude::headerExtraInclude().

Here is the caller graph for this function: