package com.se.nsl.utils;

import org.gdal.gdal.Dataset;
import org.gdal.gdal.Driver;
import org.gdal.gdal.gdal;
import org.gdal.gdalconst.gdalconstConstants;

public class ZarrReader {

    static {
        // 注册GDAL驱动
        try {
            gdal.AllRegister();
        } catch (Exception e) {
            System.err.println("GDAL驱动注册失败: " + e.getMessage());
        }
    }

    /**
     * 读取Zarr数据集并打印基本信息
     * @param zarrPath Zarr数据集路径（本地路径或URL）
     */
    public static void readZarr(String zarrPath) {
        // 打开Zarr数据集
        Driver driver = gdal.GetDriverByName("Zarr");
        if (driver == null) {
            System.out.println("zarr驱动不可用");
        }
        Dataset dataset = gdal.Open(zarrPath, gdalconstConstants.GA_ReadOnly);


        if (dataset == null) {
            System.err.println("无法打开Zarr数据集: " + zarrPath);
            System.err.println(gdal.GetLastErrorMsg());
            return;
        }

        try {
            // 打印数据集基本信息
            System.out.println("数据集信息:");
            System.out.println("  驱动: " + dataset.GetDriver().getShortName());
            System.out.println("  宽度: " + dataset.GetRasterXSize() + " 像素");
            System.out.println("  高度: " + dataset.GetRasterYSize() + " 像素");
            System.out.println("  波段数: " + dataset.GetRasterCount());

            // 获取地理变换参数
            double[] geoTransform = dataset.GetGeoTransform();
            if (geoTransform != null && geoTransform.length >= 6) {
                System.out.println("  地理变换参数:");
                System.out.printf("    左上角X: %.6f%n", geoTransform[0]);
                System.out.printf("    左上角Y: %.6f%n", geoTransform[3]);
                System.out.printf("    X方向分辨率: %.6f%n", geoTransform[1]);
                System.out.printf("    Y方向分辨率: %.6f%n", geoTransform[5]);
            }

            // 获取投影信息
            String projection = dataset.GetProjection();
            if (projection != null && !projection.isEmpty()) {
                System.out.println("  投影信息: " + projection.substring(0, Math.min(100, projection.length())) + "...");
            }

            // 读取第一个波段的数据
            if (dataset.GetRasterCount() > 0) {
                int width = dataset.GetRasterXSize();
                int height = dataset.GetRasterYSize();
                float[] buffer = new float[width * height];

                // 读取波段数据
                dataset.GetRasterBand(1).ReadRaster(0, 0, width, height, buffer);

                // 统计基本信息
                float min = Float.MAX_VALUE;
                float max = Float.MIN_VALUE;
                float sum = 0;
                int validCount = 0;

                for (float value : buffer) {
                    if (!Float.isNaN(value) && !Float.isInfinite(value)) {
                        min = Math.min(min, value);
                        max = Math.max(max, value);
                        sum += value;
                        validCount++;
                    }
                }

                if (validCount > 0) {
                    float mean = sum / validCount;
                    System.out.printf("  第一个波段统计信息: 最小值=%.2f, 最大值=%.2f, 平均值=%.2f%n", min, max, mean);
                }
            }

        } catch (Exception e) {
            System.err.println("读取Zarr数据时发生错误: " + e.getMessage());
            e.printStackTrace();
        } finally {
            // 释放资源
            if (dataset != null) {
                dataset.delete();
            }
        }
    }

    public static void main(String[] args) {
        // 示例Zarr数据集路径 - 请替换为实际路径
//        String zarrPath = "C:\\Users\\Deng\\Desktop\\just_test\\result.zarr";
        String zarrPath = "D:\\other\\simu\\uwsolver\\20250516174452\\result.zarr";

        // 读取并打印Zarr信息
        readZarr(zarrPath);

    }
}