Skip to content

Commit 7f9fe1e

Browse files
committed
Minor parser refractor
1 parent 1712472 commit 7f9fe1e

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

mlir/lib/Tools/mlir-query/Matcher/Parser.cpp

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -46,12 +46,12 @@ struct Parser::TokenInfo {
4646
TokenInfo() = default;
4747

4848
// Method to set the kind and text of the token
49-
void set(TokenKind newKind, StringRef newText) {
49+
void set(TokenKind newKind, llvm::StringRef newText) {
5050
kind = newKind;
5151
text = newText;
5252
}
5353

54-
StringRef text;
54+
llvm::StringRef text;
5555
TokenKind kind = TK_Eof;
5656
SourceRange range;
5757
VariantValue value;
@@ -60,13 +60,13 @@ struct Parser::TokenInfo {
6060
class Parser::CodeTokenizer {
6161
public:
6262
// Constructor with matcherCode and error
63-
explicit CodeTokenizer(StringRef &matcherCode, Diagnostics *error)
63+
explicit CodeTokenizer(llvm::StringRef matcherCode, Diagnostics *error)
6464
: code(matcherCode), startOfLine(matcherCode), line(1), error(error) {
6565
nextToken = getNextToken();
6666
}
6767

6868
// Constructor with matcherCode, error, and codeCompletionOffset
69-
CodeTokenizer(StringRef &matcherCode, Diagnostics *error,
69+
CodeTokenizer(llvm::StringRef matcherCode, Diagnostics *error,
7070
unsigned codeCompletionOffset)
7171
: code(matcherCode), startOfLine(matcherCode), error(error),
7272
codeCompletionLocation(matcherCode.data() + codeCompletionOffset) {
@@ -102,9 +102,9 @@ class Parser::CodeTokenizer {
102102
private:
103103
// Helper function to get the first character as a new StringRef and drop it
104104
// from the original string
105-
StringRef firstCharacterAndDrop(StringRef &str) {
105+
llvm::StringRef firstCharacterAndDrop(llvm::StringRef &str) {
106106
assert(!str.empty());
107-
StringRef firstChar = str.substr(0, 1);
107+
llvm::StringRef firstChar = str.substr(0, 1);
108108
str = str.drop_front();
109109
return firstChar;
110110
}
@@ -118,7 +118,7 @@ class Parser::CodeTokenizer {
118118
// Code completion case
119119
if (codeCompletionLocation && codeCompletionLocation <= code.data()) {
120120
result.set(TokenInfo::TK_CodeCompletion,
121-
StringRef(codeCompletionLocation, 0));
121+
llvm::StringRef(codeCompletionLocation, 0));
122122
codeCompletionLocation = nullptr;
123123
return result;
124124
}
@@ -186,7 +186,7 @@ class Parser::CodeTokenizer {
186186
return;
187187
}
188188
}
189-
StringRef errorText = code;
189+
llvm::StringRef errorText = code;
190190
code = code.drop_front(code.size());
191191
SourceRange range;
192192
range.start = result->range.start;
@@ -228,7 +228,7 @@ class Parser::CodeTokenizer {
228228
// Consume all leading whitespace from code, except newlines
229229
void consumeWhitespace() {
230230
code = code.drop_while(
231-
[](char c) { return StringRef(" \t\v\f\r").contains(c); });
231+
[](char c) { return llvm::StringRef(" \t\v\f\r").contains(c); });
232232
}
233233

234234
// Returns the current location in the source code
@@ -239,8 +239,8 @@ class Parser::CodeTokenizer {
239239
return location;
240240
}
241241

242-
StringRef code;
243-
StringRef startOfLine;
242+
llvm::StringRef code;
243+
llvm::StringRef startOfLine;
244244
unsigned line = 1;
245245
Diagnostics *error;
246246
TokenInfo nextToken;
@@ -486,7 +486,7 @@ bool Parser::parseMatcherExpressionImpl(const TokenInfo &nameToken,
486486
// completions minus the prefix.
487487
void Parser::addCompletion(const TokenInfo &compToken,
488488
const MatcherCompletion &completion) {
489-
if (StringRef(completion.typedText).startswith(compToken.text)) {
489+
if (llvm::StringRef(completion.typedText).startswith(compToken.text)) {
490490
completions.emplace_back(completion.typedText.substr(compToken.text.size()),
491491
completion.matcherDecl);
492492
}
@@ -572,7 +572,7 @@ Parser::Parser(CodeTokenizer *tokenizer, Sema *sema,
572572
Parser::RegistrySema::~RegistrySema() = default;
573573

574574
std::optional<MatcherCtor>
575-
Parser::RegistrySema::lookupMatcherCtor(StringRef matcherName) {
575+
Parser::RegistrySema::lookupMatcherCtor(llvm::StringRef matcherName) {
576576
return Registry::lookupMatcherCtor(matcherName);
577577
}
578578

@@ -603,7 +603,7 @@ Parser::RegistrySema::buildMatcherCtor(MatcherCtor ctor, SourceRange nameRange,
603603
return Registry::buildMatcherCtor(ctor, nameRange, args, error);
604604
}
605605

606-
bool Parser::parseExpression(StringRef &code, Sema *sema,
606+
bool Parser::parseExpression(llvm::StringRef &code, Sema *sema,
607607
const NamedValueMap *namedValues,
608608
VariantValue *value, Diagnostics *error) {
609609
CodeTokenizer tokenizer(code, error);
@@ -619,7 +619,7 @@ bool Parser::parseExpression(StringRef &code, Sema *sema,
619619
}
620620

621621
std::vector<MatcherCompletion>
622-
Parser::completeExpression(StringRef &code, unsigned completionOffset,
622+
Parser::completeExpression(llvm::StringRef &code, unsigned completionOffset,
623623
Sema *sema, const NamedValueMap *namedValues) {
624624
Diagnostics error;
625625
CodeTokenizer tokenizer(code, &error, completionOffset);
@@ -631,7 +631,7 @@ Parser::completeExpression(StringRef &code, unsigned completionOffset,
631631
}
632632

633633
std::optional<DynMatcher>
634-
Parser::parseMatcherExpression(StringRef &code, Sema *sema,
634+
Parser::parseMatcherExpression(llvm::StringRef &code, Sema *sema,
635635
const NamedValueMap *namedValues,
636636
Diagnostics *error) {
637637
VariantValue value;

0 commit comments

Comments
 (0)