GDAL读写矢量文件——Java

importorg.gdal.osr.*;
importorg.gdal.ogr.*;
importorg.gdal.gdal.*;

1．读取矢量

staticvoid ReadVectorFile()
{
String strVectorFile = "F:\\Work\\OGR测试数据\\beijing.shp";
// 注册所有的驱动
ogr.RegisterAll();
// 为了支持中文路径，请添加下面这句代码
gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8","YES");
// 为了使属性表字段支持中文，请添加下面这句
gdal.SetConfigOption("SHAPE_ENCODING","");
//打开数据
DataSource ds = ogr.Open(strVectorFile,0);
if (ds == null)
{
System.out.println("打开文件【"+ strVectorFile + "】失败！" );
return;
}
System.out.println("打开文件【"+ strVectorFile + "】成功！" );
// 获取该数据源中的图层个数，一般shp数据图层只有一个，如果是mdb、dxf等图层就会有多个
int iLayerCount = ds.GetLayerCount();
// 获取第一个图层
Layer oLayer = ds.GetLayerByIndex(0);
if (oLayer == null)
{
System.out.println("获取第0个图层失败！\n");
return;
}
// 对图层进行初始化，如果对图层进行了过滤操作，执行这句后，之前的过滤全部清空
oLayer.ResetReading();
// 通过属性表的SQL语句对图层中的要素进行筛选，这部分详细参考SQL查询章节内容
//oLayer.SetAttributeFilter("\"NAME99\"LIKE \"北京市市辖区\"");
// 通过指定的几何对象对图层中的要素进行筛选
//oLayer.SetSpatialFilter();
// 通过指定的四至范围对图层中的要素进行筛选
//oLayer.SetSpatialFilterRect();
// 获取图层中的属性表表头并输出
System.out.println("属性表结构信息：");
FeatureDefn oDefn =oLayer.GetLayerDefn();
int iFieldCount =oDefn.GetFieldCount();
for (int iAttr = 0; iAttr <iFieldCount; iAttr++)
{
FieldDefn oField =oDefn.GetFieldDefn(iAttr);
System.out.println(oField.GetNameRef()+ ": " +
oField.GetFieldTypeName(oField.GetFieldType())+ "(" +
oField.GetWidth()+"."+ oField.GetPrecision() + ")");
}
// 输出图层中的要素个数
System.out.println("要素个数 = " + oLayer.GetFeatureCount(0));
Feature oFeature = null;
// 下面开始遍历图层中的要素
while ((oFeature =oLayer.GetNextFeature()) != null)
{
System.out.println("当前处理第" + oFeature.GetFID() + "个:\n属性值：");
// 获取要素中的属性表内容
for (int iField = 0; iField< iFieldCount; iField++)
{
FieldDefn oFieldDefn= oDefn.GetFieldDefn(iField);
int type =oFieldDefn.GetFieldType();
switch (type)
{
caseogr.OFTString:
System.out.println(oFeature.GetFieldAsString(iField)+ "\t");
break;
caseogr.OFTReal:
System.out.println(oFeature.GetFieldAsDouble(iField)+ "\t");
break;
caseogr.OFTInteger:
System.out.println(oFeature.GetFieldAsInteger(iField)+ "\t");
break;
default:
System.out.println(oFeature.GetFieldAsString(iField)+ "\t");
break;
}
}
// 获取要素中的几何体
Geometry oGeometry =oFeature.GetGeometryRef();
// 为了演示，只输出一个要素信息
break;
}
System.out.println("数据集关闭！");
}

2．写入矢量

