java使用gdal将shp栅格化为tif,并根据特定属性值生成灰度值
package com.leador.gsp.module.parse.test;
import org.gdal.gdal.*;
import org.gdal.gdal.Driver;
import org.gdal.gdalconst.gdalconstConstants;
import org.gdal.ogr.*;
import java.util.Vector;
public class ShpToDem {
public static void main(String[] args) {
ogr.RegisterAll();
String vectorPath = "C:\\Users\\82305\\Desktop\\1\\L_shp\\dxt_element_l_polygon.shp";
DataSource dataSource = ogr.Open(vectorPath);
if (dataSource == null) {
System.out.println("Failed to open the data source.");
return;
}
Layer layer = dataSource.GetLayer(0);
Feature feature = layer.GetNextFeature();
while (feature != null){
String codePixel = feature.GetFieldAsString("codePixel");
System.out.println(codePixel);
feature = layer.GetNextFeature();
}
// shp文件的投影
String shpProjection = layer.GetSpatialRef().ExportToWkt();
//获取shp图层边界
double[] shpTransform = layer.GetExtent();
//横向像素分辨率 =(右上角 x 坐标 - 左上角 x 坐标) / 宽度
double pixel_sizeX = (shpTransform[1] - shpTransform[0])/1000;
//纵向像素分辨率 =(左下角 y 坐标 - 左上角 y 坐标) / 高度
double pixel_sizeY = (shpTransform[2] - shpTransform[3])/1000;
// shp边界范围转换tif
// 左上角x坐标, 水平(X)分辨率,旋转参数, 左上角y坐标,旋转参数,竖直(Y)分辨率。
double[] targetTransform = new double[]{shpTransform[0], pixel_sizeX, 0, shpTransform[3], 0, pixel_sizeY};
System.out.println();
for(int i =0;i<targetTransform.length;i++){
System.out.print(targetTransform[i]+" ");
}
String rasterPath = "C:\\Users\\82305\\Desktop\\澳门\\1.tif";
Driver driver = gdal.GetDriverByName("GTiff");
Dataset dataset = driver.Create(rasterPath, 1000, 1000, 1, gdalconstConstants.GDT_Byte);
dataset.SetGeoTransform(targetTransform);
dataset.SetProjection(shpProjection);
dataset.FlushCache();
// gdal_rasterize -burn 255 -ot Byte -ts 1000 1000 -l Macao_Buildings_DWG-Polygon C:\Users\82305\Desktop\澳门\Macao_Buildings_DWG-Polygon.shp C:\Users\82305\Desktop\澳门\mask.tif
Vector<String> options = new Vector<>();
// 根据高度属性来插入
options.add("ATTRIBUTE=codePixel");
int result = gdal.RasterizeLayer(dataset, new int[]{1}, layer,null,options);
if (result != 0) {
System.out.println("转换过程中出现错误");
}
if (result == 0) {
System.out.println("shp数据转换tif数据成功");
}
// 释放资源
dataset.delete();
dataSource.delete();
}
}