自动将一种 c++ 类型转换为另一种

Question

我正在尝试使自定义类和 OpenCV 之间的转换更容易。

Converting constructors 允许我们声明一个没有explicit 关键字的类并从备用数据类型构造。

例如，我们可以声明两个类：

class Foo
{
public:
    std::vector<int32_t> data;

    Foo(int k) : data(k) {}
};

class FooProxy : public cv::Mat
{
public:
    FooProxy(const Foo & foo) :  cv::Mat(foo.data.size(), 1, CV_32SC1, (void*)foo.data.data()) {}
};

然后我们可以创建一个FooProxy，方法是将其设置为相等，然后使用我们想要的任何 OpenCV 函数：

Foo myFoo(3);
myFoo.data = { 1,2,3 };

FooProxy fp = myFoo;
fp.setTo(1);
float myNorm = cv::norm(fp);

std::cout <<  "myFoo: " << myFoo.data.at(0) << ", " << myFoo.data.at(1) << ", " << myFoo.data.at(2) << "\n";
std::cout << "fp: " << fp.at<int32_t>(0) << ", " << fp.at<int32_t>(1) << ", " << fp.at<int32_t>(2) << "\n";
std::cout << "The Norm is: " << myNorm << "\n";

带输出：

myFoo: 1, 1, 1
fp: 1, 1, 1
The Norm is: 1.73205

但是，我更愿意写：

float myNorm = cv::norm(myFoo);

并让 c++ 自动转换 myFoo。这可能吗？

这行得通：

 float myNorm2 = cv::norm(FooProxy(myFoo));

但不是更简单的版本。

Answer 1

你可以创建一个conversion operator

class Foo
{
  ...

  operator cv::Mat()
  {
    return cv::Mat(data.size(), 1, CV_32SC1, (void*)data.data());
  }
};

编辑 cv::norm 有以下重载（注意没有cv::Mat）：

cv::norm(const Matx< _Tp, m, n > &M)
cv::norm(const Matx< _Tp, m, n > &M, int normType)
cv::norm(InputArray src1, int normType=NORM_L2, InputArray mask=noArray()) // this is the one we want
cv::norm(InputArray src1, InputArray src2, int normType=NORM_L2, InputArray mask=noArray())
cv::norm(const SparseMat &src, int normType)

cv::InputArray 可以从 cv::Mat 构造，但是您只能隐式地进行 1 次用户定义的转换。因此，如果您希望 Foo 与 cv::norm 一起使用，您实际上需要 cv::InputArray 的转换运算符。

Answer 2

那么你应该像这样在 namespace cv 中定义一个函数：

namespace cv{
float norm(const Foo& foo){
    return cv::norm(FooProxy(foo));
}
}

现在你可以打电话了：

float res =  cv::norm(myFoo);

---

代码：

//! 2018.01.17 09:42:16 CST

#include <opencv2/opencv.hpp>
#include <iostream>
using namespace std;
using namespace cv;

class Foo
{
    public:
        std::vector<int32_t> data;

        Foo(int k) : data(k) {}
};

class FooProxy : public cv::Mat
{
    public:
        FooProxy(const Foo & foo) :  cv::Mat(foo.data.size(), 1, CV_32SC1, (void*)foo.data.data()) {}
};

namespace cv {
float norm(const Foo& foo) {
    return cv::norm(FooProxy(foo));
}
}

int main() {
    Foo myFoo(3);
    myFoo.data = { 1,2,3 };

    FooProxy fp = myFoo;
    fp.setTo(1);
    float myNorm = cv::norm(fp);

    std::cout <<  "myFoo: " << myFoo.data.at(0) << ", " << myFoo.data.at(1) << ", " << myFoo.data.at(2) << "\n";
    std::cout << "fp: " << fp.at<int32_t>(0) << ", " << fp.at<int32_t>(1) << ", " << fp.at<int32_t>(2) << "\n";
    std::cout << "The Norm is: " << myNorm << "\n";


    float myNorm2 = cv::norm(FooProxy(myFoo));
    std::cout << "The Norm2 is:"<<myNorm2<<endl;

    float myNorm3 = cv::norm(myFoo);
    std::cout << "The Norm3 is:"<<myNorm2<<endl;
}

结果：

myFoo: 1, 1, 1
fp: 1, 1, 1
The Norm is: 1.73205
The Norm2 is:1.73205
The Norm3 is:1.73205