JAVA用geotools读写shape格式文件

转自：http://toplchx.iteye.com/blog/1335007

JAVA用geotools读写shape格式文件 （对应geotools版本：2.7.2）

（后面添加对应geotools 10.0版本的写法）

 

读shape文件。

shape格式文件最少包含3个文件，他们的后缀是：.shp, .dbf, .shx。

.shp存储地理形状和位置信息，.dbf存储属性信息，.shx是索引文件。

 

单独读取DBF文件

public void readDBF(String path) {

Java代码  收藏代码
  1.     DbaseFileReader reader = null;  
  2.     try {  
  3.         reader = new DbaseFileReader(new ShpFiles(path), false, Charset.forName("GBK"));  
  4.         DbaseFileHeader header = reader.getHeader();  
  5.         int numFields = header.getNumFields();  
  6.         //迭代读取记录  
  7.         while (reader.hasNext()) {  
  8.             try {  
  9.                 Object[] entry = reader.readEntry();  
  10.                 for (int i=0; i<numFields; i++) {  
  11.                     String title = header.getFieldName(i);  
  12.                     Object value = entry[i];  
  13.                     System.out.println(title+"="+value);  
  14.                 }  
  15.             } catch (Exception e) {  
  16.                 e.printStackTrace();  
  17.             }  
  18.         }  
  19.     } catch (Exception e) {  
  20.         e.printStackTrace();  
  21.     } finally {  
  22.         if (reader != null) {  
  23.             //关闭  
  24.             try {reader.close();} catch (Exception e) {}  
  25.         }  
  26.     }  
  27. }  

 

 

读取3个文件，以point为例：

public void readSHP(String path) {

Java代码  收藏代码
  1.     ShapefileDataStore shpDataStore = null;  
  2.     try{  
  3.         shpDataStore = new ShapefileDataStore(new File(path).toURI().toURL());  
  4.         shpDataStore.setStringCharset(Charset.forName("GBK"));  
  5.         String typeName = shpDataStore.getTypeNames()[0];  
  6.         FeatureSource<SimpleFeatureType, SimpleFeature> featureSource = null;   
  7.         featureSource = (FeatureSource<SimpleFeatureType, SimpleFeature>)shpDataStore.getFeatureSource(typeName);  
  8.         FeatureCollection<SimpleFeatureType, SimpleFeature> result = featureSource.getFeatures();  
  9.         System.out.println(result.size());  
  10.         FeatureIterator<SimpleFeature> itertor = result.features();  
  11.         while(itertor.hasNext()){  
  12.             SimpleFeature feature = itertor.next();  
  13.             Collection<Property> p = feature.getProperties();  
  14.             Iterator<Property> it = p.iterator();  
  15.             while(it.hasNext()) {  
  16.                 Property pro = it.next();  
  17.                 if (pro.getValue() instanceof Point) {  
  18.                     System.out.println("PointX = " + ((Point)(pro.getValue())).getX());  
  19.                     System.out.println("PointY = " + ((Point)(pro.getValue())).getY());  
  20.                 } else {  
  21.                     System.out.println(pro.getName() + " = " + pro.getValue());  
  22.                 }  
  23.             }  
  24.         }  
  25.         itertor.close();  
  26.     } catch (MalformedURLException e) {  
  27.         e.printStackTrace();  
  28.     } catch(IOException e) { e.printStackTrace(); }  
  29. }  

 

 

写shape文件，以point为例：

 

