用java gdal实现影像转shp的代码

以下是用Java GDAL实现影像转shp的代码：

import org.gdal.gdal.Dataset;
import org.gdal.gdal.gdal;
import org.gdal.ogr.DataSource;
import org.gdal.ogr.Driver;
import org.gdal.ogr.ogr;

public class ImageToShpConverter {
   public static void main(String[] args) {
      gdal.AllRegister();
      ogr.RegisterAll();
      
      //输入影像文件路径
      String input = "path/to/input/image.tif";
      //输出矢量文件路径
      String output = "path/to/output/shp.shp";
      
      //打开影像文件
      Dataset ds = gdal.Open(input);
      //获取影像的空间参考
      String srs = ds.GetProjection();
      
      //创建矢量驱动
      Driver driver = ogr.GetDriverByName("ESRI Shapefile");
      //创建输出矢量文件
      DataSource dsOut = driver.CreateDataSource(output);
      //创建输出矢量图层
      ogr.Layer layer = dsOut.CreateLayer("output", null, ogr.wkbPolygon);
      
      //设置输出矢量图层的空间参考
      layer.SetSpatialRef(ogr.CreateSpatialReference(srs));
      
      //从影像创建多边形
      int width = ds.getRasterXSize();
      int height = ds.getRasterYSize();
      double[] geotransform = ds.GetGeoTransform();
      
      for(int i = 0; i < width; i++) {
         for(int j = 0; j < height; j++) {
            //根据像素坐标计算地理坐标
            double[] point = new double[2];
            point[0] = geotransform[0] + i * geotransform[1] + j * geotransform[2];
            point[1] = geotransform[3] + i * geotransform[4] + j * geotransform[5];
            
            //读取像素值
            double[] pixelValue = new double[1];
            int result = ds.GetRasterBand(1).RasterIO(gdalconst.GF_Read, i, j, 1, 1, pixelValue, 1, 1, gdalconst.GDT_Float64, 0, 0);
            
            //如果像素值符合条件，创建多边形并添加到输出矢量图层
            if(result == gdalconst.CE_None && pixelValue[0] > 0) {
               //创建多边形
               ogr.Geometry geometry = ogr.Geometry.CreateFromWkt("POLYGON((" + point[0] + " " + point[1] + ", " + point[0] + " " + (point[1] + geotransform[5]) + ", " + (point[0] + geotransform[1]) + " " + (point[1] + geotransform[5]) + ", " + (point[0] + geotransform[1]) + " " + point[1] + ", " + point[0] + " " + point[1] + "))");
               
               //添加到输出矢量图层
               ogr.Feature feature = new ogr.Feature(layer.GetLayerDefn());
               feature.SetGeometry(geometry);
               layer.CreateFeature(feature);
               
               //释放内存
               feature.delete();
               geometry.delete();
            }
         }
      }
      
      //关闭文件
      dsOut.delete();
      ds.delete();
   }
}