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 java.util.logging.Level;
  13 import java.util.logging.Logger;
  14 import org.apache.commons.lang.StringUtils;
  15 import saccubus.ConvertStopFlag;
  16 import saccubus.conv.ConvertToVideoHook;
  17 import yukihane.saccubus.converter.profile.NgSetting;
  18 import yukihane.saccubus.converter.profile.ConvertProfile;
  19 import yukihane.saccubus.converter.profile.FfmpegOption;
  20 import yukihane.saccubus.converter.profile.GeneralProfile;
  21 import saccubus.net.TextProgressListener;
  22 import yukihane.mediainfowrapper.Info;
  23 import yukihane.mediainfowrapper.MediaInfo;
  24 import yukihane.mediainfowrapper.Size;
  25 import yukihane.swf.Cws2Fws;
  26
  27 /**
  28  *
  29  * @author yuki
  30  */
  31 public class FfmpegCommand extends AbstractCommand {
  32     private static final Logger logger = Logger.getLogger(FfmpegCommand.class.getName());
  33     private final File commentMiddleFile;
  34     private final File tcommMiddleFile;
  35     private final File TMP_CWS;
  36     private final File commentFile;
  37     private final File videoFile;
  38     private final File convertedVideoFile;
  39     private final ConvertProfile ffmpeg;
  40
  41     FfmpegCommand(TextProgressListener listener, ConvertStopFlag flag, File commentFile,
  42             File videoFile, File convertedVideoFile, ConvertProfile ffmpeg, GeneralProfile general) throws IOException {
  43         super(listener, flag);
  44         this.commentFile = commentFile;
  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                 false, tcommMiddleFile.getPath(), getFfmpeg().getFfmpegOption())) == 0) {
  96             sendText("変換が正常に終了しました。");
  97             return true;
  98         } else {
  99             sendText("変換エラー:" + convertedVideoFile.getPath());
 100         }
 101         return false;
 102     }
 103
 104     private int converting_video(File videoFile, File convertedVideoFile, boolean addComment, String commPath,
 105             boolean addTcomment, String tcommPath, FfmpegOption ov) throws InterruptedException, IOException {
 106         File fwsFile = Cws2Fws.createFws(videoFile, TMP_CWS);
 107
 108         List<String> cmdList = new ArrayList<String>();
 109         cmdList.add(getFfmpeg().getFfmpeg().getPath());
 110         cmdList.add("-y");
 111         String[] mainOptions = ov.getMainOption().split(" +");
 112         for (String opt : mainOptions) {
 113             if (StringUtils.isNotBlank(opt)) {
 114                 cmdList.add(opt);
 115             }
 116         }
 117         String[] inOptions = ov.getInOption().split(" +");
 118         for (String opt : inOptions) {
 119             if (StringUtils.isNotBlank(opt)) {
 120                 cmdList.add(opt);
 121             }
 122         }
 123         cmdList.add("-i");
 124
 125         if (fwsFile == null) {
 126             cmdList.add(videoFile.getPath());
 127         } else {
 128             cmdList.add(fwsFile.getPath());
 129         }
 130         String[] outOptions = ov.getOutOption().split(" +");
 131         for (String opt : outOptions) {
 132             if (StringUtils.isNotBlank(opt)) {
 133                 cmdList.add(opt);
 134             }
 135         }
 136
 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);
 140         if (ov.isResize()) {
 141             final Size scaled = (ov.isAdjustRatio()) ? MediaInfo.adjustSize(info, ov.getResizeWidth(), ov.
 142                     getResizeHeight()) : new Size(info.getWidth(), info.getHeight());
 143             cmdList.add("-s");
 144             cmdList.add(scaled.getWidth() + "x" + scaled.getHeight());
 145         }
 146
 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);
 152             }
 153         }
 154
 155         if (!avfilterArgs.isEmpty()) {
 156             cmdList.add("-vfilters");
 157             final String args = "\"" + StringUtils.join(avfilterArgs, ", ") + "\"";
 158             cmdList.add(args);
 159         }
 160
 161         cmdList.add(convertedVideoFile.getPath());
 162
 163         final StringBuilder argMsg = new StringBuilder();
 164         argMsg.append("arg:");
 165         for (String s : cmdList) {
 166             argMsg.append(" ").append(s);
 167         }
 168         logger.log(Level.INFO, argMsg.toString());
 169
 170         try {
 171             logger.log(Level.INFO, "\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                     logger.log(Level.INFO, 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().isDisableFontSizeArrange()) {
 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 ConvertProfile getFfmpeg() {
 258         return ffmpeg;
 259     }
 260 }