qt&gdal

转自：http://blog.csdn.net/deirjie/article/details/37872743

使用需要自行配置Qt和GDAL路径。

    近期写了一个高光谱图像光谱曲线匹配的算法，想封装到软件当中方便观察效果，也便于做后期算法改进和实际应用，并且以后的算法可以直接集成上来。于是打算自己写一个基本的框架实现图像浏览的一些基本功能。在网上各种找，利用GDAL进行遥感图像显示的代码很多，但不是有问题就是写的不太清楚，不够简洁，并且大多基于MFC。经过几天的奋战，成功实现了利用Qt框架进行遥感图像显示的基本功能，于是分享出来，对自己是总结，并且希望对别人有所帮助。

开发环境：VS2010， Qt4.8.4，GDAL 1.10。

       说明：这里并没有Qt和GDAL，以及C++的一些基本的东西，这里只关注功能的实现做说明。如果对于Qt和GDAL有问题的，请参见其他资料。并不高深，高手绕道！

   重要说明：Qt发音叫做“Q特”，不是“Q替”，不要再读错了，官网有专门说明的！不要搞半天还说我发音有问题！！！  还有要写成Qt，不是QT，QT是QuickTime，不要混淆了。

         OK，先说说程序的功能：

• 能够打开图像，并读取图像元数据信息；
• 能够实现图像的正确显示，包括灰度图和RGB真彩色图的显示；
• 支持图像的缩放，漫游；

 

    功能非常简单基础，但是这当中很多细节是需要认真参透的。下面开始逐一进行讲解。(PS：UI的设计就不讲了。)

        1.程序简单设计

    其实这个程序的设计相对简单，除了显示图像的控件需要自定义，其他的都可以直接在Qt Designer里面拖放就行了。那就来说说图像显示控件吧。先看头文件：

//***********************************************************************
//Assembly         : ImgTest
//Author           : Jacory
//Created          : 07-10-2014
//
// LastModified By : Jacory
// LastModified On : 07-10-2014
//***********************************************************************
//<copyright file="MapCanvas.h" company="">
//     Copyright (c) . All rights reserved.
//</copyright>
//<summary>图像显示窗口类，负责图像的读取和显示等功能。</summary>
//***********************************************************************
#ifndefMAPCANVAS_H
#defineMAPCANVAS_H

#include<QtGui/QGraphicsView>
#include<QStandardItemModel>
#include<gdal_priv.h>


///<summary>
/// ClassMapCanvas.
///</summary>
classMapCanvas : public QGraphicsView
{
    Q_OBJECT

public:
    MapCanvas( QWidget *parent = 0 );
    ~MapCanvas();
    void ReadImg( const QString imgPath );
    void CloseCurrentImg();
    /// <summary>
    /// 返回图像元数据信息模型.
    /// </summary>
    ///<returns>图像元数据信息模型.</returns>
    QStandardItemModel* ImgMetaModel()
    {
        return imgMetaModel;
    };
    /// <summary>
    /// 设置图像元数据信息模型
    /// </summary>
    /// <paramname="model">图像元数据信息模型.</param>
    void SetMetaModel( QStandardItemModel*model )
    {
        this->imgMetaModel = model;
    };
    /// <summary>
    /// 返回文件列表数据模型
    /// </summary>
    ///<returns>文件列表数据模型.</returns>
    QStandardItemModel* FileListModel()
    {
        return fileListModel;
    };
    /// <summary>
    /// 设置fileListModel图像文件列表数据模型
    /// </summary>
    /// <paramname="model">文件列表数据模型.</param>
    void SetFileListModel( QStandardItemModel*model )
    {
        this->fileListModel = model;
    };

    QSize sizeHint() const;

publicslots:
    /// <summary>
    /// 放大图像
    /// </summary>
    void ZoomIn()
    {
        ScaleImg( 1.2 );
    };
    /// <summary>
    /// 缩小图像
    /// </summary>
    void ZoomOut()
    {
        ScaleImg( 0.8 );
    };

protected:
    void wheelEvent( QWheelEvent *event );
    void mousePressEvent( QMouseEvent *event );
    void mouseMoveEvent( QMouseEvent *event );
    void mouseReleaseEvent( QMouseEvent *event);

private:
    void ShowBand( GDALRasterBand* band );
    void ShowImg( QList<GDALRasterBand*>*imgBand );
    void ShowImgInfor( const QString filename);
    void ShowFileList( const QString filename);
    unsigned char* ImgSketch( float* buffer ,GDALRasterBand* currentBand, int size, double noValue );
    /// <summary>
    /// 图像缩放
    /// </summary>
    /// <paramname="factor">缩放因子</param>
    void ScaleImg( double factor )
    {
        m_scaleFactor *= factor;
        QMatrix matrix;
        matrix.scale( m_scaleFactor,m_scaleFactor );
        this->setMatrix( matrix );
    };

    /// <summary>
    /// 图像元数据模型
    /// </summary>
    QStandardItemModel *imgMetaModel;
    /// <summary>
    /// 图像数据集
    /// </summary>
    GDALDataset *poDataset;

    /// <summary>
    /// 文件列表数据模型
    /// </summary>
    QStandardItemModel *fileListModel;

    /// <summary>
    /// 缩放系数
    /// </summary>
    float m_scaleFactor;

    /// <summary>
    /// 判断是否显示RGB彩色图像
    /// </summary>
    bool m_showColor;

    /// <summary>
    /// 上一个鼠标事件触发时鼠标的位置
    /// </summary>
    QPoint lastEventCursorPos;
};

