JAVA用geotools读写shape格式文件

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) {

    DbaseFileReader reader = null;
    try {
        reader = new DbaseFileReader(new ShpFiles(path), false, Charset.forName("GBK"));
        DbaseFileHeader header = reader.getHeader();
        int numFields = header.getNumFields();
        //迭代读取记录
        while (reader.hasNext()) {
            try {
                Object[] entry = reader.readEntry();
                for (int i=0; i<numFields; i++) {
                    String title = header.getFieldName(i);
                    Object value = entry[i];
                    System.out.println(title+"="+value);
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    } catch (Exception e) {
        e.printStackTrace();
    } finally {
        if (reader != null) {
            //关闭
            try {reader.close();} catch (Exception e) {}
        }
    }
}

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

public void readSHP(String path) {

    ShapefileDataStore shpDataStore = null;
    try{
        shpDataStore = new ShapefileDataStore(new File(path).toURI().toURL());
        shpDataStore.setStringCharset(Charset.forName("GBK"));
        String typeName = shpDataStore.getTypeNames()[0];
        FeatureSource<SimpleFeatureType, SimpleFeature> featureSource = null; 
        featureSource = (FeatureSource<SimpleFeatureType, SimpleFeature>)shpDataStore.getFeatureSource(typeName);
        FeatureCollection<SimpleFeatureType, SimpleFeature> result = featureSource.getFeatures();
        System.out.println(result.size());
        FeatureIterator<SimpleFeature> itertor = result.features();
        while(itertor.hasNext()){
            SimpleFeature feature = itertor.next();
            Collection<Property> p = feature.getProperties();
            Iterator<Property> it = p.iterator();
            while(it.hasNext()) {
                Property pro = it.next();
                if (pro.getValue() instanceof Point) {
                    System.out.println("PointX = " + ((Point)(pro.getValue())).getX());
                    System.out.println("PointY = " + ((Point)(pro.getValue())).getY());
                } else {
                    System.out.println(pro.getName() + " = " + pro.getValue());
                }
            }
        }
        itertor.close();
    } catch (MalformedURLException e) {
        e.printStackTrace();
    } catch(IOException e) { e.printStackTrace(); }
}

写shape文件，以point为例：

public static void main(String[] args) {
    try{  
        //定义属性
        final SimpleFeatureType TYPE = DataUtilities.createType("Location",
            "location:Point," + // <- the geometry attribute: Point type
            "POIID:String," + // <- a String attribute
            "MESHID:String," + // a number attribute
            "OWNER:String"
        );
        SimpleFeatureCollection collection = FeatureCollections.newCollection();
        GeometryFactory geometryFactory = new GeometryFactory();
        SimpleFeatureBuilder featureBuilder = new SimpleFeatureBuilder(TYPE);

        double latitude = Double.parseDouble("116.123456789");
        double longitude = Double.parseDouble("39.120001");
        String POIID = "2050003092";
        String MESHID = "0";
        String OWNER = "340881";

        /* Longitude (= x coord) first ! */
        Point point = geometryFactory.createPoint(new Coordinate(longitude, latitude));
        Object[] obj = {point, POIID, MESHID, OWNER};
        SimpleFeature feature = featureBuilder.buildFeature(null, obj);
        collection.add(feature);
        feature = featureBuilder.buildFeature(null, obj);
        collection.add(feature);

        File newFile = new File("D:/newPoi.shp");
        ShapefileDataStoreFactory dataStoreFactory = new ShapefileDataStoreFactory();
        Map<String, Serializable> params = new HashMap<String, Serializable>();
        params.put("url", newFile.toURI().toURL());
        params.put("create spatial index", Boolean.TRUE);
        ShapefileDataStore newDataStore = (ShapefileDataStore) dataStoreFactory.createNewDataStore(params);
        newDataStore.createSchema(TYPE);
        newDataStore.forceSchemaCRS(DefaultGeographicCRS.WGS84);

        Transaction transaction = new DefaultTransaction("create");
        String typeName = newDataStore.getTypeNames()[0];
        SimpleFeatureSource featureSource = newDataStore.getFeatureSource(typeName);

        if (featureSource instanceof SimpleFeatureStore) {
            SimpleFeatureStore featureStore = (SimpleFeatureStore) featureSource;
            featureStore.setTransaction(transaction);
            try {
                featureStore.addFeatures(collection);
                transaction.commit();
            } catch (Exception problem) {
                problem.printStackTrace();
	        transaction.rollback();
            } finally {
                transaction.close();
            }
        } else {
            System.out.println(typeName + " does not support read/write access");
        }
    } catch (Exception e) {
        e.printStackTrace();
    }
}

 以下代码对应geotools10.0版本

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

import java.io.File;
import java.nio.charset.Charset;
import java.util.Iterator;

import org.geotools.data.shapefile.ShapefileDataStore;
import org.geotools.data.shapefile.ShapefileDataStoreFactory;
import org.geotools.data.simple.SimpleFeatureIterator;
import org.geotools.data.simple.SimpleFeatureSource;
import org.opengis.feature.Property;
import org.opengis.feature.simple.SimpleFeature;

public class ShpNew {
	
	public static void main(String[] args) {
		ShapefileDataStoreFactory dataStoreFactory = new ShapefileDataStoreFactory();
		try {
			ShapefileDataStore sds = (ShapefileDataStore)dataStoreFactory.createDataStore(new File("D:\\work\\shpdir\\Poi.shp").toURI().toURL());
			sds.setCharset(Charset.forName("GBK"));
			SimpleFeatureSource featureSource = sds.getFeatureSource();
			SimpleFeatureIterator itertor = featureSource.getFeatures().features();

			while(itertor.hasNext()) {  
				SimpleFeature feature = itertor.next();  
				Iterator<Property> it = feature.getProperties().iterator();

				while(it.hasNext()) {  
					Property pro = it.next();
					System.out.println(pro);  
		            }
		        }  
		        itertor.close();  
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
}

 二、读图形信息

    try {
        ShpFiles sf = new ShpFiles("D:\\Poi.shp");
        ShapefileReader r = new ShapefileReader( sf, false, false, new GeometryFactory() );
        while (r.hasNext()) {
            Geometry shape = (Geometry) r.nextRecord().shape();  //com.vividsolutions.jts.geom.Geometry;
            System.out.println(shape.toString());
        } 
        r.close();
    } catch (Exception e) {
        e.printStackTrace();
    }

三、读dbf文件

	public void readDBF() {
		try {
			FileChannel in = new FileInputStream("D:\\Poi.dbf").getChannel();
			DbaseFileReader dbfReader =  new DbaseFileReader(in, false,  Charset.forName("GBK"));
			DbaseFileHeader header = dbfReader.getHeader();
			int fields = header.getNumFields();
			
			while ( dbfReader.hasNext() ){
				DbaseFileReader.Row row =  dbfReader.readRow();
//				System.out.println(row.toString());
				for (int i=0; i<fields; i++) {
					System.out.println(header.getFieldName(i) + " : " + row.read(i));	
				}
			}
			dbfReader.close();
			in.close();
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

四、写shape文件

	public void write(String filepath) {
		try {
			//创建shape文件对象
			File file = new File(filepath);
			Map<String, Serializable> params = new HashMap<String, Serializable>();
			params.put( ShapefileDataStoreFactory.URLP.key, file.toURI().toURL() );
			ShapefileDataStore ds = (ShapefileDataStore) new ShapefileDataStoreFactory().createNewDataStore(params);
			//定义图形信息和属性信息
			SimpleFeatureTypeBuilder tb = new SimpleFeatureTypeBuilder();
			tb.setCRS(DefaultGeographicCRS.WGS84);
			tb.setName("shapefile");
			tb.add("the_geom", Point.class);
			tb.add("POIID", Long.class);
			tb.add("NAMEC", String.class);
			ds.createSchema(tb.buildFeatureType());
			ds.setCharset(Charset.forName("GBK"));
			//设置Writer
			FeatureWriter<SimpleFeatureType, SimpleFeature> writer = ds.getFeatureWriter(ds.getTypeNames()[0], Transaction.AUTO_COMMIT);
			//写下一条
			SimpleFeature feature = writer.next();
			feature.setAttribute("the_geom", new GeometryFactory().createPoint(new Coordinate(116.123, 39.345)));
			feature.setAttribute("POIID", 1234567890l);
			feature.setAttribute("NAMEC", "某兴趣点1");
			feature = writer.next();
			feature.setAttribute("the_geom", new GeometryFactory().createPoint(new Coordinate(116.456, 39.678)));
			feature.setAttribute("POIID", 1234567891l);
			feature.setAttribute("NAMEC", "某兴趣点2");
			writer.write();
			writer.close();
			ds.dispose();
			
			//读取刚写完shape文件的图形信息
			ShpFiles shpFiles = new ShpFiles(filepath);
			ShapefileReader reader = new ShapefileReader(shpFiles, false, true, new GeometryFactory(), false);
			try {
				while (reader.hasNext()) {
					System.out.println(reader.nextRecord().shape());	
				}
			} finally {
				reader.close();
			}
		} catch (Exception e) {	}
	}

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

	public void transShape(String srcfilepath, String destfilepath) {
		try {
			//源shape文件
			ShapefileDataStore shapeDS = (ShapefileDataStore) new ShapefileDataStoreFactory().createDataStore(new File(srcfilepath).toURI().toURL());
			//创建目标shape文件对象
			Map<String, Serializable> params = new HashMap<String, Serializable>();
	        FileDataStoreFactorySpi factory = new ShapefileDataStoreFactory();
	        params.put(ShapefileDataStoreFactory.URLP.key, new File(destfilepath).toURI().toURL());
	        ShapefileDataStore ds = (ShapefileDataStore) factory.createNewDataStore(params);
	        // 设置属性
	        SimpleFeatureSource fs = shapeDS.getFeatureSource(shapeDS.getTypeNames()[0]);
	        //下面这行还有其他写法，根据源shape文件的simpleFeatureType可以不用retype，而直接用fs.getSchema设置
	        ds.createSchema(SimpleFeatureTypeBuilder.retype(fs.getSchema(), DefaultGeographicCRS.WGS84));
	        
	        //设置writer
	        FeatureWriter<SimpleFeatureType, SimpleFeature> writer = ds.getFeatureWriter(ds.getTypeNames()[0], Transaction.AUTO_COMMIT);
	        
	        //写记录
	        SimpleFeatureIterator it = fs.getFeatures().features();
	        try {
	            while (it.hasNext()) {
	                SimpleFeature f = it.next();
	                SimpleFeature fNew = writer.next();
	                fNew.setAttributes(f.getAttributes());
	                writer.write();
	            }
	        } finally {
	            it.close();
	        }
	        writer.close();
	        ds.dispose();
	        shapeDS.dispose();
		} catch (Exception e) { e.printStackTrace();	}
	}