lgrAppCmdLine.h

Go to the documentation of this file.

/******************************************************************************
** Copyright (c) 1992-2010 VT MAK
** All rights reserved.
******************************************************************************/


#pragma once

#include <cmdLine/cmdLine.h>
#include "lgrConsole.h"
#include "vlutil/vlFilename.h"

namespace MAKLogger
{
   class DtLgrCmdLine : public DtCmdLine::CmdLine
   {
   public:
      explicit DtLgrCmdLine(const std::string& message,
         const char delimiter = ' ',
         const std::string& version = "none");

      void addUnlabeled( DtCmdLine::Arg* a );

      inline bool parseInput(int argc, char* const * argv);

   protected:

      DtCmdLine::Arg* myUnlabeledArgument;
   };

   inline DtLgrCmdLine::DtLgrCmdLine(const std::string& message,
      const char delimiter, const std::string& version) :
      CmdLine(message,delimiter,version),
      myUnlabeledArgument(0)
   {
   }

   inline void DtLgrCmdLine::addUnlabeled( DtCmdLine::Arg* a )
   {
      myUnlabeledArgument = a;
      add(a);
   }

   inline bool DtLgrCmdLine::parseInput(int argc, char* const * argv)
   {
      try
      {
         std::vector<std::string> args;
         int requiredCount = 0;
         int i=1;

         DtFilename progName = argv[0];
         progName.stripPath();
         _progName = progName.c_str();

         for (i = 1; i < argc; i++)
            args.push_back(argv[i]);

         for (i = 0; static_cast<unsigned>(i) < args.size(); i++)
         {
            bool matched = false;
            for (DtCmdLine::ArgListIterator it = _argList.begin(); !matched && it != _argList.end(); it++)
            {
               if ( (*it)->processArg( &i, args ) )
               {
                  if ((*it) == myUnlabeledArgument
                     && args[i].size() > 0 && args[i][0] == myUnlabeledArgument->flagStartChar())
                  {
                     throw(DtCmdLine::CmdLineParseException("Is not recognized!",
                        args[i]));
                  }
                  requiredCount += _xorHandler.check( *it );
                  matched = true;
               }
            }

            if ( !matched && _emptyCombined( args[i] ) )
               matched = true;

            if ( !matched && !DtCmdLine::Arg::ignoreRest() )
               throw(DtCmdLine::CmdLineParseException("Couldn't find match for argument",
               args[i]));
         }

         if ( requiredCount < _numRequired )
            throw(DtCmdLine::CmdLineParseException("One or more required arguments missing!"));

         if ( requiredCount > _numRequired )
            throw(DtCmdLine::CmdLineParseException("Too many arguments!"));

         return true;

      }
      catch ( DtCmdLine::ArgException& e )
      {
         DtLgrConsole::allocateConsole();
         _output->failure(*this,e);

         int oldNotifyLevel = DtNotifyLevel;
         DtNotifyLevel = 3;

         if (myPressAnyKey)
         {
            DtInfo << "Press any key to continue" << std::endl;

            while (DtPollBlockingInputChar() == '\0')
            {
               DtSleep(0.1);
            }
         }

         DtNotifyLevel = oldNotifyLevel;
         return false;
      }
   }
}