#endif //MAPCANVAS_H

        mapView类集成自QGraphicsView类。代码里面对于成员变量和inline函数我已经写了非常详细的注释了，这里再说明一下几个函数的功能。void ReadImg( const QStringimgPath )是读图函数，通过UI来调用，它并不直接读取图像，而是将相应的参数传递给ShowBand、ShowImg、ShowImgInfor、ShowFileList来分别实现各自相应的功能。void ShowBand( GDALRasterBand*band )函数是显示单个波段图像，当文件中的波段数不为3个时，没法组成RGB显示，默认就显示单个波段图像，事实上它只是把单个波段加入3次到QList<GDALRasterBand*>，然后把这个波段列表传递给ShowImg进行显示。而void ShowImg( QList<GDALRasterBand*>*imgBand )是显示图像的核心函数，它的内部通过判断m_showColor的真假来决定将要显示的图像的组织方式，并通过unsigned char* ImgSketch(float* buffer , GDALRasterBand* currentBand, int size, double noValue )函数对图像波段进行拉伸，详细说明放在后面，这里只要知道是什么用就行了。void ShowImgInfor( constQString filename )、  void ShowFileList( const QString filename )这两个函数，分别是为了读取并显示图像文件的元数据信息和文件结构树，分别位于主界面的两侧。void CloseCurrentImg()这个函数很简单，就是关闭当前图像。void wheelEvent( QWheelEvent*event )、void mousePressEvent(QMouseEvent *event )、void mouseMoveEvent(QMouseEvent *event )、void mouseReleaseEvent(QMouseEvent *event )这4个函数都是重写的鼠标事件，具体后面再讨论。

看完了头文件，已经基本知道这个MapView是一个什么样的对象了，事实上所有与图像显示有关的操作都属于这个类的职责，自然功能函数也就应该写在这个类里。下面我们根据需求来逐步实现这些功能函数。

 

      2.打开图像并读取图像元数据信息；

   这一步先不显示图像，只要求能够正确打开图像并读取出图像的基本信息即可。主要的函数就是ShowImgInfor和ShowFileList这两个函数，来看代码：

///<summary>
///显示图像基本信息
///</summary>
///<param name="filename">文件名</param>
voidMapCanvas::ShowImgInfor( const QString filename )
{
    if ( filename == "" || poDataset== NULL )
    {
        return;
    }
    int row = 0; // 用来记录数据模型的行号

    // 图像的格式信息
    imgMetaModel->setItem( row, 0, newQStandardItem( tr( "Description" ) ) );
    imgMetaModel->setItem( row++, 1, newQStandardItem( poDataset->GetDriver()->GetDescription() ) );
    imgMetaModel->setItem( row, 0, newQStandardItem( tr( "Meta Infor" ) ) );
    imgMetaModel->setItem( row++, 1, newQStandardItem( poDataset->GetDriver()->GetMetadataItem( GDAL_DMD_LONGNAME) ) ) ;
    imgMetaModel->setItem( row, 0, newQStandardItem( tr( "Data Type" ) ) );
    imgMetaModel->setItem( row++, 1, newQStandardItem( GDALGetDataTypeName( ( poDataset->GetRasterBand( 1)->GetRasterDataType() ) ) ) );

    // 图像的大小和波段个数
    imgMetaModel->setItem( row, 0, newQStandardItem( tr( "X Size" ) ) );
    imgMetaModel->setItem( row++, 1, newQStandardItem( QString::number( poDataset->GetRasterXSize() ) ) );
    imgMetaModel->setItem( row, 0, newQStandardItem( tr( "Y Size" ) ) );
    imgMetaModel->setItem( row++, 1, newQStandardItem( QString::number( poDataset->GetRasterYSize() ) ) );
    imgMetaModel->setItem( row, 0, newQStandardItem( tr( "Band Count" ) ) );
    imgMetaModel->setItem( row++, 1, newQStandardItem( QString::number( poDataset->GetRasterCount() ) ) );

    // 图像的投影信息
    imgMetaModel->setItem( row, 0, newQStandardItem( tr( "Projection" ) ) );
    imgMetaModel->setItem( row++, 1, newQStandardItem( poDataset->GetProjectionRef() ) );

    // 图像的坐标和分辨率信息
    double adfGeoTransform[6];
    QString origin = "";
    QString pixelSize = "";
    if( poDataset->GetGeoTransform(adfGeoTransform ) == CE_None )
    {
        origin = QString::number(adfGeoTransform[0] ) + ", " + QString::number( adfGeoTransform[3] );
        pixelSize = QString::number(adfGeoTransform[1] ) + ", " + QString::number( adfGeoTransform[5] );
    }
    imgMetaModel->setItem( row, 0, newQStandardItem( tr( "Origin" ) ) );
    imgMetaModel->setItem( row++, 1, newQStandardItem( origin ) );
    imgMetaModel->setItem( row, 0, newQStandardItem( tr( "Pixel Size" ) ) );
    imgMetaModel->setItem( row++, 1, newQStandardItem( pixelSize ) );
}

