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


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;
    }
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值