ssphys/SSPhys/CommandLine.cpp

   1 // CommandLine.cpp: implementation of the CCommandLine class.
   2 //
   3 //////////////////////////////////////////////////////////////////////
   4
   5 #include "stdafx.h"
   6 #include "CommandLine.h"
   7 #include "Command.h"
   8
   9 //---------------------------------------------------------------------------
  10 tristate ToTristate (const char ch)
  11 {
  12   if (!ch)
  13     return undefined;
  14   if (ch == '-')
  15     return cleared;
  16   else if (ch == '+')
  17     return set;
  18
  19   return undefined;
  20 }
  21
  22
  23 //////////////////////////////////////////////////////////////////////
  24 // Construction/Destruction
  25 //////////////////////////////////////////////////////////////////////
  26 CCommandLine::CCommandLine ()
  27 {
  28 }
  29
  30 CCommandLine::~CCommandLine ()
  31 {
  32 }
  33
  34
  35 void CCommandLine::Parse (int argc, char* argv[])
  36 {
  37   for (int i = 1; i< argc; ++i)
  38   {
  39     if (argv[i][0] == '-')
  40     {
  41       COptionInfoList::const_iterator itor = m_OptionsInfo.begin ();
  42       COption option;
  43       while (itor != m_OptionsInfo.end ())
  44       {
  45         const COptionInfo& info = *itor;
  46         if (IsArgChar (argv[i][1], info.m_shortOption) || IsLongArgument (&argv[i][1], info.m_longOption.c_str()))
  47         {
  48           option.id = info.m_Id;
  49           switch (info.m_needArg)
  50           {
  51           case COptionInfo::requiredArgument:
  52             if (++i < argc)
  53               option.value = argv[i];
  54             else
  55               throw SSException ("missing command line parameter");
  56             break;
  57           case COptionInfo::optionalArgument:
  58             if (++i < argc)
  59               option.value = argv[i];
  60             break;
  61           case COptionInfo::tristateArgument:
  62             option.value = ToTristate (argv[i][2]);
  63             break;
  64           }
  65
  66           break;
  67         }
  68         ++itor;
  69       }
  70
  71       if (itor == m_OptionsInfo.end ())
  72         throw SSException ("unknown command line parameter");
  73
  74       m_Options.push_back (option);
  75     }
  76     else
  77     {
  78       if (m_Command.empty ())
  79       {
  80         m_Command = argv[i];
  81         std::auto_ptr <CCommand> pCommand (m_pCommandFactory->MakeCommand (m_Command));
  82         COptionInfoList commandOptions = pCommand->GetOptionsInfo ();
  83         m_OptionsInfo.insert (m_OptionsInfo.end (), commandOptions.begin (), commandOptions.end ());
  84       }
  85       else
  86       {
  87         m_Args.push (argv[i]);
  88       }
  89     }
  90   }
  91 }
  92
  93
  94 bool CCommandLine::IsLongArgument (const char* ch, const char* arg)
  95 {
  96   if (!ch)
  97     return false;
  98
  99   std::string str = (ch[0] == '-') ? &ch[1] : &ch[0];
 100   if (str == arg)
 101     return true;
 102
 103   return false;
 104 }
 105
 106 bool CCommandLine::IsArgChar (const char ch, const char arg)
 107 {
 108   if (!ch)
 109     return false;
 110   else if (ch == arg)
 111     return true;
 112   else if (tolower(ch) == arg)
 113     return true;
 114   else if (toupper(ch) == arg)
 115     return true;
 116   return false;
 117 }