///<summary>
///显示文件结构树
///</summary>
///<param name="filename">文件名</param>
voidMapCanvas::ShowFileList( const QString filename )
{
    if ( filename == "" || poDataset== NULL )
    {
        return;
    }
    QFileInfo fileInfo( filename );
    QStandardItem *rootItem = newQStandardItem( fileInfo.fileName() );
    for ( int i = 0; i <poDataset->GetRasterCount(); i++ )
    {
        QStandardItem *childItem = newQStandardItem( tr( "Band %1" ).arg( i + 1 ) );
        rootItem->setChild( i, childItem );
    }
    fileListModel->setItem( 0, rootItem );
}

其实很简单，就是用了Qt的Model-View模型，显示的控件分别是QTableView和QTreeView，数据模型分别是这里的imgMetaModel和fileListModel，只用将模型适当的初始化并绑定到显示控件上就行了。以上初始化比较简单，这里就不啰嗦了，都是一些基本的GDAL数据模型和Qt的东西。

 

        3.实现图像的显示功能

    这一步就是核心了，我打算通过函数调用的顺序来逐步讲解这里牵涉到的功能函数。首先是UI选择文件后，将文件路径传递给ReadImg函数进行文件读取，那我们先来看看ReadImg函数的实现。

///<summary>
/// 读取图像文件
///</summary>
///<param name="imgPath">图像文件</param>
voidMapCanvas::ReadImg( const QString imgPath )
{
    GDALAllRegister();
    CPLSetConfigOption("GDAL_FILENAME_IS_UTF8", "NO" );
    poDataset = ( GDALDataset* )GDALOpen(imgPath.toStdString().c_str(), GA_ReadOnly );
    if ( poDataset == NULL )
    {
        QMessageBox::critical( this, tr("Error!" ), tr( "Can not open file %1" ).arg( imgPath ) );
        return;
    }
    ShowFileList( imgPath );
    ShowImgInfor( imgPath );
    // 如果图像文件并非三个波段，则默认只显示第一波段灰度图像
    if ( poDataset->GetRasterCount() != 3 )
    {
        m_showColor = false;
        ShowBand( poDataset->GetRasterBand(1 ) );
    }
    // 如果图像正好三个波段，则默认以RGB的顺序显示彩色图
    else
    {
        m_showColor = true;
        QList<GDALRasterBand*> bandList;
        bandList.append(poDataset->GetRasterBand( 1 ) );
        bandList.append(poDataset->GetRasterBand( 2 ) );
        bandList.append(poDataset->GetRasterBand( 3 ) );
        ShowImg( &bandList );
    }
    GDALClose( poDataset );
}

逻辑很清晰，调用ShowFileList和ShowImgInfor初始化两个model以供相应的显示View控件来显示。再判断如果图像文件波段数不为3，则默认显示第一波段灰度图像，否则显示彩色图。

再来看ShowBand函数：

///<summary>
///显示单波段图像
///</summary>
///<param name="band">图像波段</param>
voidMapCanvas::ShowBand( GDALRasterBand* band )
{
    if ( band == NULL )
    {
        return;
    }

    QList<GDALRasterBand*> myBand;
    myBand.append( band );
    myBand.append( band );
    myBand.append( band );

    ShowImg( &myBand );

}

可以看到，它只是把单个波段用列表的方式存储了三次，传递给ShowImg来进行显示。那直接看ShowImg吧。

