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

 * 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 testWrap.cpp
 * @brief Unit test for wrap.c
 * @author Frank Dellaert
 **/

#include <stdlib.h>
#include <iostream>
#include <fstream>
#include <sstream>
#include <boost/assign/std/vector.hpp>
#include <boost/filesystem.hpp>
#include <CppUnitLite/TestHarness.h>

#include <wrap/utilities.h>
#include <wrap/Module.h>

using namespace std;
using namespace boost::assign;
using namespace wrap;
namespace fs = boost::filesystem;
static bool enable_verbose = false;
#ifdef TOPSRCDIR
static string topdir = TOPSRCDIR;
#else
static string topdir = "TOPSRCDIR_NOT_CONFIGURED"; // If TOPSRCDIR is not defined, we error
#endif

typedef vector<string> strvec;

// NOTE: as this path is only used to generate makefiles, it is hardcoded here for testing
// In practice, this path will be an absolute system path, which makes testing it more annoying
static const std::string headerPath = "/not_really_a_real_path/borg/gtsam/wrap";

/* ************************************************************************* */
TEST( wrap, ArgumentList ) {
  ArgumentList args;
  Argument arg1; arg1.type = "double"; arg1.name = "x";
  Argument arg2; arg2.type = "double"; arg2.name = "y";
  Argument arg3; arg3.type = "double"; arg3.name = "z";
  args.push_back(arg1);
  args.push_back(arg2);
  args.push_back(arg3);
  EXPECT(assert_equal("ddd", args.signature()));
  EXPECT(assert_equal("double,double,double", args.types()));
  EXPECT(assert_equal("x,y,z", args.names()));
}

/* ************************************************************************* */
TEST_UNSAFE( wrap, check_exception ) {
  THROWS_EXCEPTION(Module("/notarealpath", "geometry",enable_verbose));
  CHECK_EXCEPTION(Module("/alsonotarealpath", "geometry",enable_verbose), CantOpenFile);

  // clean out previous generated code
  fs::remove_all("actual_deps");

  string path = topdir + "/wrap/tests";
  Module module(path.c_str(), "testDependencies",enable_verbose);
  CHECK_EXCEPTION(module.matlab_code("actual_deps", headerPath), DependencyMissing);
}

/* ************************************************************************* */
TEST( wrap, small_parse ) {
  string moduleName("gtsam");
  Module module(moduleName, true);

  string markup(
      string("class Point2 {                \n") +
      string(" double x() const;            \n") +   // Method 1
      string(" Matrix returnChar() const;   \n") +   // Method 2
      string(" Point2 returnPoint2() const; \n") +   // Method 3
      string("};\n"));
  module.parseMarkup(markup);

  // check return types
  LONGS_EQUAL(1, module.classes.size());
  Class cls = module.classes.front();
  EXPECT(assert_equal("Point2", cls.name));
  EXPECT(!cls.isVirtual);
  EXPECT(cls.namespaces.empty());
  EXPECT(cls.static_methods.empty());
  LONGS_EQUAL(3, cls.methods.size());

  // Method 1
  Method m1 = cls.methods.at("x");
  EXPECT(assert_equal("x", m1.name));
  EXPECT(m1.is_const_);
  LONGS_EQUAL(1, m1.argLists.size());
  LONGS_EQUAL(1, m1.returnVals.size());

  ReturnValue rv1 = m1.returnVals.front();
  EXPECT(!rv1.isPair);
  EXPECT(!rv1.isPtr1);
  EXPECT(assert_equal("double", rv1.type1));
  EXPECT_LONGS_EQUAL(ReturnValue::BASIS, rv1.category1);

  // Method 2
  Method m2 = cls.methods.at("returnChar");
  EXPECT(assert_equal("returnChar", m2.name));
  EXPECT(m2.is_const_);
  LONGS_EQUAL(1, m2.argLists.size());
  LONGS_EQUAL(1, m2.returnVals.size());

  ReturnValue rv2 = m2.returnVals.front();
  EXPECT(!rv2.isPair);
  EXPECT(!rv2.isPtr1);
  EXPECT(assert_equal("Matrix", rv2.type1));
  EXPECT_LONGS_EQUAL(ReturnValue::EIGEN, rv2.category1);

  // Method 3
  Method m3 = cls.methods.at("returnPoint2");
  EXPECT(assert_equal("returnPoint2", m3.name));
  EXPECT(m3.is_const_);
  LONGS_EQUAL(1, m3.argLists.size());
  LONGS_EQUAL(1, m3.returnVals.size());

  ReturnValue rv3 = m3.returnVals.front();
  EXPECT(!rv3.isPair);
  EXPECT(!rv3.isPtr1);
  EXPECT(assert_equal("Point2", rv3.type1));
  EXPECT_LONGS_EQUAL(ReturnValue::CLASS, rv3.category1);

}

