testCommandLineArguments.cxx 8.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208
  1. /* Distributed under the OSI-approved BSD 3-Clause License. See accompanying
  2. file Copyright.txt or https://cmake.org/licensing#kwsys for details. */
  3. #include "kwsysPrivate.h"
  4. #include KWSYS_HEADER(CommandLineArguments.hxx)
  5. // Work-around CMake dependency scanning limitation. This must
  6. // duplicate the above list of headers.
  7. #if 0
  8. #include "CommandLineArguments.hxx.in"
  9. #endif
  10. #include <iostream>
  11. #include <vector>
  12. #include <stddef.h> /* size_t */
  13. #include <string.h> /* strcmp */
  14. static void* random_ptr = reinterpret_cast<void*>(0x123);
  15. static int argument(const char* arg, const char* value, void* call_data)
  16. {
  17. std::cout << "Got argument: \"" << arg << "\" value: \""
  18. << (value ? value : "(null)") << "\"" << std::endl;
  19. if (call_data != random_ptr) {
  20. std::cerr << "Problem processing call_data" << std::endl;
  21. return 0;
  22. }
  23. return 1;
  24. }
  25. static int unknown_argument(const char* argument, void* call_data)
  26. {
  27. std::cout << "Got unknown argument: \"" << argument << "\"" << std::endl;
  28. if (call_data != random_ptr) {
  29. std::cerr << "Problem processing call_data" << std::endl;
  30. return 0;
  31. }
  32. return 1;
  33. }
  34. static bool CompareTwoItemsOnList(bool i1, bool i2)
  35. {
  36. return i1 == i2;
  37. }
  38. static bool CompareTwoItemsOnList(int i1, int i2)
  39. {
  40. return i1 == i2;
  41. }
  42. static bool CompareTwoItemsOnList(double i1, double i2)
  43. {
  44. return i1 == i2;
  45. }
  46. static bool CompareTwoItemsOnList(const char* i1, const char* i2)
  47. {
  48. return strcmp(i1, i2) == 0;
  49. }
  50. static bool CompareTwoItemsOnList(const std::string& i1, const std::string& i2)
  51. {
  52. return i1 == i2;
  53. }
  54. int testCommandLineArguments(int argc, char* argv[])
  55. {
  56. // Example run: ./testCommandLineArguments --some-int-variable 4
  57. // --another-bool-variable --some-bool-variable=yes
  58. // --some-stl-string-variable=foobar --set-bool-arg1 --set-bool-arg2
  59. // --some-string-variable=hello
  60. int res = 0;
  61. kwsys::CommandLineArguments arg;
  62. arg.Initialize(argc, argv);
  63. // For error handling
  64. arg.SetClientData(random_ptr);
  65. arg.SetUnknownArgumentCallback(unknown_argument);
  66. int some_int_variable = 10;
  67. double some_double_variable = 10.10;
  68. char* some_string_variable = KWSYS_NULLPTR;
  69. std::string some_stl_string_variable = "";
  70. bool some_bool_variable = false;
  71. bool some_bool_variable1 = false;
  72. bool bool_arg1 = false;
  73. int bool_arg2 = 0;
  74. std::vector<int> numbers_argument;
  75. int valid_numbers[] = { 5, 1, 8, 3, 7, 1, 3, 9, 7, 1 };
  76. std::vector<double> doubles_argument;
  77. double valid_doubles[] = { 12.5, 1.31, 22 };
  78. std::vector<bool> bools_argument;
  79. bool valid_bools[] = { true, true, false };
  80. std::vector<char*> strings_argument;
  81. const char* valid_strings[] = { "andy", "bill", "brad", "ken" };
  82. std::vector<std::string> stl_strings_argument;
  83. std::string valid_stl_strings[] = { "ken", "brad", "bill", "andy" };
  84. typedef kwsys::CommandLineArguments argT;
  85. arg.AddArgument("--some-int-variable", argT::SPACE_ARGUMENT,
  86. &some_int_variable, "Set some random int variable");
  87. arg.AddArgument("--some-double-variable", argT::CONCAT_ARGUMENT,
  88. &some_double_variable, "Set some random double variable");
  89. arg.AddArgument("--some-string-variable", argT::EQUAL_ARGUMENT,
  90. &some_string_variable, "Set some random string variable");
  91. arg.AddArgument("--some-stl-string-variable", argT::EQUAL_ARGUMENT,
  92. &some_stl_string_variable,
  93. "Set some random stl string variable");
  94. arg.AddArgument("--some-bool-variable", argT::EQUAL_ARGUMENT,
  95. &some_bool_variable, "Set some random bool variable");
  96. arg.AddArgument("--another-bool-variable", argT::NO_ARGUMENT,
  97. &some_bool_variable1, "Set some random bool variable 1");
  98. arg.AddBooleanArgument("--set-bool-arg1", &bool_arg1,
  99. "Test AddBooleanArgument 1");
  100. arg.AddBooleanArgument("--set-bool-arg2", &bool_arg2,
  101. "Test AddBooleanArgument 2");
  102. arg.AddArgument("--some-multi-argument", argT::MULTI_ARGUMENT,
  103. &numbers_argument, "Some multiple values variable");
  104. arg.AddArgument("-N", argT::SPACE_ARGUMENT, &doubles_argument,
  105. "Some explicit multiple values variable");
  106. arg.AddArgument("-BB", argT::CONCAT_ARGUMENT, &bools_argument,
  107. "Some explicit multiple values variable");
  108. arg.AddArgument("-SS", argT::EQUAL_ARGUMENT, &strings_argument,
  109. "Some explicit multiple values variable");
  110. arg.AddArgument("-SSS", argT::MULTI_ARGUMENT, &stl_strings_argument,
  111. "Some explicit multiple values variable");
  112. arg.AddCallback("-A", argT::NO_ARGUMENT, argument, random_ptr,
  113. "Some option -A. This option has a multiline comment. It "
  114. "should demonstrate how the code splits lines.");
  115. arg.AddCallback("-B", argT::SPACE_ARGUMENT, argument, random_ptr,
  116. "Option -B takes argument with space");
  117. arg.AddCallback("-C", argT::EQUAL_ARGUMENT, argument, random_ptr,
  118. "Option -C takes argument after =");
  119. arg.AddCallback("-D", argT::CONCAT_ARGUMENT, argument, random_ptr,
  120. "This option takes concatinated argument");
  121. arg.AddCallback("--long1", argT::NO_ARGUMENT, argument, random_ptr, "-A");
  122. arg.AddCallback("--long2", argT::SPACE_ARGUMENT, argument, random_ptr, "-B");
  123. arg.AddCallback("--long3", argT::EQUAL_ARGUMENT, argument, random_ptr,
  124. "Same as -C but a bit different");
  125. arg.AddCallback("--long4", argT::CONCAT_ARGUMENT, argument, random_ptr,
  126. "-C");
  127. if (!arg.Parse()) {
  128. std::cerr << "Problem parsing arguments" << std::endl;
  129. res = 1;
  130. }
  131. std::cout << "Help: " << arg.GetHelp() << std::endl;
  132. std::cout << "Some int variable was set to: " << some_int_variable
  133. << std::endl;
  134. std::cout << "Some double variable was set to: " << some_double_variable
  135. << std::endl;
  136. if (some_string_variable &&
  137. strcmp(some_string_variable, "test string with space") == 0) {
  138. std::cout << "Some string variable was set to: " << some_string_variable
  139. << std::endl;
  140. delete[] some_string_variable;
  141. } else {
  142. std::cerr << "Problem setting string variable" << std::endl;
  143. res = 1;
  144. }
  145. size_t cc;
  146. #define CompareTwoLists(list1, list_valid, lsize) \
  147. if (list1.size() != lsize) { \
  148. std::cerr << "Problem setting " #list1 ". Size is: " << list1.size() \
  149. << " should be: " << lsize << std::endl; \
  150. res = 1; \
  151. } else { \
  152. std::cout << #list1 " argument set:"; \
  153. for (cc = 0; cc < lsize; ++cc) { \
  154. std::cout << " " << list1[cc]; \
  155. if (!CompareTwoItemsOnList(list1[cc], list_valid[cc])) { \
  156. std::cerr << "Problem setting " #list1 ". Value of " << cc \
  157. << " is: [" << list1[cc] << "] <> [" << list_valid[cc] \
  158. << "]" << std::endl; \
  159. res = 1; \
  160. break; \
  161. } \
  162. } \
  163. std::cout << std::endl; \
  164. }
  165. CompareTwoLists(numbers_argument, valid_numbers, 10);
  166. CompareTwoLists(doubles_argument, valid_doubles, 3);
  167. CompareTwoLists(bools_argument, valid_bools, 3);
  168. CompareTwoLists(strings_argument, valid_strings, 4);
  169. CompareTwoLists(stl_strings_argument, valid_stl_strings, 4);
  170. std::cout << "Some STL String variable was set to: "
  171. << some_stl_string_variable << std::endl;
  172. std::cout << "Some bool variable was set to: " << some_bool_variable
  173. << std::endl;
  174. std::cout << "Some bool variable was set to: " << some_bool_variable1
  175. << std::endl;
  176. std::cout << "bool_arg1 variable was set to: " << bool_arg1 << std::endl;
  177. std::cout << "bool_arg2 variable was set to: " << bool_arg2 << std::endl;
  178. std::cout << std::endl;
  179. for (cc = 0; cc < strings_argument.size(); ++cc) {
  180. delete[] strings_argument[cc];
  181. strings_argument[cc] = KWSYS_NULLPTR;
  182. }
  183. return res;
  184. }