///<summary>
/// 显示图像
///</summary>
///<param name="imgBand">图像波段</param>
voidMapCanvas::ShowImg( QList<GDALRasterBand*> *imgBand )
{
    if ( imgBand->size() != 3 )
    {
        return;
    }

    int imgWidth = imgBand->at( 0)->GetXSize();
    int imgHeight = imgBand->at( 0)->GetYSize();

    m_scaleFactor = this->height() * 1.0 /imgHeight;

    int iScaleWidth = ( int )( imgWidth *m_scaleFactor - 1 );
    int iScaleHeight = ( int )( imgHeight *m_scaleFactor - 1 );

    GDALDataType dataType = imgBand->at( 0)->GetRasterDataType();

    // 首先分别读取RGB三个波段
    float* rBand = new float[iScaleWidth *iScaleHeight];
    float* gBand = new float[iScaleWidth *iScaleHeight];
    float* bBand = new float[iScaleWidth *iScaleHeight];

    unsigned char *rBandUC, *gBandUC, *bBandUC;

    // 根据是否显示彩色图像，判断RGB三个波段的组成方式，并分别读取
    if ( m_showColor == true )
    {
        imgBand->at( 0 )->RasterIO(GF_Read, 0, 0, imgWidth, imgHeight, rBand , iScaleWidth, iScaleHeight,GDT_Float32, 0, 0 );
        imgBand->at( 1 )->RasterIO(GF_Read, 0, 0, imgWidth, imgHeight, gBand, iScaleWidth, iScaleHeight,GDT_Float32, 0, 0 );
        imgBand->at( 2 )->RasterIO(GF_Read, 0, 0, imgWidth, imgHeight, bBand, iScaleWidth, iScaleHeight,GDT_Float32, 0, 0 );

        // 分别拉伸每个波段并将Float转换为unsigned char
        rBandUC = ImgSketch( rBand,imgBand->at( 0 ), iScaleWidth * iScaleHeight, imgBand->at( 0)->GetNoDataValue() );
        gBandUC = ImgSketch( gBand,imgBand->at( 1 ), iScaleWidth * iScaleHeight, imgBand->at( 1)->GetNoDataValue() );
        bBandUC = ImgSketch( bBand,imgBand->at( 2 ), iScaleWidth * iScaleHeight, imgBand->at( 2)->GetNoDataValue() );
    }
    else
    {
        imgBand->at( 0 )->RasterIO(GF_Read, 0, 0, imgWidth, imgHeight, rBand , iScaleWidth, iScaleHeight,GDT_Float32, 0, 0 );

        rBandUC = ImgSketch( rBand,imgBand->at( 0 ), iScaleWidth * iScaleHeight, imgBand->at( 0)->GetNoDataValue() );
        gBandUC = rBandUC;
        bBandUC = rBandUC;
    }

    // 将三个波段组合起来
    int bytePerLine = ( iScaleWidth * 24 + 31 )/ 8;
    unsigned char* allBandUC = new unsignedchar[bytePerLine * iScaleHeight * 3];
    for( int h = 0; h < iScaleHeight; h++ )
    {
        for( int w = 0; w < iScaleWidth; w++)
        {
            allBandUC[h * bytePerLine + w * 3 +0] = rBandUC[h * iScaleWidth + w];
            allBandUC[h * bytePerLine + w * 3 +1] = gBandUC[h * iScaleWidth + w];
            allBandUC[h * bytePerLine + w * 3 +2] = bBandUC[h * iScaleWidth + w];
        }
    }

    // 构造图像并显示
    QGraphicsPixmapItem *imgItem = newQGraphicsPixmapItem(  QPixmap::fromImage(QImage( allBandUC, iScaleWidth, iScaleHeight, bytePerLine,QImage::Format_RGB888  ) ) );
    QGraphicsScene *myScene = newQGraphicsScene();
    myScene->addItem( imgItem );
    this->setScene( myScene );
}

这个函数的核心在于构造QGraphicsPixmapItem，也就是这里的imgItem这个对象，之前的代码都是为了正确构造这个对象而做的工作。这里的imgWidth和imgHeight分别是图像文件的宽和高，而iScaleWidth和iScaleHeight是缩放后的宽和高，缩放的条件正是显示窗口的大小。读取图像时，采用分别读取RGB三个波段，最后再组合起来的方式，读取波段都是采用RasterIO函数，采用float的数组来进行读取避免一些格式问题带来的精度截断，读取之后，通过ImgSketch函数进行图像拉伸，把图像值拉伸到0~255灰度级。

///<summary>
/// 图像线性拉伸
///</summary>
///<param name="buffer">图像缓存</param>
///<param name="currentBand">当前波段</param>
///<param name="size">The size.</param>
///<param name="noValue">图像中的异常值</param>
///<returns>经过拉伸的8位图像缓存</returns>
unsignedchar* MapCanvas::ImgSketch( float* buffer , GDALRasterBand* currentBand, intbandSize, double noValue )
{
    unsigned char* resBuffer = new unsignedchar[bandSize];
    double max, min;
    double minmax[2];

    currentBand->ComputeRasterMinMax( 1,minmax );
    min = minmax[0];
    max = minmax[1];
    if( min <= noValue && noValue<= max )
    {
        min = 0;
    }
    for ( int i = 0; i < bandSize; i++ )
    {
        if ( buffer[i] > max )
        {
            resBuffer[i] = 255;
        }
        else if ( buffer[i] <= max&& buffer[i] >= min )
        {
            resBuffer[i] =static_cast<uchar>( 255 - 255 * ( max - buffer[i] ) / ( max - min ) );
        }
        else
        {
            resBuffer[i] = 0;
        }
    }

    return resBuffer;
}