Java代码  收藏代码
  1. public static void main(String[] args) {  
  2.     try{    
  3.         //定义属性  
  4.         final SimpleFeatureType TYPE = DataUtilities.createType("Location",  
  5.             "location:Point," + // <- the geometry attribute: Point type  
  6.             "POIID:String," + // <- a String attribute  
  7.             "MESHID:String," + // a number attribute  
  8.             "OWNER:String"  
  9.         );  
  10.         SimpleFeatureCollection collection = FeatureCollections.newCollection();  
  11.         GeometryFactory geometryFactory = new GeometryFactory();  
  12.         SimpleFeatureBuilder featureBuilder = new SimpleFeatureBuilder(TYPE);  
  13.   
  14.         double latitude = Double.parseDouble("116.123456789");  
  15.         double longitude = Double.parseDouble("39.120001");  
  16.         String POIID = "2050003092";  
  17.         String MESHID = "0";  
  18.         String OWNER = "340881";  
  19.   
  20.         /* Longitude (= x coord) first ! */  
  21.         Point point = geometryFactory.createPoint(new Coordinate(longitude, latitude));  
  22.         Object[] obj = {point, POIID, MESHID, OWNER};  
  23.         SimpleFeature feature = featureBuilder.buildFeature(null, obj);  
  24.         collection.add(feature);  
  25.         feature = featureBuilder.buildFeature(null, obj);  
  26.         collection.add(feature);  
  27.   
  28.         File newFile = new File("D:/newPoi.shp");  
  29.         ShapefileDataStoreFactory dataStoreFactory = new ShapefileDataStoreFactory();  
  30.         Map<String, Serializable> params = new HashMap<String, Serializable>();  
  31.         params.put("url", newFile.toURI().toURL());  
  32.         params.put("create spatial index", Boolean.TRUE);  
  33.         ShapefileDataStore newDataStore = (ShapefileDataStore) dataStoreFactory.createNewDataStore(params);  
  34.         newDataStore.createSchema(TYPE);  
  35.         newDataStore.forceSchemaCRS(DefaultGeographicCRS.WGS84);  
  36.   
  37.         Transaction transaction = new DefaultTransaction("create");  
  38.         String typeName = newDataStore.getTypeNames()[0];  
  39.         SimpleFeatureSource featureSource = newDataStore.getFeatureSource(typeName);  
  40.   
  41.         if (featureSource instanceof SimpleFeatureStore) {  
  42.             SimpleFeatureStore featureStore = (SimpleFeatureStore) featureSource;  
  43.             featureStore.setTransaction(transaction);  
  44.             try {  
  45.                 featureStore.addFeatures(collection);  
  46.                 transaction.commit();  
  47.             } catch (Exception problem) {  
  48.                 problem.printStackTrace();  
  49.             transaction.rollback();  
  50.             } finally {  
  51.                 transaction.close();  
  52.             }  
  53.         } else {  
  54.             System.out.println(typeName + " does not support read/write access");  
  55.         }  
  56.     } catch (Exception e) {  
  57.         e.printStackTrace();  
  58.     }  
  59. }  

 

 以下代码对应geotools10.0版本

一、读shp文件（图形信息+属性信息）的写法：

Java代码  收藏代码
  1. import java.io.File;  
  2. import java.nio.charset.Charset;  
  3. import java.util.Iterator;  
  4.   
  5. import org.geotools.data.shapefile.ShapefileDataStore;  
  6. import org.geotools.data.shapefile.ShapefileDataStoreFactory;  
  7. import org.geotools.data.simple.SimpleFeatureIterator;  
  8. import org.geotools.data.simple.SimpleFeatureSource;  
  9. import org.opengis.feature.Property;  
  10. import org.opengis.feature.simple.SimpleFeature;  
  11.   
  12. public class ShpNew {  
  13.       
  14.     public static void main(String[] args) {  
  15.         ShapefileDataStoreFactory dataStoreFactory = new ShapefileDataStoreFactory();  
  16.         try {  
  17.             ShapefileDataStore sds = (ShapefileDataStore)dataStoreFactory.createDataStore(new File("D:\\work\\shpdir\\Poi.shp").toURI().toURL());  
  18.             sds.setCharset(Charset.forName("GBK"));  
  19.             SimpleFeatureSource featureSource = sds.getFeatureSource();  
  20.             SimpleFeatureIterator itertor = featureSource.getFeatures().features();  
  21.   
  22.             while(itertor.hasNext()) {    
  23.                 SimpleFeature feature = itertor.next();    
  24.                 Iterator<Property> it = feature.getProperties().iterator();  
  25.   
  26.                 while(it.hasNext()) {    
  27.                     Property pro = it.next();  
  28.                     System.out.println(pro);    
  29.                     }  
  30.                 }    
  31.                 itertor.close();    
  32.         } catch (Exception e) {  
  33.             e.printStackTrace();  
  34.         }  
  35.     }  
  36. }  

 

 二、读图形信息

Java代码  收藏代码
  1. try {  
  2.     ShpFiles sf = new ShpFiles("D:\\Poi.shp");  
  3.     ShapefileReader r = new ShapefileReader( sf, false, false, new GeometryFactory() );  
  4.     while (r.hasNext()) {  
  5.         Geometry shape = (Geometry) r.nextRecord().shape();  //com.vividsolutions.jts.geom.Geometry;  
  6.         System.out.println(shape.toString());  
  7.     }   
  8.     r.close();  
  9. } catch (Exception e) {  
  10.     e.printStackTrace();  
  11. }  

 

三、读dbf文件

Java代码  收藏代码
  1.     public void readDBF() {  
  2.         try {  
  3.             FileChannel in = new FileInputStream("D:\\Poi.dbf").getChannel();  
  4.             DbaseFileReader dbfReader =  new DbaseFileReader(in, false,  Charset.forName("GBK"));  
  5.             DbaseFileHeader header = dbfReader.getHeader();  
  6.             int fields = header.getNumFields();  
  7.               
  8.             while ( dbfReader.hasNext() ){  
  9.                 DbaseFileReader.Row row =  dbfReader.readRow();  
  10. //              System.out.println(row.toString());  
  11.                 for (int i=0; i<fields; i++) {  
  12.                     System.out.println(header.getFieldName(i) + " : " + row.read(i));     
  13.                 }  
  14.             }  
  15.             dbfReader.close();  
  16.             in.close();  
  17.         } catch (Exception e) {  
  18.             e.printStackTrace();  
  19.         }  
  20.     }  

 

