e60140b63ab0dc5e72b35721805792c4d85d908f
[meviz.git] / src / main / java / eu / svjatoslav / meviz / encoder / Encoder.java
1 /*
2  * Meviz - Various tools collection to work with multimedia.
3  * Copyright (C) 2012, 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 2 of the GNU General Public License
7  * as published by the Free Software Foundation.
8  */
9
10 package eu.svjatoslav.meviz.encoder;
11
12 import java.io.File;
13 import java.util.List;
14
15 import eu.svjatoslav.commons.file.FilePathParser;
16 import eu.svjatoslav.commons.string.WildCardMatcher;
17 import eu.svjatoslav.meviz.Module;
18 import eu.svjatoslav.meviz.encoder.converters.AbstractConverter;
19
20 public class Encoder implements Module {
21
22         public FormatsRegistry encoderRegistry = new FormatsRegistry();
23
24         public EncodingOptions encodingOptions;
25
26         public EncodingPlan encodingPlan;
27
28         CommandlineHandler commandlineHandler = new CommandlineHandler();
29
30         /**
31          * Generate encoding plan
32          * 
33          * @param sourceFile
34          *            Source directory of file
35          */
36         public void compileEncodingPlan(final File sourceFile) {
37                 if (!sourceFile.exists()) {
38                         System.out.println("Error: file \"" + sourceFile.getAbsolutePath()
39                                         + "\" does not exist.");
40                         return;
41                 }
42
43                 if (sourceFile.isDirectory()) {
44                         // handle directory
45                         for (final File subFile : sourceFile.listFiles())
46                                 if (subFile.isDirectory()) {
47                                         if (encodingOptions.recursive)
48                                                 compileEncodingPlan(subFile);
49                                 } else
50                                         compileEncodingPlan(subFile);
51                 } else if (sourceFile.isFile())
52                         if (fileMatchesInputPattern(sourceFile)) {
53
54                                 // System.out.println("Processing file: " +
55                                 // file.getAbsolutePath());
56
57                                 final String sourceFileExtension = FilePathParser
58                                                 .getFileExtension(sourceFile);
59
60                                 // encode source file into every desired target format
61                                 for (final String targetFormat : encodingOptions.outputFormats) {
62
63                                         // construct target file
64                                         final File targetFile = getTargetFile(sourceFile,
65                                                         targetFormat);
66
67                                         // System.out.println("target path: " +
68                                         // targetFilePath.toString());
69
70                                         if (!targetFile.exists()) {
71
72                                                 final List<eu.svjatoslav.meviz.encoder.converters.AbstractConverter> formats = encoderRegistry
73                                                                 .getEncoders(sourceFileExtension, targetFormat);
74
75                                                 if (formats.size() == 0)
76                                                         System.out
77                                                                         .println("Error: no encoders found to convert file \""
78                                                                                         + sourceFile.getAbsolutePath()
79                                                                                         + "\" into "
80                                                                                         + targetFormat
81                                                                                         + " format.");
82                                                 else if (formats.size() > 1)
83                                                         System.out
84                                                                         .println("Error: Encoders piping not yet supported to convert file \""
85                                                                                         + sourceFile.getAbsolutePath()
86                                                                                         + "\" into "
87                                                                                         + targetFormat
88                                                                                         + " format.");
89                                                 else {
90                                                         final AbstractConverter chosenFormat = formats.get(0);
91                                                         final EncodingTask encodingTask = new EncodingTask(
92                                                                         sourceFile, targetFile, chosenFormat);
93
94                                                         if (chosenFormat.isTerminalMandatory())
95                                                                 encodingTask.setUseTerminal(true);
96                                                         encodingPlan.scheduleTask(encodingTask);
97                                                 }
98                                         }
99                                 }
100                         }
101
102         }
103
104         /**
105          * @param file
106          *            single file candidate to potentially be encoded
107          * 
108          * @return <code>true</code> if file shall be encoded.
109          */
110         public boolean fileMatchesInputPattern(final File file) {
111                 final String fileName = file.getName().toLowerCase();
112
113                 for (final String inputPattern : encodingOptions.inputPatterns)
114                         if (WildCardMatcher.match(fileName, inputPattern.toLowerCase()))
115                                 return true;
116
117                 return false;
118         }
119
120         @Override
121         public String getDescription() {
122                 return "Convert between various media formats.";
123         }
124
125         @Override
126         public String getModuleCommand() {
127                 return "encode";
128         }
129
130         private File getTargetFile(final File sourceFile, final String targetFormat) {
131                 final StringBuffer targetFilePath = new StringBuffer();
132                 targetFilePath.append(sourceFile.getParent());
133                 targetFilePath.append("/");
134                 targetFilePath.append(FilePathParser
135                                 .getFileNameWithoutExtension(sourceFile));
136                 targetFilePath.append(".");
137                 targetFilePath.append(targetFormat);
138
139                 return new File(targetFilePath.toString());
140         }
141
142         @Override
143         public void run(final String[] args) {
144
145                 // parse incoming commandline arguments
146                 encodingOptions = commandlineHandler.parseCommandlineArguments(args);
147
148                 if (encodingOptions == null) {
149                         showCommandlineHelp();
150                         return;
151                 }
152
153                 encodingPlan = new EncodingPlan();
154
155                 compileEncodingPlan(encodingOptions.workingDirectory);
156
157                 if (!encodingOptions.testOnly)
158                         try {
159                                 encodingPlan.execute(encodingOptions);
160                         } catch (final Exception exception) {
161                                 exception.printStackTrace();
162                         }
163
164         }
165
166         @Override
167         public void showCommandlineHelp() {
168                 commandlineHandler.parser.showHelp();
169                 System.out.println("Example commands:");
170                 System.out
171                                 .println("    Convert all MTS files in the current directory into MP4's.");
172                 System.out.println("        meviz encode -o mp4 -i *.MTS");
173         }
174
175 }