package org.springblade.common.utils;

import cn.hutool.core.io.FileUtil;
import com.alibaba.fastjson.JSONObject;
import org.geotools.data.DataStore;
import org.geotools.data.DataStoreFinder;
import org.geotools.data.FeatureSource;
import org.geotools.feature.FeatureCollection;
import org.geotools.feature.FeatureIterator;
import org.geotools.geojson.feature.FeatureJSON;
import org.locationtech.jts.geom.Geometry;
import org.locationtech.jts.geom.Point;
import org.locationtech.jts.io.ParseException;
import org.opengis.feature.simple.SimpleFeature;
import org.opengis.feature.simple.SimpleFeatureType;
import org.opengis.filter.Filter;
import java.io.File;
import java.io.IOException;
import java.io.StringWriter;
import java.util.*;

public class ShapeFileUtil {

	/**
	 * 将文件解压
	 * @param zipFile
	 * @return
	 * @throws IOException
	 */
	public static FeatureCollection getFeatureCollectionByShpFile(File zipFile) throws IOException {
		try {
			String tempDir = FileUtil.getTmpDirPath();
			File shapeDir = new File(tempDir + File.separator + new Date().getTime());
			shapeDir.mkdir();
			List<String> files = ZipUtil.unZipFiles(zipFile, shapeDir.getPath() + File.separator);
			String shapFileName = "";
			for (String fileName : files) {
				String suffix = fileName.substring(fileName.lastIndexOf(".") + 1);
				if ("shp".equals(suffix)) {
					shapFileName = fileName;
				}
			}
			File shapeFile = new File(shapFileName);
			Map<String, Object> shapeFileParams = new HashMap();
			shapeFileParams.put("url", shapeFile.toURI().toURL());
			// 设置编码
			shapeFileParams.put("charset", "utf-8");
			DataStore dataStore = DataStoreFinder.getDataStore(shapeFileParams);
			if (dataStore == null) {
				throw new RuntimeException("couldn't load the damn data store: " + shapeFileParams);
			}
			String typeName = dataStore.getTypeNames()[0];
			FeatureSource<SimpleFeatureType, SimpleFeature> source = dataStore.getFeatureSource(typeName);
			Filter filter = Filter.INCLUDE;
			FeatureCollection<SimpleFeatureType, SimpleFeature> collection = source.getFeatures(filter);
			return collection;
		} catch (Exception e) {
			throw e;
		}
	}

	/**
	 * 解析shp文件返回数据
	 * @param zipFile:
	 * @return JSONObject
	 * @description 通过shp压缩文件，将其转换为GeoJson格式
	 */
	//将解压后的文件转换成GeoJson格式
	public static List<Map<String,Object>> shpToGeoJson(File zipFile) throws IOException{
		List<Map<String,Object>> dtoList = new ArrayList<>();
		try {
			FeatureJSON featureJSON = new FeatureJSON();
			JSONObject geoJsonObject = new JSONObject();
			geoJsonObject.put("type", "FeatureCollection");
			// 解压zip文件，获取FeatureCollection
			FeatureCollection collection = getFeatureCollectionByShpFile(zipFile);
			// 获取迭代对象
			FeatureIterator iterator = collection.features();
			// 遍历feature转为json对象
			while (iterator.hasNext()) {
				Map<String,Object> shpDTO = new HashMap<>();
				SimpleFeature feature = (SimpleFeature) iterator.next();
				StringWriter writer = new StringWriter();
				featureJSON.writeFeature(feature, writer);
				String temp = writer.toString();
				JSONObject json = JSONObject.parseObject(temp);

				try {
					// 空间坐标
					shpDTO.put("geometry",json.getJSONObject("geometry").getString("coordinates"));
					// 空间类型
					if (json.getJSONObject("geometry").get("type") != null) {
						shpDTO.put("geoType",json.getJSONObject("geometry").getString("type"));
					}
					dtoList.add(shpDTO);
				} catch (NumberFormatException e) {
					System.err.println("处理单个特征时发生错误：" + e.getMessage());
				} catch (Exception e) {
					System.err.println("处理单个特征时发生错误：" + e.getMessage());
				}
			}
			iterator.close();
		} catch (IOException e) {
			System.err.println("读取或解析文件时发生错误：" + e.getMessage());
			throw e;
		}
		return dtoList;
	}


	public static void main(String[] args) throws IOException, ParseException {
		File file = new File("F:\\test4jdata\\shp\\zrq.zip");
		List<Map<String, Object>> list = shpToGeoJson(file);
		System.out.println("list = " + list);
	}

	public static double[] getLongitudeLatitude(Geometry wktPoint) throws ParseException {
		Geometry geometry = wktPoint;
		if (geometry instanceof Point) {
			Point point = (Point) geometry;
			double longitude = point.getX();
			double latitude = point.getY();
			return new double[]{longitude, latitude};
		} else {
			throw new IllegalArgumentException("The geometry is not a point.");
		}
	}
}