317 lines
9.8 KiB
C++
317 lines
9.8 KiB
C++
/* ----------------------------------------------------------------------------
|
|
|
|
* GTSAM Copyright 2010, Georgia Tech Research Corporation,
|
|
* Atlanta, Georgia 30332-0415
|
|
* All Rights Reserved
|
|
* Authors: Frank Dellaert, et al. (see THANKS for the full author list)
|
|
|
|
* See LICENSE for the license information
|
|
|
|
* -------------------------------------------------------------------------- */
|
|
|
|
/**
|
|
* @file Argument.ccp
|
|
* @author Frank Dellaert
|
|
* @author Andrew Melim
|
|
* @author Richard Roberts
|
|
**/
|
|
|
|
#include "Argument.h"
|
|
#include "Class.h"
|
|
|
|
#include <boost/lexical_cast.hpp>
|
|
|
|
#include <iostream>
|
|
#include <fstream>
|
|
#include <sstream>
|
|
|
|
using namespace std;
|
|
using namespace wrap;
|
|
|
|
/* ************************************************************************* */
|
|
Argument Argument::expandTemplate(const TemplateSubstitution& ts) const {
|
|
Argument instArg = *this;
|
|
instArg.type = ts.tryToSubstitite(type);
|
|
return instArg;
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
ArgumentList ArgumentList::expandTemplate(
|
|
const TemplateSubstitution& ts) const {
|
|
ArgumentList instArgList;
|
|
for(const Argument& arg: *this) {
|
|
Argument instArg = arg.expandTemplate(ts);
|
|
instArgList.push_back(instArg);
|
|
}
|
|
return instArgList;
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
string Argument::matlabClass(const string& delim) const {
|
|
string result;
|
|
for(const string& ns: type.namespaces())
|
|
result += ns + delim;
|
|
if (type.name() == "string" || type.name() == "unsigned char"
|
|
|| type.name() == "char")
|
|
return result + "char";
|
|
if (type.name() == "Vector" || type.name() == "Matrix")
|
|
return result + "double";
|
|
if (type.name() == "int" || type.name() == "size_t")
|
|
return result + "numeric";
|
|
if (type.name() == "bool")
|
|
return result + "logical";
|
|
return result + type.name();
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
void Argument::matlab_unwrap(FileWriter& file, const string& matlabName) const {
|
|
file.oss << " ";
|
|
|
|
string cppType = type.qualifiedName("::");
|
|
string matlabUniqueType = type.qualifiedName();
|
|
bool isNotScalar = !type.isScalar();
|
|
|
|
// We cannot handle scalar non const references
|
|
if (!isNotScalar && is_ref && !is_const) {
|
|
throw std::runtime_error("Cannot unwrap a scalar non-const reference");
|
|
}
|
|
|
|
if (is_ptr && type.category != Qualified::EIGEN)
|
|
// A pointer: emit an "unwrap_shared_ptr" call which returns a pointer
|
|
file.oss << "boost::shared_ptr<" << cppType << "> " << name
|
|
<< " = unwrap_shared_ptr< ";
|
|
else if (is_ref && isNotScalar && type.category != Qualified::EIGEN)
|
|
// A reference: emit an "unwrap_shared_ptr" call and de-reference the pointer
|
|
file.oss << cppType << "& " << name << " = *unwrap_shared_ptr< ";
|
|
else
|
|
// Not a pointer, or a reference to a scalar type. Therefore, emit an "unwrap" call
|
|
// unwrap is specified in matlab.h as a series of template specializations
|
|
// that know how to unpack the expected MATLAB object
|
|
// example: double tol = unwrap< double >(in[2]);
|
|
// example: Vector v = unwrap< Vector >(in[1]);
|
|
file.oss << cppType << " " << name << " = unwrap< ";
|
|
|
|
file.oss << cppType << " >(" << matlabName;
|
|
if( (is_ptr || is_ref) && isNotScalar && type.category != Qualified::EIGEN)
|
|
file.oss << ", \"ptr_" << matlabUniqueType << "\"";
|
|
file.oss << ");" << endl;
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
void Argument::proxy_check(FileWriter& proxyFile, const string& s) const {
|
|
proxyFile.oss << "isa(" << s << ",'" << matlabClass(".") << "')";
|
|
if (type.name() == "Vector")
|
|
proxyFile.oss << " && size(" << s << ",2)==1";
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
void Argument::emit_cython_pxd(
|
|
FileWriter& file, const std::string& className,
|
|
const std::vector<std::string>& templateArgs) const {
|
|
string cythonType = type.pxdClassName();
|
|
if (cythonType == "This") cythonType = className;
|
|
else if (type.isEigen())
|
|
cythonType = "const " + cythonType + "&";
|
|
else if (type.match(templateArgs))
|
|
cythonType = type.name();
|
|
|
|
// add modifier
|
|
if (!type.isEigen()) {
|
|
if (is_ptr) cythonType = "shared_ptr[" + cythonType + "]&";
|
|
if (is_ref) cythonType = cythonType + "&";
|
|
if (is_const) cythonType = "const " + cythonType;
|
|
}
|
|
|
|
file.oss << cythonType << " " << name;
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
void Argument::emit_cython_pyx(FileWriter& file) const {
|
|
file.oss << type.pyxArgumentType() << " " << name;
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
std::string Argument::pyx_convertEigenTypeAndStorageOrder() const {
|
|
if (!type.isEigen())
|
|
return "";
|
|
return name + " = " + name + ".astype(float, order=\'F\', copy=False)";
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
std::string Argument::pyx_asParam() const {
|
|
string cythonType = type.pxdClassName();
|
|
string cythonVar;
|
|
if (type.isNonBasicType()) {
|
|
cythonVar = name + "." + type.shared_pxd_obj_in_pyx();
|
|
if (!is_ptr) cythonVar = "deref(" + cythonVar + ")";
|
|
} else if (type.isEigen()) {
|
|
cythonVar = "<" + cythonType + ">" + "(Map[" + cythonType + "](" + name + "))";
|
|
} else {
|
|
cythonVar = name;
|
|
}
|
|
return cythonVar;
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
string ArgumentList::types() const {
|
|
string str;
|
|
bool first = true;
|
|
for(Argument arg: *this) {
|
|
if (!first)
|
|
str += ",";
|
|
str += arg.type.name();
|
|
first = false;
|
|
}
|
|
return str;
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
string ArgumentList::signature() const {
|
|
string sig;
|
|
bool cap = false;
|
|
|
|
for(Argument arg: *this) {
|
|
for(char ch: arg.type.name())
|
|
if (isupper(ch)) {
|
|
sig += ch;
|
|
//If there is a capital letter, we don't want to read it below
|
|
cap = true;
|
|
}
|
|
if (!cap)
|
|
sig += arg.type.name()[0];
|
|
//Reset to default
|
|
cap = false;
|
|
}
|
|
|
|
return sig;
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
string ArgumentList::names() const {
|
|
string str;
|
|
bool first = true;
|
|
for(Argument arg: *this) {
|
|
if (!first)
|
|
str += ",";
|
|
str += arg.name;
|
|
first = false;
|
|
}
|
|
return str;
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
bool ArgumentList::allScalar() const {
|
|
for(Argument arg: *this)
|
|
if (!arg.type.isScalar())
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
void ArgumentList::matlab_unwrap(FileWriter& file, int start) const {
|
|
int index = start;
|
|
for(Argument arg: *this) {
|
|
stringstream buf;
|
|
buf << "in[" << index << "]";
|
|
arg.matlab_unwrap(file, buf.str());
|
|
index++;
|
|
}
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
void ArgumentList::emit_prototype(FileWriter& file, const string& name) const {
|
|
file.oss << name << "(";
|
|
bool first = true;
|
|
for(Argument arg: *this) {
|
|
if (!first)
|
|
file.oss << ", ";
|
|
file.oss << arg.type.name() << " " << arg.name;
|
|
first = false;
|
|
}
|
|
file.oss << ")";
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
void ArgumentList::emit_cython_pxd(
|
|
FileWriter& file, const std::string& className,
|
|
const std::vector<std::string>& templateArgs) const {
|
|
for (size_t j = 0; j<size(); ++j) {
|
|
at(j).emit_cython_pxd(file, className, templateArgs);
|
|
if (j<size()-1) file.oss << ", ";
|
|
}
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
void ArgumentList::emit_cython_pyx(FileWriter& file) const {
|
|
for (size_t j = 0; j < size(); ++j) {
|
|
at(j).emit_cython_pyx(file);
|
|
if (j < size() - 1) file.oss << ", ";
|
|
}
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
std::string ArgumentList::pyx_convertEigenTypeAndStorageOrder(const std::string& indent) const {
|
|
string ret, conversion;
|
|
for (size_t j = 0; j < size(); ++j) {
|
|
conversion = at(j).pyx_convertEigenTypeAndStorageOrder();
|
|
if (!conversion.empty())
|
|
ret += indent + conversion + "\n";
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
std::string ArgumentList::pyx_asParams() const {
|
|
string ret;
|
|
for (size_t j = 0; j < size(); ++j) {
|
|
ret += at(j).pyx_asParam();
|
|
if (j < size() - 1) ret += ", ";
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
std::string ArgumentList::pyx_paramsList() const {
|
|
string s;
|
|
for (size_t j = 0; j < size(); ++j) {
|
|
s += "'" + at(j).name + "'";
|
|
if (j < size() - 1) s += ", ";
|
|
}
|
|
return s;
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
std::string ArgumentList::pyx_castParamsToPythonType(
|
|
const std::string& indent) const {
|
|
if (size() == 0)
|
|
return "";
|
|
|
|
// cast params to their correct python argument type to pass in the function call later
|
|
string s;
|
|
for (size_t j = 0; j < size(); ++j)
|
|
s += indent + at(j).name + " = <" + at(j).type.pyxArgumentType()
|
|
+ ">(__params[" + std::to_string(j) + "])\n";
|
|
return s;
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
void ArgumentList::proxy_check(FileWriter& proxyFile) const {
|
|
// Check nr of arguments
|
|
proxyFile.oss << "if length(varargin) == " << size();
|
|
if (size() > 0)
|
|
proxyFile.oss << " && ";
|
|
// ...and their type.names
|
|
bool first = true;
|
|
for (size_t i = 0; i < size(); i++) {
|
|
if (!first)
|
|
proxyFile.oss << " && ";
|
|
string s = "varargin{" + boost::lexical_cast<string>(i + 1) + "}";
|
|
(*this)[i].proxy_check(proxyFile, s);
|
|
first = false;
|
|
}
|
|
proxyFile.oss << "\n";
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
|