修改降雨文件的jar

13693261870

2024-10-30 495292b588949bcb2cb15eaa9657619aca7f5579

 src/main/java/com/se/simu/helper/ShpHelper.java

@@ -7,10 +7,16 @@
import com.se.simu.domain.dto.GeLayer;
import lombok.extern.slf4j.Slf4j;
import org.gdal.ogr.*;
import org.gdal.osr.CoordinateTransformation;
import org.gdal.osr.SpatialReference;
import org.gdal.osr.osr;

import java.math.BigDecimal;
import java.sql.Timestamp;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.util.List;
import java.util.Map;
import java.util.Vector;

/**
@@ -31,6 +37,113 @@
        }

        return options;
    }

    public static boolean createShp(String filePath, Map<String, Object> map, SpatialReference sr, Double minx, Double miny, Double maxx, Double maxy) {
        Driver driver = null;
        DataSource dataSource = null;
        Layer layer = null;
        try {
            driver = ogr.GetDriverByName("ESRI shapefile");
            if (null == driver) return false;

            dataSource = driver.CreateDataSource(filePath, null);
            if (null == dataSource) return false;

            layer = dataSource.CreateLayer(FileUtil.getName(filePath), sr, ogr.wkbPolygon, getOptions());
            if (null == layer) return false;

            if (null != map) createFields(layer, map);

            Feature f = new Feature(layer.GetLayerDefn());
            Geometry g = createPolygon(sr, minx, miny, maxx, maxy);
            f.SetGeometry(g);
            if (null != map) setValues(f, map);

            layer.CreateFeature(f);

            return true;
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            return false;
        } finally {
            GdalHelper.delete(layer, dataSource, driver);
        }
    }

    private static void createFields(Layer layer,Map<String, Object> map) {
        for (String key : map.keySet()) {
            Object val = map.get(key);
            switch (val.getClass().getTypeName()) {
                //case "java.math.BigDecimal":
                case "java.lang.Double":
                case "double":
                    layer.CreateField(new FieldDefn(key, ogr.OFTReal));
                    break;
                case "java.lang.Long":
                case "long":
                    layer.CreateField(new FieldDefn(key, ogr.OFTInteger64));
                    break;
                case "java.lang.Integer":
                case "int":
                    layer.CreateField(new FieldDefn(key, ogr.OFTInteger));
                    break;
                //case "java.sql.Timestamp":
                //case "java.time.LocalDate":
                //    layer.CreateField(new FieldDefn(key, ogr.OFTDateTime));
                //    break;
                default:
                    layer.CreateField(new FieldDefn(key, ogr.OFTString));
                    break;
            }
        }
    }

    private static void setValues(Feature f, Map<String, Object> map) {
        for (String key : map.keySet()) {
            Object val = map.get(key);
            switch (val.getClass().getTypeName()) {
                case "java.lang.Double":
                case "double":
                    f.SetField(key, Double.parseDouble(val.toString()));
                    break;
                case "java.lang.Long":
                case "long":
                    f.SetField(key, Long.parseLong(val.toString()));
                    break;
                case "java.lang.Integer":
                case "int":
                    f.SetField(key, Integer.parseInt(val.toString()));
                    break;
                default:
                    f.SetField(key, val.toString());
                    break;
            }
        }
    }

    private static Geometry createPolygon(SpatialReference sr, Double minx, Double miny, Double maxx, Double maxy) {
        String epsg = sr.GetAuthorityCode(null);
        if (!("4326".equals(epsg) || "4490".equals(epsg))) {
            double[] dmin = fromWgs84(sr, minx, miny);
            double[] dmax = fromWgs84(sr, maxx, maxy);
            minx = dmin[0];
            miny = dmin[1];
            maxx = dmax[0];
            maxy = dmax[1];
        }

        Geometry ring = new Geometry(ogr.wkbLinearRing);
        ring.AddPoint_2D(minx, maxy);
        ring.AddPoint_2D(maxx, maxy);
        ring.AddPoint_2D(maxx, miny);
        ring.AddPoint_2D(minx, miny);
        ring.AddPoint_2D(minx, maxy);

        Geometry poly = new Geometry(ogr.wkbPolygon);
        poly.AddGeometry(ring);

        return poly;
    }

    public static boolean createShp(String filePath, GeLayer geLayer) {
@@ -186,4 +299,35 @@
                return ogr.OFTString;
        }
    }

    private String getEpsg(SpatialReference sr) {
        return sr.GetAuthorityCode(null);
    }

    /**
     * 转换为WGS84坐标
     */
    public static Geometry toWgs84(SpatialReference sr, double x, double y) {
        Geometry point = new Geometry(ogr.wkbPoint);
        point.AssignSpatialReference(sr);
        point.AddPoint(x, y);

        point.TransformTo(GdalHelper.SR4326);
        //point.SwapXY();

        return point;
    }

    /**
     * WGS84转换为目标坐标
     */
    public static double[] fromWgs84(SpatialReference sr, double x, double y) {
        // https://blog.csdn.net/weixin_34910922/article/details/129208661
        CoordinateTransformation ct = new CoordinateTransformation(GdalHelper.SR4326, sr);
        if (sr.IsProjected() != 1) {
            sr.SetAxisMappingStrategy(osr.OAMS_TRADITIONAL_GIS_ORDER);
        }

        return ct.TransformPoint(x, y);
    }
}