使用qt处理图像的枕形畸变和桶形畸变代码

处理图像的枕形畸变和桶形畸变可以使用OpenCV库中的函数来实现。

下面是一个使用Qt和OpenCV库处理图像的枕形畸变和桶形畸变的示例代码：

#include <opencv2/opencv.hpp>
#include <QImage>
#include <QPixmap>

using namespace cv;

// 枕形畸变
Mat barrelDistortion(Mat input, double k)
{
    Mat output = input.clone();
    int w = input.cols;
    int h = input.rows;
    int cx = w / 2;
    int cy = h / 2;
    double dx, dy, r, theta;
    double dtheta = 1.0 / w;

    for (int y = 0; y < h; y++) {
        for (int x = 0; x < w; x++) {
            dx = x - cx;
            dy = y - cy;
            r = sqrt(dx*dx + dy*dy);
            theta = atan(r * dtheta);
            double rn = r * (1 + k * theta*theta);
            double scale = rn / r;
            int nx = dx * scale + cx;
            int ny = dy * scale + cy;

            if (nx >= 0 && nx < w && ny >= 0 && ny < h) {
                output.at<Vec3b>(y, x) = input.at<Vec3b>(ny, nx);
            }
        }
    }

    return output;
}

// 桶形畸变
Mat pincushionDistortion(Mat input, double k)
{
    Mat output = input.clone();
    int w = input.cols;
    int h = input.rows;
    int cx = w / 2;
    int cy = h / 2;
    double dx, dy, r, theta;
    double dtheta = 1.0 / w;

    for (int y = 0; y < h; y++) {
        for (int x = 0; x < w; x++) {
            dx = x - cx;
            dy = y - cy;
            r = sqrt(dx*dx + dy*dy);
            theta = atan(r * dtheta);
            double rn = r / (1 + k * theta*theta);
            double scale = rn / r;
            int nx = dx * scale + cx;
            int ny = dy * scale + cy;

            if (nx >= 0 && nx < w && ny >= 0 && ny < h) {
                output.at<Vec3b>(y, x) = input.at<Vec3b>(ny, nx);
            }
        }
    }

    return output;
}

// 将Mat转换为QImage
QImage Mat2QImage(Mat input)
{
    cv::Mat mat;
    cvtColor(input, mat, CV_BGR2RGB);
    QImage image((const uchar*)mat.data, mat.cols, mat.rows, mat.cols * mat.channels(), QImage::Format_RGB888);
    return image;
}

// 将QImage转换为Mat
Mat QImage2Mat(QImage input)
{
    cv::Mat mat(input.height(), input.width(), CV_8UC4, (uchar*)input.bits(), input.bytesPerLine());
    cv::Mat mat2;
    cvtColor(mat, mat2, CV_RGBA2BGR);
    return mat2;
}

// 示例函数
void testDistortion()
{
    // 读取图像
    Mat input = imread("input.jpg");

    // 枕形畸变
    Mat output1 = barrelDistortion(input, 0.5);

    // 桶形畸变
    Mat output2 = pincushionDistortion(input, 0.5);

    // 显示图像
    QImage image1 = Mat2QImage(output1);
    QPixmap pixmap1 = QPixmap::fromImage(image1);
    QImage image2 = Mat2QImage(output2);
    QPixmap pixmap2 = QPixmap::fromImage(image2);

    QLabel label1;
    label1.setPixmap(pixmap1.scaled(640, 480, Qt::KeepAspectRatio));
    QLabel label2;
    label2.setPixmap(pixmap2.scaled(640, 480, Qt::KeepAspectRatio));

    QVBoxLayout layout;
    layout.addWidget(&label1);
    layout.addWidget(&label2);

    QWidget window;
    window.setLayout(&layout);
    window.show();
}

int main(int argc, char *argv[])
{
    testDistortion();
    return 0;
}

在上面的代码中，`barrelDistortion`函数和`pincushionDistortion`函数分别实现了枕形畸变和桶形畸变的处理。这两个函数都接受一个`Mat`类型的输入图像和畸变系数`k`，并返回处理后的图像。

`testDistortion`函数演示了如何使用上述函数来处理输入图像并显示结果。在该函数中，先读取输入图像，然后分别对其进行枕形和桶形畸变的处理，并将处理后的结果显示出来。为了方便，这里使用了Qt的GUI框架来显示图像。

注意，上述代码中使用的是OpenCV 3.x或更早版本。如果您使用的是OpenCV 4.x或更新版本，可能需要进行一些修改。