@xingjs@20250220@添加处理sww相关接口类

xingjinshuang

2025-02-20 0890b7861feae74bdcfd1851e577db6b9f31d484

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

@@ -8,6 +8,7 @@
import lombok.extern.slf4j.Slf4j;
import org.gdal.gdal.Dataset;
import org.gdal.gdal.gdal;
import org.gdal.gdalconst.gdalconst;
import org.gdal.ogr.*;
import org.gdal.osr.SpatialReference;

@@ -17,12 +18,6 @@
import java.util.Map;
import java.util.Vector;

/**
 * ShapeFile帮助类
 *
 * @author WWW
 * @date 2024-09-13
 */
@Slf4j
@SuppressWarnings("ALL")
public class ShpHelper {
@@ -120,7 +115,7 @@
        }
    }

    private static Geometry createPolygon(SpatialReference sr, Double minx, Double miny, Double maxx, Double maxy) {
    public 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 = GdalHelper.fromWgs84(sr, minx, miny);
@@ -140,6 +135,7 @@

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

        return poly;
    }
@@ -215,10 +211,7 @@
        f.SetField(i, local.getYear(), local.getMonthValue(), local.getDayOfMonth(), local.getHour(), local.getMinute(), local.getSecond(), 8);
    }

    /**
     * 创建Geometry对象
     */
    private static Geometry createGeometry(GeLayer geLayer, JSONObject geom) {
    public static Geometry createGeometry(GeLayer geLayer, JSONObject geom) {
        String type = geom.getStr("type");
        JSONArray cs = geom.getJSONArray("coordinates");

@@ -302,7 +295,7 @@
        return sr.GetAuthorityCode(null);
    }

    public static boolean polygonize(Dataset ds, String filePath) {
    public static boolean polygonize2Shp(Dataset ds, String filePath) {
        Driver driver = null;
        DataSource dataSource = null;
        Layer layer = null;
@@ -316,10 +309,9 @@
            layer = dataSource.CreateLayer(FileUtil.getName(filePath), ds.GetSpatialRef(), ogr.wkbPolygon, getOptions());
            if (null == layer) return false;

            FieldDefn field = new FieldDefn("val", ogr.OFTReal);
            layer.CreateField(field);
            layer.CreateField(new FieldDefn("val", ogr.OFTReal));

            gdal.Polygonize(ds.GetRasterBand(1), null, layer, 0, null);
            gdal.Polygonize(ds.GetRasterBand(1), ds.GetRasterBand(1).GetMaskBand(), layer, 0);

            return true;
        } catch (Exception ex) {
@@ -329,4 +321,39 @@
            GdalHelper.delete(layer, dataSource, driver);
        }
    }

    public static boolean polygonize2Geojson(Dataset ds, String filePath) {
        Driver driver = null;
        DataSource dataSource = null;
        Layer layer = null;
        try {
            driver = ogr.GetDriverByName("GeoJSON");
            if (null == driver) return false;

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

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

            layer.CreateField(new FieldDefn("val", ogr.OFTReal));

            gdal.Polygonize(ds.GetRasterBand(1), ds.GetRasterBand(1).GetMaskBand(), layer, 0);

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

    public static void test() {
        String path = "D:\\simu\\out\\20241010095328\\waters\\1730217635000\\";

        Dataset ds = gdal.Open(path + "water.tif", gdalconst.GA_ReadOnly);
        ds.SetSpatialRef(GdalHelper.createSpatialReference(4548));
        polygonize2Geojson(ds, path + "water.geojson");
    }
}