OSDN Git Service

a02524e369fef66428ab626407448eba7e217336
[coroid/inqubus.git] / frontend / src / saccubus / worker / impl / convert / Convert.java
1 /* $Id$ */
2 package saccubus.worker.impl.convert;
3
4 import static org.apache.commons.io.FilenameUtils.getBaseName;
5 import static org.apache.commons.lang.StringUtils.*;
6 import static saccubus.worker.impl.convert.ConvertStatus.*;
7
8 import java.io.BufferedReader;
9 import java.io.File;
10 import java.io.IOException;
11 import java.io.InputStreamReader;
12 import java.io.UnsupportedEncodingException;
13 import java.net.URLEncoder;
14 import java.util.ArrayList;
15 import java.util.EnumMap;
16 import java.util.EnumSet;
17 import java.util.List;
18 import java.util.Map;
19 import java.util.Map.Entry;
20 import org.slf4j.Logger;
21 import org.slf4j.LoggerFactory;
22 import java.util.regex.Matcher;
23 import java.util.regex.Pattern;
24 import saccubus.conv.ConvertToVideoHook;
25 import saccubus.conv.CommentType;
26 import saccubus.util.FfmpegUtil;
27 import saccubus.worker.Worker;
28 import saccubus.worker.WorkerListener;
29 import saccubus.worker.profile.ConvertProfile;
30 import saccubus.worker.profile.ConvertProfile.HideCondition;
31 import saccubus.worker.profile.FfmpegProfile;
32 import saccubus.worker.profile.GeneralProfile;
33 import saccubus.worker.profile.OutputProfile;
34 import yukihane.inqubus.util.OutputNamePattern;
35 import yukihane.mediainfowrapper.Info;
36 import yukihane.mediainfowrapper.MediaInfo;
37 import yukihane.mediainfowrapper.Size;
38 import yukihane.swf.Cws2Fws;
39
40 /**
41  * 動画を(コメント付きに)変換するワーカクラス.
42  * @author yuki
43  */
44 public class Convert extends Worker<ConvertResult, ConvertProgress> {
45
46     private static final Logger logger = LoggerFactory.getLogger(Convert.class);
47     private final ConvertProfile profile;
48     private final File videoFile;
49     private final File commentFile;
50
51     public Convert(ConvertProfile profile, File video, File comment) {
52         this(profile, video, comment, null);
53     }
54
55     /**
56      * 変換ワーカコンストラクタ.
57      * @param profile 変換用プロファイル.
58      * @param video 変換元動画.
59      * @param comment 変換元コメント. コメントを付与しない場合はnull.
60      * @param output 変換後出力動画.
61      * @throws IOException 変換失敗.
62      */
63     public Convert(ConvertProfile profile, File video, File comment,
64             WorkerListener<ConvertResult, ConvertProgress> listener) {
65         super(listener);
66         this.profile = profile;
67         this.videoFile = video;
68         this.commentFile = comment;
69         logger.info("convert video:{}, comment:{}", videoFile, commentFile);
70     }
71
72     @Override
73     protected ConvertResult work() throws Exception {
74         if (!profile.isConvert()) {
75             return new ConvertResult(true, "");
76         }
77
78         final GeneralProfile gene = profile.getGeneralProfile();
79         final OutputProfile outprof = profile.getOutputProfile();
80         final OutputNamePattern outputPattern = new OutputNamePattern(outprof.getFileName());
81         final String id = outprof.getVideoId();
82         outputPattern.setId(isNotEmpty(id) ? id : "");
83         final String title = outprof.getTitile();
84         outputPattern.setTitle(isNotEmpty(title) ? title : "");
85         final String fileName = getBaseName(videoFile.getPath());
86         outputPattern.setFileName(fileName);
87         outputPattern.setReplaceFrom(gene.getReplaceFrom());
88         outputPattern.setReplaceFrom(gene.getReplaceTo());
89         final File outputFile = new File(outprof.getDir(),
90                 outputPattern.createFileName() + profile.getFfmpegOption().getExtOption());
91
92         final Map<CommentType, File> tmpComments = new EnumMap<>(CommentType.class);
93         try {
94
95             if (profile.isCommentOverlay()) {
96                 for (CommentType ct : CommentType.values()) {
97                     tmpComments.put(ct, File.createTempFile("vhk", ".tmp", profile.getTempDir()));
98                 }
99
100                 final HideCondition hide = profile.getNgSetting();
101
102                 for (CommentType ct : CommentType.values()) {
103                     publish(new ConvertProgress(PROCESS, -1.0, ct.toString() + "の中間ファイルへの変換中"));
104                     ConvertToVideoHook.convert(EnumSet.of(ct), commentFile, tmpComments.get(ct),
105                             hide.getId(), hide.getWord());
106                 }
107             }
108
109             checkStop();
110             publish(new ConvertProgress(PROCESS, -1.0, "動画の変換を開始"));
111
112             final int code = convert(outputFile, new EnumMap<>(tmpComments));
113             if (code != 0) {
114                 throw new IOException("ffmpeg実行失敗(code " + code + "): " + outputFile.getPath());
115             }
116             publish(new ConvertProgress(PROCESS, 100.0, "変換が正常に終了しました。"));
117             return new ConvertResult(true, outputFile.getName());
118         } finally {
119             for(File f : tmpComments.values()) {
120                 if(f != null && f.exists()) {
121                     f.delete();
122                 }
123             }
124         }
125     }
126
127     private int convert(File outputFile, Map<CommentType,File> tmpComments) throws InterruptedException, IOException {
128         File fwsFile = null;
129         try {
130             final File tmpCws = File.createTempFile("cws", ".swf", profile.getTempDir());
131             fwsFile = Cws2Fws.createFws(videoFile, tmpCws);
132             tmpCws.delete();
133             final File target = (fwsFile != null) ? fwsFile : videoFile;
134
135             final List<String> arguments = createArguments(target, outputFile, tmpComments);
136             final FfmpegUtil util = new FfmpegUtil(profile.getFfmpeg(), target);
137             int duration;
138             try {
139                 duration = util.getDuration();
140             } catch (IOException ex) {
141                 logger.info("動画再生時間を取得できませんでした: {}", target);
142                 duration = Integer.MAX_VALUE;
143             }
144             return executeFfmpeg(arguments, duration);
145         } finally {
146             if (fwsFile != null && fwsFile.exists()) {
147                 fwsFile.delete();
148             }
149         }
150     }
151
152     private List<String> createArguments(final File targetVideoFile, File output, Map<CommentType,File> comments)
153             throws IOException, UnsupportedEncodingException {
154         final ConvertProfile prof = profile;
155         final FfmpegProfile ffop = prof.getFfmpegOption();
156
157         final List<String> cmdList = new ArrayList<>();
158         cmdList.add(prof.getFfmpeg().getPath());
159         cmdList.add("-y");
160         final String[] mainOptions = ffop.getMainOption().split(" +");
161         for (String opt : mainOptions) {
162             if (isNotBlank(opt)) {
163                 cmdList.add(opt);
164             }
165         }
166         final String[] inOptions = ffop.getInOption().split(" +");
167         for (String opt : inOptions) {
168             if (isNotBlank(opt)) {
169                 cmdList.add(opt);
170             }
171         }
172         cmdList.add("-i");
173         cmdList.add(targetVideoFile.getPath());
174         final String[] outOptions = ffop.getOutOption().split(" +");
175         for (String opt : outOptions) {
176             if (isNotBlank(opt)) {
177                 cmdList.add(opt);
178             }
179         }
180         final Info info = MediaInfo.getInfo(profile.getMediaInfo(), targetVideoFile);
181         // 4:3 なら1.33, 16:9 なら1.76
182         final boolean isHD = ((double) info.getWidth() / (double) info.getHeight() > 1.5);
183         if (ffop.isResize()) {
184             final Size scaled = (ffop.isAdjustRatio()) ? MediaInfo.adjustSize(info, ffop.getResizeWidth(), ffop.
185                     getResizeHeight()) : new Size(info.getWidth(), info.getHeight());
186             cmdList.add("-s");
187             cmdList.add(scaled.getWidth() + "x" + scaled.getHeight());
188         }
189         final List<String> avfilterArgs = createAvfilterOptions(ffop.getAvfilterOption());
190         if (!prof.isVhookDisabled()) {
191             final String vhookArg = getVhookArg(prof, comments, isHD);
192             if (isNotBlank(vhookArg)) {
193                 avfilterArgs.add(vhookArg);
194             }
195         }
196         if (!avfilterArgs.isEmpty()) {
197             cmdList.add("-vfilters");
198             final String args = join(avfilterArgs, ", ");
199             cmdList.add(args);
200         }
201         cmdList.add(output.getPath());
202
203         logger.info("arg: {}", cmdList);
204         return cmdList;
205     }
206     private static final Pattern PATTERN_TIME = Pattern.compile("time=(\\d+):(\\d+):(\\d+)");
207
208     private int executeFfmpeg(final List<String> cmdList, int duration) throws InterruptedException, IOException {
209         Process process = null;
210         try {
211             logger.info("Processing FFmpeg...");
212             process = Runtime.getRuntime().exec(cmdList.toArray(new String[0]));
213             BufferedReader ebr = new BufferedReader(new InputStreamReader(
214                     process.getErrorStream()));
215             String msg;
216             while ((msg = ebr.readLine()) != null) {
217                 if (msg.startsWith("frame=")) {
218                     final Matcher m = PATTERN_TIME.matcher(msg);
219                     double per = -1.0;
220                     if (m.find()) {
221                         final double hour = Integer.parseInt(m.group(1));
222                         final double min = Integer.parseInt(m.group(2));
223                         final double sec = Integer.parseInt(m.group(3));
224                         final double time = ((hour * 60) + min) * 60 + sec;
225                         per = 100.0 * time / duration;
226                         if (logger.isTraceEnabled()) {
227                             logger.trace("time:{}, duration:{}", time, duration);
228                             logger.trace(msg);
229                         }
230                     }
231                     publish(new ConvertProgress(PROCESS, per, msg));
232                 } else if (!msg.endsWith("No accelerated colorspace conversion found")) {
233                     logger.warn(msg);
234                 } else {
235                     logger.info(msg);
236                 }
237
238                 checkStop();
239             }
240
241             process.waitFor();
242             return process.exitValue();
243         } finally {
244             if (process != null) {
245                 process.destroy();
246             }
247         }
248     }
249
250     private static List<String> createAvfilterOptions(String avfilterOption) {
251         final List<String> avfilterArgs = new ArrayList<>();
252         if (isNotBlank(avfilterOption)) {
253             avfilterArgs.add(avfilterOption);
254         }
255         return avfilterArgs;
256     }
257
258     private static String getVhookArg(ConvertProfile prof, Map<CommentType, File> comments, boolean isHD) throws
259             UnsupportedEncodingException {
260         StringBuilder sb = new StringBuilder();
261         sb.append("vhext=");
262         sb.append(prof.getVhook().getPath().replace("\\", "/"));
263         if (prof.isCommentOverlay()) {
264             for(Entry<CommentType, File> e : comments.entrySet()) {
265                 sb.append("|");
266                 sb.append(e.getKey().getVhookOptionPrefix());
267                 sb.append(URLEncoder.encode(e.getValue().getPath().replace("\\", "/"), "Shift_JIS"));
268             }
269         }
270         sb.append("|");
271         sb.append("--font:");
272         sb.append(URLEncoder.encode(
273                 prof.getFont().getPath().replace("\\", "/"), "Shift_JIS"));
274         sb.append("|");
275         sb.append("--font-index:");
276         sb.append(prof.getFontIndex());
277         sb.append("|");
278         sb.append("--show-user:");
279         final int dispNum = (prof.getMaxNumOfComment() < 0) ? 30 : prof.getMaxNumOfComment();
280         sb.append(dispNum);
281         sb.append("|");
282         sb.append("--shadow:");
283         sb.append(prof.getShadowIndex());
284         sb.append("|");
285         if (prof.isShowConverting()) {
286             sb.append("--enable-show-video");
287             sb.append("|");
288         }
289         if (!prof.isDisableFontSizeArrange()) {
290             sb.append("--enable-fix-font-size");
291             sb.append("|");
292         }
293         if (prof.isCommentOpaque()) {
294             sb.append("--enable-opaque-comment");
295             sb.append("|");
296         }
297         if (isHD) {
298             sb.append("--aspect-mode:1");
299             sb.append("|");
300         }
301         return sb.toString();
302     }
303
304     protected void checkStop() throws InterruptedException {
305         if (Thread.interrupted()) {
306             throw new InterruptedException("中止要求を受け付けました");
307         }
308     }
309 }