package com.xindao.ocr.swingui.swing.utils;

import org.bytedeco.opencv.global.opencv_imgcodecs;
import org.bytedeco.opencv.global.opencv_imgproc;
import org.bytedeco.opencv.opencv_core.Mat;
import org.bytedeco.opencv.opencv_core.Size;
import org.springframework.web.multipart.MultipartFile;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.*;
import java.nio.file.Files;
import java.util.Objects;
import java.util.Optional;

/**
 * @author sy
 * @date 2022/11/23 22:03
 */
public class ToFile {

    /**
     * MultipartFile转File
     * @param file
     * @return
     * @throws IOException
     */
    public static File multipartFiletoFile(MultipartFile file) throws IOException {
        File toFile = null;
        if((!file.equals("")) && (file.getSize() > 0)) {
            String filePath = "/tmp/img";
            if(!new File(filePath).exists()) {
                new File(filePath).mkdirs();
            }
            InputStream inputStream = file.getInputStream();
            String fileFullName = file.getOriginalFilename();
            String fileName = fileFullName.substring(0, fileFullName.lastIndexOf("."));
            String prefix = fileFullName.substring(fileFullName.lastIndexOf("."));
            toFile = new File(filePath + fileName + "_" + System.currentTimeMillis() + prefix);
            intputStreamToFile(inputStream, toFile);
            inputStream.close();
        }
        return toFile;
    }

    /**
     * 获取文件流
     * @param inputStream
     * @param file
     */
    private static void intputStreamToFile(InputStream inputStream, File file) {
        try (OutputStream outputStream = new FileOutputStream(file)) {
            int bytesRead = 0;
            byte[] buffer = new byte[8192];
            while((bytesRead = inputStream.read(buffer, 0, 8192)) != -1) {
                outputStream.write(buffer, 0, bytesRead);
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 删除临时文件
     * @param file
     */
    public static void deleteTempFile(File file) {
        if(Optional.ofNullable(file).isPresent()) {
            File del = new File(file.toURI());
            del.delete();
        }
    }

    /**
     * 删除临时目录下的所有文件
     * @param cacheDir
     */
    public static void deleteTempFiles(File cacheDir) {
        //删除临时目录
        if (cacheDir.exists() && cacheDir.isDirectory()) {
            for (File tempFile : Objects.requireNonNull(cacheDir.listFiles())) {
                if (tempFile.isFile()) {
                    try {
                        Files.delete(tempFile.toPath());
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
            }
        }
    }


    /**
     * 保存图像到指定路径
     * @param image 图像对象
     * @param filePath 保存路径
     * @param formatName 图像格式（如 "png", "jpg"）
     * @return 是否保存成功
     */
    public static boolean saveImage(BufferedImage image, String filePath, String formatName) {
        if (image == null || filePath == null || formatName == null) {
            System.err.println("参数不能为空");
            return false;
        }

        try {
            // 创建文件对象
            File outputFile = new File(filePath);

            // 确保父目录存在
            File parentDir = outputFile.getParentFile();
            if (parentDir != null && !parentDir.exists()) {
                parentDir.mkdirs(); // 递归创建目录
            }

            // 保存图片
            return ImageIO.write(image, formatName, outputFile);
        } catch (IOException e) {
            System.err.println("保存图片失败：" + e.getMessage());
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 预处理图像以提高OCR识别率
     * @param inputPath 输入图像路径
     */
    public static void preprocessImage(String inputPath) {
        Mat src = opencv_imgcodecs.imread(inputPath);
        if (src.empty()) {
            System.err.println("无法读取图像: " + inputPath);
            return ;
        }

        // 1. 灰度化
        Mat gray = new Mat();
        opencv_imgproc.cvtColor(src, gray, opencv_imgproc.COLOR_BGR2GRAY);

        // 2. 轻量去噪（避免过度模糊H的边缘）
//        Mat blurred = new Mat();
//        opencv_imgproc.GaussianBlur(gray, blurred, new Size(1, 1), 0); // 缩小模糊核，保留字符细节

        // 3. 对比度增强（改用CLAHE，更精细控制对比度）
        Mat enhanced = new Mat();
        opencv_imgproc.createCLAHE(3, new Size(3, 3)).apply(gray, enhanced); // clipLimit调整对比度强度

//        // 4. 二值化（调整阈值参数，让H的轮廓更锐利）
//        Mat binary = new Mat();
//        opencv_imgproc.adaptiveThreshold(
//                enhanced,
//                binary,
//                255,
//                opencv_imgproc.ADAPTIVE_THRESH_GAUSSIAN_C,
//                opencv_imgproc.THRESH_BINARY_INV,
//                3,   // blockSize缩小，提升局部阈值精度
//                3     // C值调整，控制阈值偏移
//        );
//
//        // 5. 形态学操作（膨胀+腐蚀，让H的笔画更粗壮）
//        Mat kernel = opencv_imgproc.getStructuringElement(opencv_imgproc.MORPH_RECT, new Size(1, 1)); // 增大核尺寸
//        Mat morph = new Mat();
//        opencv_imgproc.dilate(binary, morph,kernel);   // 先膨胀（加粗字符）
//        opencv_imgproc.erode(morph, morph,kernel);    // 再腐蚀（修复膨胀后的边缘，保持字符形状）

        // 保存并释放资源
        opencv_imgcodecs.imwrite(inputPath, enhanced);
        src.release();
        gray.release();
//        blurred.release();
        enhanced.release();
//        binary.release();
//        kernel.release();
//        morph.release();

    }

}