From 0ab6ee5612205a3780aeef4919f0a4035654684a Mon Sep 17 00:00:00 2001
From: Heiko Strathmann <heiko.strathmann@gmail.com>
Date: Mon, 19 Feb 2018 16:52:47 +0000
Subject: [PATCH] move swig specific put type dispatchers to swig interface

---
 src/interfaces/swig/shogun.i | 20 +++++++++++++++++++-
 src/shogun/base/SGObject.cpp | 18 ------------------
 src/shogun/base/SGObject.h   |  5 -----
 3 files changed, 19 insertions(+), 24 deletions(-)

diff --git a/src/interfaces/swig/shogun.i b/src/interfaces/swig/shogun.i
index 88407aa7ba5..f1ce36cb7f8 100644
--- a/src/interfaces/swig/shogun.i
+++ b/src/interfaces/swig/shogun.i
@@ -144,10 +144,28 @@ namespace shogun
 		else
 			$self->put(tag_t, value);
 	}
+	
+#ifdef SWIGJAVA
+	template <typename T, typename T2 = typename std::enable_if<std::is_same<SGMatrix<float64_t>, T>::value, T>::type>
+	void put_vector_or_matrix_dispatcher(const std::string& name, T value)
+	{
+		Tag<SGVector<float64_t>> tag_vec(name);
+		Tag<SGMatrix<float64_t>> tag_mat(name);
+	
+		if ($self->has(tag_mat))
+			$self->put(tag_mat, value);
+		else if ((value.num_rows==1 || value.num_cols==1) && $self->has(tag_vec))
+			$self->put(tag_vec, SGVector<float64_t>(value.data()));
+		else
+			$self->put(tag_mat, value);
+	}
+#endif // SWIGJAVA
 }
 
 %template(put) CSGObject::put_scalar_dispatcher<int32_t, int32_t>;
+#ifndef SWIGJAVA
 %template(put) CSGObject::put_scalar_dispatcher<int64_t, int64_t>;
+#endif // SWIGJAVA
 %template(put) CSGObject::put_scalar_dispatcher<float64_t, float64_t>;
 
 
@@ -155,7 +173,7 @@ namespace shogun
 %template(put) CSGObject::put<SGVector<float64_t>, SGVector<float64_t>>;
 %template(put) CSGObject::put<SGMatrix<float64_t>, SGMatrix<float64_t>>;
 #else // SWIGJAVA
-%template(put) CSGObject::put_vector_or_matrix<SGMatrix<float64_t>, SGMatrix<float64_t>>;
+%template(put) CSGObject::put_vector_or_matrix_dispatcher<SGMatrix<float64_t>, SGMatrix<float64_t>>;
 #endif // SWIGJAVA
 
 } // namespace shogun
diff --git a/src/shogun/base/SGObject.cpp b/src/shogun/base/SGObject.cpp
index c931b2fc6dc..def57d14184 100644
--- a/src/shogun/base/SGObject.cpp
+++ b/src/shogun/base/SGObject.cpp
@@ -1039,21 +1039,3 @@ void CSGObject::put(const std::string& name, CSGObject* value)
 	put(Tag<CSGObject*>(name), value);
 }
 
-namespace shogun
-{
-// TODO: Move this to SGBase.i and make it SGMatrix<T> (T=float64) rather than T=SGMatrix<float64>
-template<>
-void CSGObject::put_vector_or_matrix(const std::string& name, SGMatrix<float64_t> value)
-{
-	Tag<SGVector<float64_t>> tag_vec(name);
-	Tag<SGMatrix<float64_t>> tag_mat(name);
-
-	if (has(tag_mat))
-		put(tag_mat, value);
-	else if ((value.num_rows==1 || value.num_cols==1) && has(tag_vec))
-		put(tag_vec, SGVector<float64_t>(value.data()));
-	else
-		put(tag_mat, value);
-}
-}
-
diff --git a/src/shogun/base/SGObject.h b/src/shogun/base/SGObject.h
index 67d3d37381f..bb46e9be948 100644
--- a/src/shogun/base/SGObject.h
+++ b/src/shogun/base/SGObject.h
@@ -421,11 +421,6 @@ class CSGObject
 		put(Tag<T>(name), value);
 	}
 
-	// FIXME: move to swig interface.i, can be moved once the typemaps match
-	// also should be void put_vector_or_matrix(const std::string& name, SGMatrix<T> value);
-	template <typename T, typename T2 = typename std::enable_if<!std::is_base_of<CSGObject, typename std::remove_pointer<T>::type>::value, T>::type>
-	void put_vector_or_matrix(const std::string& name, T value);
-
 	/** Getter for a class parameter, identified by a Tag.
 	 * Throws an exception if the class does not have such a parameter.
 	 *