GetOpt.cxx
上传用户:sy_wanhua
上传日期:2013-07-25
资源大小:3048k
文件大小:8k
源码类别:

流媒体/Mpeg4/MP4

开发平台:

C/C++

  1. /* ====================================================================
  2.  * The Vovida Software License, Version 1.0 
  3.  * 
  4.  * Copyright (c) 2001 Vovida Networks, Inc.  All rights reserved.
  5.  * 
  6.  * Redistribution and use in source and binary forms, with or without
  7.  * modification, are permitted provided that the following conditions
  8.  * are met:
  9.  * 
  10.  * 1. Redistributions of source code must retain the above copyright
  11.  *    notice, this list of conditions and the following disclaimer.
  12.  * 
  13.  * 2. Redistributions in binary form must reproduce the above copyright
  14.  *    notice, this list of conditions and the following disclaimer in
  15.  *    the documentation and/or other materials provided with the
  16.  *    distribution.
  17.  * 
  18.  * 3. The names "VOCAL", "Vovida Open Communication Application Library",
  19.  *    and "Vovida Open Communication Application Library (VOCAL)" must
  20.  *    not be used to endorse or promote products derived from this
  21.  *    software without prior written permission. For written
  22.  *    permission, please contact vocal@vovida.org.
  23.  *
  24.  * 4. Products derived from this software may not be called "VOCAL", nor
  25.  *    may "VOCAL" appear in their name, without prior written
  26.  *    permission of Vovida Networks, Inc.
  27.  * 
  28.  * THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESSED OR IMPLIED
  29.  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  30.  * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, TITLE AND
  31.  * NON-INFRINGEMENT ARE DISCLAIMED.  IN NO EVENT SHALL VOVIDA
  32.  * NETWORKS, INC. OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT DAMAGES
  33.  * IN EXCESS OF ,000, NOR FOR ANY INDIRECT, INCIDENTAL, SPECIAL,
  34.  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  35.  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  36.  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  37.  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  38.  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
  39.  * USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
  40.  * DAMAGE.
  41.  * 
  42.  * ====================================================================
  43.  * 
  44.  * This software consists of voluntary contributions made by Vovida
  45.  * Networks, Inc. and many individuals on behalf of Vovida Networks,
  46.  * Inc.  For more information on Vovida Networks, Inc., please see
  47.  * <http://www.vovida.org/>.
  48.  *
  49.  */
  50. static const char* const GetOpt_cxx_Version = 
  51.   "$Id: GetOpt.cxx,v 1.7 2001/05/15 03:48:55 icahoon Exp $";
  52.  
  53.  
  54. #include "GetOpt.hxx"
  55. #include <cstring>
  56. #include <cassert>
  57. using Vocal::Configuration::GetOpt;
  58. using Vocal::Configuration::ValueType;
  59. using Vocal::Configuration::Value;
  60. using Vocal::Configuration::NameValueMap;
  61. using Vocal::ReturnCode;
  62. using Vocal::SUCCESS;
  63. using std::list;
  64. using std::string;
  65. GetOpt::GetOpt()
  66.     : myOptString(""),
  67.         myPrintError(0),
  68.         myRemainingArgc(0),
  69.         myRemainingArgv(0)
  70. {
  71.     option   nullOption = { 0, 0, 0, 0 };
  72.     myLongOpts.insert(myLongOpts.begin(), nullOption);
  73.     myAllOpts.insert(myAllOpts.begin(), nullOption);
  74. }
  75. GetOpt::~GetOpt()
  76. {
  77.     size_t  size =  myLongOpts.size();
  78.     
  79.     for ( size_t i = 0; i < size; i++ )
  80.     {
  81.      delete [] myLongOpts[i].name;
  82. myLongOpts[i].name = 0;
  83.     }
  84. }
  85. ReturnCode
  86. GetOpt::add
  87. (
  88.     const char *            longOption,
  89.     char           optionChar,
  90.     ValueType               valType
  91. )
  92. {
  93.     // Make sure we don't already have this option.
  94.     //
  95.     const option & opt = find(longOption, optionChar);
  96.     if ( opt.name != 0 || opt.val != 0 )
  97.     {
  98.         return ( !SUCCESS );
  99.     }
  100.     bool    haveOption = false;
  101.     size_t  size = 0;
  102.     if ( longOption != 0 )
  103.     {    
  104.         size = strlen(longOption);
  105.     }
  106.     // Build the long options
  107.     //
  108.     option  newOption = { 0, 0, 0, 0 };
  109.     
  110.     if ( size != 0 )
  111.     {
  112.         // If we have a long option, copy the name.
  113.         //
  114.      char * newName = new char[size+1]; 
  115.      strncpy(newName, longOption, size+1);
  116.      newOption.name =  newName;
  117.         haveOption = true;
  118.     }
  119.             
  120.     // If we have a option character, add it to the option string
  121.     // and add it to the new option.
  122.     //
  123.     if ( optionChar != '' )
  124.     {    
  125.         myOptString += optionChar;
  126.         newOption.val = optionChar;
  127.         haveOption = true;
  128.     }
  129.     
  130.     if ( haveOption != true )
  131.     {
  132.         return ( !SUCCESS );
  133.     }
  134.     newOption.has_arg = static_cast<int>(valType);
  135.         
  136.     switch ( valType )
  137.     {
  138.      case VALUE_PRESENT:
  139. {
  140.             if ( optionChar )
  141.             {
  142.         myOptString += ':';
  143.             }
  144.     break;
  145. }
  146. case VALUE_OPTIONAL:
  147. {
  148.             if ( optionChar )
  149.             {
  150.         myOptString += "::";
  151.             }
  152.     break;
  153. }
  154. default:
  155. {
  156.     break;
  157. }
  158.     }
  159.     if ( size != 0 )
  160.     {
  161.         vector<option>::iterator i = myLongOpts.end();
  162.         myLongOpts.insert(--i, newOption);
  163.     }
  164.     vector<option>::iterator i = myAllOpts.end();
  165.     myAllOpts.insert(--i, newOption);
  166.     return ( SUCCESS );
  167. }
  168. void          
  169. GetOpt::printError(bool error)
  170. {
  171.     myPrintError = ( error ? 1 : 0 );
  172. }
  173. ReturnCode        
  174. GetOpt::parse(int argc, char **  argv)
  175. {
  176.     ReturnCode  rc = SUCCESS;
  177.     
  178.     opterr = myPrintError;
  179.     
  180.     int c = 0;
  181.     int optionIndex = -1;
  182.           
  183.     while   (  ( c = getopt_long(argc, argv, myOptString.c_str(), 
  184.                    &myLongOpts[0], &optionIndex) )
  185.           != -1 
  186.     )
  187.     {
  188.         if ( c == ':' || c == '?' )
  189.         {
  190.             rc = !SUCCESS;
  191.             continue;
  192.         }
  193. const option  & opt =   c == 0 
  194.                                 ? myLongOpts[optionIndex] 
  195.                                 : find(0, c);
  196.         if ( opt.name == 0 && opt.val == 0 )
  197.         {
  198.             rc = !SUCCESS;
  199.             continue;
  200.         }
  201.         string  name;
  202.         if ( opt.name )
  203.         {
  204.             name = opt.name;
  205.         }
  206.         else
  207.         {
  208.             name = opt.val;
  209.         }
  210.         addValue(name, optarg, static_cast<ValueType>(opt.has_arg));
  211.     }
  212.     myRemainingArgc = argc - optind;
  213.     myRemainingArgv = ( myRemainingArgc ? &argv[optind] : 0 );
  214.     
  215.     myRemaining.clear();
  216.     
  217.     argv = myRemainingArgv;
  218.     
  219.     for ( argc = myRemainingArgc; argc > 0; --argc )
  220.     {
  221.         string  arg = *argv;
  222.         myRemaining.push_back(arg);
  223.         ++argv;
  224.     }
  225.     
  226.     return ( rc );
  227. }
  228. void
  229. GetOpt::addValue
  230. (
  231.     const string    &   name, 
  232.     const char      *   value,
  233.     ValueType           type
  234. )
  235. {
  236.     Value & val = myOptionMap[name];
  237.     ++val.count;
  238.     
  239.     switch ( type )
  240.     {
  241. case VALUE_OPTIONAL:
  242. {
  243.             if ( value )
  244.             {
  245.                 val.type = VALUE_PRESENT;
  246.                 val.value.push_back(string(value));
  247.             }
  248.     break;
  249. }
  250. case VALUE_PRESENT:
  251. {
  252.     assert(value);
  253.             val.type = VALUE_PRESENT;
  254.             val.value.push_back(string(value));
  255.     break;
  256. }
  257. default:
  258.         {
  259.             val.type = NO_VALUE;
  260.     break;
  261.         }
  262.     }
  263. }
  264. void
  265. GetOpt::addValue
  266. (
  267.     const string    &   name, 
  268.     const Value     &   value
  269. )
  270. {
  271.     Value & val = myOptionMap[name];
  272.     if ( val.count == 0 )
  273.     {
  274.         val = value;
  275.         return;
  276.     }
  277.     
  278.     val.count += value.count;
  279.     
  280.     if  (   value.type == VALUE_PRESENT 
  281.         ||  value.type == VALUE_OPTIONAL
  282.         )
  283.     {
  284.         val.type = VALUE_PRESENT;
  285.         
  286.         for (   list<string>::const_iterator i = value.value.begin();
  287.                 i != value.value.end();
  288.                 ++i
  289.             )
  290.         {
  291.             val.value.push_back(*i);
  292.         }
  293.     }
  294. }
  295. const NameValueMap &   
  296. GetOpt::options() const
  297. {
  298.     return ( myOptionMap );
  299. }
  300. int         
  301. GetOpt::remainingArgc() const
  302. {
  303.     return ( myRemainingArgc );
  304. }
  305. char **     
  306. GetOpt::remainingArgv() const
  307. {
  308.     return ( myRemainingArgv );
  309. }
  310. const list<string> &    
  311. GetOpt::remaining() const
  312. {
  313.     return ( myRemaining );
  314. }
  315. const option &      
  316. GetOpt::find(const char * longOpt, char c) const
  317. {
  318.     size_t  size = myAllOpts.size();
  319.     for ( size_t i = 0; i < size; ++i )
  320.     {
  321.         if  ( c != 0 && c == myAllOpts[i].val )
  322.         {
  323.             return ( myAllOpts[i] );
  324.         }
  325.         if ( longOpt != 0 && myAllOpts[i].name != 0 )
  326.         {
  327.             size_t  n = strlen(longOpt);
  328.             
  329.             if ( strncmp(myAllOpts[i].name, longOpt, n) == 0 )
  330.             {
  331.                 return ( myAllOpts[i] );
  332.             }
  333.         }
  334.     }
  335.     // Not found, so return null last option
  336.     //
  337.     return ( myAllOpts[size-1] );
  338. }