基于GDAL的栅格图像预处理
前言
栅格数据和矢量数据构成空间数据的主要来源,怎样以开源方式读取并处理这些空间数据?目前有多种开源支持包,这里只介绍GDAL包。GDAL包的优点是支持库简洁、支持栅格和矢量、与多种开发平台结合。OpenGis方式读取空间数据,有利于自己编写程序进行图像预处理和智能识别等等,比如:遥感影像的降噪、锐化;红外图像的林火识别;工厂监控视频识别等等。本文中利用GDAL包读取高程栅格DEM,并添加气象自动站点的数据,进行空间插值研究。
一、程序主要程序功能实现过程
第一步:读取栅格数据,包含坡向和DEM
第二步:读入站点信息数据
第三步:按照行列号读取栅格单元到内存,不考虑高程为0的单元
第三步:每一个坡向情况中,参考固定数目的气象站点。首先确定搜索范围,获取定量数目的监测站点。
第四步:在同一个计算窗口内,通过给各个因子赋权重,依据海拔高程回归关系与加权回归分析得到
温度递减率。
第五步:计算单个站点的温度预测值,然后计算所有站点的距离权重因子,根据因子大小确定综合
影响后的温度值
第四步:开辟新内存存储处理后的栅格数据,然后新建一个tiff格式的文件,把内存
数据导出到该文件中
二、代码示例
int _tmain(int argc, _TCHAR* argv[]) { /* DEM、坡向栅格数据的数据框大小 */ int XsizeDEM; int YsizeDEM; int XsizeAspect; int YsizeAspect; //double geoTransform[6]; //double Xp,Yp; /* DEM、坡向栅格单元对象的VALUE值 */ short int *pmemDEM; float *pmemAspect; float *pmemNew; //GDAL注册 GDALAllRegister(); /* 栅格单元的底图来源文件名:DEM/ASPECT */ const char *pszFileDEM; pszFileDEM="F:\\beijing_dem\\bj25_CopyRaster11.img"; const char *pszFileAspect; pszFileAspect="F:\\beijing_dem\\aspect_bj.tif"; /*读取站点信息*/ int n,m; float site[32][6]; FILE *fp; if((fp=fopen("F:\\beijing_dem\\site_36\\information.txt","r"))== NULL) { printf("cannot open this file\n"); exit(0); } for(n=0;n<32;n++) { for(m=0;m<6;m++) { fscanf(fp,"%f",&site[n][m]); } } for(n=0;n<32;n++) { for(m=0;m<6;m++) { printf("%f",site[n][m]);/*注意这里*/; printf(" "); } printf("\n"); //每输出一行,输出一个换行符 } fclose(fp); /* DEM/ASPECT的数据集读取器 */ GDALDataset *poDatasetDEM; GDALRasterBand *poBandDEM; GDALDataset *poDatasetAspect; GDALRasterBand *poBandAspect; /* 判断DEM/ASPECT文件是否存在,不存在错误提示 */ poDatasetDEM=(GDALDataset*)GDALOpen(pszFileDEM,GA_ReadOnly); if(poDatasetDEM==NULL) { printf("File: %s不能打开",pszFileDEM); return 0; } poDatasetAspect=(GDALDataset*)GDALOpen(pszFileAspect,GA_ReadOnly); if(poDatasetAspect==NULL) { printf("File: %s不能打开",pszFileAspect); return 0; } /* 判断DEM/ASPECT文件如果存在,就把文件输入到波段第一层 */ poBandDEM=poDatasetDEM->GetRasterBand(1); poBandAspect=poDatasetAspect->GetRasterBand(1); /* 被处理栅格单元对象的数据框大小的提取 */ XsizeDEM=poBandDEM->GetXSize(); YsizeDEM=poBandDEM->GetYSize(); XsizeAspect=poBandAspect->GetXSize(); YsizeAspect=poBandAspect->GetYSize(); /* 被处理栅格单元对象的内存开辟 */ pmemDEM=(short int *)CPLMalloc(sizeof(short int)*XsizeDEM*YsizeDEM); poBandDEM->RasterIO(GF_Read,0,0,XsizeDEM,YsizeDEM,pmemDEM,XsizeDEM,YsizeDEM,GDT_Int16,0,0); pmemAspect=(float*)CPLMalloc(sizeof(float)*XsizeAspect*YsizeAspect); poBandAspect->RasterIO(GF_Read,0,0,XsizeAspect,YsizeAspect,pmemAspect,XsizeAspect,YsizeAspect,GDT_Float32,0,0); /* 被处理栅格单元对象的类型提示 */ printf("Type is: %s\n",GDALGetDataTypeName(poBandDEM->GetRasterDataType())); //开辟新栅格内存空间 pmemNew=(float *)CPLMalloc(sizeof(float)*XsizeDEM*YsizeDEM); for(int i=0;i<YsizeDEM;i++) { for(int j=0;j<XsizeDEM;j++) { int flag=0; float H_value; float A_value; H_value=pmemDEM[i*XsizeDEM+j]; A_value=pmemAspect[i*XsizeDEM+j]; //单个栅格插值处理 //高程没有值的特殊情况 if(H_value==0) { pmemNew[i*XsizeDEM+j]=0; } else { //平地无坡向的情况 if(A_value==-1) { //处理站点和插值单元重合的情况,插值单元的值等于站点监测值 for(n=0;n<32;n++) { if(((site[n][4]-i)*(site[n][4]-i)+(site[n][5]-j)*(site[n][5]-j))==0) { float x1=site[n][0],x2=site[n][3]; pmemNew[i*XsizeDEM+j]=site[n][3]; printf("%f 平地站点数据: ",x1); printf("%f\n",x2); flag=1; } } if(flag==0) { pmemNew[i*XsizeDEM+j]=plain_calculate(site,H_value,i,j); //pmemNew[i*XsizeDEM+j]=3; } } else { //处理站点和插值单元重合的情况,插值单元的值等于站点监测值 for(n=0;n<32;n++) { if(((site[n][4]-i)*(site[n][4]-i)+(site[n][5]-j)*(site[n][5]-j))==0) { float x1=site[n][0],x2=site[n][3]; pmemNew[i*XsizeDEM+j]=site[n][3]; printf("%f 非平地站点数据: ",x1); printf("%f\n",x2); flag=1; } } if(flag==0) { //每一个栅格单元进行插值计算 pmemNew[i*XsizeDEM+j]=calculate(site,H_value,A_value,i,j); //pmemNew[i*XsizeDEM+j]=1; } } } //poDatasetDEM->GetGeoTransform( geoTransform); //Xp = geoTransform[0] +j*geoTransform[1]+i*geoTransform[2]; //Yp = geoTransform[3] + j*geoTransform[4] + i*geoTransform[5]; } } /** 创建新的空TIFF栅格文件 */ GDALAllRegister(); GDALDriver *poDriver; poDriver=GetGDALDriverManager()->GetDriverByName("GTiff");//AAIGrid(Arc/Info ASCII Grid) HFA (img no lim) if(poDriver==NULL) exit(1); GDALDataset *poDstDS; poDstDS=poDriver->Create("F:\\beijing_dem\\beijing_mix513.tiff",XsizeDEM,YsizeDEM,1,GDT_Float32,NULL); double trans[6]={219323.300357,100.00,0.00,4680250.0000,0.0,-100.000}; //如果图像不含地理坐标信息,默认返回值是:(0,1,0,0,0,1) //In a north up image, //左上角点坐标(padfGeoTransform[0],padfGeoTransform[3]); //padfGeoTransform[1]是像元宽度(影像在宽度上的分辨率); //padfGeoTransform[5]是像元高度(影像在高度上的分辨率); //如果影像是指北的,padfGeoTransform[2]和padfGeoTransform[4]这两个参数的值为0。 poDstDS->SetGeoTransform(trans); GDALClose((GDALDatasetH)poDstDS); /* //处理后的数据层保存 */ GDALDataset *poDatasetNew; poDatasetNew=(GDALDataset*)GDALOpen("F:\\beijing_dem\\beijing_mix513.tiff",GA_Update); GDALRasterBand *poBandNew; poBandNew=poDatasetNew->GetRasterBand(1); poBandNew->RasterIO(GF_Write,0,0,XsizeDEM,YsizeDEM,pmemNew,XsizeDEM,YsizeDEM,GDT_Float32,0,0); GDALClose((GDALDatasetH)poDatasetNew); //释放数据 CPLFree(pmemDEM); CPLFree(pmemNew); printf("处理结束\n"); }
三、插值结果