libstdc++-v3/testsuite/util/io/verified_cmd_line_input.cc

   1 // -*- C++ -*-
   2
   3 // Copyright (C) 2005, 2006, 2007 Free Software Foundation, Inc.
   4 //
   5 // This file is part of the GNU ISO C++ Library.  This library is free
   6 // software; you can redistribute it and/or modify it under the terms
   7 // of the GNU General Public License as published by the Free Software
   8 // Foundation; either version 2, or (at your option) any later
   9 // version.
  10
  11 // This library is distributed in the hope that it will be useful, but
  12 // WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 // General Public License for more details.
  15
  16 // You should have received a copy of the GNU General Public License
  17 // along with this library; see the file COPYING.  If not, write to
  18 // the Free Software Foundation, 59 Temple Place - Suite 330, Boston,
  19 // MA 02111-1307, USA.
  20
  21 // As a special exception, you may use this file as part of a free
  22 // software library without restriction.  Specifically, if other files
  23 // instantiate templates or use macros or inline functions from this
  24 // file, or you compile this file and link it with other files to
  25 // produce an executable, this file does not by itself cause the
  26 // resulting executable to be covered by the GNU General Public
  27 // License.  This exception does not however invalidate any other
  28 // reasons why the executable file might be covered by the GNU General
  29 // Public License.
  30
  31 // Copyright (C) 2004 Ami Tavory and Vladimir Dreizin, IBM-HRL.
  32
  33 // Permission to use, copy, modify, sell, and distribute this software
  34 // is hereby granted without fee, provided that the above copyright
  35 // notice appears in all copies, and that both that copyright notice
  36 // and this permission notice appear in supporting documentation. None
  37 // of the above authors, nor IBM Haifa Research Laboratories, make any
  38 // representation about the suitability of this software for any
  39 // purpose. It is provided "as is" without express or implied
  40 // warranty.
  41
  42 /**
  43  * @file verified_cmd_line_input.cpp
  44  * Contains definitions for tests - verified command line input.
  45  */
  46
  47 #include <util/io/verified_cmd_line_input.hpp>
  48 #include <limits.h>
  49 #include <utility>
  50 #include <stdlib.h>
  51 #include <bits/functexcept.h>
  52
  53 namespace __gnu_pbds
  54 {
  55   namespace test
  56   {
  57     void
  58     verify_argc(size_t given, size_t required)
  59     {
  60       if (given != required)
  61         __throw_illegal_input_error();
  62     }
  63
  64     void
  65     verify_prob(double prob)
  66     {
  67       if (prob < 0 || prob > 1)
  68         __throw_illegal_input_error();
  69     }
  70
  71     std::string
  72     get_cmd_line_str(int argc, char* a_p_argv[], int argn)
  73     {
  74       if (argc <= argn)
  75         __throw_illegal_input_error();
  76       const std::string ret(a_p_argv[argn]);
  77       return ret;
  78     }
  79
  80     double
  81     get_cmd_line_prob(int argc, char* a_p_argv[], int argn)
  82     {
  83       if (argc <= argn)
  84         __throw_illegal_input_error();
  85       const double ret = ::atof(a_p_argv[argn]);
  86       verify_prob(ret);
  87       return ret;
  88     }
  89
  90     size_t
  91     get_cmd_line_size(int argc, char* a_p_argv[], int argn)
  92     {
  93       if (argc <= argn)
  94         __throw_illegal_input_error();
  95       const size_t ret = static_cast<size_t>(::atoi(a_p_argv[argn]));
  96       return ret;
  97     }
  98
  99     bool
 100     get_cmd_line_bool(int argc, char* a_p_argv[], int argn)
 101     {
 102       if (argc <= argn)
 103         __throw_illegal_input_error();
 104
 105       const std::string opt(a_p_argv[argn]);
 106       if (opt.size() != 1)
 107         __throw_illegal_input_error();
 108       if (opt[0] == 't')
 109         return true;
 110       if (opt[0] == 'f')
 111         return false;
 112       __throw_illegal_input_error();
 113       return false;
 114     }
 115   } // namespace test
 116 } // namespace __gnu_pbds