source/Interpreter/OptionValueEnumeration.cpp

   1 //===-- OptionValueEnumeration.cpp ------------------------------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #include "lldb/Interpreter/OptionValueEnumeration.h"
  11
  12 // C Includes
  13 // C++ Includes
  14 // Other libraries and framework includes
  15 // Project includes
  16 #include "lldb/Utility/StringList.h"
  17
  18 using namespace lldb;
  19 using namespace lldb_private;
  20
  21 OptionValueEnumeration::OptionValueEnumeration(
  22     const OptionEnumValueElement *enumerators, enum_type value)
  23     : OptionValue(), m_current_value(value), m_default_value(value),
  24       m_enumerations() {
  25   SetEnumerations(enumerators);
  26 }
  27
  28 OptionValueEnumeration::~OptionValueEnumeration() {}
  29
  30 void OptionValueEnumeration::DumpValue(const ExecutionContext *exe_ctx,
  31                                        Stream &strm, uint32_t dump_mask) {
  32   if (dump_mask & eDumpOptionType)
  33     strm.Printf("(%s)", GetTypeAsCString());
  34   if (dump_mask & eDumpOptionValue) {
  35     if (dump_mask & eDumpOptionType)
  36       strm.PutCString(" = ");
  37     const size_t count = m_enumerations.GetSize();
  38     for (size_t i = 0; i < count; ++i) {
  39       if (m_enumerations.GetValueAtIndexUnchecked(i).value == m_current_value) {
  40         strm.PutCString(m_enumerations.GetCStringAtIndex(i).GetStringRef());
  41         return;
  42       }
  43     }
  44     strm.Printf("%" PRIu64, (uint64_t)m_current_value);
  45   }
  46 }
  47
  48 Status OptionValueEnumeration::SetValueFromString(llvm::StringRef value,
  49                                                   VarSetOperationType op) {
  50   Status error;
  51   switch (op) {
  52   case eVarSetOperationClear:
  53     Clear();
  54     NotifyValueChanged();
  55     break;
  56
  57   case eVarSetOperationReplace:
  58   case eVarSetOperationAssign: {
  59     ConstString const_enumerator_name(value.trim());
  60     const EnumerationMapEntry *enumerator_entry =
  61         m_enumerations.FindFirstValueForName(const_enumerator_name);
  62     if (enumerator_entry) {
  63       m_current_value = enumerator_entry->value.value;
  64       NotifyValueChanged();
  65     } else {
  66       StreamString error_strm;
  67       error_strm.Printf("invalid enumeration value '%s'", value.str().c_str());
  68       const size_t count = m_enumerations.GetSize();
  69       if (count) {
  70         error_strm.Printf(", valid values are: %s",
  71                           m_enumerations.GetCStringAtIndex(0).GetCString());
  72         for (size_t i = 1; i < count; ++i) {
  73           error_strm.Printf(", %s",
  74                             m_enumerations.GetCStringAtIndex(i).GetCString());
  75         }
  76       }
  77       error.SetErrorString(error_strm.GetString());
  78     }
  79     break;
  80   }
  81
  82   case eVarSetOperationInsertBefore:
  83   case eVarSetOperationInsertAfter:
  84   case eVarSetOperationRemove:
  85   case eVarSetOperationAppend:
  86   case eVarSetOperationInvalid:
  87     error = OptionValue::SetValueFromString(value, op);
  88     break;
  89   }
  90   return error;
  91 }
  92
  93 void OptionValueEnumeration::SetEnumerations(
  94     const OptionEnumValueElement *enumerators) {
  95   m_enumerations.Clear();
  96   if (enumerators) {
  97     for (size_t i = 0; enumerators[i].string_value != nullptr; ++i) {
  98       ConstString const_enumerator_name(enumerators[i].string_value);
  99       EnumeratorInfo enumerator_info = {enumerators[i].value,
 100                                         enumerators[i].usage};
 101       m_enumerations.Append(const_enumerator_name,
 102                             enumerator_info);
 103     }
 104     m_enumerations.Sort();
 105   }
 106 }
 107
 108 lldb::OptionValueSP OptionValueEnumeration::DeepCopy() const {
 109   return OptionValueSP(new OptionValueEnumeration(*this));
 110 }
 111
 112 size_t OptionValueEnumeration::AutoComplete(
 113     CommandInterpreter &interpreter, llvm::StringRef s, int match_start_point,
 114     int max_return_elements, bool &word_complete, StringList &matches) {
 115   word_complete = false;
 116   matches.Clear();
 117
 118   const uint32_t num_enumerators = m_enumerations.GetSize();
 119   if (!s.empty()) {
 120     for (size_t i = 0; i < num_enumerators; ++i) {
 121       llvm::StringRef name = m_enumerations.GetCStringAtIndex(i).GetStringRef();
 122       if (name.startswith(s))
 123         matches.AppendString(name);
 124     }
 125   } else {
 126     // only suggest "true" or "false" by default
 127     for (size_t i = 0; i < num_enumerators; ++i)
 128       matches.AppendString(m_enumerations.GetCStringAtIndex(i).GetStringRef());
 129   }
 130   return matches.GetSize();
 131 }