Skip to content

Commit

Permalink
[PEx] Minor corrections to IR
Browse files Browse the repository at this point in the history
  • Loading branch information
aman-goel committed May 7, 2024
1 parent cbcbd70 commit d739b61
Showing 1 changed file with 8 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -1645,26 +1645,29 @@ private void WriteExpr(CompilationContext context, StringWriter output, IPExpr e
context.Write(output, $".getValues()");
break;
case MapAccessExpr mapAccessExpr:
context.Write(output, $"(({GetPExplicitType(mapAccessExpr.MapExpr.Type)})");
WriteExpr(context, output, mapAccessExpr.MapExpr);
context.Write(output, ".get(");
context.Write(output, ").get(");
WriteExpr(context, output, mapAccessExpr.IndexExpr);
context.Write(output, ")");
break;
case SeqAccessExpr seqAccessExpr:
context.Write(output, $"(({GetPExplicitType(seqAccessExpr.SeqExpr.Type)})");
WriteExpr(context, output, seqAccessExpr.SeqExpr);
context.Write(output, ".get(");
context.Write(output, ").get(");
WriteExpr(context, output, seqAccessExpr.IndexExpr);
context.Write(output, ")");
break;
case SetAccessExpr setAccessExpr:
context.Write(output, $"(({GetPExplicitType(setAccessExpr.SetExpr.Type)})");
WriteExpr(context, output, setAccessExpr.SetExpr);
context.Write(output, ".get(");
context.Write(output, ").get(");
WriteExpr(context, output, setAccessExpr.IndexExpr);
context.Write(output, ")");
break;
case NamedTupleAccessExpr namedTupleAccessExpr:
context.Write(output, $"(({GetPExplicitType(namedTupleAccessExpr.Type)})(");
context.Write(output, "(");
context.Write(output, $"(({GetPExplicitType(namedTupleAccessExpr.SubExpr.Type)})");
WriteExpr(context, output, namedTupleAccessExpr.SubExpr);
context.Write(output, $").getField(\"{namedTupleAccessExpr.FieldName}\")))");
break;
Expand All @@ -1674,7 +1677,7 @@ private void WriteExpr(CompilationContext context, StringWriter output, IPExpr e
case TupleAccessExpr tupleAccessExpr:
context.Write(output, $"({GetPExplicitType(tupleAccessExpr.Type)})(");
var tupleType = (tupleAccessExpr.SubExpr.Type.Canonicalize() as TupleType);
context.Write(output, "(");
context.Write(output, $"(({GetPExplicitType(tupleAccessExpr.SubExpr.Type)})");
WriteExpr(context, output, tupleAccessExpr.SubExpr);
context.Write(output, $").getField({tupleAccessExpr.FieldNo}))");
break;
Expand Down

0 comments on commit d739b61

Please sign in to comment.