Rename CommandLineHandle class

tools/PlaYUVerToolsCmdParser.cpp

@@ -0,0 +1,189 @@
+/*    This file is a part of plaYUVer project
+ *    Copyright (C) 2014-2015  by Luis Lucas      (luisfrlucas@gmail.com)
+ *                                Joao Carreira   (jfmcarreira@gmail.com)
+ *
+ *    This program is free software; you can redistribute it and/or modify
+ *    it under the terms of the GNU General Public License as published by
+ *    the Free Software Foundation; either version 2 of the License, or
+ *    (at your option) any later version.
+ *
+ *    This program is distributed in the hope that it will be useful,
+ *    but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *    GNU General Public License for more details.
+ *
+ *   You should have received a copy of the GNU General Public License along
+ *   with this program; if not, write to the Free Software Foundation, Inc.,
+ *   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+/**
+ * \file     PlaYUVerToolsCmdParser.cpp
+ * \brief    Handle for command line
+ */
+
+#include <cstring>
+#include <cstdarg>
+#include "PlaYUVerToolsCmdParser.h"
+#include "lib/PlaYUVerFrame.h"
+#include "lib/PlaYUVerStream.h"
+#include "lib/PlaYUVerModuleIf.h"
+#include "modules/PlaYUVerModuleFactory.h"
+
+namespace plaYUVer
+{
+
+PlaYUVerToolsCmdParser::PlaYUVerToolsCmdParser()
+{
+  m_uiLogLevel = 0;
+  m_bQuiet = false;
+  m_strPelFmt = std::string( "YUV420p" );
+  m_iFrames = -1;
+}
+
+PlaYUVerToolsCmdParser::~PlaYUVerToolsCmdParser()
+{
+
+}
+
+Void PlaYUVerToolsCmdParser::log( UInt level, const char *fmt, ... )
+{
+  if( level >= m_uiLogLevel )
+  {
+    std::va_list args;
+    va_start( args, fmt );
+    vfprintf( stdout, fmt, args );
+    va_end( args );
+  }
+}
+
+Int PlaYUVerToolsCmdParser::parseToolsArgs( Int argc, Char *argv[] )
+{
+  Int iRet = 0;
+
+  Opts().addOptions()/**/
+  ( "quiet,q", m_bQuiet, "disable verbose" );
+
+  Opts().addOptions()/**/
+  ( "input,i", m_apcInputs, "input file" ) /**/
+  //( "input,i", m_strInput, "input file" ) /**/
+  ( "output,o", m_strOutput, "output file" ) /**/
+  ( "size,s", m_strResolution, "size (WxH)" ) /**/
+  ( "pel_fmt", m_strPelFmt, "pixel format" ) /**/
+  ( "bits_pel", m_uiBitsPerPixel, "bits per pixel" ) /**/
+  ( "frames,f", m_iFrames, "number of frames to parse" );
+
+  Opts().addOptions()/**/
+  ( "quality", m_strQualityMetric, "select a quality metric" ) /**/
+  ( "module", m_strModule, "select a module (use internal name)" );
+
+  config( argc, argv );
+  if( !parse() )
+  {
+    iRet = 1;
+  }
+
+  if( m_bQuiet )
+  {
+    m_uiLogLevel = LOG_RESULT;
+  }
+
+  if( Opts()["module"]->count() && Opts()["help"]->count() )
+  {
+    listModuleHelp();
+    iRet = 1;
+  }
+  else if( Opts()["help"]->count() )
+  {
+    printf( "Usage: %s modules/quality [options] -input=input_file [--output=output_file]\n", argv[0] );
+    Opts().doHelp( std::cout );
+    iRet = 1;
+  }
+
+  if( Opts()["module_list"]->count() || Opts()["module_list_full"]->count() )
+  {
+    listModules();
+    iRet = 1;
+  }
+  return iRet;
+}
+
+Void PlaYUVerToolsCmdParser::listModules()
+{
+  Bool bDetailed = false;
+
+  if( Opts()["module_list_full"]->count() )
+    bDetailed = true;
+
+  PlaYUVerModuleIf* pcCurrModuleIf;
+  PlaYUVerModuleFactoryMap& PlaYUVerModuleFactoryMap = PlaYUVerModuleFactory::Get()->getMap();
+  PlaYUVerModuleFactoryMap::iterator it = PlaYUVerModuleFactoryMap.begin();
+
+  printf( "PlaYUVer available modules: \n" );
+  //printf( "                                           " );
+  printf( "   [Internal Name]               " );
+  if( bDetailed )
+  {
+    printf( "   [Full Name]                             " );
+    printf( "   [Type]        " );
+    printf( "   [Description]" );
+  }
+  printf( " \n" );
+
+  Char ModuleNameString[40];
+
+  for( UInt i = 0; it != PlaYUVerModuleFactoryMap.end(); ++it, i++ )
+  {
+    printf( "   " );
+    printf( "%-30s", it->first );
+    if( bDetailed )
+    {
+      ModuleNameString[0] = '\0';
+      pcCurrModuleIf = it->second();
+
+      if( pcCurrModuleIf->m_pchModuleCategory )
+      {
+        strcat( ModuleNameString, pcCurrModuleIf->m_pchModuleCategory );
+        strcat( ModuleNameString, "/" );
+      }
+      strcat( ModuleNameString, pcCurrModuleIf->m_pchModuleName );
+      printf( "   %-40s", ModuleNameString );
+      switch( pcCurrModuleIf->m_iModuleType )
+      {
+      case FRAME_PROCESSING_MODULE:
+        printf( "   Processing    " );
+        break;
+      case FRAME_MEASUREMENT_MODULE:
+        printf( "   Measurement   " );
+        break;
+      }
+      printf( "   %s", pcCurrModuleIf->m_pchModuleTooltip );
+      pcCurrModuleIf->Delete();
+    }
+    printf( "\n" );
+  }
+}
+
+Void PlaYUVerToolsCmdParser::listModuleHelp()
+{
+  std::string moduleName = m_strModule;
+  PlaYUVerModuleIf* pcCurrModuleIf = NULL;
+
+  PlaYUVerModuleFactoryMap& PlaYUVerModuleFactoryMap = PlaYUVerModuleFactory::Get()->getMap();
+  PlaYUVerModuleFactoryMap::iterator it = PlaYUVerModuleFactoryMap.begin();
+  for( UInt i = 0; it != PlaYUVerModuleFactoryMap.end(); ++it, i++ )
+  {
+    if( strcmp( it->first, moduleName.c_str() ) == 0 )
+    {
+      pcCurrModuleIf = it->second();
+      break;
+    }
+  }
+  if( pcCurrModuleIf )
+  {
+    printf( "Usage: playuverTools --module=%s options:\n", it->first );
+    pcCurrModuleIf->m_cModuleOptions.doHelp( std::cout );
+  }
+}
+
+}  // NAMESPACE