private static final String GEOMETRY_TYPE_POINT = "Point";
private static final String GEOMETRY_TYPE_MULTIPOINT = "MultiPoint";
private static final String GEOMETRY_TYPE_LINESTRING = "LineString";
private static final String GEOMETRY_TYPE_MULTILINESTRING = "MultiLineString";
private static final String GEOMETRY_TYPE_POLYGON = "Polygon";
private static final String GEOMETRY_TYPE_MULTIPOLYGON = "MultiPolygon";
public static void write2Shape(String shpPath, String encode, String authority,String geoType, List geoms) {
try {
//创建shape文件对象
File file = new File(shpPath);
Map params = new HashMap<>();
ShapefileDataStore ds = new ShapefileDataStore(file.toURI().toURL());
//定义图形信息和属性信息
SimpleFeatureTypeBuilder tb = new SimpleFeatureTypeBuilder();
//坐标框架
CoordinateReferenceSystem sourceCRS = CRS.decode(authority);
tb.setCRS(sourceCRS);
tb.setName("shapefile");
if (GEOMETRY_TYPE_POINT.equals(geoType)) {
tb.add("the_geom", Point.class);
} else if (GEOMETRY_TYPE_MULTIPOINT.equals(geoType)) {
tb.add("the_geom", MultiPoint.class);
} else if (GEOMETRY_TYPE_LINESTRING.equals(geoType)) {
tb.add("the_geom", LineString.class);
} else if (GEOMETRY_TYPE_MULTILINESTRING.equals(geoType)) {
tb.add("the_geom", MultiLineString.class);
}else if (GEOMETRY_TYPE_POLYGON.equals(geoType)) {
tb.add("the_geom", Polygon.class);
} else if (GEOMETRY_TYPE_MULTIPOLYGON.equals(geoType)) {
tb.add("the_geom", MultiPolygon.class);
} else {
throw new Exception("Geometry中没有该类型：" + geoType);
}
ds.createSchema(tb.buildFeatureType());
//设置编码
Charset charset = Charset.forName(encode);
ds.setCharset(charset);
//设置Writer
FeatureWriter writer = ds.getFeatureWriter(ds.getTypeNames()[0], Transaction.AUTO_COMMIT);
for (Geometry geom : geoms) {
//String type = geom.getGeometryType();
SimpleFeature feature = writer.next();
feature.setAttribute("the_geom", geom);
}
writer.write();
writer.close();
ds.dispose();
} catch (Exception e) {
e.printStackTrace();
}
}
public static void write2Shape(String shpPath, String encode, String prjWKT, String geoType, String shpKey, List attrKeys, List> data) throws Exception{
if (data == null || data.size() == 0) {
return;
}
//创建shape文件对象
File file = new File(shpPath);
ShapefileDataStore ds = new ShapefileDataStore(file.toURI().toURL());
//定义图形信息和属性信息
SimpleFeatureTypeBuilder tb = new SimpleFeatureTypeBuilder();
//坐标框架
//CoordinateReferenceSystem sourceCRS = CRS.decode(authority);
CoordinateReferenceSystem sourceCRS = CRS.parseWKT(prjWKT);
tb.setCRS(sourceCRS);
tb.setName("shapefile");
CoordinateReferenceSystem middleCRS = CRS.decode("EPSG:4490");
MathTransform transform = CRS.findMathTransform(middleCRS, sourceCRS);
//MathTransform transform = CRS.findMathTransform(DefaultGeographicCRS.WGS84, sourceCRS);
if (GEOMETRY_TYPE_POINT.equals(geoType)) {
tb.add("the_geom", Point.class);
} else if (GEOMETRY_TYPE_MULTIPOINT.equals(geoType)) {
tb.add("the_geom", MultiPoint.class);
} else if (GEOMETRY_TYPE_LINESTRING.equals(geoType)) {
tb.add("the_geom", LineString.class);
} else if (GEOMETRY_TYPE_MULTILINESTRING.equals(geoType)) {
tb.add("the_geom", MultiLineString.class);
}else if (GEOMETRY_TYPE_POLYGON.equals(geoType)) {
tb.add("the_geom", Polygon.class);
} else if (GEOMETRY_TYPE_MULTIPOLYGON.equals(geoType)) {
tb.add("the_geom", MultiPolygon.class);
} else {
throw new Exception("Geometry中没有该类型：" + geoType);
}
for (String field : attrKeys) {
tb.add(field.toUpperCase(), String.class);
}
ds.createSchema(tb.buildFeatureType());
//设置编码
Charset charset = Charset.forName(encode);
ds.setCharset(charset);
//设置Writer
FeatureWriter writer = ds.getFeatureWriter(ds.getTypeNames()[0], Transaction.AUTO_COMMIT);
//写入文件信息
for (int i = 0; i < data.size(); i++) {
SimpleFeature feature = writer.next();
Map row = data.get(i);
Geometry geom = (Geometry) row.get(shpKey);
for(int j=0;j getAttrOfShp(String shpFile) throws Exception {
List> result = new ArrayList>();
SimpleFeatureSource featureSource = null;
File file = new File(shpFile);
ShapefileDataStore shpDataStore = null;
shpDataStore = new ShapefileDataStore(file.toURI().toURL());
//设置编码
//Charset charset = Charset.forName("UTF-8");
Charset charset = Charset.forName("GBK");
shpDataStore.setCharset(charset);
String tableName = shpDataStore.getTypeNames()[0];
featureSource = shpDataStore.getFeatureSource(tableName);
SimpleFeatureType schema = featureSource.getSchema();
ArrayList fields = new ArrayList<>();
schema.getAttributeDescriptors().forEach(attr-> {
Class cls = attr.getType().getBinding();
String length = GetLength(attr.getType().getRestrictions());
//获取属性名称
String clsName = cls.getName();
if(!attr.getLocalName().equals("the_geom")) {
clsName = clsName.substring(clsName.lastIndexOf(".") + 1).toLowerCase();
if (length == null || "".equals(length)) {
Fields field = new Fields(attr.getLocalName(), clsName, 0);
field = typeJudgment(field);
fields.add(field);
} else {
Fields field = new Fields(attr.getLocalName(), clsName, Integer.parseInt(length));
field = typeJudgment(field);
fields.add(field);
}
}
});
System.out.println(fields);
return fields;
}
public static List> readSHP(String shpFile, int spatialRef, List arrKeys) throws Exception{
List> result = new ArrayList>();
SimpleFeatureSource featureSource = null;
File file = new File(shpFile);
ShapefileDataStore shpDataStore = null;
shpDataStore = new ShapefileDataStore(file.toURI().toURL());
//设置编码
//Charset charset = Charset.forName("UTF-8");
Charset charset = Charset.forName("GBK");
shpDataStore.setCharset(charset);
String tableName = shpDataStore.getTypeNames()[0];
featureSource = shpDataStore.getFeatureSource(tableName);
SimpleFeatureCollection featureCollection = featureSource.getFeatures();
SimpleFeatureType featureType = featureCollection.getSchema();
CoordinateReferenceSystem coordinateReferenceSystem = featureType.getCoordinateReferenceSystem();
SimpleFeatureIterator features = featureCollection.features();
while(features.hasNext()){
SimpleFeature feature = features.next();
ArrayList
