package com.dy.pipIrrWebFile.util;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.FileUtils;
import org.jcodec.api.FrameGrab;
import org.jcodec.common.model.Picture;
import org.jcodec.scale.AWTUtil;
import org.springframework.web.multipart.MultipartFile;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.*;

/**
 * @Author: liurunyu
 * @Date: 2025/1/13 11:22
 * @Description
 */

/**
 * 视频操作工具类
 */
@Slf4j
public class VideoUtils {

    /*** 图片格式*/
    private static final String FILE_EXT = "jpg";

    /*** 帧数*/
    private static final int THUMB_FRAME = 5;

    /*** 图片格式*/
    private String fileExt ;

    /*** 帧数*/
    private Integer thumbFrame ;

    public VideoUtils(){
        this.fileExt = FILE_EXT;
        this.thumbFrame = THUMB_FRAME;
    }


    public VideoUtils(String fileExt, int thumbFrame){
        this.fileExt = fileExt;
        this.thumbFrame = thumbFrame;
        if(this.fileExt == null || this.fileExt.trim().equals("")){
            this.fileExt = FILE_EXT;
        }
        if(this.thumbFrame == null || this.thumbFrame.intValue() <= 0){
            this.thumbFrame = THUMB_FRAME;
        }
    }

    /**
     * 获取指定视频的帧并保存为图片至指定目录
     * @param videoFilePath 源视频文件路径
     * @param frameFilePath 截取帧的图片存放路径
     */
    public void fetchFrame(String videoFilePath, String frameFilePath) throws Exception {
        File videoFile = new File(videoFilePath);
        File frameFile = new File(frameFilePath);
        getThumbnail(videoFile, frameFile);
    }


    /**
     * 获取指定视频的帧并保存为图片至指定目录
     * @param videoFilePath 源视频文件路径
     */
    public BufferedImage fetchFrame(String videoFilePath) throws Exception {
        File videoFile = new File(videoFilePath);
        return getThumbnail(videoFile);
    }

    /**
     * 获取指定视频的帧并保存为图片至指定目录
     *
     * @param videoFile  源视频文件
     * @param targetFile 截取帧的图片
     */
    public void fetchFrame(MultipartFile videoFile, File targetFile) throws Exception {
        File file = new File(videoFile.getName());
        FileUtils.copyInputStreamToFile(videoFile.getInputStream(), file);
        getThumbnail(file, targetFile);
    }

    /**
     * 获取指定视频的帧并保存为图片至指定目录
     *
     * @param videoFile  源视频文件
     */
    public BufferedImage fetchFrame(MultipartFile videoFile) throws Exception {
        File file = new File(videoFile.getName());
        FileUtils.copyInputStreamToFile(videoFile.getInputStream(), file);
        return getThumbnail(file);
    }

    /**
     * 获取指定视频的帧并保存为图片至指定目录
     *
     * @param videoFile 源视频文件

    public File fetchFrame2(MultipartFile videoFile) {
        String originalFilename = videoFile.getOriginalFilename();
        File file = new File(originalFilename);
        File targetFile = null;
        try {
            FileUtils.copyInputStreamToFile(videoFile.getInputStream(), file);

            int i = originalFilename.lastIndexOf(".");
            String imageName;

            if (i > 0) {
                imageName = originalFilename.substring(0, i);
            } else {
                imageName = UUID.randomUUID().toString().replace("-", "");
            }
            imageName = imageName + ".jpg";
            targetFile = new File(imageName);
            getThumbnail(file, targetFile);
        } catch (Exception e) {
            log.error("获取视频指定帧异常：", e);
        } finally {
            if (file.exists()) {
                file.delete();
            }
        }
        log.debug("视频文件 - 帧截取 - 处理结束");
        return targetFile;
    }
    */

    /**
     * 获取第一帧缩略图
     *
     * @param videoFile  视频路径
     * @param targetFile 缩略图目标路径
     */
    public void getThumbnail(File videoFile, File targetFile) throws Exception {
        // 根据扩展名创建一个新文件路径
        Picture picture = FrameGrab.getFrameFromFile(videoFile, thumbFrame);
        BufferedImage bufferedImage = AWTUtil.toBufferedImage(picture);
        ImageIO.write(bufferedImage, fileExt, targetFile);
    }


    /**
     * 获取第一帧缩略图
     *
     * @param videoFile  视频路径
     */
    public BufferedImage getThumbnail(File videoFile) throws Exception {
        // 根据扩展名创建一个新文件路径
        Picture picture = FrameGrab.getFrameFromFile(videoFile, thumbFrame);
        BufferedImage bufferedImage = AWTUtil.toBufferedImage(picture);
        return bufferedImage;
    }


    public static void main(String[] args) {
        VideoUtils util = new VideoUtils();
        try {
            long startTime = System.currentTimeMillis();
            util.getThumbnail(new File("D:\\mp4\\test.mp4"), new File("D:\\mp4\\test.jpg"));
            System.out.println("截取图片耗时：" + (System.currentTimeMillis() - startTime));
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

}