栅格矢量化

13693261870

2024-11-06 deeef732865c7dd901f07b7123e8f7a3c9f189c9

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

@@ -6,6 +6,8 @@
import com.se.simu.domain.dto.GeField;
import com.se.simu.domain.dto.GeLayer;
import lombok.extern.slf4j.Slf4j;
import org.gdal.gdal.Dataset;
import org.gdal.gdal.gdal;
import org.gdal.ogr.*;
import org.gdal.osr.SpatialReference;

@@ -299,4 +301,32 @@
    private String getEpsg(SpatialReference sr) {
        return sr.GetAuthorityCode(null);
    }

    public static boolean polygonize(Dataset ds, String filePath) {
        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), ds.GetSpatialRef(), ogr.wkbPolygon, getOptions());
            if (null == layer) return false;

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

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

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