这里的转换核心为

resBuffer[i] = static_cast<uchar>( 255 - 255 * ( max -buffer[i] ) / ( max - min ) );

转换公式也可以变为((buffer[i]-min)/(max-min))*255，是一样的，当然还有其他的拉伸方法，可以自行使用。

拉伸之后组合波段，需要用到int bytePerLine = (iScaleWidth * 24 + 31 ) / 8;这句，因为要显示的图像为8位的，需要把图像值变换为unsigned char型，这个工作在imgSketch里面已经做了，但是由于要将3个波段组织正确，还需要字节对齐，也就是这里的bytePerLine，这个是byte型表示的这个图像的宽度，当然是包含了RGB三个波段。然后就是将每个波段的值一一对齐放入allBandUC这个数组中，最后根据这个组合后的数组构QImage，QPixmap以及QGraphicsPixmapItem，最后把这个Item加入到scene里面用MapView显示就行了。需要注意的是，这个过程看似简单，但里面有很多细节的地方需要认真对待，仔细思考，不然很容易得不到正确的结果。

 

4.实现图像的漫游功能

这一部分我在网上看了很多代码，大部分人都是从底层原理开始又来写一遍，但其实对于漫游的操作其实Qt在后台已经支持的非常好了，我们的MapView继承自QGraphicsView，这个父类已经把很多操作封装到位了，我们只要合理组织就好了，没有必要再去写一遍。那就很简单了，来分别看看实现代码吧。

///<summary>
///鼠标滚轮事件，实现图像缩放
///</summary>
///<param name="event">滚轮事件</param>
voidMapCanvas::wheelEvent( QWheelEvent *event )
{
    // 滚轮向上滑动，放大图像
    if ( event->delta() > 0 )
    {
        ZoomIn();
    }
    // 滚轮向下滑动，缩小图像
    if ( event->delta() < 0 )
    {
        ZoomOut();
    }
}

///<summary>
///鼠标按键按下事件
///</summary>
///<param name="event">鼠标事件.</param>
voidMapCanvas::mousePressEvent( QMouseEvent *event )
{
    // 滚轮键按下，平移图像
    if ( event->button() == Qt::MidButton )
    {
        this->setDragMode(QGraphicsView::ScrollHandDrag );
        this->setInteractive( false );
        lastEventCursorPos = event->pos();
    }
}

///<summary>
/// 鼠标移动事件
///</summary>
///<param name="event">鼠标事件</param>
voidMapCanvas::mouseMoveEvent( QMouseEvent *event )
{
    if ( this->dragMode() ==QGraphicsView::ScrollHandDrag )
    {
        QPoint delta = ( event->pos() -lastEventCursorPos ) / 10;
       this->horizontalScrollBar()->setValue(this->horizontalScrollBar()->value() + ( isRightToLeft() ? delta.x() :-delta.x() ) );
       this->verticalScrollBar()->setValue(this->verticalScrollBar()->value() - delta.y() );
        this->viewport()->setCursor(Qt::ClosedHandCursor );
    }

}

///<summary>
///鼠标按键释放事件
///</summary>
///<param name="event">鼠标事件</param>
voidMapCanvas::mouseReleaseEvent( QMouseEvent *event )
{
    if ( event->button() == Qt::MidButton )
    {
        this->setDragMode(QGraphicsView::NoDrag );
    }
}

重写鼠标事件，这里需要说明的是，漫游的操作Qt采用鼠标左键按下并移动来实现视图平移，但是我这里强制定义为了鼠标中键按下，所以还比较麻烦相对来说，如果你只想用默认的漫游操作，直接把this->setDragMode(QGraphicsView::NoDrag );这句加入到你的构造函数中就行了，非常简单。事实上我这个平移函数有问题，我后面再解释。这里我们先关注放大缩小功能，这部分代码我是通过inline的方式添加到头文件中的。

///<summary>
    /// 放大图像
    /// </summary>
    void ZoomIn()
    {
        ScaleImg( 1.2 );
    };
    /// <summary>
    /// 缩小图像
    /// </summary>
    void ZoomOut()
    {
        ScaleImg( 0.8 );
    };

///<summary>
    /// 图像缩放
    /// </summary>
    /// <paramname="factor">缩放因子</param>
    void ScaleImg( double factor )
    {
        m_scaleFactor *= factor;
        QMatrix matrix;
        matrix.scale( m_scaleFactor,m_scaleFactor );
        this->setMatrix( matrix );
    };

图像缩放也非常简单，让你的QGraphicsView去处理细节，不要自己又去从底层计算然后写一大堆代码，还不一定就比Qt默认的方式效率高。

做到这里基本就搞定了，我刚刚说的平移函数那里，实际上我在mouseMoveEvent里采用计算当前坐标点和事件发生的坐标点差值的方式，然后去设置水平和竖直滑动条的值来实现平移

