Added support for hiding command-line options

1 files changed, 25 insertions, 4 deletions

diff --git a/lib/Basic/CmdLineOptions.cpp b/lib/Basic/CmdLineOptions.cpp
index 324f349e..da333884 100644
--- a/lib/Basic/CmdLineOptions.cpp
+++ b/lib/Basic/CmdLineOptions.cpp
@@ -15,6 +15,12 @@
 #include "klee/CommandLine.h"
 #include "klee/Config/Version.h"
 
+#include "llvm/ADT/ArrayRef.h"
+#include "llvm/ADT/StringMap.h"
+#include "llvm/Support/CommandLine.h"
+
+using namespace llvm;
+
 namespace klee {
 
 llvm::cl::opt<bool>
@@ -81,6 +87,25 @@ llvm::cl::opt<bool>
     UseAssignmentValidatingSolver("debug-assignment-validating-solver",
                                   llvm::cl::init(false));
 
+void KCommandLine::HideUnrelatedOptions(cl::OptionCategory &Category) {
+  StringMap<cl::Option *> map;
+  cl::getRegisteredOptions(map);
+  for (StringMap<cl::Option *>::iterator i = map.begin(), e = map.end(); i != e;
+       i++) {
+    if (i->second->Category != &Category) {
+      i->second->setHiddenFlag(cl::Hidden);
+    }
+  }
+}
+
+void KCommandLine::HideUnrelatedOptions(
+    llvm::ArrayRef<const llvm::cl::OptionCategory *> Categories) {
+  for (ArrayRef<const cl::OptionCategory *>::iterator i = Categories.begin(),
+                                                      e = Categories.end();
+       i != e; i++)
+    HideUnrelatedOptions(*i);
+}
+
 #ifdef ENABLE_METASMT
 
 #ifdef METASMT_DEFAULT_BACKEND_IS_BTOR
@@ -156,7 +181,3 @@ llvm::cl::opt<CoreSolverType> DebugCrossCheckCoreSolverWith(
 #undef METASMT_IS_DEFAULT_STR
 #undef Z3_IS_DEFAULT_STR
 #undef DEFAULT_CORE_SOLVER
-
-
-
-