/* ************************************************************************* */
TEST( wrap, parse_geometry ) {
  string markup_header_path = topdir + "/wrap/tests";
  Module module(markup_header_path.c_str(), "geometry",enable_verbose);
  EXPECT_LONGS_EQUAL(3, module.classes.size());

  // forward declarations
  LONGS_EQUAL(2, module.forward_declarations.size());
  EXPECT(assert_equal("VectorNotEigen", module.forward_declarations[0].name));
  EXPECT(assert_equal("ns::OtherClass", module.forward_declarations[1].name));

  // includes
  strvec exp_includes; exp_includes += "folder/path/to/Test.h";
  EXPECT(assert_equal(exp_includes, module.includes));

  LONGS_EQUAL(3, module.classes.size());

  // Key for ReturnValue::return_category
//  CLASS = 1,
//  EIGEN = 2,
//  BASIS = 3,
//  VOID  = 4,
//  DUMMY = 5

  {
    // check first class
    //  class Point2 {
    //   Point2();
    //   Point2(double x, double y);
    //   double x() const;
    //   double y() const;
    //   int dim() const;
    //   char returnChar() const;
    //   void argChar(char a) const;
    //   void argUChar(unsigned char a) const;
    //   VectorNotEigen vectorConfusion();
    //  };

    Class cls = module.classes.at(0);
    EXPECT(assert_equal("Point2", cls.name));
    EXPECT_LONGS_EQUAL(2, cls.constructor.args_list.size());
    EXPECT_LONGS_EQUAL(7, cls.methods.size());

    {
      //   char returnChar() const;
      CHECK(cls.methods.find("returnChar") != cls.methods.end());
      Method m1 = cls.methods.find("returnChar")->second;
      LONGS_EQUAL(1, m1.returnVals.size());
      EXPECT(assert_equal("char", m1.returnVals.front().type1));
      EXPECT_LONGS_EQUAL(ReturnValue::BASIS, m1.returnVals.front().category1); // FAIL: gets 1 instead of 3
      EXPECT(!m1.returnVals.front().isPair);
      EXPECT(assert_equal("returnChar", m1.name));
      LONGS_EQUAL(1, m1.argLists.size());
      EXPECT_LONGS_EQUAL(0, m1.argLists.front().size());
      EXPECT(m1.is_const_);
    }

    {
      //   VectorNotEigen vectorConfusion();
      CHECK(cls.methods.find("vectorConfusion") != cls.methods.end());
      Method m1 = cls.methods.find("vectorConfusion")->second;
      LONGS_EQUAL(1, m1.returnVals.size());
      EXPECT(assert_equal("VectorNotEigen", m1.returnVals.front().type1));
      EXPECT_LONGS_EQUAL(ReturnValue::CLASS, m1.returnVals.front().category1); // FAIL: gets 0 instead of 1
      EXPECT(!m1.returnVals.front().isPair);
      EXPECT(assert_equal("vectorConfusion", m1.name));
      LONGS_EQUAL(1, m1.argLists.size());
      EXPECT_LONGS_EQUAL(0, m1.argLists.front().size());
      EXPECT(!m1.is_const_);
    }

    EXPECT_LONGS_EQUAL(0, cls.static_methods.size());
    EXPECT_LONGS_EQUAL(0, cls.namespaces.size());
  }

  // check second class, Point3
  {
    Class cls = module.classes.at(1);
    EXPECT(assert_equal("Point3", cls.name));
    EXPECT_LONGS_EQUAL(1, cls.constructor.args_list.size());
    EXPECT_LONGS_EQUAL(1, cls.methods.size());
    EXPECT_LONGS_EQUAL(2, cls.static_methods.size());
    EXPECT_LONGS_EQUAL(0, cls.namespaces.size());

    // first constructor takes 3 doubles
    ArgumentList c1 = cls.constructor.args_list.front();
    EXPECT_LONGS_EQUAL(3, c1.size());

    // check first double argument
    Argument a1 = c1.front();
    EXPECT(!a1.is_const);
    EXPECT(assert_equal("double", a1.type));
    EXPECT(!a1.is_ref);
    EXPECT(assert_equal("x", a1.name));

    // check method
    CHECK(cls.methods.find("norm") != cls.methods.end());
    Method m1 = cls.methods.find("norm")->second;
    LONGS_EQUAL(1, m1.returnVals.size());
    EXPECT(assert_equal("double", m1.returnVals.front().type1));
    EXPECT_LONGS_EQUAL(ReturnValue::BASIS, m1.returnVals.front().category1); // FAIL: gets 1 instead of 3
    EXPECT(assert_equal("norm", m1.name));
    LONGS_EQUAL(1, m1.argLists.size());
    EXPECT_LONGS_EQUAL(0, m1.argLists.front().size());
    EXPECT(m1.is_const_);
  }

  // Test class is the third one
  {
    Class testCls = module.classes.at(2);
    EXPECT_LONGS_EQUAL( 2, testCls.constructor.args_list.size());
    EXPECT_LONGS_EQUAL(19, testCls.methods.size());
    EXPECT_LONGS_EQUAL( 0, testCls.static_methods.size());
    EXPECT_LONGS_EQUAL( 0, testCls.namespaces.size());

    // function to parse: pair<Vector,Matrix> return_pair (Vector v, Matrix A) const;
    CHECK(testCls.methods.find("return_pair") != testCls.methods.end());
    Method m2 = testCls.methods.find("return_pair")->second;
    LONGS_EQUAL(1, m2.returnVals.size());
    EXPECT(m2.returnVals.front().isPair);
    EXPECT_LONGS_EQUAL(ReturnValue::EIGEN, m2.returnVals.front().category1); // FAIL: gets 1 instead of 2
    EXPECT(assert_equal("Vector", m2.returnVals.front().type1));
    EXPECT_LONGS_EQUAL(ReturnValue::EIGEN, m2.returnVals.front().category2); // FAIL: gets 6201089 instead of 2
    EXPECT(assert_equal("Matrix", m2.returnVals.front().type2));
  }

  // evaluate global functions
//  Vector aGlobalFunction();
  LONGS_EQUAL(1, module.global_functions.size());
  CHECK(module.global_functions.find("aGlobalFunction") != module.global_functions.end());
  {
    GlobalFunction gfunc = module.global_functions.at("aGlobalFunction");
    EXPECT(assert_equal("aGlobalFunction", gfunc.name));
    LONGS_EQUAL(1, gfunc.returnVals.size());
    EXPECT(assert_equal("Vector", gfunc.returnVals.front().type1));
    EXPECT_LONGS_EQUAL(1, gfunc.argLists.size());
    LONGS_EQUAL(1, gfunc.namespaces.size());
    EXPECT(gfunc.namespaces.front().empty());
  }
}