四、写shape文件

Java代码  收藏代码
  1. public void write(String filepath) {  
  2.     try {  
  3.         //创建shape文件对象  
  4.         File file = new File(filepath);  
  5.         Map<String, Serializable> params = new HashMap<String, Serializable>();  
  6.         params.put( ShapefileDataStoreFactory.URLP.key, file.toURI().toURL() );  
  7.         ShapefileDataStore ds = (ShapefileDataStore) new ShapefileDataStoreFactory().createNewDataStore(params);  
  8.         //定义图形信息和属性信息  
  9.         SimpleFeatureTypeBuilder tb = new SimpleFeatureTypeBuilder();  
  10.         tb.setCRS(DefaultGeographicCRS.WGS84);  
  11.         tb.setName("shapefile");  
  12.         tb.add("the_geom", Point.class);  
  13.         tb.add("POIID", Long.class);  
  14.         tb.add("NAMEC", String.class);  
  15.         ds.createSchema(tb.buildFeatureType());  
  16.         ds.setCharset(Charset.forName("GBK"));  
  17.         //设置Writer  
  18.         FeatureWriter<SimpleFeatureType, SimpleFeature> writer = ds.getFeatureWriter(ds.getTypeNames()[0], Transaction.AUTO_COMMIT);  
  19.         //写下一条  
  20.         SimpleFeature feature = writer.next();  
  21.         feature.setAttribute("the_geom", new GeometryFactory().createPoint(new Coordinate(116.123, 39.345)));  
  22.         feature.setAttribute("POIID", 1234567890l);  
  23.         feature.setAttribute("NAMEC", "某兴趣点1");  
  24.         feature = writer.next();  
  25.         feature.setAttribute("the_geom", new GeometryFactory().createPoint(new Coordinate(116.456, 39.678)));  
  26.         feature.setAttribute("POIID", 1234567891l);  
  27.         feature.setAttribute("NAMEC", "某兴趣点2");  
  28.         writer.write();  
  29.         writer.close();  
  30.         ds.dispose();  
  31.           
  32.         //读取刚写完shape文件的图形信息  
  33.         ShpFiles shpFiles = new ShpFiles(filepath);  
  34.         ShapefileReader reader = new ShapefileReader(shpFiles, false, true, new GeometryFactory(), false);  
  35.         try {  
  36.             while (reader.hasNext()) {  
  37.                 System.out.println(reader.nextRecord().shape());      
  38.             }  
  39.         } finally {  
  40.             reader.close();  
  41.         }  
  42.     } catch (Exception e) { }  
  43. }  

 

五、由源shape文件创建新的shape文件

 

Java代码  收藏代码
  1. public void transShape(String srcfilepath, String destfilepath) {  
  2.     try {  
  3.         //源shape文件  
  4.         ShapefileDataStore shapeDS = (ShapefileDataStore) new ShapefileDataStoreFactory().createDataStore(new File(srcfilepath).toURI().toURL());  
  5.         //创建目标shape文件对象  
  6.         Map<String, Serializable> params = new HashMap<String, Serializable>();  
  7.         FileDataStoreFactorySpi factory = new ShapefileDataStoreFactory();  
  8.         params.put(ShapefileDataStoreFactory.URLP.key, new File(destfilepath).toURI().toURL());  
  9.         ShapefileDataStore ds = (ShapefileDataStore) factory.createNewDataStore(params);  
  10.         // 设置属性  
  11.         SimpleFeatureSource fs = shapeDS.getFeatureSource(shapeDS.getTypeNames()[0]);  
  12.         //下面这行还有其他写法，根据源shape文件的simpleFeatureType可以不用retype，而直接用fs.getSchema设置  
  13.         ds.createSchema(SimpleFeatureTypeBuilder.retype(fs.getSchema(), DefaultGeographicCRS.WGS84));  
  14.           
  15.         //设置writer  
  16.         FeatureWriter<SimpleFeatureType, SimpleFeature> writer = ds.getFeatureWriter(ds.getTypeNames()[0], Transaction.AUTO_COMMIT);  
  17.           
  18.         //写记录  
  19.         SimpleFeatureIterator it = fs.getFeatures().features();  
  20.         try {  
  21.             while (it.hasNext()) {  
  22.                 SimpleFeature f = it.next();  
  23.                 SimpleFeature fNew = writer.next();  
  24.                 fNew.setAttributes(f.getAttributes());  
  25.                 writer.write();  
  26.             }  
  27.         } finally {  
  28.             it.close();  
  29.         }  
  30.         writer.close();  
  31.         ds.dispose();  
  32.         shapeDS.dispose();  
  33.     } catch (Exception e) { e.printStackTrace();    }  
  34. }  

posted on 2014-05-09 16:36  打油小牧童  阅读(21818)  评论(0)  编辑  收藏  举报

导航