QPointdelta = ( event->pos() - lastEventCursorPos ) / 10;
       this->horizontalScrollBar()->setValue(this->horizontalScrollBar()->value() + ( isRightToLeft() ? delta.x() :-delta.x() ) );
       this->verticalScrollBar()->setValue(this->verticalScrollBar()->value() - delta.y() );

我这样写后，我的平移实际上并不是非常流畅，而是有点像浏览网页时按住鼠标中键往下浏览的效果，但实际上很多GIS和遥感软件的平移方式都不是这样的。这里应该是我对Qt还有些不熟悉，也希望大家知道的话告诉我一下，我认为Qt应该是支持的。

好了，差不多了，其他的功能我也还在完善，先写到这里吧，希望对大家有所帮助。最后还是把MapView的实现文件整体粘上来。VS2010的工程文件我放到CSDN的资源里面大家可以下载。

#include"MapCanvas.h"
#include<QtGui/QMessageBox>
#include<QtCore/QFileInfo>
#include<QtGui/QImage>
#include<QtGui/QPixmap>
#include<QtGui/QGraphicsPixmapItem>
#include<QtGui/QMatrix>
#include<QtGui/QWheelEvent>
#include<QtGui/QScrollBar>

MapCanvas::MapCanvas(QWidget *parent /*= 0 */ )
    : QGraphicsView( parent )
{
    poDataset = NULL;
    m_scaleFactor = 1.0;
    m_showColor = true;
    imgMetaModel = new QStandardItemModel;
    imgMetaModel->setColumnCount( 2 );
    fileListModel = new QStandardItemModel;
    QSizePolicy policy( QSizePolicy::Preferred,QSizePolicy::Preferred );
    this->setSizePolicy( policy );
}

///<summary>
///Finalizes an instance of the <see cref="MapCanvas" /> class.
///</summary>
MapCanvas::~MapCanvas()
{

}

///<summary>
/// 读取图像文件
///</summary>
///<param name="imgPath">图像文件</param>
voidMapCanvas::ReadImg( const QString imgPath )
{
    GDALAllRegister();
    CPLSetConfigOption("GDAL_FILENAME_IS_UTF8", "NO" );
    poDataset = ( GDALDataset* )GDALOpen(imgPath.toStdString().c_str(), GA_ReadOnly );
    if ( poDataset == NULL )
    {
        QMessageBox::critical( this, tr("Error!" ), tr( "Can not open file %1" ).arg( imgPath ) );
        return;
    }
    ShowFileList( imgPath );
    ShowImgInfor( imgPath );
    // 如果图像文件并非三个波段，则默认只显示第一波段灰度图像
    if ( poDataset->GetRasterCount() != 3 )
    {
        m_showColor = false;
        ShowBand( poDataset->GetRasterBand(1 ) );
    }
    // 如果图像正好三个波段，则默认以RGB的顺序显示彩色图
    else
    {
        m_showColor = true;
        QList<GDALRasterBand*> bandList;
        bandList.append(poDataset->GetRasterBand( 1 ) );
        bandList.append(poDataset->GetRasterBand( 2 ) );
        bandList.append(poDataset->GetRasterBand( 3 ) );
        ShowImg( &bandList );
    }
    GDALClose( poDataset );
}

///<summary>
///关闭当前图像文件
///</summary>
voidMapCanvas::CloseCurrentImg()
{
    poDataset = NULL;
    imgMetaModel->clear();
    fileListModel->clear();
}

///<summary>
///显示单波段图像
///</summary>
///<param name="band">图像波段</param>
voidMapCanvas::ShowBand( GDALRasterBand* band )
{
    if ( band == NULL )
    {
        return;
    }

    QList<GDALRasterBand*> myBand;
    myBand.append( band );
    myBand.append( band );
    myBand.append( band );

    ShowImg( &myBand );

}

