OSDN Git Service

da8641afc630e1ed946ee734f58fe0ccf6b13e15
[coroid/inqubus.git] / frontend / src / saccubus / converter / FfmpegCommand.java
1 /* $Id$ */
2 package saccubus.converter;
3
4 import java.io.BufferedReader;
5 import java.io.File;
6 import java.io.IOException;
7 import java.io.InputStreamReader;
8 import java.io.UnsupportedEncodingException;
9 import java.net.URLEncoder;
10 import java.util.ArrayList;
11 import java.util.List;
12 import org.apache.commons.lang.StringUtils;
13 import saccubus.ConvertStopFlag;
14 import saccubus.conv.ConvertToVideoHook;
15 import saccubus.converter.profile.Ffmpeg;
16 import saccubus.converter.profile.FfmpegOption;
17 import saccubus.converter.profile.GeneralSetting;
18 import saccubus.converter.profile.NgSetting;
19 import saccubus.net.TextProgressListener;
20 import yukihane.mediainfowrapper.Info;
21 import yukihane.mediainfowrapper.MediaInfo;
22 import yukihane.mediainfowrapper.Size;
23 import yukihane.swf.Cws2Fws;
24
25 /**
26  *
27  * @author yuki
28  */
29 public class FfmpegCommand extends AbstractCommand {
30
31     private final File commentMiddleFile;
32     private final File tcommMiddleFile;
33     private final File TMP_CWS;
34     private final File commentFile;
35     private final File tcommFile;
36     private final File videoFile;
37     private final File convertedVideoFile;
38     private final Ffmpeg ffmpeg;
39
40     FfmpegCommand(TextProgressListener listener, ConvertStopFlag flag, File commentFile, File tcommFile,
41             File videoFile, File convertedVideoFile, Ffmpeg ffmpeg, GeneralSetting general) throws IOException {
42         super(listener, flag);
43         this.commentFile = commentFile;
44         this.tcommFile = tcommFile;
45         this.videoFile = videoFile;
46         this.convertedVideoFile = convertedVideoFile;
47         this.ffmpeg = ffmpeg;
48
49         File tmpDir = general.getTempDir();
50         commentMiddleFile = File.createTempFile("vhk", ".tmp", tmpDir);
51         tcommMiddleFile = File.createTempFile("tcom", ".tmp", tmpDir);
52         TMP_CWS = File.createTempFile("cws", ".swf", tmpDir);
53     }
54
55     public boolean execute() throws InterruptedException, IOException {
56         try {
57             return exec();
58         } finally {
59             if (commentMiddleFile.exists()) {
60                 commentMiddleFile.delete();
61             }
62             if (tcommMiddleFile.exists()) {
63                 tcommMiddleFile.delete();
64             }
65             if (TMP_CWS.exists()) {
66                 TMP_CWS.delete();
67             }
68         }
69     }
70
71     private boolean exec() throws InterruptedException, IOException {
72         final NgSetting ngSetting = getFfmpeg().getNgSetting();
73         if (commentFile != null) {
74             sendText("コメントの中間ファイルへの変換中");
75             boolean conv = ConvertToVideoHook.convert(commentFile, commentMiddleFile, ngSetting.getId(), ngSetting.
76                     getWord());
77             if (!conv) {
78                 sendText("コメント変換に失敗。ファイル名に使用できない文字が含まれているか正規表現の間違い?");
79                 return false;
80             }
81         }
82         stopFlagReturn();
83         if (tcommFile != null) {
84             sendText("投稿者コメントの中間ファイルへの変換中");
85             boolean conv = ConvertToVideoHook.convert(tcommFile, tcommMiddleFile, ngSetting.getId(), ngSetting.getWord());
86             if (!conv) {
87                 sendText("コメント変換に失敗。ファイル名に使用できない文字が含まれているか正規表現の間違い?");
88                 return false;
89             }
90         }
91         stopFlagReturn();
92         sendText("動画の変換を開始");
93         int code;
94         if ((code = converting_video(videoFile, convertedVideoFile, (commentFile != null), commentMiddleFile.getPath(),
95                 (tcommFile
96                 != null), tcommMiddleFile.getPath(), getFfmpeg().getFfmpegOption())) == 0) {
97             sendText("変換が正常に終了しました。");
98             return true;
99         } else {
100             sendText("変換エラー:" + convertedVideoFile.getPath());
101         }
102         return false;
103     }
104
105     private int converting_video(File videoFile, File convertedVideoFile, boolean addComment, String commPath,
106             boolean addTcomment, String tcommPath, FfmpegOption ov) throws InterruptedException, IOException {
107         File fwsFile = Cws2Fws.createFws(videoFile, TMP_CWS);
108
109         List<String> cmdList = new ArrayList<String>();
110         cmdList.add(getFfmpeg().getFfmpeg().getPath());
111         cmdList.add("-y");
112         String[] mainOptions = ov.getMainOption().split(" +");
113         for (String opt : mainOptions) {
114             if (StringUtils.isNotBlank(opt)) {
115                 cmdList.add(opt);
116             }
117         }
118         String[] inOptions = ov.getInOption().split(" +");
119         for (String opt : inOptions) {
120             if (StringUtils.isNotBlank(opt)) {
121                 cmdList.add(opt);
122             }
123         }
124         cmdList.add("-i");
125
126         if (fwsFile == null) {
127             cmdList.add(videoFile.getPath());
128         } else {
129             cmdList.add(fwsFile.getPath());
130         }
131         String[] outOptions = ov.getOutOption().split(" +");
132         for (String opt : outOptions) {
133             if (StringUtils.isNotBlank(opt)) {
134                 cmdList.add(opt);
135             }
136         }
137
138         final Info info = MediaInfo.getInfo(new File("bin", "MediaInfo"), videoFile);
139         // 4:3 なら1.33, 16:9 なら1.76
140         boolean isHD = ((double)info.getWidth()/(double)info.getHeight() > 1.5);
141         if (ov.isResize()) {
142             final Size scaled = (ov.isAdjustRatio()) ? MediaInfo.adjustSize(info, ov.getResizeWidth(), ov.
143                     getResizeHeight()) : new Size(info.getWidth(), info.getHeight());
144             cmdList.add("-s");
145             cmdList.add(scaled.getWidth() + "x" + scaled.getHeight());
146         }
147
148         List<String> avfilterArgs = getAvfilterOptions(ov.getAvfilterOption());
149         if (!getFfmpeg().isVhookDisabled()) {
150             String vhookArg = getVhookArg(addComment, commPath, addTcomment, tcommPath, isHD);
151             if(StringUtils.isNotBlank(vhookArg)){
152                 avfilterArgs.add(vhookArg);
153             }
154         }
155
156         if (!avfilterArgs.isEmpty()) {
157             cmdList.add("-vfilters");
158             final String args = "\"" + StringUtils.join(avfilterArgs, ", ") + "\"";
159             cmdList.add(args);
160         }
161
162         cmdList.add(convertedVideoFile.getPath());
163
164         System.out.print("arg:");
165         for (String s : cmdList) {
166             System.out.print(" " + s);
167         }
168         System.out.println();
169
170         try {
171             System.out.println("\n\n----\nProcessing FFmpeg...\n----\n\n");
172             Process process = Runtime.getRuntime().exec(cmdList.toArray(new String[0]));
173             BufferedReader ebr = new BufferedReader(new InputStreamReader(
174                     process.getErrorStream()));
175             String e;
176             while ((e = ebr.readLine()) != null) {
177                 String state = e;
178                 if (state.startsWith("frame=")) {
179                     sendText(state);
180                 } else if (!state.endsWith("No accelerated colorspace conversion found")) {
181                     System.out.println(e);
182                 }
183
184                 try {
185                     stopFlagReturn();
186                 } catch (InterruptedException ex) {
187                     process.destroy();
188                     throw ex;
189                 }
190
191             }
192             process.waitFor();
193             return process.exitValue();
194         } finally {
195             if (fwsFile != null) {
196                 fwsFile.delete();
197             }
198         }
199     }
200
201     private List<String> getAvfilterOptions(String avfilterOption) {
202         final List<String> avfilterArgs = new ArrayList<String>();
203         if (StringUtils.isNotBlank(avfilterOption)) {
204             avfilterArgs.add(avfilterOption);
205         }
206         return avfilterArgs;
207     }
208
209     private String getVhookArg(boolean addComment, String commPath, boolean addTcomment,
210             String tcommPath, boolean isHD) throws UnsupportedEncodingException {
211         StringBuilder sb = new StringBuilder();
212         sb.append("vhext=");
213         sb.append(getFfmpeg().getVhook().getPath().replace("\\", "/"));
214         if (addComment) {
215             sb.append("|");
216             sb.append("--data-user:");
217             sb.append(URLEncoder.encode(commPath.replace("\\", "/"), "Shift_JIS"));
218         }
219         if (addTcomment) {
220             sb.append("|");
221             sb.append("--data-owner:");
222             sb.append(URLEncoder.encode(tcommPath.replace("\\", "/"), "Shift_JIS"));
223         }
224         sb.append("|");
225         sb.append("--font:");
226         sb.append(URLEncoder.encode(
227                 getFfmpeg().getFont().getPath().replace("\\", "/"), "Shift_JIS"));
228         sb.append("|");
229         sb.append("--font-index:");
230         sb.append(getFfmpeg().getFontIndex());
231         sb.append("|");
232         sb.append("--show-user:");
233         sb.append(getFfmpeg().getMaxNumOfComment());
234         sb.append("|");
235         sb.append("--shadow:");
236         sb.append(getFfmpeg().getShadowIndex());
237         sb.append("|");
238         if (getFfmpeg().isShowConverting()) {
239             sb.append("--enable-show-video");
240             sb.append("|");
241         }
242         if (getFfmpeg().isSelfAdjustFontSize()) {
243             sb.append("--enable-fix-font-size");
244             sb.append("|");
245         }
246         if (getFfmpeg().isCommentOpaque()) {
247             sb.append("--enable-opaque-comment");
248             sb.append("|");
249         }
250         if(isHD){
251             sb.append("--aspect-mode:1");
252             sb.append("|");
253         }
254         return sb.toString();
255     }
256
257     private Ffmpeg getFfmpeg() {
258         return ffmpeg;
259     }
260 }