关于Java配置GDAL库参考之前的博文。
在Java中使用OGR读写矢量数据时,需要导入ogr库,如果需要对中文路径的支持还有空间参考的支持,需要导入gdal和osr库,代码如下:
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)); }