src/main/java/eu/svjatoslav/meviz/encoder/CommandlineHandler.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 eu.svjatoslav.commons.commandline.parameterparser.Parser;
  13 import eu.svjatoslav.commons.commandline.parameterparser.parameter.DirectoryParameter;
  14 import eu.svjatoslav.commons.commandline.parameterparser.parameter.NullParameter;
  15 import eu.svjatoslav.commons.commandline.parameterparser.parameter.StringParameter;
  16
  17 public class CommandlineHandler {
  18
  19         Parser parser = new Parser();
  20
  21         StringParameter outputFormatParameter = parser
  22                         .createStringParameter("Encoding output format.")
  23                         .addAliases("-o", "--output-format").setMandatory();
  24
  25         NullParameter deinterlaceParameter = parser.createNullParameter(
  26                         "Deinterlace video.").addAliases("--deinterlace", "-d");
  27
  28         NullParameter testParameter = parser.createNullParameter(
  29                         "Simulate file encoding.").addAliases("-t", "--test");
  30
  31         NullParameter recursiveParameter = parser.createNullParameter(
  32                         "Enable recursive mode.").addAliases("-r", "--recursive");
  33
  34         StringParameter inputPatternParameter = parser
  35                         .createStringParameter("File input pattern.")
  36                         .addAliases("-i", "--input-pattern").setMandatory();
  37
  38         DirectoryParameter workingDirectoryParameter = parser
  39                         .createDirectoryParameter("Working directory.")
  40                         .addAliases("-w", "--working-directory").mustExist();
  41
  42         BitrateParameter videoBitrateParameter = new BitrateParameter(
  43                         "Video bitrate");
  44
  45         public CommandlineHandler() {
  46                 videoBitrateParameter.addAliases("-v", "--video-bitrate");
  47                 parser.addParameter(videoBitrateParameter);
  48         }
  49
  50         /**
  51          * @return {@link EncodingOptions} if commandline arguments were
  52          *         successfully parsed, or <code>null</code> if parsing error
  53          *         occurred.
  54          */
  55         public EncodingOptions parseCommandlineArguments(final String[] args) {
  56
  57                 final EncodingOptions options = new EncodingOptions();
  58
  59                 parser.parse(args);
  60
  61                 if (recursiveParameter.isParameterSpecified())
  62                         options.recursive = true;
  63
  64                 if (deinterlaceParameter.isParameterSpecified())
  65                         options.deinterlace = true;
  66
  67                 if (testParameter.isParameterSpecified())
  68                         options.testOnly = true;
  69
  70                 if (outputFormatParameter.isParameterSpecified())
  71                         options.outputFormats.addAll(outputFormatParameter
  72                                         .getArgumentsAsStrings());
  73
  74                 if (workingDirectoryParameter.isParameterSpecified())
  75                         options.workingDirectory = workingDirectoryParameter
  76                                         .getArgumentsAsFiles().get(0);
  77
  78                 if (inputPatternParameter.isParameterSpecified())
  79                         options.inputPatterns.addAll(inputPatternParameter
  80                                         .getArgumentsAsStrings());
  81
  82                 if (videoBitrateParameter.isParameterSpecified())
  83                         try {
  84                                 options.videoBitrate = videoBitrateParameter.getValue();
  85                         } catch (final Exception e) {
  86                                 System.out
  87                                                 .println("Invalid video bitrate. Valid values are: LOW, MEDIUM, HIGH.");
  88                                 return null;
  89                         }
  90
  91                 return options;
  92         }
  93 }