GDAL输出影像

[cpp] view
plain copy

//输出影像  

/** 

* @brief    ReadImageToBuff. 

* 把数据从内存中输出到文件中,默认保存为tif格式。 

* @created: 2012/03/25 

* @author:  Zhengwen.Fu 

* @param    szPath      输出路径 

* @param    img         数据块内存块指针 

* @param    nSamples    数据块内存块代表的影像的列数 

* @param    nLines      数据块内存块代表的影像的行数 

* @param    nBandCount  数据块内存块代表的影像的波段数 

* @param    pszFormat   GDAL文件格式标识串，默认是"GTiff"，表示存储为tiff格式，其它格式还有：HDF，ENVI等，具体可参考GDAL说明文档中的：GDAL Raster Formats专题,在index.html页面的Supported Formats里； 

* @exception     

* @return   true    成功 

* @return   false   失败 

* @note     注意szPath指定的文件后缀名要和pszFormat指定的文件格式要求的后缀名一致； 

* @remarks   

* @history: 1.Created by Zhengwen.Fu on 3 25th,2012.         

*/  

//template<typename T>  

bool SaveImageToFile(char *szPath,float* img,int nSamples,int nLines,int nBandCount,const char *pszFormat ="GTiff")  

{  

    GDALAllRegister();         //利用GDAL读取图片，先要进行注册  

    CPLSetConfigOption("GDAL_FILENAME_IS_UTF8","NO");   //设置支持中文路径  

  

    //准备保存图片  

    GDALDriver *poDriver=GetGDALDriverManager()->GetDriverByName(pszFormat);  

    if( poDriver == NULL )   

    {  

        return false;  

    }  

    char **papszOptions = NULL;   

    GDALDataset *WriteDataSet = poDriver->Create(szPath, nSamples,nLines,nBandCount,GDT_Float32,papszOptions);  

  

    if(WriteDataSet->RasterIO(GF_Write,0,0,nSamples,nLines,img,nSamples,nLines,GDT_Float32,nBandCount,NULL,0,0,0)==CE_Failure )  

    {  

        return false;  

    }  

    delete WriteDataSet;WriteDataSet=NULL;  

    return true;  

}  

用法示例：

[cpp] view
plain copy

float data[10*10*1];  

memset(data,0,100*sizeof(float));  

  

//输出为tiff格式  

SaveImageToFile("F:\\Work\\数据\\tif.tif",data,10,10,1,"GTiff");  

  

//HDF4  

SaveImageToFile("F:\\Work\\数据\\HDF4.hdf4",data,10,10,1,"HDF4");  

//不支持  

  

//HDF5  

SaveImageToFile("F:\\Work\\数据\\HDF5.hdf5",data,10,10,1,"HDF5");  

//不支持  

  

//ENVI标准影像  

SaveImageToFile("F:\\Work\\ENVI.img",data,10,10,1,"ENVI");  

//ENVI .hdr Labelled Raster采用头文件加主文件的方式保存，头文件以.hdr做后缀名，主文件的后缀可以任意，但通常都使用.img，这正是造成与Erdas Imagine混淆的原因。  

//文件名不能用hdr做后缀，因为写img时会写同名的hdr文件，如果用hdr则造成头文件和主文件是同一文件，造成读写异常  

  

SaveImageToFile("F:\\Work\\数据\\ENVI2.envi",data,10,10,1,"ENVI");  

//主文件可以使用除hdr之外的任何后缀做文件后缀名  

  

//Erdas Image文件  

SaveImageToFile("F:\\Work\\数据\\ErdasImagine.img",data,10,10,1,"HFA");  

//输出的文件仅有一个.img文件  

  

SaveImageToFile("F:\\Work\\数据\\ErdasImagine.eds",data,10,10,1,"HFA");  

//可以不使用.img做后缀，但是你必须知道你输出的这个文件时Erdas Image文件，否则可能用Envi等是无法打开的。