/* ************************************************************************* */
TEST( wrap, parse_namespaces ) {
  string header_path = topdir + "/wrap/tests";
  Module module(header_path.c_str(), "testNamespaces",enable_verbose);
  EXPECT_LONGS_EQUAL(6, module.classes.size());

  {
    strvec module_exp_includes;
    module_exp_includes += "path/to/ns1.h";
    module_exp_includes += "path/to/ns1/ClassB.h";
    module_exp_includes += "path/to/ns2.h";
    module_exp_includes += "path/to/ns2/ClassA.h";
    module_exp_includes += "path/to/ns3.h";
    EXPECT(assert_equal(module_exp_includes, module.includes));
  }

  {
    Class cls = module.classes.at(0);
    EXPECT(assert_equal("ClassA", cls.name));
    strvec exp_namespaces; exp_namespaces += "ns1";
    EXPECT(assert_equal(exp_namespaces, cls.namespaces));
  }

  {
    Class cls = module.classes.at(1);
    EXPECT(assert_equal("ClassB", cls.name));
    strvec exp_namespaces; exp_namespaces += "ns1";
    EXPECT(assert_equal(exp_namespaces, cls.namespaces));
  }

  {
    Class cls = module.classes.at(2);
    EXPECT(assert_equal("ClassA", cls.name));
    strvec exp_namespaces; exp_namespaces += "ns2";
    EXPECT(assert_equal(exp_namespaces, cls.namespaces));
  }

  {
    Class cls = module.classes.at(3);
    EXPECT(assert_equal("ClassB", cls.name));
    strvec exp_namespaces; exp_namespaces += "ns2", "ns3";
    EXPECT(assert_equal(exp_namespaces, cls.namespaces));
  }

  {
    Class cls = module.classes.at(4);
    EXPECT(assert_equal("ClassC", cls.name));
    strvec exp_namespaces; exp_namespaces += "ns2";
    EXPECT(assert_equal(exp_namespaces, cls.namespaces));
  }

  {
    Class cls = module.classes.at(5);
    EXPECT(assert_equal("ClassD", cls.name));
    strvec exp_namespaces;
    EXPECT(assert_equal(exp_namespaces, cls.namespaces));
  }

  // evaluate global functions
//  Vector ns1::aGlobalFunction();
//  Vector ns2::aGlobalFunction();
  LONGS_EQUAL(1, module.global_functions.size());
  CHECK(module.global_functions.find("aGlobalFunction") != module.global_functions.end());
  {
    GlobalFunction gfunc = module.global_functions.at("aGlobalFunction");
    EXPECT(assert_equal("aGlobalFunction", gfunc.name));
    LONGS_EQUAL(2, gfunc.returnVals.size());
    EXPECT(assert_equal("Vector", gfunc.returnVals.front().type1));
    EXPECT_LONGS_EQUAL(2, gfunc.argLists.size());

    // check namespaces
    LONGS_EQUAL(2, gfunc.namespaces.size());
    strvec exp_namespaces1; exp_namespaces1 += "ns1";
    EXPECT(assert_equal(exp_namespaces1, gfunc.namespaces.at(0)));

    strvec exp_namespaces2; exp_namespaces2 += "ns2";
    EXPECT(assert_equal(exp_namespaces2, gfunc.namespaces.at(1)));
  }
}

