source: icGREP/icgrep-devel/icgrep/re/re_name.h

Last change on this file was 5847, checked in by cameron, 3 weeks ago

Unify Parabix CC compiler and Direct CC compiler with a common superclass

File size: 5.8 KB
Line 
1#ifndef RE_NAME_H
2#define RE_NAME_H
3
4#include <re/re_re.h>
5#include <re/re_cc.h>
6#include <llvm/Support/Casting.h>
7
8namespace UCD {
9    class UnicodeSet;
10}
11
12namespace re {
13
14class Name : public RE {
15public:
16    static inline bool classof(const RE * re) {
17        return re->getClassTypeId() == ClassTypeId::Name;
18    }
19    static inline bool classof(const void *) {
20        return false;
21    }
22    using length_t = std::string::size_type;
23    enum class Type {
24        Unicode
25        , UnicodeProperty
26        , Capture
27        , Reference
28        , ZeroWidth
29        , Unknown
30    };
31    std::string getNamespace() const;
32    bool hasNamespace() const;
33    std::string getName() const;
34    Type getType() const;
35    RE * getDefinition() const;
36    bool operator<(const Name & other) const;
37    bool operator<(const CC & other) const;
38    bool operator>(const CC & other) const;
39    void setDefinition(RE * definition);
40    virtual ~Name() {}
41protected:
42    friend Name * makeName(const std::string & name, RE * cc);
43    friend Name * makeCapture(const std::string & name, RE * captured);
44    friend Name * makeReference(const std::string & name, RE * captureName);
45    friend Name * makeZeroWidth(const std::string & name, RE * zerowidth);
46    friend Name * makeName(CC * const cc);
47    friend Name * makeName(const std::string &, Type);
48    friend Name * makeName(const std::string &, const std::string &, Type);
49    friend Name * makeName(const std::string & nm, const Name::Type type, RE * defn); 
50    Name(const char * nameSpace, const length_t namespaceLength, const char * name, const length_t nameLength, Type type, RE * defn)
51    : RE(ClassTypeId::Name)
52    , mNamespaceLength(namespaceLength)
53    , mNamespace(replicateString(nameSpace, namespaceLength))
54    , mNameLength(nameLength)
55    , mName(replicateString(name, nameLength))
56    , mType(type)
57    , mDefinition(defn) {
58
59    }
60    inline const char * replicateString(const char * string, const length_t length) {
61        if (string && (length > 0)) {
62            char * allocated = reinterpret_cast<char*>(mAllocator.allocate(length));
63            std::memcpy(allocated, string, length);
64            return allocated;
65        }
66        return nullptr;
67    }
68
69private:
70    const length_t      mNamespaceLength;
71    const char * const  mNamespace;
72    const length_t      mNameLength;
73    const char * const  mName;
74    Type                mType;
75    RE *                mDefinition;
76};
77
78inline std::string Name::getNamespace() const {
79    return std::string(mNamespace, mNamespaceLength);
80}
81
82inline bool Name::hasNamespace() const {
83    return (mNamespaceLength != 0);
84}
85
86inline std::string Name::getName() const {
87    return std::string(mName, mNameLength);
88}
89   
90inline Name::Type Name::getType() const {
91    return mType;
92}
93
94inline RE * Name::getDefinition() const {
95    return mDefinition;
96}
97
98inline void Name::setDefinition(RE * definition) {
99    assert (definition != nullptr);
100    assert (definition != this);
101    mDefinition = definition;
102}
103
104inline bool Name::operator < (const Name & other) const {
105    if (LLVM_LIKELY(mDefinition && other.mDefinition && llvm::isa<CC>(mDefinition) && llvm::isa<CC>(other.mDefinition))) {
106        return *llvm::cast<CC>(mDefinition) < *llvm::cast<CC>(other.mDefinition);
107    } else if (mNamespaceLength < other.mNamespaceLength) {
108        return true;
109    } else if (mNamespaceLength > other.mNamespaceLength) {
110        return false;
111    } else if (mNameLength < other.mNameLength) {
112        return true;
113    } else if (mNameLength > other.mNameLength) {
114        return false;
115    }
116    const auto diff = std::memcmp(mNamespace, other.mNamespace, mNamespaceLength);
117    if (diff < 0) {
118        return true;
119    } else if (diff > 0) {
120        return false;
121    }
122    return (std::memcmp(mName, other.mName, mNameLength) < 0);
123}
124
125inline bool Name::operator < (const CC & other) const {
126    if (mDefinition && llvm::isa<CC>(mDefinition)) {
127        return *llvm::cast<CC>(mDefinition) < other;
128    }
129    return RE::ClassTypeId::Name < RE::ClassTypeId::CC;
130}
131
132inline bool Name::operator > (const CC & other) const {
133    if (mDefinition && llvm::isa<CC>(mDefinition)) {
134        return other < *llvm::cast<CC>(mDefinition);
135    }
136    return RE::ClassTypeId::CC < RE::ClassTypeId::Name;
137}
138
139inline Name * makeName(const std::string & name, const Name::Type type) {
140    return new Name(nullptr, 0, name.c_str(), name.length(), type, nullptr);
141}
142
143inline Name * makeName(const std::string & property, const std::string & value, const Name::Type type) {
144    return new Name(property.c_str(), property.length(), value.c_str(), value.length(), type, nullptr);
145}
146   
147inline Name * makeName(const std::string & nm, const Name::Type type, RE * defn) {
148    return new Name(nullptr, 0, nm.c_str(), nm.length(), type, defn);
149}
150
151inline Name * makeName(const std::string & name, RE * cc) {
152    if (llvm::isa<Name>(cc)) {
153        return llvm::cast<Name>(cc);
154    }
155    else if (llvm::isa<CC>(cc)) {
156        return new Name(nullptr, 0, name.c_str(), name.length(), Name::Type::Unicode, cc);
157    }
158    else return new Name(nullptr, 0, name.c_str(), name.length(), Name::Type::Unknown, cc);
159}
160
161inline Name * makeName(CC * const cc) {
162    const std::string name = cc->canonicalName();
163    return new Name(nullptr, 0, name.c_str(), name.length(), Name::Type::Unicode, cc);
164}
165
166inline Name * makeCapture(const std::string & name, RE * captured) {
167    return new Name(nullptr, 0, name.c_str(), name.length(), Name::Type::Capture, captured);
168}
169   
170inline Name * makeReference(const std::string & name, RE * captureName) {
171    return new Name(nullptr, 0, name.c_str(), name.length(), Name::Type::Reference, captureName);
172}
173
174inline Name * makeZeroWidth(const std::string & name, RE * zerowidth = NULL) {
175    return new Name(nullptr, 0, name.c_str(), name.length(), Name::Type::ZeroWidth, zerowidth);
176}
177}
178
179#endif // RE_NAME_H
Note: See TracBrowser for help on using the repository browser.