From 399c5e555167df20ea8845a37dccaf27bcab07d2 Mon Sep 17 00:00:00 2001
From: dellaert <dellaert@cc.gatech.edu>
Date: Sun, 25 May 2014 16:28:59 -0400
Subject: [PATCH] Added test for overloaded global functions

---
 wrap/tests/expected/geometry_wrapper.cpp      | 19 +++++++++++++++++++
 .../tests/expected/overloadedGlobalFunction.m |  8 ++++++++
 wrap/tests/geometry.h                         |  4 ++++
 wrap/tests/testWrap.cpp                       |  5 +++--
 4 files changed, 34 insertions(+), 2 deletions(-)
 create mode 100644 wrap/tests/expected/overloadedGlobalFunction.m

diff --git a/wrap/tests/expected/geometry_wrapper.cpp b/wrap/tests/expected/geometry_wrapper.cpp
index e00004d57..b34112718 100644
--- a/wrap/tests/expected/geometry_wrapper.cpp
+++ b/wrap/tests/expected/geometry_wrapper.cpp
@@ -502,6 +502,19 @@ void aGlobalFunction_42(int nargout, mxArray *out[], int nargin, const mxArray *
   checkArguments("aGlobalFunction",nargout,nargin,0);
   out[0] = wrap< Vector >(aGlobalFunction());
 }
+void overloadedGlobalFunction_43(int nargout, mxArray *out[], int nargin, const mxArray *in[])
+{
+  checkArguments("overloadedGlobalFunction",nargout,nargin,1);
+  int a = unwrap< int >(in[0]);
+  out[0] = wrap< Vector >(overloadedGlobalFunction(a));
+}
+void overloadedGlobalFunction_44(int nargout, mxArray *out[], int nargin, const mxArray *in[])
+{
+  checkArguments("overloadedGlobalFunction",nargout,nargin,2);
+  int a = unwrap< int >(in[0]);
+  double b = unwrap< double >(in[1]);
+  out[0] = wrap< Vector >(overloadedGlobalFunction(a,b));
+}
 
 void mexFunction(int nargout, mxArray *out[], int nargin, const mxArray *in[])
 {
@@ -643,6 +656,12 @@ void mexFunction(int nargout, mxArray *out[], int nargin, const mxArray *in[])
     case 42:
       aGlobalFunction_42(nargout, out, nargin-1, in+1);
       break;
+    case 43:
+      overloadedGlobalFunction_43(nargout, out, nargin-1, in+1);
+      break;
+    case 44:
+      overloadedGlobalFunction_44(nargout, out, nargin-1, in+1);
+      break;
     }
   } catch(const std::exception& e) {
     mexErrMsgTxt(("Exception from gtsam:\n" + std::string(e.what()) + "\n").c_str());
diff --git a/wrap/tests/expected/overloadedGlobalFunction.m b/wrap/tests/expected/overloadedGlobalFunction.m
new file mode 100644
index 000000000..5b086b15e
--- /dev/null
+++ b/wrap/tests/expected/overloadedGlobalFunction.m
@@ -0,0 +1,8 @@
+function varargout = overloadedGlobalFunction(varargin)
+      if length(varargin) == 1 && isa(varargin{1},'numeric')
+        varargout{1} = geometry_wrapper(43, varargin{:});
+      elseif length(varargin) == 2 && isa(varargin{1},'numeric') && isa(varargin{2},'double')
+        varargout{1} = geometry_wrapper(44, varargin{:});
+      else
+        error('Arguments do not match any overload of function overloadedGlobalFunction');
+      end
diff --git a/wrap/tests/geometry.h b/wrap/tests/geometry.h
index bdced45ed..bc233763d 100644
--- a/wrap/tests/geometry.h
+++ b/wrap/tests/geometry.h
@@ -87,6 +87,10 @@ class Test {
 
 Vector aGlobalFunction();
 
+// An overloaded global function
+Vector overloadedGlobalFunction(int a);
+Vector overloadedGlobalFunction(int a, double b);
+
 // comments at the end!
 
 // even more comments at the end!
diff --git a/wrap/tests/testWrap.cpp b/wrap/tests/testWrap.cpp
index 2cf152a7d..290372c09 100644
--- a/wrap/tests/testWrap.cpp
+++ b/wrap/tests/testWrap.cpp
@@ -307,8 +307,8 @@ TEST( wrap, parse_geometry ) {
   }
 
   // evaluate global functions
-//  Vector aGlobalFunction();
-  LONGS_EQUAL(1, module.global_functions.size());
+  //  Vector aGlobalFunction();
+  LONGS_EQUAL(2, module.global_functions.size());
   CHECK(module.global_functions.find("aGlobalFunction") != module.global_functions.end());
   {
     GlobalFunction gfunc = module.global_functions.at("aGlobalFunction");
@@ -447,6 +447,7 @@ TEST( wrap, matlab_code_geometry ) {
   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"    ));
+  EXPECT(files_equal(epath + "overloadedGlobalFunction.m"    , apath + "overloadedGlobalFunction.m"    ));
 }
 
 /* ************************************************************************* */