从GDALDataSet创建RBG Color QImage

时间:2015-11-19 15:15:59

标签: c++ qt gdal

我使用GDAL来阅读一些图像文件,并希望使用Qt显示它们。到目前为止,我设法为QImage中的每个GDALRasterBand创建了灰度GDALDataSet,但我不知道如何创建单个RGB图像。

这就是我所做的:

#include <gdal_priv.h>
#include <QtGui\QImage>

int main(int argc, char *argv[])
{
    GDALAllRegister();

    GDALDataset* dataset = static_cast<GDALDataset*>(GDALOpen("path_to_some_image.tif", GA_ReadOnly));
    int size_out = 200;

    for (int i = 1; i <= 3; ++i)
    { 
        GDALRasterBand* band = dataset->GetRasterBand(i);

        std::vector<uchar> band_data(size_out * size_out);
        band->RasterIO(GF_Read, 0, 0, size_out, size_out, band_data.data(), size_out, size_out, GDT_Byte, 0, 0);

        QImage band_image(band_data.data(), size_out, size_out, QImage::Format_Grayscale8);
        band_image.save(QString("C:\\band_%1.png").arg(i));
    }

    return 0;
}

如何读取数据,以便创建单个RGB QImage

1 个答案:

答案 0 :(得分:2)

你快到了。第一项是QImage采用格式标志的缓冲区。因此,该格式标志需要匹配您从文件加载的图像,否则需要进行转换。以下示例假定为4通道图像。

QImage格式标记文档:http://doc.qt.io/qt-5/qimage.html#Format-enum

下一个组件是GDAL的RasterIO方法分别处理每个波段,这意味着您必须单独交错像素或者失去逐个加载光栅所带来的效率。

RasterIO:http://gdal.org/classGDALRasterBand.html#a30786c81246455321e96d73047b8edf1

我喜欢OpenCV的merge方法。只需为每个波段创建一个灰度图像,并将它们merge组合在一起。

OpenCV合并:http://docs.opencv.org/3.0.0/d2/de8/group__core__array.html#ga61f2f2bde4a0a0154b2333ea504fab1d

例如,给定RGBA GeoTiff,

// OpenCV Libraries
#include <opencv2/core/core.hpp>
#include <opencv2/imgproc/imgproc.hpp>

// GDAL Libraries
#include <gdal.h>

// QT Libraries
#include <QtGui\QImage>

using namespace cv;


int main( int argc, char* argv[] )
{

    // Initialize GDAL
    GDALAllRegister();

    // Load image
    GDALDataset* dataset = GDALOpen("path_to_some_image.tif", GA_ReadOnly);

    // Get raster image size
    int rows = dataset->GetRasterYSize();
    int cols = dataset->GetRasterXSize();
    int channels = dataset->GetRasterCount();

    // Create each separate image as grayscale
    std::vector<cv::Mat> image_list(channels, cv::Mat( rows, cols, CV_8UC1 ));
    cv::Mat output_image;

    // Iterate over each channel
    for (int i = 1; i <= channels; ++i)
    { 
        // Fetch the band
        GDALRasterBand* band = dataset->GetRasterBand(i);

        // Read the data
        band->RasterIO( GF_Read, 0, 0, 
                        cols, rows, 
                        image_list[i-1].data, 
                        cols, rows, 
                        GDT_Byte, 0, 0);

    }

    // Merge images
    cv::merge( image_list, output_image );

    // Create the QImage
    QImage qt_image( band_data.data(), 
                     cols, 
                     rows,
                     QImage::Format_RGBA8888);

    // Do some stuff with the image

    return 0;
 }