strip_type.hpp

Go to the documentation of this file.

#ifndef MLPACK_BINDINGS_UTIL_STRIP_TYPE_HPP
#define MLPACK_BINDINGS_UTIL_STRIP_TYPE_HPP

namespace mlpack {
namespace util {

inline std::string StripType(std::string cppType)
{
  // Basically what we need to do is strip any '<' (template bits) from the
  // type.  We'll try first by removing any instances of <>.
  const size_t loc = cppType.find("<>");
  if (loc != std::string::npos)
    cppType.replace(loc, 2, "");

  // Let's just replace any invalid characters with valid '_' characters.
  std::replace(cppType.begin(), cppType.end(), '<', '_');
  std::replace(cppType.begin(), cppType.end(), '>', '_');
  std::replace(cppType.begin(), cppType.end(), ' ', '_');
  std::replace(cppType.begin(), cppType.end(), ',', '_');

  return cppType;
}

} // namespace util
} // namespace mlpack

#endif