java通过gdal把单波段tiff文件转换为shp矢量文件并压缩

最新推荐文章于 2024-06-22 06:00:00 发布

WRY_

最新推荐文章于 2024-06-22 06:00:00 发布

阅读量882

点赞数

分类专栏： gis java 文章标签： java 开发语言

本文链接：https://blog.csdn.net/qq_36871430/article/details/128664570

版权

java 同时被 2 个专栏收录

37 篇文章 0 订阅

订阅专栏

gis

7 篇文章 0 订阅

订阅专栏


import cn.hutool.core.util.ZipUtil;
import org.gdal.gdal.Band;
import org.gdal.gdal.Dataset;
import org.gdal.gdal.Driver;
import org.gdal.gdal.gdal;
import org.gdal.gdalconst.gdalconst;
import org.gdal.gdalconst.gdalconstConstants;
import org.gdal.ogr.Feature;
import org.gdal.ogr.FieldDefn;
import org.gdal.ogr.Layer;
import org.gdal.ogr.ogr;
import org.gdal.osr.SpatialReference;

import java.io.File;

/**
     * tiff文件转矢量文件
     * @return 默认返回shp矢量文件  隐藏的会生成shp文件同名的zip压缩文件
     */
    public static String tiffConvertShp(String inRaster) {
        String outShp = inRaster.replace(".tif",".shp");
        //载入栅格，读取相关信息
        Dataset dataset = gdal.Open(inRaster, gdalconstConstants.GA_ReadOnly);
        Band band = dataset.GetRasterBand(1);//栅格转矢量需要的波段信息
        SpatialReference prj = new SpatialReference();
        if (!dataset.GetProjectionRef().isEmpty()) {
            prj.ImportFromWkt(dataset.GetProjectionRef());//栅格数据的坐标系作为矢量化后的坐标系
        }
        //创建输出矢量
        String fileName = outShp.substring(outShp.lastIndexOf("\\") + 1);//带后缀的文件名
        String name = fileName.substring(0, fileName.lastIndexOf("."));//不带后缀
        Driver driver = gdal.GetDriverByName("ESRI Shapefile");
        Dataset create = driver.Create(outShp, dataset.getRasterXSize(), dataset.getRasterYSize(), 1, gdalconst.GDT_Byte);

        Layer layer = create.CreateLayer(name, prj);
        FieldDefn field = new FieldDefn("value", ogr.OFTReal);//创建一个字段用来存储栅格的像素值
        layer.CreateField(field);

        //矢量化
        gdal.Polygonize(band, null, layer, 0);

        Feature feature = layer.GetNextFeature();
        //删除无用要素
        while (null!=feature){
            int value = feature.GetFieldAsInteger("value");
            if(value<1||value>4){
                layer.DeleteFeature(feature.GetFID());
            }
            feature=layer.GetNextFeature();
        }

        layer.SyncToDisk();

        File[] strings = new File[4];
        strings[0]=new File(outShp);
        strings[1]=new File(outShp.replace(".shp",".dbf"));
        strings[2]=new File(outShp.replace(".shp",".prj"));
        strings[3]=new File(outShp.replace(".shp",".shx"));

        ZipUtil.zip(new File(outShp.replace(".shp", ".zip")), false, strings);

        dataset.delete();
        create.delete();

        return outShp;
    }