]>
Commit | Line | Data |
---|---|---|
4569a895 AT |
1 | // -*- C++ -*- |
2 | ||
949d9ae1 | 3 | // Copyright (C) 2005, 2006, 2007 Free Software Foundation, Inc. |
4569a895 AT |
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> | |
949d9ae1 | 51 | #include <bits/functexcept.h> |
4569a895 | 52 | |
5e11f978 | 53 | namespace __gnu_pbds |
4569a895 | 54 | { |
4569a895 AT |
55 | namespace test |
56 | { | |
4569a895 AT |
57 | void |
58 | verify_argc(size_t given, size_t required) | |
59 | { | |
60 | if (given != required) | |
949d9ae1 | 61 | __throw_illegal_input_error(); |
4569a895 AT |
62 | } |
63 | ||
64 | void | |
65 | verify_prob(double prob) | |
66 | { | |
67 | if (prob < 0 || prob > 1) | |
949d9ae1 | 68 | __throw_illegal_input_error(); |
4569a895 AT |
69 | } |
70 | ||
71 | std::string | |
72 | get_cmd_line_str(int argc, char* a_p_argv[], int argn) | |
73 | { | |
74 | if (argc <= argn) | |
949d9ae1 | 75 | __throw_illegal_input_error(); |
4569a895 | 76 | const std::string ret(a_p_argv[argn]); |
949d9ae1 | 77 | return ret; |
4569a895 AT |
78 | } |
79 | ||
80 | double | |
81 | get_cmd_line_prob(int argc, char* a_p_argv[], int argn) | |
82 | { | |
83 | if (argc <= argn) | |
949d9ae1 | 84 | __throw_illegal_input_error(); |
4569a895 | 85 | const double ret = ::atof(a_p_argv[argn]); |
4569a895 | 86 | verify_prob(ret); |
949d9ae1 | 87 | return ret; |
4569a895 AT |
88 | } |
89 | ||
90 | size_t | |
91 | get_cmd_line_size(int argc, char* a_p_argv[], int argn) | |
92 | { | |
93 | if (argc <= argn) | |
949d9ae1 | 94 | __throw_illegal_input_error(); |
4569a895 | 95 | const size_t ret = static_cast<size_t>(::atoi(a_p_argv[argn])); |
949d9ae1 | 96 | return ret; |
4569a895 AT |
97 | } |
98 | ||
99 | bool | |
100 | get_cmd_line_bool(int argc, char* a_p_argv[], int argn) | |
101 | { | |
102 | if (argc <= argn) | |
949d9ae1 | 103 | __throw_illegal_input_error(); |
4569a895 AT |
104 | |
105 | const std::string opt(a_p_argv[argn]); | |
4569a895 | 106 | if (opt.size() != 1) |
949d9ae1 | 107 | __throw_illegal_input_error(); |
4569a895 | 108 | if (opt[0] == 't') |
949d9ae1 | 109 | return true; |
4569a895 | 110 | if (opt[0] == 'f') |
949d9ae1 BK |
111 | return false; |
112 | __throw_illegal_input_error(); | |
113 | return false; | |
4569a895 | 114 | } |
4569a895 | 115 | } // namespace test |
5e11f978 | 116 | } // namespace __gnu_pbds |