adding overload for Sqrt(Integer_type)

to cast the integer type to double
FlexibleSUSY · Apr 24, 2016 · 1ed9c0a · 1ed9c0a
1 parent 242e94f
commit 1ed9c0a
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/src/wrappers.hpp b/src/wrappers.hpp
@@ -26,6 +26,7 @@
 #include <limits>
 #include <sstream>
 #include <string>
+#include <type_traits>
 #include <vector>
 #include <Eigen/Core>
 #include <boost/lexical_cast.hpp>
@@ -460,12 +461,18 @@ Derived SignedAbsSqrt(const Eigen::ArrayBase<Derived>& m)
    return m.unaryExpr(std::ptr_fun(SignedAbsSqrt_d));
 }
 
-template <class T>
+template <class T, typename = typename std::enable_if<std::is_floating_point<T>::value,T>::type>
 T Sqrt(T a)
 {
    return std::sqrt(a);
 }
 
+template <class T, typename = typename std::enable_if<std::is_integral<T>::value,T>::type>
+double Sqrt(T a)
+{
+   return std::sqrt(static_cast<double>(a));
+}
+
 template <typename Scalar, int M, int N>
 Eigen::Array<Scalar, M, N> Sqrt(const Eigen::Array<Scalar, M, N>& m)
 {

diff --git a/test/test_wrappers.cpp b/test/test_wrappers.cpp
@@ -381,6 +381,14 @@ BOOST_AUTO_TEST_CASE(test_Sqr_vector_Eigen_Array)
    Sqr(v);
 }
 
+BOOST_AUTO_TEST_CASE(test_Sqrt_overloads)
+{
+   BOOST_CHECK_EQUAL(Sqrt(2.f), std::sqrt(2.f));
+   BOOST_CHECK_EQUAL(Sqrt(2.) , std::sqrt(2.));
+   BOOST_CHECK_EQUAL(Sqrt(2.L), std::sqrt(2.L));
+   BOOST_CHECK_EQUAL(Sqrt(2)  , std::sqrt(2.));
+}
+
 BOOST_AUTO_TEST_CASE(test_Sqrt_vector)
 {
    std::vector<double> v(3);