1 files changed, 121 insertions, 0 deletions
diff --git a/tools/slang-cpp-parser/identifier-lookup.h b/tools/slang-cpp-parser/identifier-lookup.h
new file mode 100644
index 000000000..226d00667
--- /dev/null
+++ b/tools/slang-cpp-parser/identifier-lookup.h
@@ -0,0 +1,121 @@
+#pragma once
+
+#include "diagnostics.h"
+
+namespace CppParse
+{
+using namespace Slang;
+
+enum class IdentifierStyle
+{
+    None, ///< It's not an identifier
+
+    Identifier, ///< Just an identifier
+
+    PreDeclare, ///< Declare a type (not visible in C++ code)
+    TypeSet,    ///< TypeSet
+
+    TypeModifier, ///< const, volatile etc
+    Keyword,      ///< A keyword C/C++ keyword that is not another type
+
+    Class,     ///< class
+    Struct,    ///< struct
+    Namespace, ///< namespace
+    Enum,      ///< enum
+
+    TypeDef, ///< typedef
+
+    Access, ///< public, protected, private
+
+    Reflected,
+    Unreflected,
+
+    CallingConvention, ///< Used on a method
+    Virtual,           ///<
+
+    Template,
+
+    Static,
+
+    IntegerModifier,
+
+    Extern,
+
+    CallableMisc, ///< For SLANG_NO_THROW etc
+
+    IntegerType, ///< Built in integer type
+
+    Default, /// default
+
+    CountOf,
+};
+
+typedef uint32_t IdentifierFlags;
+struct IdentifierFlag
+{
+    enum Enum : IdentifierFlags
+    {
+        StartScope = 0x1, ///< namespace, struct or class
+        ClassLike = 0x2,  ///< Struct or class
+        Keyword = 0x4,
+        Reflection = 0x8,
+    };
+};
+
+
+class IdentifierLookup
+{
+public:
+    struct Pair
+    {
+        const char* name;
+        IdentifierStyle style;
+    };
+
+    IdentifierStyle get(const UnownedStringSlice& slice) const
+    {
+        Index index = m_pool.findIndex(slice);
+        return (index >= 0) ? m_styles[index] : IdentifierStyle::None;
+    }
+
+    void set(const char* name, IdentifierStyle style) { set(UnownedStringSlice(name), style); }
+
+    void set(const UnownedStringSlice& name, IdentifierStyle style);
+
+    void set(const char* const* names, size_t namesCount, IdentifierStyle style);
+
+    void set(const Pair* pairs, Index pairsCount);
+
+    void reset()
+    {
+        m_styles.clear();
+        m_pool.clear();
+    }
+
+    void initDefault(const UnownedStringSlice& markPrefix);
+
+    IdentifierLookup()
+        : m_pool(StringSlicePool::Style::Empty)
+    {
+        SLANG_ASSERT(m_pool.getSlicesCount() == 0);
+    }
+
+    static const IdentifierFlags kIdentifierFlags[Index(IdentifierStyle::CountOf)];
+
+protected:
+    List<IdentifierStyle> m_styles;
+    StringSlicePool m_pool;
+};
+
+
+SLANG_FORCE_INLINE IdentifierFlags getFlags(IdentifierStyle style)
+{
+    return IdentifierLookup::kIdentifierFlags[Index(style)];
+}
+
+SLANG_FORCE_INLINE bool hasFlag(IdentifierStyle style, IdentifierFlag::Enum flag)
+{
+    return (getFlags(style) & flag) != 0;
+}
+
+} // namespace CppParse