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

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

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. double latitude = Double.parseDouble("116.123456789");
14. double longitude = Double.parseDouble("39.120001");
15. String POIID = "2050003092";
16. String MESHID = "0";
17. String OWNER = "340881";
18. /* Longitude (= x coord) first ! */
19. Point point = geometryFactory.createPoint(new Coordinate(longitude, latitude));
20. Object[] obj = {point, POIID, MESHID, OWNER};
21. SimpleFeature feature = featureBuilder.buildFeature(null, obj);
22. collection.add(feature);
23. feature = featureBuilder.buildFeature(null, obj);
24. collection.add(feature);
25. File newFile = new File("D:/newPoi.shp");
26. ShapefileDataStoreFactory dataStoreFactory = new ShapefileDataStoreFactory();
27. Map<String, Serializable> params = new HashMap<String, Serializable>();
28. params.put("url", newFile.toURI().toURL());
29. params.put("create spatial index", Boolean.TRUE);
30. ShapefileDataStore newDataStore = (ShapefileDataStore) dataStoreFactory.createNewDataStore(params);
31. newDataStore.createSchema(TYPE);
32. newDataStore.forceSchemaCRS(DefaultGeographicCRS.WGS84);
33. Transaction transaction = new DefaultTransaction("create");
34. String typeName = newDataStore.getTypeNames()[0];
35. SimpleFeatureSource featureSource = newDataStore.getFeatureSource(typeName);
36. if (featureSource instanceof SimpleFeatureStore) {
37. SimpleFeatureStore featureStore = (SimpleFeatureStore) featureSource;
38. featureStore.setTransaction(transaction);
39. try {
40. featureStore.addFeatures(collection);
41. transaction.commit();
42. } catch (Exception problem) {
43. problem.printStackTrace();
44. transaction.rollback();
45. } finally {
46. transaction.close();
47. }
48. } else {
49. System.out.println(typeName + " does not support read/write access");
50. }
51. } catch (Exception e) {
52. e.printStackTrace();
53. }
54. }

以下代码对应geotools10.0版本

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

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

二、读图形信息

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文件

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. while ( dbfReader.hasNext() ){
8. DbaseFileReader.Row row =  dbfReader.readRow();
9. //              System.out.println(row.toString());
10. for (int i=0; i<fields; i++) {
11. System.out.println(header.getFieldName(i) + " : " + row.read(i));
12. }
13. }
14. dbfReader.close();
15. in.close();
16. } catch (Exception e) {
17. e.printStackTrace();
18. }
19. }

四、写shape文件

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. //读取刚写完shape文件的图形信息
32. ShpFiles shpFiles = new ShpFiles(filepath);
33. ShapefileReader reader = new ShapefileReader(shpFiles, false, true, new GeometryFactory(), false);
34. try {
35. while (reader.hasNext()) {
36. System.out.println(reader.nextRecord().shape());
37. }
38. } finally {
39. reader.close();
40. }
41. } catch (Exception e) { }
42. }

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

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. //设置writer
15. FeatureWriter<SimpleFeatureType, SimpleFeature> writer = ds.getFeatureWriter(ds.getTypeNames()[0], Transaction.AUTO_COMMIT);
16. //写记录
17. SimpleFeatureIterator it = fs.getFeatures().features();
18. try {
19. while (it.hasNext()) {
20. SimpleFeature f = it.next();
21. SimpleFeature fNew = writer.next();
22. fNew.setAttributes(f.getAttributes());
23. writer.write();
24. }
25. } finally {
26. it.close();
27. }
28. writer.close();
29. ds.dispose();
30. shapeDS.dispose();
31. } catch (Exception e) { e.printStackTrace();    }
32. }