geojson文件与shape文件的相互转换

一.geojson转shape

pom文件:

	<dependencies>
		<dependency>
			<groupId>nl.cloudfarming.client</groupId>
			<artifactId>lib-geotools</artifactId>
			<version>2.7.5</version>
		</dependency>
		<dependency>
			<groupId>org.geotools</groupId>
			<artifactId>gt-geojson</artifactId>
			<version>18.0</version>
		</dependency>
		<dependency>
			<groupId>com.googlecode.json-simple</groupId>
			<artifactId>json-simple</artifactId>
			<version>1.1.1</version>
		</dependency>
		<dependency>
			<groupId>commons-lang</groupId>
			<artifactId>commons-lang</artifactId>
			<version>2.6</version>
		</dependency>
		<dependency>
			<groupId>com.alibaba</groupId>
			<artifactId>fastjson</artifactId>
			<version>1.2.62</version>
		</dependency>
	</dependencies>

java代码:

import com.alibaba.fastjson.JSONObject;
import com.vividsolutions.jts.geom.*;
import org.geotools.data.FeatureWriter;
import org.geotools.data.Transaction;
import org.geotools.data.shapefile.ShapefileDataStore;
import org.geotools.data.shapefile.ShapefileDataStoreFactory;
import org.geotools.feature.simple.SimpleFeatureTypeBuilder;
import org.geotools.geojson.geom.GeometryJSON;
import org.geotools.referencing.crs.DefaultGeographicCRS;
import org.opengis.feature.simple.SimpleFeature;
import org.opengis.feature.simple.SimpleFeatureType;
import java.io.*;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class GeojsonToShape {

	public static void main(String[] args) {

		geojson2Shape( "E:\\points.geojson", "E:\\college\\college.shp");

	}

	public static void geojson2Shape(String jsonPath, String shpPath){
		GeometryJSON geojson = new GeometryJSON();
		try{
			String geojsonStr = readJson(jsonPath);
			Map<String, Object> geojsonMap = JSONObject.parseObject(geojsonStr, Map.class);
			List<Map> features = (List<Map>) geojsonMap.get("features");
			Map geojsonExemple = features.get(0);
			String geojsonType = ((Map) geojsonExemple.get("geometry")).get("type").toString();
			Map<String, Class> mapFields = new HashMap();
			for (int i = 0; i < features.size(); i++) {
				Map<String, Object> attributes = (Map<String, Object>) features.get(i).get("properties");
				for (String key : attributes.keySet()) {
					Class type = attributes.get(key).getClass();
					mapFields.put(key, type);
				}
			}

			Class<?> geoType = null;
			switch(geojsonType){
				case "Point":
					geoType = Point.class;
					break;
				case "MultiPoint":
					geoType = MultiPoint.class;
					break;
				case "LineString":
					geoType = LineString.class;
					break;
				case "MultiLineString":
					geoType = MultiLineString.class;
					break;
				case "Polygon":
					geoType = Polygon.class;
					break;
				case "MultiPolygon":
					geoType = MultiPolygon.class;
					break;
			}

			//创建shape文件对象
			File file = new File(shpPath);
			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", geoType);

			for (String key : mapFields.keySet()) {
				tb.add(key, mapFields.get(key));
			}

			ds.createSchema(tb.buildFeatureType());
			//设置Writer
			FeatureWriter<SimpleFeatureType, SimpleFeature> writer = ds.
					getFeatureWriter(ds.getTypeNames()[0], Transaction.AUTO_COMMIT);

			for(int i=0,len=features.size();i<len;i++){
				Map oneGeojson = features.get(i);
				Map<String,Object> attributes = (Map<String, Object>) oneGeojson.get("properties");
				String strFeature = JSONObject.toJSONString(oneGeojson);
				Reader reader = new StringReader(strFeature);
				SimpleFeature feature = writer.next();

				switch(geojsonType){
					case "Point":
						feature.setAttribute("the_geom",geojson.readPoint(reader));
						break;
					case "MultiPoint":
						feature.setAttribute("the_geom",geojson.readMultiPoint(reader));
						break;
					case "LineString":
						feature.setAttribute("the_geom",geojson.readLine(reader));
						break;
					case "MultiLineString":
						feature.setAttribute("the_geom",geojson.readMultiLine(reader));
						break;
					case "Polygon":
						feature.setAttribute("the_geom",geojson.readPolygon(reader));
						break;
					case "MultiPolygon":
						feature.setAttribute("the_geom",geojson.readMultiPolygon(reader));
						break;
				}

				for (String key : attributes.keySet()) {
					feature.setAttribute(key,attributes.get(key));
				}
				writer.write();
			}
			writer.close();
			ds.dispose();
		}
		catch(Exception e){
			System.out.println("转换失败");
			e.printStackTrace();
		}
	}


	/**
	 * 读取文件
	 * @param filepath 文件路径
	 * @throws IOException
	 */
	public static String readJson(String filepath) {
		FileReader re = null;
		BufferedReader buff = null;
		String line = "";
		try {
			File file = new File(filepath);
			re = new FileReader(file);
			buff = new BufferedReader(re);
			String tempString = null;
			while ((tempString = buff.readLine()) != null) {
				line += tempString;
			}
			return line;
		} catch (Exception e) {
			System.out.println("失败了");
		} finally {
			try {
				re.close();
				buff.close();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
		return line;
	}

}

二.shape转geojson

pom:

	<dependencies>
		<dependency>
			<groupId>nl.cloudfarming.client</groupId>
			<artifactId>lib-geotools</artifactId>
			<version>2.7.5</version>
		</dependency>
		<dependency>
			<groupId>org.geotools</groupId>
			<artifactId>gt-geojson</artifactId>
			<version>18.0</version>
		</dependency>
		<dependency>
			<groupId>com.googlecode.json-simple</groupId>
			<artifactId>json-simple</artifactId>
			<version>1.1.1</version>
		</dependency>
		<dependency>
			<groupId>commons-lang</groupId>
			<artifactId>commons-lang</artifactId>
			<version>2.6</version>
		</dependency>
		<dependency>
			<groupId>com.alibaba</groupId>
			<artifactId>fastjson</artifactId>
			<version>1.2.62</version>
		</dependency>
	</dependencies>

java代码:

import org.geotools.data.shapefile.ShapefileDataStore;
import org.geotools.data.simple.SimpleFeatureCollection;
import org.geotools.data.simple.SimpleFeatureIterator;
import org.geotools.data.simple.SimpleFeatureSource;
import org.geotools.geojson.feature.FeatureJSON;
import org.opengis.feature.simple.SimpleFeature;

import java.io.*;
import java.nio.charset.Charset;

public class ShapeToGeojson {


    public static void main(String[] args) {
        shape2Geojson("E:\\jingwei\\china_2000.shp","E:\\points.geojson");
    }

    /**
     * shp转换为Geojson
     * @param shpPath
     * @return
     */
    public static void shape2Geojson(String shpPath, String jsonPath){
        FeatureJSON featureJson = new FeatureJSON();
        try{
            BufferedWriter out =new BufferedWriter(new OutputStreamWriter(new FileOutputStream(jsonPath),"UTF-8"));

            out.write("{\"type\": \"FeatureCollection\",\"features\": [");

            File file = new File(shpPath);
            ShapefileDataStore shpDataStore = null;
            shpDataStore = new ShapefileDataStore(file.toURL());

            // 设置编码,防止中文乱码
            shpDataStore.setStringCharset(Charset.forName("gbk"));

            String typeName = shpDataStore.getTypeNames()[0];
            SimpleFeatureSource featureSource = null;
            featureSource =  shpDataStore.getFeatureSource (typeName);
            SimpleFeatureCollection result = featureSource.getFeatures();

            System.out.println(result.size());

            SimpleFeatureIterator itertor = result.features();
            int count = 1;
            while (itertor.hasNext()) {
                SimpleFeature feature = itertor.next();
                StringWriter writer = new StringWriter();
                featureJson.writeFeature(feature, writer);
                out.write(writer.toString());
                System.out.println(count);
                if (count != result.size() ){
                    out.write( ",");
                }
                count ++;
            }
            out.write( "]}");
            itertor.close();
            out.flush();
            out.close();
        }
        catch(Exception e){
            System.out.println("转换失败");
            e.printStackTrace();
        }

    }

}