使用enable_if添加重载运算符==

时间:2018-04-27 09:53:35

标签: c++ c++11 templates operator-overloading enable-if

我有一个工作模板类点,其中包含operator==的重载。

由于浮点比较,我试图为浮点添加第二个带有enable_if的重载,以使用几乎相等的函数。

这是我的尝试:

template<typename T>
class Point2D
{
public:
   Point2D(T x, T y);

   Point2D& operator= (const Point2D& point);
   bool     operator==(const Point2D& point) const;
   bool     operator!=(const Point2D& point) const;
};

template<typename T>
Point2D<T>::Point2D(T x, T y) : x_(x), y_(y)
{
}

template<typename T>
Point2D<T>& Point2D<T>::operator=(const Point2D& point)
{
   if(this != &point)
   {
      x_ = point.x_;
      y_ = point.y_;
   }

   return *this;
}

template<typename T>
bool Point2D<T>::operator==(const Point2D& point) const
{
   return (x_ == point.x_) && (y_ == point.y_);
}

template<typename T>
typename std::enable_if<std::is_floating_point<T>::value, bool>::type
Point2D<T>::operator==(const Point2D& point) const
{
   return Traits::almost_equal(x_, point.x_) &&
          Traits::almost_equal(y_, point.y_);
}

注意:

  • 这是一个简化示例。

  • 代码实际上没有启用enable_if重载

  • 我必须将声明和实现分开(都在.h中),所以请按原样参考。

编译器给我的错误是

error: prototype for typename std::enable_if<std::is_floating_point<_Tp>::value, bool>::type Point2D<T>::operator==(const Point2D<T>&) const
  does not match any in class Point2D<T>
  Point2D<T>::operator==(const Point2D& point) const
  ^

error: candidate is: bool Point2D<T>::operator==(const Point2D<T>&)
const bool Point2D<T>::operator==(const Point2D& point) const
      ^

我不明白错误是指什么。

2 个答案:

答案 0 :(得分:2)

真的,不要那样做。这样做:

template<typename T>
class Point2D
{
  bool equals(const Point2D& other, std::true_type is_floating_point ) const;
  bool equals(const Point2D& other, std::false_type is_floating_point ) const;
public:
  Point2D(T x, T y);

  Point2D& operator= (const Point2D& point);
  bool     operator==(const Point2D& point) const;
  bool     operator!=(const Point2D& point) const;
};

现在:

template<typename T>
bool Point2D<T>::operator==(const Point2D& point) const
{
  return this->equals(point, std::is_floating_point<T>{});
}
template<class T>
bool Point2D<T>::equals(const Point2D& point, std::false_type /*is_floating_point*/ ) const {
  return (x_ == point.x_) && (y_ == point.y_);
}
template<class T>
bool Point2D<T>::equals(const Point2D& point, std::true_type /*is_floating_point*/ ) const {
  return Traits::almost_equal(x_, point.x_) &&
      Traits::almost_equal(y_, point.y_);
}

这是标签调度。它比SFINAE enable-if更清晰,更容易,编译时间更短。

答案 1 :(得分:1)

错误很明确 - 您的定义与您的声明不符。在原型和定义中,您需要完全相同的enable_if。此外,您的enable_if无效,因为它需要在替换(SFINAE)期间发生。

为了使事情更具可读性,您可以定义别名并使用尾随返回类型

template<typename T>
class Point2D
{
   template <typename U>
   using EnableIfFloat = 
      typename std::enable_if<std::is_floating_point<U>::value, bool>::type;

public:
   Point2D(T x, T y);

   template <typename U = T>
   auto operator==(const Point2D& point) const -> EnableIfFloat<U>&;
   bool operator==(const Point2D& point) const;
   bool operator!=(const Point2D& point) const;
};

定义:

template<typename T>
template<typename U>
auto Point2D<T>::operator==(const Point2D& point) const -> EnableIfFloat<U>&
{
   return Traits::almost_equal(x_, point.x_) &&
          Traits::almost_equal(y_, point.y_);
}
相关问题