/* ----------------------------------------------------------------------------

 * 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
 **/

#include <iostream>
#include <fstream>
#include <sstream>
#include <boost/foreach.hpp>
#include <boost/regex.hpp>

#include "Argument.h"

using namespace std;
using namespace wrap;

/* ************************************************************************* */
string Argument::matlabClass(const string& delim) const {
	string result;
	BOOST_FOREACH(const string& ns, namespaces)
	result += ns + delim;
	if (type=="string")
		return result + "char";
	if (type=="bool" || type=="int" || type=="size_t" || type=="Vector" || type=="Matrix")
		return result + "double";
	return result + type;
}

/* ************************************************************************* */
void Argument::matlab_unwrap(FileWriter& file, const string& matlabName) const {
  file.oss << "  ";

  string cppType = qualifiedType("::");
  string matlabType = qualifiedType();

  if (is_ptr)
		// 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)
		// 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: 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) file.oss << ", \"" << matlabType << "\"";
  file.oss << ");" << endl;
}

/* ************************************************************************* */
string Argument::qualifiedType(const string& delim) const {
	string result;
	BOOST_FOREACH(const string& ns, namespaces) result += ns + delim;
	return result + type;
}

/* ************************************************************************* */
string ArgumentList::types() const {
  string str;
  bool first=true;
  BOOST_FOREACH(Argument arg, *this) {
    if (!first) str += ","; str += arg.type; first=false;
  }
  return str;
}

/* ************************************************************************* */
string ArgumentList::signature() const {
  string sig;
  bool cap=false;

  BOOST_FOREACH(Argument arg, *this) {
  	BOOST_FOREACH(char ch, arg.type)
        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[0];
    //Reset to default
    cap = false;
  }

  return sig;
}

/* ************************************************************************* */
string ArgumentList::names() const {
  string str;
  bool first=true;
  BOOST_FOREACH(Argument arg, *this) {
    if (!first) str += ","; str += arg.name; first=false;
  }
  return str;
}

/* ************************************************************************* */
void ArgumentList::matlab_unwrap(FileWriter& file, int start) const {
  int index = start;
  BOOST_FOREACH(Argument arg, *this) {
    stringstream buf;
    buf << "in[" << index << "]";
    arg.matlab_unwrap(file,buf.str());
    index++;
  }
}

/* ************************************************************************* */