/* ************************************************************************* */
TEST( wrap, matlab_code_namespaces ) {
  string header_path = topdir + "/wrap/tests";
  Module module(header_path.c_str(), "testNamespaces",enable_verbose);
  EXPECT_LONGS_EQUAL(6, module.classes.size());
  string path = topdir + "/wrap";

  // clean out previous generated code
  fs::remove_all("actual_namespaces");

  // emit MATLAB code
  string exp_path = path + "/tests/expected_namespaces/";
  string act_path = "actual_namespaces/";
  module.matlab_code("actual_namespaces", headerPath);


  EXPECT(files_equal(exp_path + "ClassD.m"                    , act_path + "ClassD.m"                   ));
  EXPECT(files_equal(exp_path + "+ns1/ClassA.m"               , act_path + "+ns1/ClassA.m"              ));
  EXPECT(files_equal(exp_path + "+ns1/ClassB.m"               , act_path + "+ns1/ClassB.m"              ));
  EXPECT(files_equal(exp_path + "+ns2/ClassA.m"               , act_path + "+ns2/ClassA.m"              ));
  EXPECT(files_equal(exp_path + "+ns2/ClassC.m"               , act_path + "+ns2/ClassC.m"              ));
  EXPECT(files_equal(exp_path + "+ns2/+ns3/ClassB.m"          , act_path + "+ns2/+ns3/ClassB.m"         ));
  EXPECT(files_equal(exp_path + "testNamespaces_wrapper.cpp"  , act_path + "testNamespaces_wrapper.cpp" ));
}

/* ************************************************************************* */
TEST( wrap, matlab_code_geometry ) {
  // Parse into class object
  string header_path = topdir + "/wrap/tests";
  Module module(header_path,"geometry",enable_verbose);
  string path = topdir + "/wrap";

  // clean out previous generated code
  fs::remove_all("actual");

  // emit MATLAB code
  // make_geometry will not compile, use make testwrap to generate real make
  module.matlab_code("actual", headerPath);
  string epath = path + "/tests/expected/";
  string apath = "actual/";

  EXPECT(files_equal(epath + "geometry_wrapper.cpp" , apath + "geometry_wrapper.cpp" ));
  EXPECT(files_equal(epath + "Point2.m"             , apath + "Point2.m"             ));
  EXPECT(files_equal(epath + "Point3.m"             , apath + "Point3.m"             ));
  EXPECT(files_equal(epath + "Test.m"               , apath + "Test.m"               ));
  EXPECT(files_equal(epath + "aGlobalFunction.m"    , apath + "aGlobalFunction.m"    ));
}

/* ************************************************************************* */
int main() { TestResult tr; return TestRegistry::runAllTests(tr); }
/* ************************************************************************* */