5bf756267b153d07fdf8650d7c3ac8b2f3f3e833
[meviz.git] / src / main / java / eu / svjatoslav / meviz / encoder / converters / FFMpegVideo.java
1 /*
2  * Meviz - Various tools collection to work with multimedia.
3  * Copyright (C) 2012 -- 2019, Svjatoslav Agejenko, svjatoslav@svjatoslav.eu
4  *
5  * This program is free software; you can redistribute it and/or
6  * modify it under the terms of version 3 of the GNU Lesser General Public License
7  * or later as published by the Free Software Foundation.
8 */
9
10 package eu.svjatoslav.meviz.encoder.converters;
11
12 import eu.svjatoslav.meviz.encoder.BitrateParameter.Bitrate;
13 import eu.svjatoslav.meviz.encoder.EncodingOptions;
14
15 import java.io.File;
16 import java.util.List;
17
18 public class FFMpegVideo extends AbstractConverter {
19
20     private String constructCodecParamsString(final EncodingOptions options,
21                                               final int videoBitrate, final int audioBitrate,
22                                               final String videoCodec, final String audioCodec) {
23
24         final StringBuilder codecParams = new StringBuilder();
25
26         if (audioCodec == null)
27             codecParams.append("-an ");
28         else
29             codecParams.append("-acodec " + audioCodec + " ");
30
31         codecParams.append("-vcodec " + videoCodec + " ");
32
33         if (videoBitrate != -1)
34             codecParams.append("-b " + videoBitrate + "k ");
35
36         if (audioBitrate != -1)
37             codecParams.append("-b:a " + audioBitrate + "k ");
38
39         if (options.isDeinterlace())
40             codecParams.append("-filter:v yadif ");
41
42         if (options.getTargetFps() != null)
43             codecParams.append("-r " + options.getTargetFps() + " ");
44
45         return codecParams.toString();
46     }
47
48     private int getAudioBitrateValue(final Bitrate bitRate) {
49         switch (bitRate) {
50             case LOW:
51                 return 128;
52
53             case MEDIUM:
54                 return 160;
55
56             case HIGH:
57                 return 320;
58
59             case COPY:
60                 return -1;
61
62             case NONE:
63                 return -1;
64
65             case LOSSLESS:
66                 return -1;
67
68             default:
69                 throw new RuntimeException("Audio bitrate: " + bitRate
70                         + " is not supported.");
71         }
72     }
73
74     @Override
75     public String getCommand(final File inputFile, final File targetFile,
76                              final EncodingOptions options, final String targetFormat) {
77
78         int videoBitrate = getVideoBitrateValue(options.getVideoBitrate());
79         int audioBitrate = getAudioBitrateValue(options.getAudioBitrate());
80
81         // convert
82         final StringBuilder codecParams = new StringBuilder();
83
84         String videoCodec = "libx264";
85         String audioCodec = "libmp3lame";
86
87         if (targetFormat.equals("webm")) {
88             videoCodec = "vp8";
89             audioCodec = "opus";
90             codecParams.append("-s 800x450 ");
91         }
92
93         if (options.getVideoBitrate() == Bitrate.COPY)
94             videoCodec = "copy";
95
96         if (options.getAudioBitrate() == Bitrate.COPY)
97             audioCodec = "copy";
98
99         if (options.getAudioBitrate() == Bitrate.LOSSLESS){
100             audioCodec = "flac";
101             codecParams.append("-strict -2 ");
102         }
103
104         if (options.getAudioBitrate() == Bitrate.NONE)
105             audioCodec = null;
106
107         if (options.isForPortablePlayer()) {
108             videoBitrate = 1000;
109             audioBitrate = 128;
110             videoCodec = "libxvid";
111
112             // reduce resolution
113             codecParams.append("-s 640x480 ");
114
115             // enforce maximum keyframe interval
116             codecParams.append("-g 150 ");
117         }
118
119         codecParams.append(constructCodecParamsString(options, videoBitrate,
120                 audioBitrate, videoCodec, audioCodec));
121
122         return "ffmpeg -i \"" + inputFile.getAbsolutePath() + "\" "
123                 + codecParams.toString() + "\"" + targetFile.getAbsolutePath()
124                 + "\"";
125     }
126
127     @Override
128     public List<String> getSourceFileExtensions() {
129         return getSupportedExtensions();
130     }
131
132     private List<String> getSupportedExtensions() {
133         return toList("mkv", "mts", "mp4", "avi", "mpg", "mpeg", "vob", "m4v",
134                 "webm", "mov", "asf", "3gp");
135     }
136
137     @Override
138     public List<String> getTargetFileExtensions() {
139         return getSupportedExtensions();
140     }
141
142     private int getVideoBitrateValue(final Bitrate bitRate) {
143         switch (bitRate) {
144             case LOW:
145                 return 1000;
146
147             case MEDIUM:
148                 return 4000;
149
150             case HIGH:
151                 return 16000;
152
153             case COPY:
154                 return -1;
155
156             default:
157                 throw new RuntimeException("Video bitrate: " + bitRate
158                         + " is not supported.");
159         }
160     }
161
162     @Override
163     public boolean isTerminalMandatory() {
164         return true;
165     }
166
167 }