转自: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) {
- 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(); }
- }
来源:http://www.cnblogs.com/cugwx/p/3719195.html