注册 |登录

linqy的个人博客分享 http://blog.sciencenet.cn/u/linqy

博文

java利用geotools读写shp以及相关的文件

已有 9704 次阅读 2017-10-25 10:07 |个人分类:java|系统分类:科研笔记

利用java读取shp以及相关的文件

对应geotools版本：2.7.2

读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为例：  

  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. }  

参考：http://toplchx.iteye.com/blog/1335007




https://blog.sciencenet.cn/blog-3134052-1082365.html

上一篇：java中vector的用法简单总结
下一篇：Java对MySQL数据库进行连接、查询和修改
收藏 IP: 111.195.160.*| 热度|

0

该博文允许注册用户评论 请点击登录 评论 (0 个评论)

数据加载中...

Archiver|手机版|科学网 ( 京ICP备07017567号-12 )

GMT+8, 2024-4-27 03:30

Powered by ScienceNet.cn

Copyright © 2007- 中国科学报社

返回顶部