2 package saccubus.converter;
4 import java.io.BufferedReader;
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 java.util.logging.Level;
13 import java.util.logging.Logger;
14 import javax.swing.SwingWorker;
15 import org.apache.commons.lang.StringUtils;
16 import saccubus.conv.ConvertToVideoHook;
17 import saccubus.converter.profile.ConvertProfile;
18 import saccubus.converter.profile.FfmpegProfile;
19 import saccubus.converter.profile.GeneralProfile;
20 import yukihane.mediainfowrapper.Info;
21 import yukihane.mediainfowrapper.MediaInfo;
22 import yukihane.mediainfowrapper.Size;
23 import saccubus.converter.profile.ConvertProfile.HideCondition;
24 import saccubus.converter.profile.Profile;
25 import yukihane.swf.Cws2Fws;
31 public class Ffmpeg extends SwingWorker<FfmpegResult, FfmpegProgress> {
33 private static final Logger logger = Logger.getLogger(Ffmpeg.class.getName());
34 private final File commentMiddleFile;
35 private final File tcommMiddleFile;
36 private final File TMP_CWS;
37 private final File commentFile;
38 private final File videoFile;
39 private final File convertedVideoFile;
40 private final ConvertProfile ffmpeg;
42 Ffmpeg(File commentFile, File videoFile, File convertedVideoFile, ConvertProfile ffmpeg) throws IOException {
43 this.commentFile = commentFile;
44 this.videoFile = videoFile;
45 this.convertedVideoFile = convertedVideoFile;
48 File tmpDir = ffmpeg.getTempDir();
49 commentMiddleFile = File.createTempFile("vhk", ".tmp", tmpDir);
50 tcommMiddleFile = File.createTempFile("tcom", ".tmp", tmpDir);
51 TMP_CWS = File.createTempFile("cws", ".swf", tmpDir);
55 protected FfmpegResult doInBackground() throws Exception {
59 if (commentMiddleFile.exists()) {
60 commentMiddleFile.delete();
62 if (tcommMiddleFile.exists()) {
63 tcommMiddleFile.delete();
65 if (TMP_CWS.exists()) {
71 private FfmpegResult exec() throws InterruptedException, IOException {
72 final HideCondition ngSetting = getFfmpeg().getNgSetting();
73 if (commentFile != null) {
74 publish(new FfmpegProgress("コメントの中間ファイルへの変換中"));
75 boolean conv = ConvertToVideoHook.convert(commentFile, commentMiddleFile, ngSetting.getId(), ngSetting.
78 publish(new FfmpegProgress("コメント変換に失敗。ファイル名に使用できない文字が含まれているか正規表現の間違い?"));
79 return new FfmpegResult(false);
83 // if (tcommFile != null) {
84 // sendText("投稿者コメントの中間ファイルへの変換中");
85 // boolean conv = ConvertToVideoHook.convert(tcommFile, tcommMiddleFile, ngSetting.getId(), ngSetting.getWord());
87 // sendText("コメント変換に失敗。ファイル名に使用できない文字が含まれているか正規表現の間違い?");
92 publish(new FfmpegProgress("動画の変換を開始"));
94 if ((code = converting_video(videoFile, convertedVideoFile, (commentFile != null), commentMiddleFile.getPath(),
95 false, tcommMiddleFile.getPath(), getFfmpeg().getFfmpegOption())) == 0) {
96 publish(new FfmpegProgress("変換が正常に終了しました。"));
97 return new FfmpegResult(true);
99 publish(new FfmpegProgress("変換エラー:" + convertedVideoFile.getPath()));
101 return new FfmpegResult(false);
104 private int converting_video(File videoFile, File convertedVideoFile, boolean addComment, String commPath,
105 boolean addTcomment, String tcommPath, FfmpegProfile ov) throws InterruptedException, IOException {
106 File fwsFile = Cws2Fws.createFws(videoFile, TMP_CWS);
108 List<String> cmdList = new ArrayList<String>();
109 cmdList.add(getFfmpeg().getFfmpeg().getPath());
111 String[] mainOptions = ov.getMainOption().split(" +");
112 for (String opt : mainOptions) {
113 if (StringUtils.isNotBlank(opt)) {
117 String[] inOptions = ov.getInOption().split(" +");
118 for (String opt : inOptions) {
119 if (StringUtils.isNotBlank(opt)) {
125 if (fwsFile == null) {
126 cmdList.add(videoFile.getPath());
128 cmdList.add(fwsFile.getPath());
130 String[] outOptions = ov.getOutOption().split(" +");
131 for (String opt : outOptions) {
132 if (StringUtils.isNotBlank(opt)) {
137 final Info info = MediaInfo.getInfo(new File("bin", "MediaInfo"), videoFile);
138 // 4:3 なら1.33, 16:9 なら1.76
139 boolean isHD = ((double) info.getWidth() / (double) info.getHeight() > 1.5);
141 final Size scaled = (ov.isAdjustRatio()) ? MediaInfo.adjustSize(info, ov.getResizeWidth(), ov.
142 getResizeHeight()) : new Size(info.getWidth(), info.getHeight());
144 cmdList.add(scaled.getWidth() + "x" + scaled.getHeight());
147 List<String> avfilterArgs = getAvfilterOptions(ov.getAvfilterOption());
148 if (!getFfmpeg().isVhookDisabled()) {
149 String vhookArg = getVhookArg(addComment, commPath, addTcomment, tcommPath, isHD);
150 if (StringUtils.isNotBlank(vhookArg)) {
151 avfilterArgs.add(vhookArg);
155 if (!avfilterArgs.isEmpty()) {
156 cmdList.add("-vfilters");
157 final String args = "\"" + StringUtils.join(avfilterArgs, ", ") + "\"";
161 cmdList.add(convertedVideoFile.getPath());
163 final StringBuilder argMsg = new StringBuilder();
164 argMsg.append("arg:");
165 for (String s : cmdList) {
166 argMsg.append(" ").append(s);
168 logger.log(Level.INFO, argMsg.toString());
170 Process process = null;
172 logger.log(Level.INFO, "\n\n----\nProcessing FFmpeg...\n----\n\n");
173 process = Runtime.getRuntime().exec(cmdList.toArray(new String[0]));
174 BufferedReader ebr = new BufferedReader(new InputStreamReader(
175 process.getErrorStream()));
177 while ((e = ebr.readLine()) != null) {
179 if (state.startsWith("frame=")) {
180 publish(new FfmpegProgress(state));
181 } else if (!state.endsWith("No accelerated colorspace conversion found")) {
182 logger.log(Level.INFO, e);
187 } catch (InterruptedException ex) {
193 return process.exitValue();
195 // TODO 正常終了した場合もdestroyしていいのか?
196 if (process != null) {
199 if (fwsFile != null) {
205 private List<String> getAvfilterOptions(String avfilterOption) {
206 final List<String> avfilterArgs = new ArrayList<String>();
207 if (StringUtils.isNotBlank(avfilterOption)) {
208 avfilterArgs.add(avfilterOption);
213 private String getVhookArg(boolean addComment, String commPath, boolean addTcomment,
214 String tcommPath, boolean isHD) throws UnsupportedEncodingException {
215 StringBuilder sb = new StringBuilder();
217 sb.append(getFfmpeg().getVhook().getPath().replace("\\", "/"));
220 sb.append("--data-user:");
221 sb.append(URLEncoder.encode(commPath.replace("\\", "/"), "Shift_JIS"));
225 sb.append("--data-owner:");
226 sb.append(URLEncoder.encode(tcommPath.replace("\\", "/"), "Shift_JIS"));
229 sb.append("--font:");
230 sb.append(URLEncoder.encode(
231 getFfmpeg().getFont().getPath().replace("\\", "/"), "Shift_JIS"));
233 sb.append("--font-index:");
234 sb.append(getFfmpeg().getFontIndex());
236 sb.append("--show-user:");
237 sb.append(getFfmpeg().getMaxNumOfComment());
239 sb.append("--shadow:");
240 sb.append(getFfmpeg().getShadowIndex());
242 if (getFfmpeg().isShowConverting()) {
243 sb.append("--enable-show-video");
246 if (!getFfmpeg().isDisableFontSizeArrange()) {
247 sb.append("--enable-fix-font-size");
250 if (getFfmpeg().isCommentOpaque()) {
251 sb.append("--enable-opaque-comment");
255 sb.append("--aspect-mode:1");
258 return sb.toString();
261 private ConvertProfile getFfmpeg() {
265 protected void checkStop() throws InterruptedException {
266 if (Thread.interrupted()) {
267 throw new InterruptedException("中止要求を受け付けました");