///<summary>
/// 显示图像
///</summary>
///<param name="imgBand">图像波段</param>
voidMapCanvas::ShowImg( QList<GDALRasterBand*> *imgBand )
{
    if ( imgBand->size() != 3 )
    {
        return;
    }

    int imgWidth = imgBand->at( 0)->GetXSize();
    int imgHeight = imgBand->at( 0)->GetYSize();

    m_scaleFactor = this->height() * 1.0 /imgHeight;

    int iScaleWidth = ( int )( imgWidth *m_scaleFactor - 1 );
    int iScaleHeight = ( int )( imgHeight *m_scaleFactor - 1 );

    GDALDataType dataType = imgBand->at( 0)->GetRasterDataType();

    // 首先分别读取RGB三个波段
    float* rBand = new float[iScaleWidth *iScaleHeight];
    float* gBand = new float[iScaleWidth *iScaleHeight];
    float* bBand = new float[iScaleWidth *iScaleHeight];

    unsigned char *rBandUC, *gBandUC, *bBandUC;

    // 根据是否显示彩色图像，判断RGB三个波段的组成方式，并分别读取
    if ( m_showColor == true )
    {
        imgBand->at( 0 )->RasterIO(GF_Read, 0, 0, imgWidth, imgHeight, rBand , iScaleWidth, iScaleHeight,GDT_Float32, 0, 0 );
        imgBand->at( 1 )->RasterIO(GF_Read, 0, 0, imgWidth, imgHeight, gBand, iScaleWidth, iScaleHeight,GDT_Float32, 0, 0 );
        imgBand->at( 2 )->RasterIO(GF_Read, 0, 0, imgWidth, imgHeight, bBand, iScaleWidth, iScaleHeight,GDT_Float32, 0, 0 );

        // 分别拉伸每个波段并将Float转换为unsigned char
        rBandUC = ImgSketch( rBand,imgBand->at( 0 ), iScaleWidth * iScaleHeight, imgBand->at( 0)->GetNoDataValue() );
        gBandUC = ImgSketch( gBand,imgBand->at( 1 ), iScaleWidth * iScaleHeight, imgBand->at( 1)->GetNoDataValue() );
        bBandUC = ImgSketch( bBand,imgBand->at( 2 ), iScaleWidth * iScaleHeight, imgBand->at( 2)->GetNoDataValue() );
    }
    else
    {
        imgBand->at( 0 )->RasterIO(GF_Read, 0, 0, imgWidth, imgHeight, rBand , iScaleWidth, iScaleHeight,GDT_Float32, 0, 0 );

        rBandUC = ImgSketch( rBand,imgBand->at( 0 ), iScaleWidth * iScaleHeight, imgBand->at( 0)->GetNoDataValue() );
        gBandUC = rBandUC;
        bBandUC = rBandUC;
    }

    // 将三个波段组合起来
    int bytePerLine = ( iScaleWidth * 24 + 31 )/ 8;
    unsigned char* allBandUC = new unsignedchar[bytePerLine * iScaleHeight * 3];
    for( int h = 0; h < iScaleHeight; h++ )
    {
        for( int w = 0; w < iScaleWidth; w++)
        {
            allBandUC[h * bytePerLine + w * 3 +0] = rBandUC[h * iScaleWidth + w];
            allBandUC[h * bytePerLine + w * 3 +1] = gBandUC[h * iScaleWidth + w];
            allBandUC[h * bytePerLine + w * 3 +2] = bBandUC[h * iScaleWidth + w];
        }
    }

    // 构造图像并显示
    QGraphicsPixmapItem *imgItem = newQGraphicsPixmapItem(  QPixmap::fromImage(QImage( allBandUC, iScaleWidth, iScaleHeight, bytePerLine,QImage::Format_RGB888  ) ) );
    QGraphicsScene *myScene = newQGraphicsScene();
    myScene->addItem( imgItem );
    this->setScene( myScene );
}

///<summary>
///显示图像基本信息
///</summary>
///<param name="filename">文件名</param>
voidMapCanvas::ShowImgInfor( const QString filename )
{
    if ( filename == "" || poDataset== NULL )
    {
        return;
    }
    int row = 0; // 用来记录数据模型的行号

    // 图像的格式信息
    imgMetaModel->setItem( row, 0, newQStandardItem( tr( "Description" ) ) );
    imgMetaModel->setItem( row++, 1, newQStandardItem( poDataset->GetDriver()->GetDescription() ) );
    imgMetaModel->setItem( row, 0, newQStandardItem( tr( "Meta Infor" ) ) );
    imgMetaModel->setItem( row++, 1, newQStandardItem( poDataset->GetDriver()->GetMetadataItem( GDAL_DMD_LONGNAME) ) ) ;
    imgMetaModel->setItem( row, 0, newQStandardItem( tr( "Data Type" ) ) );
    imgMetaModel->setItem( row++, 1, newQStandardItem( GDALGetDataTypeName( ( poDataset->GetRasterBand( 1)->GetRasterDataType() ) ) ) );

    // 图像的大小和波段个数
    imgMetaModel->setItem( row, 0, newQStandardItem( tr( "X Size" ) ) );
    imgMetaModel->setItem( row++, 1, newQStandardItem( QString::number( poDataset->GetRasterXSize() ) ) );
    imgMetaModel->setItem( row, 0, newQStandardItem( tr( "Y Size" ) ) );
    imgMetaModel->setItem( row++, 1, newQStandardItem( QString::number( poDataset->GetRasterYSize() ) ) );
    imgMetaModel->setItem( row, 0, newQStandardItem( tr( "Band Count" ) ) );
    imgMetaModel->setItem( row++, 1, newQStandardItem( QString::number( poDataset->GetRasterCount() ) ) );

    // 图像的投影信息
    imgMetaModel->setItem( row, 0, newQStandardItem( tr( "Projection" ) ) );
    imgMetaModel->setItem( row++, 1, newQStandardItem( poDataset->GetProjectionRef() ) );

    // 图像的坐标和分辨率信息
    double adfGeoTransform[6];
    QString origin = "";
    QString pixelSize = "";
    if( poDataset->GetGeoTransform(adfGeoTransform ) == CE_None )
    {
        origin = QString::number(adfGeoTransform[0] ) + ", " + QString::number( adfGeoTransform[3] );
        pixelSize = QString::number(adfGeoTransform[1] ) + ", " + QString::number( adfGeoTransform[5] );
    }
    imgMetaModel->setItem( row, 0, newQStandardItem( tr( "Origin" ) ) );
    imgMetaModel->setItem( row++, 1, newQStandardItem( origin ) );
    imgMetaModel->setItem( row, 0, newQStandardItem( tr( "Pixel Size" ) ) );
    imgMetaModel->setItem( row++, 1, newQStandardItem( pixelSize ) );
}

