GDAL 栅格数据转矢量数据

#include "gdal_priv.h"
#include "ogrsf_frmts.h" //for ogr
#include "gdal_alg.h"  //for GDALPolygonize
#include "cpl_conv.h" //for CPLMalloc()

int ImagePolygonize(const char * pszSrcFile,const char* pszDstFile,const char* pszFormat);

int main()
{
const char* pszSrcFile = "H:\\GDAL文件\\gdal栅格矢量化\\classify_bitmap.bmp";
const char* pszDstFile = "H:\\GDAL文件\\gdal栅格矢量化\\shape.shp";
ImagePolygonize(pszSrcFile,pszDstFile,"ESRI Shapefile");
// system("pause");
return 0;
}

int ImagePolygonize(const char * pszSrcFile,const char* pszDstFile,const char* pszFormat)
{
GDALAllRegister();
OGRRegisterAll();//记得添加驱动注册
CPLSetConfigOption("GDAL_FILENAME_IS_UTF8","NO");

GDALDataset* poSrcDS=(GDALDataset*)GDALOpen(pszSrcFile,GA_ReadOnly);
if(poSrcDS==NULL)
{
return 0;
}
// 创建输出矢量文件
OGRSFDriver *poDriver;
poDriver = OGRSFDriverRegistrar::GetRegistrar()->GetDriverByName( pszFormat );
if (poDriver == NULL)
{
GDALClose((GDALDatasetH)poSrcDS);
return 0;
}
//根据文件名创建输出矢量文件
OGRDataSource* poDstDS=poDriver->CreateDataSource(pszDstFile);
if (poDstDS==NULL)
{
GDALClose((GDALDatasetH)poSrcDS);
return 0;
}
// 定义空间参考，与输入图像相同;
OGRSpatialReference *poSpatialRef = new OGRSpatialReference(poSrcDS->GetProjectionRef());
OGRLayer* poLayer = poDstDS->CreateLayer("Result", poSpatialRef, wkbPolygon, NULL);
if (poDstDS == NULL)
{
GDALClose((GDALDatasetH)poSrcDS);
OGRDataSource::DestroyDataSource(poDstDS);
delete poSpatialRef;
poSpatialRef = NULL;
return 0;
}
OGRFieldDefn ofieldDef("Segment", OFTInteger); //创建属性表，只有一个字段即“Segment”，里面保存对应的栅格的像元值
poLayer->CreateField(&ofieldDef);
GDALRasterBandH hSrcBand = (GDALRasterBandH) poSrcDS->GetRasterBand(1); //获取图像的第一个波段
GDALPolygonize(hSrcBand, NULL, (OGRLayerH)poLayer, 0, NULL, NULL, NULL); //调用栅格矢量化
GDALClose(poSrcDS); //关闭文件
OGRDataSource::DestroyDataSource(poDstDS);
return 1;
}