Skip to content

Commit ae690fa

Browse files
committed
Refactor parsing methods to replace OperationParseHeader with OperationHeader for improved clarity and consistency across multiple classes.
1 parent 2acebe1 commit ae690fa

9 files changed

Lines changed: 20 additions & 22 deletions

File tree

src/MLIR.Generators/Emitters/AssemblyFormat/TryParseEmitter.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ public static void Emit(StringBuilder builder, OperationModel operation, Operati
5454

5555
private void EmitMethod(StringBuilder builder)
5656
{
57-
builder.AppendLine(" protected override ParseResult<OperationBodySyntax> TryParseBody(in OperationParseHeader header, OperationParsingContext context)");
57+
builder.AppendLine(" protected override ParseResult<OperationBodySyntax> TryParseBody(in OperationHeader header, OperationParsingContext context)");
5858
builder.AppendLine(" {");
5959

6060
var format = operation.AssemblyFormat!;

src/MLIR/Dialects/Extensions/FuncOperationAssemblyFormat.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ public FuncOperationAssemblyFormat(OperationDefinition definition)
2727

2828
/// <inheritdoc/>
2929
protected override ParseResult<OperationBodySyntax> TryParseBody(
30-
in OperationParseHeader header,
30+
in OperationHeader header,
3131
OperationParsingContext context)
3232
{
3333
if (header.ResultList.Count != 0 || header.EqualsToken.HasValue)

src/MLIR/Dialects/Extensions/SelectLikeOperationAssemblyFormat.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ public SelectLikeOperationAssemblyFormat(OperationDefinition definition)
3131

3232
/// <inheritdoc/>
3333
protected override ParseResult<OperationBodySyntax> TryParseBody(
34-
in OperationParseHeader header,
34+
in OperationHeader header,
3535
OperationParsingContext context)
3636
{
3737
var conditionResult = context.TryParseSsaToken();

src/MLIR/Dialects/IOperationAssemblyFormat.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ public ParseResult<OperationSyntax> TryParse(OperationParsingContext context)
4444
/// <param name="context">The parsing context.</param>
4545
/// <param name="header">The parsed operation header.</param>
4646
/// <returns>The parsed operation syntax, a no-match result, or a diagnostic-producing failure.</returns>
47-
public ParseResult<OperationSyntax> TryParseAfterHeader(OperationParseHeader header, in OperationParsingContext context)
47+
public ParseResult<OperationSyntax> TryParseAfterHeader(OperationHeader header, in OperationParsingContext context)
4848
{
4949
var result = TryParseBody(header, context);
5050
if (!result.IsSuccess)
@@ -71,7 +71,7 @@ public ParseResult<OperationSyntax> TryParseAfterHeader(OperationParseHeader hea
7171
/// <param name="context">The parsing context.</param>
7272
/// <returns>The parsed operation body syntax, a no-match result, or a diagnostic-producing failure.</returns>
7373
protected abstract ParseResult<OperationBodySyntax> TryParseBody(
74-
in OperationParseHeader header,
74+
in OperationHeader header,
7575
OperationParsingContext context);
7676

7777
/// <summary>
Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,15 @@
1-
namespace MLIR.Text;
2-
3-
using MLIR.Syntax;
1+
namespace MLIR.Syntax;
42

53
/// <summary>
64
/// Carries the operation header parsed before dispatching to a custom operation assembly format.
75
/// </summary>
86
/// <remarks>
9-
/// Initializes a new instance of the <see cref="OperationParseHeader"/> class.
7+
/// Initializes a new instance of the <see cref="OperationHeader"/> class.
108
/// </remarks>
119
/// <param name="nameToken">The parsed operation name token.</param>
1210
/// <param name="resultList">The parsed SSA result tokens with their separator tokens.</param>
1311
/// <param name="equalsToken">The parsed equals token, if present.</param>
14-
public readonly struct OperationParseHeader(
12+
public readonly struct OperationHeader(
1513
Token nameToken,
1614
SeparatedSyntaxList<Token> resultList,
1715
Token? equalsToken)

src/MLIR/Text/OperationParsingContext.cs

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ internal OperationParsingContext(Parser parser)
1818
/// <summary>
1919
/// Parses an operation header: optional SSA result list, optional equals token, and operation name token.
2020
/// </summary>
21-
public ParseResult<OperationParseHeader> TryParseHeader()
21+
public ParseResult<OperationHeader> TryParseHeader()
2222
{
2323
var resultItems = new List<Token>();
2424
var resultSeparators = new List<Token>();
@@ -29,7 +29,7 @@ public ParseResult<OperationParseHeader> TryParseHeader()
2929
var firstResultTokenResult = TryParseSsaToken();
3030
if (!firstResultTokenResult.IsSuccess)
3131
{
32-
return ParseResult<OperationParseHeader>.Failure(firstResultTokenResult.Diagnostic!);
32+
return ParseResult<OperationHeader>.Failure(firstResultTokenResult.Diagnostic!);
3333
}
3434

3535
var firstResultToken = firstResultTokenResult.Value;
@@ -40,7 +40,7 @@ public ParseResult<OperationParseHeader> TryParseHeader()
4040
var countTokenResult = Expect(TokenKind.Integer, "Expected result count after ':'.");
4141
if (!countTokenResult.IsSuccess)
4242
{
43-
return ParseResult<OperationParseHeader>.Failure(countTokenResult.Diagnostic!);
43+
return ParseResult<OperationHeader>.Failure(countTokenResult.Diagnostic!);
4444
}
4545

4646
var count = int.Parse(countTokenResult.Value.Text, CultureInfo.InvariantCulture);
@@ -56,7 +56,7 @@ public ParseResult<OperationParseHeader> TryParseHeader()
5656
var nextResultToken = TryParseSsaToken();
5757
if (!nextResultToken.IsSuccess)
5858
{
59-
return ParseResult<OperationParseHeader>.Failure(nextResultToken.Diagnostic!);
59+
return ParseResult<OperationHeader>.Failure(nextResultToken.Diagnostic!);
6060
}
6161

6262
resultItems.Add(nextResultToken.Value);
@@ -65,7 +65,7 @@ public ParseResult<OperationParseHeader> TryParseHeader()
6565
var equalsTokenResult = Expect(TokenKind.Equal, "Expected '=' after operation result list.");
6666
if (!equalsTokenResult.IsSuccess)
6767
{
68-
return ParseResult<OperationParseHeader>.Failure(equalsTokenResult.Diagnostic!);
68+
return ParseResult<OperationHeader>.Failure(equalsTokenResult.Diagnostic!);
6969
}
7070

7171
equalsToken = equalsTokenResult.Value;
@@ -82,10 +82,10 @@ public ParseResult<OperationParseHeader> TryParseHeader()
8282
}
8383
else
8484
{
85-
return ParseResult<OperationParseHeader>.Failure(CreateDiagnostic("Expected an operation name."));
85+
return ParseResult<OperationHeader>.Failure(CreateDiagnostic("Expected an operation name."));
8686
}
8787

88-
return ParseResult<OperationParseHeader>.Success(new OperationParseHeader(
88+
return ParseResult<OperationHeader>.Success(new OperationHeader(
8989
nameToken,
9090
new SeparatedSyntaxList<Token>(resultItems, resultSeparators),
9191
equalsToken));

tests/MLIR.Generators.Tests/DialectGeneratorAssemblyFormatTests.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ public void TryParseGeneratesOperandAndLiteralParsingCallsForAddIOp()
110110
AssertContainsAll(
111111
registrationSource,
112112
"protected override ParseResult<OperationBodySyntax> TryParseBody(",
113-
"in OperationParseHeader header,",
113+
"in OperationHeader header,",
114114
"var lhsResult = context.TryParseSsaToken();",
115115
"var rhsResult = context.TryParseSsaToken();",
116116
"context.Expect(TokenKind.Comma, ",

tests/MLIR.Tests/ParsingTests.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ public override SyntaxNode Rewrite(SyntaxRewriter rewriter)
6262
private sealed class PrefixConstantAssemblyFormat : BodyOnlyOperationAssemblyFormat
6363
{
6464
protected override ParseResult<OperationBodySyntax> TryParseBody(
65-
in OperationParseHeader header,
65+
in OperationHeader header,
6666
OperationParsingContext context)
6767
{
6868
if (context.Is(TokenKind.LParen))
@@ -143,7 +143,7 @@ public override SyntaxNode Rewrite(SyntaxRewriter rewriter)
143143
private sealed class SsaListCapturingAssemblyFormat : BodyOnlyOperationAssemblyFormat
144144
{
145145
protected override ParseResult<OperationBodySyntax> TryParseBody(
146-
in OperationParseHeader header,
146+
in OperationHeader header,
147147
OperationParsingContext context)
148148
{
149149
var listResult = context.TryParseSsaTokenList();

tests/MLIR.Tests/SemanticTestInfrastructure.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -444,7 +444,7 @@ public PrefixConstantAssemblyFormat(OperationDefinition definition)
444444
}
445445

446446
protected override ParseResult<OperationBodySyntax> TryParseBody(
447-
in OperationParseHeader header,
447+
in OperationHeader header,
448448
OperationParsingContext context)
449449
{
450450
if (context.Is(TokenKind.LParen))
@@ -516,7 +516,7 @@ public ContextDirectedConstantAssemblyFormat(
516516
}
517517

518518
protected override ParseResult<OperationBodySyntax> TryParseBody(
519-
in OperationParseHeader header,
519+
in OperationHeader header,
520520
OperationParsingContext context)
521521
{
522522
if (context.Is(TokenKind.LParen))

0 commit comments

Comments
 (0)