package org.sxkj.odm.utils;

import com.google.common.collect.Lists;
import org.geotools.coverage.grid.GridCoverage2D;
import org.geotools.gce.geotiff.GeoTiffReader;
import org.geotools.geometry.jts.JTS;
import org.geotools.referencing.CRS;
import org.geotools.referencing.crs.DefaultGeographicCRS;
import org.jetbrains.annotations.NotNull;
import org.locationtech.jts.geom.Coordinate;
import org.locationtech.jts.geom.GeometryFactory;
import org.locationtech.jts.geom.Point;
import org.locationtech.jts.geom.PrecisionModel;
import org.opengis.geometry.DirectPosition;
import org.opengis.geometry.Envelope;
import org.opengis.referencing.crs.CoordinateReferenceSystem;
import org.opengis.referencing.operation.MathTransform;
import org.sxkj.odm.entity.MyPoint;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.geom.Line2D;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileInputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * tif 文件读取工具
 */
public class ParseTifUtils {

    /**
	 * 获取tif四角位置
	 *
	 * @return
	 */
	public static void getTifFour(String path) {
		try {
			// 读取 TIF 文件
			GeoTiffReader reader = new GeoTiffReader(new File(path));
			//获取文件的坐标系信息
			CoordinateReferenceSystem crs = reader.getCoordinateReferenceSystem();
			GridCoverage2D coverage = reader.read(null);
			// 获取 TIF 文件的四角坐标，并转换为经纬度坐标系
			Envelope envelope = coverage.getEnvelope();
			Envelope transformedEnvelope = CRS.transform(envelope, crs);
			double maxLon = transformedEnvelope.getMaximum(0); // 最大经度
			double maxLat = transformedEnvelope.getMaximum(1); // 最大纬度
			double minLon = transformedEnvelope.getMinimum(0); // 最小经度
			double minLat = transformedEnvelope.getMinimum(1); // 最小纬度
			DirectPosition lowerCorner = transformedEnvelope.getLowerCorner();// 左下角
			DirectPosition upperCorner = transformedEnvelope.getUpperCorner();// 右上角
			//获取tif 文件的宽度，高度
			double width = coverage.getGridGeometry().getGridRange2D().getWidth();
			double height = coverage.getGridGeometry().getGridRange2D().getHeight();
			// 左上角(最小经度，最大纬度)  右下角(最大经度，最小纬度)
			// 输出结果
			System.out.println("lowerCorner: " + lowerCorner);
			System.out.println("upperCorner: " + upperCorner);
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	/**
	 * 获取经纬度在当前tif文件中的地理坐标（单点）
	 * @param path
	 * @param longitude
	 * @param latitude
	 * @return
	 */
	public static Point getTifPosByLngLat(String path, Double longitude, Double latitude) {
		try {
			//类型
			PrecisionModel precisionModel = new PrecisionModel(PrecisionModel.FLOATING);
			//84坐标
			GeometryFactory geometryFactory = new GeometryFactory(precisionModel, 4326);
			// 读取 TIF 文件
			GeoTiffReader reader = new GeoTiffReader(new File(path));
			//获取文件的坐标系信息
			GridCoverage2D coverage = reader.read(null);
			// 创建一个点对象，并定义其经纬度
			Point point = geometryFactory.createPoint(new Coordinate(longitude, latitude));
			// 转换坐标系统
			MathTransform transform = CRS.findMathTransform(DefaultGeographicCRS.WGS84, coverage.getCoordinateReferenceSystem());
			// 转为地理坐标
			Point reprojectedPoint = (Point) JTS.transform(point, transform);
			// 返回
			return reprojectedPoint;
		} catch (Exception e) {
			e.printStackTrace();
		}
		return null;
	}

	/**
	 * 获取经纬度在当前tif文件中的地理坐标(多点)
	 * @param path
	 * @param coordinateList 经纬度坐标集合
	 * @return
	 */
	public static List<Point> getTifPosByMoreLngLat(String path, List<Coordinate> coordinateList) {
		List<Point> list = new ArrayList<>();
		try {
			FileInputStream fileInputStream = new FileInputStream(path);
			//类型
			PrecisionModel precisionModel = new PrecisionModel(PrecisionModel.FLOATING);
			//84坐标
			GeometryFactory geometryFactory = new GeometryFactory(precisionModel, 4326);
			// 读取 TIF 文件
			GeoTiffReader reader = new GeoTiffReader(fileInputStream);
			//获取文件的坐标系信息
			GridCoverage2D coverage = reader.read(null);
			for (Coordinate coordinate : coordinateList) {
				// 创建一个点对象，并定义其经纬度
				Point point = geometryFactory.createPoint(coordinate);
				// 转换坐标系统
				MathTransform transform = CRS.findMathTransform(DefaultGeographicCRS.WGS84, coverage.getCoordinateReferenceSystem());
				// 转为地理坐标
				Point reprojectedPoint = (Point) JTS.transform(point, transform);
				// 加入集合
				list.add(reprojectedPoint);
			}
			// 关闭流
			fileInputStream.close();
			// 返回
			return list;
		} catch (Exception e) {
			e.printStackTrace();
		}
		return list;
	}


    /**
     * 在 tif 文件上画线
	 * @param tifFilePath tif 读取路径
	 * @param outPath 输出路径
	 * @param map
     */
	public static void drawLineTif(String tifFilePath,String outPath, Map<String, Double> map) {
		try {
			FileInputStream fileInputStream = new FileInputStream(tifFilePath);
			// 读取TIF图片
			BufferedImage bufferedImage = ImageIO.read(fileInputStream);
			// 创建Graphics2D对象
			Graphics2D g2d = (Graphics2D) bufferedImage.getGraphics();
			// 设置红色,、画笔粗细
			g2d.setColor(Color.RED);
			g2d.setStroke(new BasicStroke(3f));
			// 画线
			drawLine(g2d, map);
			// 释放图形上下文资源
			g2d.dispose();
			// 输出图片
			File outputFile = new File(outPath);
			// 写入
			ImageIO.write(bufferedImage, "tif", outputFile);
			// 关闭流
			fileInputStream.close();
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

    /**
     * 在 tif 文件上批量画线
	 * @param tifFilePath tif 读取路径
	 * @param outPath 输出路径
	 * @param points
     */
	public static void drawLineListTif(String tifFilePath,String outPath, List<MyPoint> points) {
		try {
			FileInputStream fileInputStream = new FileInputStream(tifFilePath);
			// 读取TIF图片
			BufferedImage bufferedImage = ImageIO.read(fileInputStream);
			// 创建Graphics2D对象
			Graphics2D g2d = (Graphics2D) bufferedImage.getGraphics();
			// 设置红色,、画笔粗细
			g2d.setColor(Color.RED);
			g2d.setStroke(new BasicStroke(3f));
			List<Map<String, Double>> maps = new ArrayList<>();
			int size = points.size();
			// 将点分成多个线，当前点和下一个点组成一条线，末尾和首个点组成一条线
			for (int i = 0; i < points.size(); i++) {
				Map<String, Double> map = new HashMap<>();
				MyPoint current = points.get(i);
				map.put("x1",current.getX());
				map.put("y1",current.getY());
				MyPoint next = (i == size - 1) ? points.get(0) : points.get(i + 1);
				map.put("x2",next.getX());
				map.put("y2",next.getY());
				maps.add(map);
			}
			for (Map<String, Double> map : maps) {
				// 画线
				drawLine(g2d, map);
			}
			// 释放图形上下文资源
			g2d.dispose();
			// 输出图片
			File outputFile = new File(outPath);
			// 写入
			ImageIO.write(bufferedImage, "tif", outputFile);
			// 关闭流
			fileInputStream.close();
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

    /**
     * 在 tif 文件上画面
	 * @param tifFilePath tif 读取路径
	 * @param outPath 输出路径
	 * @param points
     */
	public static boolean drawPolygonByPoints(String tifFilePath,String outPath, List<MyPoint> points, String suffix) {
		try {
			FileInputStream fileInputStream = new FileInputStream(tifFilePath);
			// 读取TIF图片
			BufferedImage bufferedImage = ImageIO.read(fileInputStream);
			// 创建Graphics2D对象
			Graphics2D g2d = (Graphics2D) bufferedImage.getGraphics();
			// 设置红色,、画笔粗细
			g2d.setColor(Color.RED);
			g2d.setStroke(new BasicStroke(3f));
			// 将点集合组装成画面的数组
			Map<String, int[]> hashMap = getStringMap(points);
			// 画面
			drawPolygon(g2d,hashMap);
			// 释放图形上下文资源
			g2d.dispose();
			// 输出图片
			File outputFile = new File(outPath);
			// 写入
			ImageIO.write(bufferedImage, suffix, outputFile);
			// 关闭流
			fileInputStream.close();
			// 返回
			return true;
		} catch (Exception e) {
			e.printStackTrace();
		}
		return false;
	}

	/**
	 * 将点集合组装成画面的数组
	 * @param points
	 * @return
	 */
	@NotNull
	private static Map<String, int[]> getStringMap(List<MyPoint> points) {
		Map<String, int []> hashMap = new HashMap<>();
		List<Integer> lngList = new ArrayList<>();
		List<Integer> latList = new ArrayList<>();
		for (MyPoint point : points) {
			lngList.add((int)Math.round(point.getX().doubleValue()));
			latList.add((int)Math.round(point.getY().doubleValue()));
		}
		int[] lngArr = lngList.stream()
			.mapToInt(Integer::intValue)
			.toArray();
		int[] latArr = latList.stream()
			.mapToInt(Integer::intValue)
			.toArray();
		hashMap.put("lngs",lngArr);
		hashMap.put("lats",latArr);
		return hashMap;
	}


	/**
	 * 画线
	 * @param g2D
	 * @param map
	 */
	public static void drawLine(Graphics2D g2D, Map<String, Double> map) {
		// 打印转换后的Map
		Line2D.Double line = new Line2D.Double(map.get("x1"), map.get("y1"),
			map.get("x2"), map.get("y2"));
		g2D.draw(line);
	}

	/**
	 * 画面
	 * @param g2D
	 * @param map
	 */
	public static void drawPolygon(Graphics2D g2D, Map<String, int []> map) {
		// 创建一个多边形
		int[] xPoints = map.get("lngs");
		int[] yPoints = map.get("lats");
		Polygon polygon = new Polygon(xPoints, yPoints, xPoints.length);
		// 画多边形
		g2D.drawPolygon(polygon);
	}




}