staticvoid WriteVectorFile()
{
String strVectorFile ="E:\\TestPolygon.shp";
// 注册所有的驱动
ogr.RegisterAll();
// 为了支持中文路径，请添加下面这句代码
gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8","NO");
// 为了使属性表字段支持中文，请添加下面这句
gdal.SetConfigOption("SHAPE_ENCODING","");
//创建数据，这里以创建ESRI的shp文件为例
String strDriverName = "ESRIShapefile";
org.gdal.ogr.Driver oDriver =ogr.GetDriverByName(strDriverName);
if (oDriver == null)
{
System.out.println(strVectorFile+ " 驱动不可用！\n");
return;
}
// 创建数据源
DataSource oDS = oDriver.CreateDataSource(strVectorFile,null);
if (oDS == null)
{
System.out.println("创建矢量文件【"+ strVectorFile +"】失败！\n" );
return;
}
// 创建图层，创建一个多边形图层，这里没有指定空间参考，如果需要的话，需要在这里进行指定
Layer oLayer =oDS.CreateLayer("TestPolygon", null, ogr.wkbPolygon, null);
if (oLayer == null)
{
System.out.println("图层创建失败！\n");
return;
}
// 下面创建属性表
// 先创建一个叫FieldID的整型属性
FieldDefn oFieldID = newFieldDefn("FieldID", ogr.OFTInteger);
oLayer.CreateField(oFieldID, 1);
// 再创建一个叫FeatureName的字符型属性，字符长度为50
FieldDefn oFieldName = newFieldDefn("FieldName", ogr.OFTString);
oFieldName.SetWidth(100);
oLayer.CreateField(oFieldName, 1);
FeatureDefn oDefn =oLayer.GetLayerDefn();
// 创建三角形要素
Feature oFeatureTriangle = newFeature(oDefn);
oFeatureTriangle.SetField(0, 0);
oFeatureTriangle.SetField(1, "三角形");
Geometry geomTriangle =Geometry.CreateFromWkt("POLYGON ((0 0,20 0,10 15,0 0))");
oFeatureTriangle.SetGeometry(geomTriangle);
oLayer.CreateFeature(oFeatureTriangle);
// 创建矩形要素
Feature oFeatureRectangle = newFeature(oDefn);
oFeatureRectangle.SetField(0, 1);
oFeatureRectangle.SetField(1, "矩形");
Geometry geomRectangle =Geometry.CreateFromWkt("POLYGON ((30 0,60 0,60 30,30 30,30 0))");
oFeatureRectangle.SetGeometry(geomRectangle);
oLayer.CreateFeature(oFeatureRectangle);
// 创建五角形要素
Feature oFeaturePentagon = newFeature(oDefn);
oFeaturePentagon.SetField(0, 2);
oFeaturePentagon.SetField(1, "五角形");
Geometry geomPentagon =Geometry.CreateFromWkt("POLYGON ((70 0,85 0,90 15,80 30,65 15,700))");
oFeaturePentagon.SetGeometry(geomPentagon);
oLayer.CreateFeature(oFeaturePentagon);
System.out.println("\n数据集创建完成！\n");
}

3．矢量数据管理

staticvoid VectorDelete(String strVectorFile)
{
// 注册所有的驱动
ogr.RegisterAll();
File file = newFile(strVectorFile);
org.gdal.ogr.Driver oDriver = null;
{
//打开矢量
DataSource oDS =ogr.Open(strVectorFile, 0);
if (oDS == null)
{
file.delete();
return;
}
oDriver = oDS.GetDriver();
if (oDriver == null)
{
file.delete();
return;
}
}
if(oDriver.DeleteDataSource(strVectorFile) == ogr.OGRERR_NONE)
return;
else
file.delete();
}
staticvoid VectorRename(String strOldFile, String strNewFile)
{
// 注册所有的驱动
ogr.RegisterAll();
File file = new File(strOldFile);
org.gdal.ogr.Driver oDriver = null;
{
//打开矢量
DataSource oDS =ogr.Open(strOldFile, 0);
if (oDS == null)
{
file.renameTo(newFile(strNewFile));
return;
}
oDriver = oDS.GetDriver();
if (oDriver == null)
{
file.renameTo(newFile(strNewFile));
return;
}
DataSource oDDS =oDriver.CopyDataSource(oDS, strNewFile, null);
if (oDDS == null)
{
file.renameTo(newFile(strNewFile));
return;
}
}
if(oDriver.DeleteDataSource(strOldFile) == ogr.OGRERR_NONE)
return;
else
file.renameTo(newFile(strNewFile));
}