package com.se.simu.utils;
|
|
import org.apache.commons.imaging.*;
|
import org.gdal.gdal.Dataset;
|
import org.gdal.gdal.Band;
|
import org.gdal.gdal.gdal;
|
import org.gdal.gdalconst.gdalconstConstants;
|
|
import javax.imageio.ImageIO;
|
import java.awt.*;
|
import java.awt.image.BufferedImage;
|
import java.io.File;
|
import java.io.IOException;
|
|
public class TiffToRGBUtil {
|
public static void main(String[] args) throws Exception{
|
tifToPng("D:\\城市内涝\\sem\\DEM.tif","D:\\城市内涝\\sem\\DEM.png");
|
}
|
public static void tifToPng(String tifPath, String pngPath) throws Exception {
|
// 注册所有的 GDAL 驱动
|
gdal.AllRegister();
|
// 输入的 TIFF 文件路径
|
// 打开输入的 TIFF 数据集
|
Dataset dataset = gdal.Open(tifPath, gdalconstConstants.GA_ReadOnly);
|
if (dataset == null) {
|
System.err.println("无法打开输入的 TIFF 文件");
|
return;
|
}
|
// 获取地理变换信息
|
double[] geoTransform = dataset.GetGeoTransform();
|
// 获取第一个波段
|
Band band = dataset.GetRasterBand(1);
|
// 获取图像宽度和高度
|
int width = dataset.getRasterXSize();
|
int height = dataset.getRasterYSize();
|
// 读取图像数据到字节数组
|
byte[] imageData = new byte[width * height];
|
band.ReadRaster(0, 0, width, height, imageData);
|
BufferedImage pngImage = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB);
|
// 假设高度信息存储在灰度值中，根据地理变换信息和像素值计算实际高度
|
for (int y = 0; y < height; y++) {
|
for (int x = 0; x < width; x++) {
|
int pixelValue = imageData[y * width + x] & 0xFF;
|
// 假设高度信息存储在像素值中
|
double pixelHeight = pixelValue * geoTransform[5];
|
//height = -10000 + ((R * 256 * 256 + G * 256 + B) * 0.1)
|
int value = (int) ((pixelHeight + 10000) * 10);
|
//value=(R * 256 * 256 + G * 256 + B);
|
int r = value / (256 * 256);
|
value = value % (256 * 256);
|
int g = value / 256;
|
int b = value % 256;
|
Color color = new Color(r, g, b);
|
System.out.printf("Pixel (%d, %d): R=%d, G=%d, B=%d%n", x, y, r, g, b);
|
int newRgb = color.getRGB();
|
// 将处理后的像素颜色值设置到新的 PNG 图像中
|
pngImage.setRGB(x, y, newRgb);
|
}
|
}
|
// 保存为 PNG 图像
|
ImageIO.write(pngImage, "png", new File(pngPath));
|
// 释放资源
|
dataset.delete();
|
}
|
|
public static void tiffToPng(String tifPath, String pngPath) throws Exception {
|
// 输出的 PNG 文件路径
|
String outputPngFilePath = pngPath;
|
try {
|
// 读取 TIF 图像
|
File tiffFile = new File(tifPath);
|
BufferedImage tifImage = Imaging.getBufferedImage(tiffFile);
|
int width = tifImage.getWidth();
|
int height = tifImage.getHeight();
|
// 创建一个新的 BufferedImage 对象，类型为 TYPE_INT_ARGB
|
BufferedImage pngImage = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB);
|
// 遍历 TIF 图像的每个像素
|
for (int y = 0; y < height; y++) {
|
for (int x = 0; x < width; x++) {
|
// 获取 TIF 图像中当前像素的颜色值
|
int pixel = tifImage.getRGB(x, y);
|
// 在这里可以对像素的颜色值进行一些处理，例如根据高度信息调整颜色
|
// 假设我们简单地将像素的红色分量根据高度进行调整
|
int red = (pixel >> 16) & 0xff;
|
int green = (pixel >> 8) & 0xff;
|
int blue = (pixel) & 0xff;
|
// 假设高度信息存储在绿色分量中，你可以根据实际情况调整
|
//height = -10000 + ((R * 256 * 256 + G * 256 + B) * 0.1)
|
int heightValue = (int) (-10000 + ((red * 256 * 256 + green * 256 + blue) * 0.1));
|
System.out.println(heightValue);
|
// 简单地将红色分量根据高度值进行调整，例如，越高越红
|
// 确保红色分量不超过 255
|
red = Math.min(red, 255);
|
green = Math.min(green, 255);
|
blue = Math.min(blue, 255);
|
// 重新组合 ARGB 值
|
Color color = new Color(red, green, blue);
|
System.out.printf("Pixel (%d, %d): R=%d, G=%d, B=%d%n", x, y, red, green, blue);
|
int newRgb = color.getRGB();
|
// 将处理后的像素颜色值设置到新的 PNG 图像中
|
pngImage.setRGB(x, y, newRgb);
|
}
|
}
|
// 保存为 PNG 图像
|
ImageIO.write(pngImage, "png", new File(outputPngFilePath));
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
