float *转QImage-Go服务器开发

要将一个指向 float 的指针转换为 Qt 中的 QImage，你需要考虑图像的宽度、高度和格式（如灰度或 RGB）。以下是如何将 float* 转换为 QImage 的一般步骤和示例代码。

1. 确定图像参数

首先，你需要知道图像的宽度、高度以及通道数。例如，RGB 图像有三个通道，而灰度图像只有一个通道。

2. 创建 QImage

然后，你可以使用这些数据来创建一个 QImage 对象。在这里，我提供两个示例，一个用于灰度图像，另一个用于 RGB 图像。

示例代码

#include <QImage>
#include <iostream>

QImage floatToGrayScaleQImage(float* data, int width, int height) {
    // 创建一个 QImage 对象，指定格式为 Format_Grayscale8 (8位灰阶)
    QImage image(width, height, QImage::Format_Grayscale8);

    // 遍历每个像素并设置其值
    for (int y = 0; y < height; ++y) {
        for (int x = 0; x < width; ++x) {
            // 假设 data 是 [0, 1] 范围内的浮点数，将其转换为 [0, 255]
            unsigned char value = static_cast<unsigned char>(data[y * width + x] * 255);
            image.setPixel(x, y, qRgb(value, value, value)); // 灰度值相同，所以用 qRgb 设置 R、G 和 B 为同样的值
        }
    }

    return image;
}

QImage floatToRGBQImage(float* data, int width, int height) {
    // 创建一个 QImage 对象，指定格式为 Format_RGB888 (RGB三通道)
    QImage image(width, height, QImage::Format_RGB888);

    // 遍历每个像素并设置其值
    for (int y = 0; y < height; ++y) {
        for (int x = 0; x < width; ++x) {
            int index = (y * width + x) * 3; // RGB 每个 pixel 有3个 float 值

            unsigned char r = static_cast<unsigned char>(data[index] * 255);   // Red channel
            unsigned char g = static_cast<unsigned char>(data[index + 1] * 255); // Green channel
            unsigned char b = static_cast<unsigned char>(data[index + 2] * 255); // Blue channel

            image.setPixel(x, y, qRgb(r, g, b)); 
        }
    }

    return image;
}

int main() {
    const int width = 256;
    const int height = 256;

    float* grayData = new float[width * height];
    
    // 填充一些数据（在这里我们使用随机数据作为示例）
    for(int i = 0; i < width * height; ++i) {
        grayData[i] = static_cast<float>(rand()) / RAND_MAX;
    }

    QImage grayImg = floatToGrayScaleQImage(grayData, width, height);
    
    grayImg.save("gray_image.png"); 

    
// 以下部分用于处理 RGB 图像
    
float* rgbData = new float[width * height * 3]; 
    
for(int i=0;i<width*height*3;i++) { 
      rgbData[i]=static_cast<float>(rand())/RAND_MAX;
}
    

QImage rgbImg= floatToRGBQImage(rgbData,width,height); 

rgbImg.save("rgb_image.png"); 

// 清理内存
delete[] grayData;
delete[] rgbData;

return 0;
}