springboot java ffmpeg 视频压缩、提取视频帧图片、获取视频分辨率

发布于:2024-12-19 ⋅ 阅读:(9) ⋅ 点赞:(0)

用到的maven依赖:

lombok依赖就不贴出来了

  <dependency>
            <groupId>org.bytedeco</groupId>
            <artifactId>ffmpeg-platform</artifactId>
            <version>4.3.2-1.5.5</version>
  </dependency>
   <dependency>
        <groupId>org.apache.commons</groupId>
        <artifactId>commons-lang3</artifactId>
        <version>3.12.0</version>
      </dependency>
 <dependency>
            <groupId>cn.hutool</groupId>
            <artifactId>hutool-all</artifactId>
            <version>5.5.2</version>
  </dependency>

工具类:

import cn.hutool.core.io.IoUtil;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.bytedeco.ffmpeg.avcodec.AVCodec;
import org.bytedeco.ffmpeg.avcodec.AVCodecContext;
import org.bytedeco.ffmpeg.avformat.AVFormatContext;
import org.bytedeco.ffmpeg.global.avcodec;
import org.bytedeco.ffmpeg.global.avformat;
import org.bytedeco.ffmpeg.global.avutil;
import org.bytedeco.javacpp.Loader;
import org.bytedeco.javacpp.PointerPointer;

import java.io.*;
import java.util.concurrent.TimeUnit;

@Slf4j
public class VideoUtils {

    static class LazyFfmpeg {
        private static final String ffmpeg = Loader.load(org.bytedeco.ffmpeg.ffmpeg.class);
    }

    public static String ffmpeg() {
        return LazyFfmpeg.ffmpeg;
    }

    /**
     * 压缩视频
     * @param inputFilePath 压缩前视频地址
     * @param outputFilePath 压缩后视频地址
     */
    public static void compressVideo(String inputFilePath, String outputFilePath) {

        if (StringUtils.isAnyBlank(inputFilePath, outputFilePath)) {
            throw new RuntimeException("输入视频路径或输出视频文件路径不能为空");
        }

        if (StringUtils.equals(inputFilePath, outputFilePath)) {
            throw new RuntimeException("outputFilePath不能和inputFilePath相同");
        }

        validIsFile(new File(inputFilePath));

        ProcessBuilder processBuilder = new ProcessBuilder(
                ffmpeg(),
                "-y",                     // 自动覆盖输出文件
                "-i", inputFilePath,      // 输入文件路径
                "-crf","30",
                "-c:v","h264",
                "-preset", "slow",            // 使用较慢的预设来提高压缩效率
//                "-b:v", "1000",  // 设置视频比特率为 1000 kbps
//                "-vf", String.format("scale=%s:%s", 1920, 1080),
//                "-c:a", "copy",           // 保持音频编码不变
                "-c:a", "aac",                  // 使用 AAC 音频编码
                "-b:a", "2k",                 // 设置音频比特率为 128 kbps
                outputFilePath            // 输出文件路径

        );

        StringBuilder stringBuilder = new StringBuilder();

        int exitCode;
        try {
            Process process = processBuilder.start();

            // 捕获错误输出
            processErrorMsg(process, stringBuilder);

            // 等待 FFmpeg 进程完成
            exitCode = process.waitFor();

        } catch (Throwable e) {
            throw new RuntimeException(e);
        }

        if (exitCode != 0) {
            throw new RuntimeException(stringBuilder.toString());
        }


    }

    /**
     * 提取图片
     *
     * @param videoPath 视频路径
     * @param second    提取指定时间图片
     * @param timeout   等待的最长时间
     * @param unit      参数的时间 timeout 单位
     * @return 图片
     */
    public static byte[] ffmpegExtractImage(String videoPath, Number second, long timeout, TimeUnit unit) {

        if (timeout <= 0) {
            throw new IllegalArgumentException("timeout不能小于等于0");
        }

        if (second == null) {
            second = 0;
        }

        if (unit == null) {
            unit = TimeUnit.MINUTES;
        }

        File videoFile = new File(videoPath);
        validIsFile(videoFile);
        ProcessBuilder extractBuilder = new ProcessBuilder(
                ffmpeg(),
                "-ss", second.toString(),
                "-i", videoPath,
                "-f", "image2pipe",
                "-vframes", "1",
//                "-vcodec", "png",//如果觉得照片不清晰,就启用此选项,但是照片会变大
                "-"
        );

        try {

            Process process = extractBuilder.start();

            try (InputStream inputStream = process.getInputStream()) {

                byte[] bytes = IoUtil.readBytes(inputStream);

                boolean result = process.waitFor(timeout, unit);
                if (!result) {
                    throw new RuntimeException("子进程退出之前已超过等待时间");
                }

                return bytes;
            }

        } catch (IOException | InterruptedException e) {
            throw new RuntimeException(e);
        }

    }

    /**
     * 获取视频分辨率
     *
     * @param videoFilePath 视频路径
     */
    public static int[] getVideoResolution(String videoFilePath) {

        validIsFile(new File(videoFilePath));

        AVFormatContext formatContext = avformat.avformat_alloc_context();
        AVCodecContext codecContext = avcodec.avcodec_alloc_context3(null);

        // 打开视频文件
        if (avformat.avformat_open_input(formatContext, videoFilePath, null, null) != 0) {
            throw new RuntimeException("无法打开视频文件");
        }

        // 获取视频流信息
        if (avformat.avformat_find_stream_info(formatContext, (PointerPointer) null) < 0) {
            throw new RuntimeException("无法获取视频流信息");
        }

        // 查找视频流
        int videoStreamIndex = -1;
        for (int i = 0; i < formatContext.nb_streams(); i++) {
            if (formatContext.streams(i).codecpar().codec_type() == avutil.AVMEDIA_TYPE_VIDEO) {
                videoStreamIndex = i;
                break;
            }
        }

        if (videoStreamIndex == -1) {
            throw new RuntimeException("视频流未找到");
        }

        // 获取视频解码器上下文
        avcodec.avcodec_parameters_to_context(codecContext, formatContext.streams(videoStreamIndex).codecpar());

        // 查找解码器
        AVCodec codec = avcodec.avcodec_find_decoder(codecContext.codec_id());
        if (codec == null) {
            throw new RuntimeException("无法找到解码器");
        }

        // 打开解码器
        if (avcodec.avcodec_open2(codecContext, codec, (PointerPointer) null) < 0) {
            throw new RuntimeException("无法打开解码器");
        }

        // 获取视频分辨率
        int width = codecContext.width();
        int height = codecContext.height();

        // 清理资源
        codecContext.close();

        return new int[]{width, height};
    }
    private static void processErrorMsg(Process process, StringBuilder stringBuilder) {

        new Thread(() -> {
            try (BufferedReader reader = new BufferedReader(new InputStreamReader(process.getErrorStream()))) {
                String line;
                while ((line = reader.readLine()) != null) {
                    stringBuilder.append(line);
                }
            } catch (IOException e) {
                log.error("打印命令行错误日志出现异常  errMsg:{}", e.getMessage(), e);
            }
        }).start();
    }

    public static void validIsFile(File file) {
        validExists(file);
        if (!file.isFile()) {
            throw new IllegalArgumentException("不是文件");
        }
    }

    public static void validExists(File file) {
        if (!file.exists()) {
            throw new IllegalArgumentException("videoPath不存在");
        }
    }
}


网站公告

今日签到

点亮在社区的每一天
去签到