diff --git a/nas-common/src/main/java/top/itning/yunshunas/common/util/CommandUtils.java b/nas-common/src/main/java/top/itning/yunshunas/common/util/CommandUtils.java index 8c47675e..1815c9b8 100644 --- a/nas-common/src/main/java/top/itning/yunshunas/common/util/CommandUtils.java +++ b/nas-common/src/main/java/top/itning/yunshunas/common/util/CommandUtils.java @@ -41,4 +41,23 @@ public static void process(List command, Consumer commandInfo) t } } } + + public static String process(List command) throws IOException { + ProcessBuilder builder = new ProcessBuilder(command); + builder.redirectErrorStream(true); + Process process = builder.start(); + + InputStream inputStream = process.getInputStream(); + InputStreamReader isr = new InputStreamReader(inputStream); + BufferedReader br = new BufferedReader(isr); + + StringBuilder sb = new StringBuilder(); + try (inputStream; isr; br) { + String line; + while ((line = br.readLine()) != null) { + sb.append(line); + } + } + return sb.toString(); + } } diff --git a/nas-video/src/main/java/top/itning/yunshunas/video/video/FFmpegUtils.java b/nas-video/src/main/java/top/itning/yunshunas/video/video/FFmpegUtils.java new file mode 100644 index 00000000..4b4c892c --- /dev/null +++ b/nas-video/src/main/java/top/itning/yunshunas/video/video/FFmpegUtils.java @@ -0,0 +1,189 @@ +package top.itning.yunshunas.video.video; + +import lombok.Builder; +import lombok.Data; +import lombok.extern.slf4j.Slf4j; +import org.apache.commons.lang3.math.NumberUtils; +import top.itning.yunshunas.common.util.CommandUtils; + +import java.io.IOException; +import java.util.ArrayList; +import java.util.Arrays; +import java.util.List; +import java.util.Optional; +import java.util.function.Consumer; + +/** + * @author itning + * @since 2024/2/12 13:42 + */ +@Slf4j +public class FFmpegUtils { + /** + * 获取视频文件的视频编码信息（第一个视频流） + * + * @param ffprobePath ffprobe文件路径 + * @param file 视频文件 + * @return 编码信息 + * @throws IOException 执行失败 + */ + public static String getVideoCodec(String ffprobePath, String file) throws IOException { + return CommandUtils.process(Arrays.asList( + ffprobePath, + "-v", "error", + "-select_streams", "v:0", + "-show_entries", "stream=codec_name", + "-of", "csv=p=0", + file + )); + } + + /** + * 获取视频文件的音频编码信息（第一个音频流） + * + * @param ffprobePath ffprobe文件路径 + * @param file 视频文件 + * @return 编码信息 + * @throws IOException 执行失败 + */ + public static String getAudioCodec(String ffprobePath, String file) throws IOException { + return CommandUtils.process(Arrays.asList( + ffprobePath, + "-v", "error", + "-select_streams", "a:0", + "-show_entries", "stream=codec_name", + "-of", "csv=p=0", + file + )); + } + + /** + * 获取视频文件的总帧数信息（第一个视频流） + * + * @param ffprobePath ffprobe文件路径 + * @param file 视频文件 + * @return 帧数信息，获取失败时返回-1 + * @throws IOException 执行失败 + */ + public static long getVideoTotalFrame(String ffprobePath, String file) throws IOException { + return NumberUtils.toLong(CommandUtils.process(Arrays.asList( + ffprobePath, + "-v", "error", + "-select_streams", "v:0", + "-show_entries", "stream=nb_frames", + "-of", "csv=p=0", + file + )), -1); + } + + /** + * 获取视频文件的平均码率信息（第一个视频流） + * + * @param ffprobePath ffprobe文件路径 + * @param file 视频文件 + * @return 码率信息，获取失败时返回-1 + * @throws IOException 执行失败 + */ + public static long getVideoBitRate(String ffprobePath, String file) throws IOException { + return NumberUtils.toLong(CommandUtils.process(Arrays.asList( + ffprobePath, + "-v", "error", + "-select_streams", "v:0", + "-show_entries", "stream=bit_rate", + "-of", "csv=p=0", + file + )), -1); + } + + public static void convertedToM3u8(String ffmpegPath, + String file, + String outFileName, + ConvertedM3u8Params params, + Consumer commandInfo) throws IOException { + List command = new ArrayList<>(); + command.add(ffmpegPath); + command.add("-y"); + command.add("-loglevel"); + command.add("quiet"); + command.add("-i"); + command.add(file); + command.add("-codec:v"); + command.add(Optional.ofNullable(params.getVideoCodec()).orElse("copy")); + if (null != params.getVideoCodec()) { + Optional.ofNullable(params.getVideoBitRate()).ifPresent(videoBitRate -> { + command.add("-b:v"); + command.add(videoBitRate); + }); + Optional.ofNullable(params.getMinVideoBitRate()).ifPresent(minVideoBitRate -> { + command.add("-minrate"); + command.add(minVideoBitRate); + }); + Optional.ofNullable(params.getMinVideoBitRate()).ifPresent(maxVideoBitRate -> { + command.add("-maxrate"); + command.add(maxVideoBitRate); + }); + Optional.ofNullable(params.getMinVideoBitRate()).ifPresent(videoBufSize -> { + command.add("-bufsize"); + command.add(videoBufSize); + }); + } + command.add("-codec:a"); + command.add(Optional.ofNullable(params.getAudioCodec()).orElse("copy")); + if (null != params.getAudioCodec()) { + Optional.ofNullable(params.getAudioBitRate()).ifPresent(audioBitRate -> { + command.add("-b:a"); + command.add(audioBitRate); + }); + } + command.add("-hls_time"); + command.add(Optional.ofNullable(params.getHlsTime()).orElse(10).toString()); + command.add("-hls_list_size"); + command.add(Optional.ofNullable(params.getHlsListSize()).orElse(0).toString()); + command.add("-start_number"); + command.add(Optional.ofNullable(params.getStartNumber()).orElse(0).toString()); + command.add(outFileName); + command.add("-progress"); + command.add("-"); + command.add("-nostats"); + log.info("The final command: {}", String.join(" ", command)); + //- `frame=388`：已处理的帧数。这表示 FFmpeg 已经读取、解码、可能重新编码和写入了388帧。 + //- `fps=127.16`：当前的处理速度，以帧每秒（Frames Per Second）计。这里的意思是 FFmpeg 正在以每秒处理约127.16帧的速度运行。 + //- `stream_0_0_q=29.0`：当前视频流的量化参数。量化参数越低，输出视频质量越高（但文件大小也越大）。这个参数的具体含义可能会根据使用的编解码器有所不同。 + //- `bitrate=1104.8kbits/s`：当前的平均比特率，以千比特每秒（kilobits per second）计。这表示视频（和可能的音频）数据的平均数据传输速率。 + //- `total_size=2097200`：到目前为止生成的输出文件大小，以字节为单位。 + //- `out_time_us=15185850`：到目前为止的输出时长，以微秒为单位。 + //- `out_time_ms=15185850`：到目前为止的输出时长，以毫秒为单位。与 `out_time_us` 相同，只不过单位不同。 + //- `out_time=00:00:15.185850`：到目前为止的输出时长，以时:分:秒.微秒的格式表示。这里显示的是已经处理了大约15秒的视频。 + //- `dup_frames=0`：到目前为止，由于编码过程中的某些需求（例如帧率转换），被复制的帧数。这里的0表示没有帧被复制。 + //- `drop_frames=0`：到目前为止，被丢弃的帧数。帧可能会因为多种原因被丢弃，例如保持同步或适应目标帧率。这里的0表示没有帧被丢弃。 + //- `speed=4.98x`：处理速度相对于实时播放速度的比率。这里的4.98x意味着 FFmpeg 正在以实时速度的近5倍处理视频。 + //- `progress=continue`：这个输出指示处理的当前状态。这里可能是一个打印错误，通常情况下应该是 `progress=continue` 表示处理正在继续。如果处理完成，这里会显示 `progress=end`。 + CommandUtils.process(command, commandInfo); + } + + @Data + @Builder + public static class ConvertedM3u8Params { + private String videoCodec; + private String audioCodec; + private Integer startNumber; + private Integer hlsTime; + private Integer hlsListSize; + private String videoBitRate; + private String audioBitRate; + private String minVideoBitRate; + private String maxVideoBitRate; + private String videoBufSize; + + public static ConvertedM3u8Params defaultParams() { + return ConvertedM3u8Params.builder() + .videoCodec("libx264") + .audioCodec("aac") + .videoBitRate("5000k") + .minVideoBitRate("2000k") + .maxVideoBitRate("10000k") + .videoBufSize("5000k") + .build(); + } + } +}