///<summary>
///显示文件结构树
///</summary>
///<param name="filename">文件名</param>
voidMapCanvas::ShowFileList( const QString filename )
{
    if ( filename == "" || poDataset== NULL )
    {
        return;
    }
    QFileInfo fileInfo( filename );
    QStandardItem *rootItem = newQStandardItem( fileInfo.fileName() );
    for ( int i = 0; i <poDataset->GetRasterCount(); i++ )
    {
        QStandardItem *childItem = newQStandardItem( tr( "Band %1" ).arg( i + 1 ) );
        rootItem->setChild( i, childItem );
    }
    fileListModel->setItem( 0, rootItem );
}

///<summary>
/// 图像线性拉伸
///</summary>
///<param name="buffer">图像缓存</param>
///<param name="currentBand">当前波段</param>
///<param name="size">The size.</param>
///<param name="noValue">图像中的异常值</param>
///<returns>经过拉伸的8位图像缓存</returns>
unsignedchar* MapCanvas::ImgSketch( float* buffer , GDALRasterBand* currentBand, intbandSize, double noValue )
{
    unsigned char* resBuffer = new unsignedchar[bandSize];
    double max, min;
    double minmax[2];


    currentBand->ComputeRasterMinMax( 1,minmax );
    min = minmax[0];
    max = minmax[1];
    if( min <= noValue && noValue<= max )
    {
        min = 0;
    }
    for ( int i = 0; i < bandSize; i++ )
    {
        if ( buffer[i] > max )
        {
            resBuffer[i] = 255;
        }
        else if ( buffer[i] <= max&& buffer[i] >= min )
        {
            resBuffer[i] =static_cast<uchar>( 255 - 255 * ( max - buffer[i] ) / ( max - min ) );
        }
        else
        {
            resBuffer[i] = 0;
        }
    }

    return resBuffer;
}

///<summary>
/// 控件大小
///</summary>
///<returns>QSize.</returns>
QSizeMapCanvas::sizeHint() const
{
    return QSize( 1024, 768 );
}

///<summary>
///鼠标滚轮事件，实现图像缩放
///</summary>
///<param name="event">滚轮事件</param>
voidMapCanvas::wheelEvent( QWheelEvent *event )
{
    // 滚轮向上滑动，放大图像
    if ( event->delta() > 0 )
    {
        ZoomIn();
    }
    // 滚轮向下滑动，缩小图像
    if ( event->delta() < 0 )
    {
        ZoomOut();
    }
}

///<summary>
///鼠标按键按下事件
///</summary>
///<param name="event">鼠标事件.</param>
voidMapCanvas::mousePressEvent( QMouseEvent *event )
{
    // 滚轮键按下，平移图像
    if ( event->button() == Qt::MidButton )
    {
        this->setDragMode(QGraphicsView::ScrollHandDrag );
        this->setInteractive( false );
        lastEventCursorPos = event->pos();
    }
}

///<summary>
/// 鼠标移动事件
///</summary>
///<param name="event">鼠标事件</param>
voidMapCanvas::mouseMoveEvent( QMouseEvent *event )
{
    if ( this->dragMode() ==QGraphicsView::ScrollHandDrag )
    {
        QPoint delta = ( event->pos() -lastEventCursorPos ) / 10;
       this->horizontalScrollBar()->setValue(this->horizontalScrollBar()->value() + ( isRightToLeft() ? delta.x() :-delta.x() ) );
       this->verticalScrollBar()->setValue(this->verticalScrollBar()->value() - delta.y() );
        this->viewport()->setCursor(Qt::ClosedHandCursor );
    }

}

///<summary>
///鼠标按键释放事件
///</summary>
///<param name="event">鼠标事件</param>
voidMapCanvas::mouseReleaseEvent( QMouseEvent *event )
{
    if ( event->button() == Qt::MidButton )
    {
        this->setDragMode(QGraphicsView::NoDrag );
    }
}