ADD FFMPEG conversion

2025-05-08 21:59:03 +02:00
parent 920bcc32b1
commit d27276b66a
8 changed files with 274 additions and 10 deletions
--- a/src/main/java/com/ddf/vodsystem/services/FfmpegService.java
+++ b/src/main/java/com/ddf/vodsystem/services/FfmpegService.java
@@ -0,0 +1,140 @@
+package com.ddf.vodsystem.services;
+
+import lombok.Data;
+
+import java.io.BufferedReader;
+import java.io.File;
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.regex.Pattern;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import static java.lang.Long.parseLong;
+
+@Data
+public class FfmpegService {
+    private static final Logger logger = LoggerFactory.getLogger(FfmpegService.class);
+
+    private List<String> command;
+
+    private File inputFile;
+    private File outputFile;
+    private Float startPoint;
+    private Float endPoint;
+    private Integer width;
+    private Integer height;
+    private Float fps;
+    private Float fileSize;
+
+    private static final float AUDIO_RATIO = 0.2f;
+    private static final float BITRATE_MULTIPLIER = 0.9f;
+
+    private Pattern timePattern = Pattern.compile("time=([\\d:.]+)");
+    private long out_time_ms;
+
+    public FfmpegService(File file, File output) {
+        command = new ArrayList<>();
+        command.add("ffmpeg");
+        command.add("-progress");
+        command.add("pipe:1");
+        command.add("-y");
+
+        this.inputFile = file;
+        this.outputFile = output;
+
+    }
+
+    public void setResolution(Integer width, Integer height) {
+        this.width = width;
+        this.height = height;
+    }
+
+
+    private void buildFilters() {
+        List<String> filters = new ArrayList<>();
+        System.out.println(fps);
+        if (fps != null) {
+            filters.add("fps=" + fps);
+        }
+
+        if ((width != null && height == null) || (height != null && width == null)) {
+            String w = (width != null) ? width.toString() : "-1";
+            String h = (height != null) ? height.toString() : "-1";
+            filters.add("scale=" + w + ":" + h);
+        }
+
+        if (!filters.isEmpty()) {
+            command.add("-vf");
+            command.add(String.join(",", filters));
+        }
+    }
+
+    private void buildBitrate() {
+        float length = endPoint - startPoint;
+        float bitrate = (fileSize / length) * BITRATE_MULTIPLIER;
+
+        float video_bitrate = bitrate * (1 - AUDIO_RATIO);
+        float audio_bitrate = bitrate * AUDIO_RATIO;
+
+        command.add("-b:v");
+        command.add(video_bitrate + "k");
+        command.add("-b:a");
+        command.add(audio_bitrate + "k");
+    }
+
+    private void buildInputs(){
+        if (startPoint != null) {
+            command.add("-ss");
+            command.add(startPoint.toString());
+        }
+
+        command.add("-i");
+        command.add(inputFile.getAbsolutePath());
+
+        if (endPoint != null) {
+            command.add("-t");
+
+            Float duration = endPoint - startPoint;
+            command.add(duration.toString());
+        }
+    }
+
+    private ProcessBuilder buildCommand() {
+        buildInputs();
+        buildFilters();
+
+        if (fileSize != null) {
+            buildBitrate();
+        }
+
+        // Output file
+        command.add(outputFile.getAbsolutePath());
+
+        logger.info("Running command: {}", String.join(" ", command));
+        return new ProcessBuilder(command);
+    }
+
+    public void run() throws IOException, InterruptedException {
+        logger.info("FFMPEG starting...");
+        ProcessBuilder pb = buildCommand();
+        pb.redirectErrorStream(true);
+        Process process = pb.start();
+
+        BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
+
+        String line;
+        while ((line = reader.readLine()) != null) {
+
+            if (line.startsWith("out_time_ms=")) {
+                out_time_ms = parseLong(line.substring("out_time_ms=".length()));
+            }
+        }
+
+        logger.info("FFMPEG finished");
+    }
+
+}