fix: 调整输出的格式 (#78)

Reviewed-on: PostGuard/Canon#78
2024-05-10 16:45:21 +08:00
parent f993d49856
commit 1908561d31
5 changed files with 122 additions and 43 deletions
--- a/Canon.Core/SemanticParser/CodeGeneratorVisitor.cs
+++ b/Canon.Core/SemanticParser/CodeGeneratorVisitor.cs
@@ -146,6 +146,19 @@ public class CodeGeneratorVisitor : TypeCheckVisitor
        Builder.EndScope();
    }

+    public override void PreVisit(Factor factor)
+    {
+        base.PreVisit(factor);
+
+        factor.OnParethnesisGenerator += (_, e) => { e.Expression.IsCondition = factor.IsCondition; };
+
+        factor.OnNotGenerator += (_, e) => { e.Factor.IsCondition = factor.IsCondition; };
+
+        factor.OnUminusGenerator += (_, e) => { e.Factor.IsCondition = factor.IsCondition; };
+
+        factor.OnPlusGenerator += (_, e) => { e.Factor.IsCondition = factor.IsCondition; };
+    }
+
    public override void PostVisit(Factor factor)
    {
        base.PostVisit(factor);
@@ -256,7 +269,23 @@ public class CodeGeneratorVisitor : TypeCheckVisitor
            return;
        }

-        variable.VariableName = variable.Identifier.IdentifierName;
+        // 虽然这里这样可能会生成来类似于 &*x 的代码
+        // 但是可以正常运行
+        variable.VariableName =
+            symbol.Reference ? $"*{variable.Identifier.IdentifierName}" : variable.Identifier.IdentifierName;
+    }
+
+    public override void PreVisit(Term term)
+    {
+        base.PreVisit(term);
+
+        term.OnMultiplyGenerator += (_, e) =>
+        {
+            e.Left.IsCondition = term.IsCondition;
+            e.Right.IsCondition = term.IsCondition;
+        };
+
+        term.OnFactorGenerator += (_, e) => { e.Factor.IsCondition = term.IsCondition; };
    }

    public override void PostVisit(Term term)
@@ -276,6 +305,19 @@ public class CodeGeneratorVisitor : TypeCheckVisitor
        };
    }

+    public override void PreVisit(SimpleExpression simpleExpression)
+    {
+        base.PreVisit(simpleExpression);
+
+        simpleExpression.OnAddGenerator += (_, e) =>
+        {
+            e.Left.IsCondition = simpleExpression.IsCondition;
+            e.Right.IsCondition = simpleExpression.IsCondition;
+        };
+
+        simpleExpression.OnTermGenerator += (_, e) => { e.Term.IsCondition = simpleExpression.IsCondition; };
+    }
+
    public override void PostVisit(SimpleExpression simpleExpression)
    {
        base.PostVisit(simpleExpression);
@@ -298,6 +340,16 @@ public class CodeGeneratorVisitor : TypeCheckVisitor
    {
        base.PreVisit(expression);

+        expression.OnSimpleExpressionGenerator += (_, e) =>
+        {
+            e.SimpleExpression.IsCondition = expression.IsCondition;
+        };
+
+        expression.OnRelationGenerator += (_, e) =>
+        {
+            e.Left.IsCondition = expression.IsCondition;
+            e.Right.IsCondition = expression.IsCondition;
+        };
        if (expression.IsWhileCondition)
        {
            GenerateWhileLabel();
@@ -306,6 +358,7 @@ public class CodeGeneratorVisitor : TypeCheckVisitor
                             """);
        }
    }
+
    public override void PostVisit(Expression expression)
    {
        base.PostVisit(expression);
@@ -422,7 +475,11 @@ public class CodeGeneratorVisitor : TypeCheckVisitor
    {
        base.PreVisit(statement);

-        statement.OnIfGenerator += (_, e) => { e.Condition.IsIfCondition = true; };
+        statement.OnIfGenerator += (_, e) =>
+        {
+            e.Condition.IsIfCondition = true;
+            e.Condition.IsCondition = true;
+        };

        statement.OnForGenerator += (_, e) =>
        {
@@ -445,16 +502,6 @@ public class CodeGeneratorVisitor : TypeCheckVisitor

        statement.OnAssignGenerator += (_, e) =>
        {
-            if (!SymbolTable.TryGetSymbol(e.Variable.Identifier.IdentifierName, out Symbol? symbol))
-            {
-                return;
-            }
-
-            if (symbol.Reference)
-            {
-                e.Variable.VariableName = "*" + e.Variable.VariableName;
-            }
-
            Builder.AddLine($"{e.Variable.VariableName} = {e.Expression.VariableName};");
        };

@@ -686,7 +733,7 @@ public class CodeGeneratorVisitor : TypeCheckVisitor

        if (procedureId == "write" || procedureId == "writeln")
        {
-            string result = $"printf(\"{GenerateFormatString(parameters) + isReturn}\"";
+            string result = $"printf(\"{GenerateFormatString(parameters, true) + isReturn}\"";

            foreach (Expression parameter in parameters)
            {
@@ -747,9 +794,24 @@ public class CodeGeneratorVisitor : TypeCheckVisitor
                 parameters.Zip(functionType.Parameters))
        {
            if (parameterType.IsVar)
+            {
+                // 这里需要判断parameter是否也为引用类型
+                if (SymbolTable.TryGetSymbol(parameter.VariableName, out Symbol? parameterSymboe))
+                {
+                    if (parameterSymboe.Reference)
+                    {
+                        parameterValue += $", {parameter.VariableName}";
+                    }
+                    else
                    {
                        parameterValue += $", &{parameter.VariableName}";
                    }
+                }
+                else
+                {
+                    parameterValue += $", &{parameter.VariableName}";
+                }
+            }
            else
            {
                parameterValue += $", {parameter.VariableName}";
@@ -768,22 +830,23 @@ public class CodeGeneratorVisitor : TypeCheckVisitor
        {
            if (expression.VariableType == PascalBasicType.Integer)
            {
-                value += "%d ";
+                value += "%d";
            }

            if (expression.VariableType == PascalBasicType.Real)
            {
                // 这里需要按照输出调整
-                value += output ? "%.6lf " : "%lf ";
+                // 在输出real的前面需要添加一个空格
+                value += output ? "%.6lf" : "%lf";
            }

            if (expression.VariableType == PascalBasicType.Character)
            {
-                value += "%c ";
+                value += "%c";
            }
        }

-        return value.Trim();
+        return value;
    }

    private static string GenerateBasicTypeString(PascalType pascalType)
--- a/Canon.Core/SyntaxNodes/Expression.cs
+++ b/Canon.Core/SyntaxNodes/Expression.cs
@@ -48,6 +48,11 @@ public class Expression : NonTerminatedSyntaxNode
    /// </summary>
    public bool IsIfCondition { get; set; }

+    /// <summary>
+    /// 是否为条件判断语句
+    /// </summary>
+    public bool IsCondition { get; set; }
+
    /// <summary>
    /// 是否为WHILE语句中的条件语句
    /// </summary>
--- a/Canon.Core/SyntaxNodes/Factor.cs
+++ b/Canon.Core/SyntaxNodes/Factor.cs
@@ -50,6 +50,11 @@ public class Factor : NonTerminatedSyntaxNode
 {
    public override NonTerminatorType Type => NonTerminatorType.Factor;

+    /// <summary>
+    /// 是否为条件判断语句
+    /// </summary>
+    public bool IsCondition { get; set; }
+
    public override void PreVisit(SyntaxNodeVisitor visitor)
    {
        visitor.PreVisit(this);
@@ -158,7 +163,8 @@ public class Factor : NonTerminatedSyntaxNode
            else
            {
                // factor -> id ( )
-                OnProcedureCallGenerator?.Invoke(this, new ProcedureCallGeneratorEventArgs
+                OnProcedureCallGenerator?.Invoke(this,
+                    new ProcedureCallGeneratorEventArgs
                    {
                        ProcedureName = terminatedSyntaxNode.Token.Convert<IdentifierSemanticToken>()
                    });
@@ -177,7 +183,6 @@ public class Factor : NonTerminatedSyntaxNode
        }
        else
        {
-
            SemanticToken token = Children[0].Convert<TerminatedSyntaxNode>().Token;
            Factor factor = Children[1].Convert<Factor>();

--- a/Canon.Core/SyntaxNodes/SimpleExpression.cs
+++ b/Canon.Core/SyntaxNodes/SimpleExpression.cs
@@ -21,6 +21,11 @@ public class SimpleExpression : NonTerminatedSyntaxNode
 {
    public override NonTerminatorType Type => NonTerminatorType.SimpleExpression;

+    /// <summary>
+    /// 是否为条件判断语句
+    /// </summary>
+    public bool IsCondition { get; set; }
+
    public override void PreVisit(SyntaxNodeVisitor visitor)
    {
        visitor.PreVisit(this);
@@ -52,14 +57,12 @@ public class SimpleExpression : NonTerminatedSyntaxNode
    {
        if (Children.Count == 1)
        {
-            OnTermGenerator?.Invoke(this, new TermGeneratorEventArgs
-            {
-                Term = Children[0].Convert<Term>()
-            });
+            OnTermGenerator?.Invoke(this, new TermGeneratorEventArgs { Term = Children[0].Convert<Term>() });
        }
        else
        {
-            OnAddGenerator?.Invoke(this, new AddGeneratorEventArgs
+            OnAddGenerator?.Invoke(this,
+                new AddGeneratorEventArgs
                {
                    Left = Children[0].Convert<SimpleExpression>(),
                    Operator = Children[1].Convert<AddOperator>(),
--- a/Canon.Core/SyntaxNodes/Term.cs
+++ b/Canon.Core/SyntaxNodes/Term.cs
@@ -21,6 +21,11 @@ public class Term : NonTerminatedSyntaxNode
 {
    public override NonTerminatorType Type => NonTerminatorType.Term;

+    /// <summary>
+    /// 是否为条件判断语句
+    /// </summary>
+    public bool IsCondition { get; set; }
+
    public override void PreVisit(SyntaxNodeVisitor visitor)
    {
        visitor.PreVisit(this);
@@ -52,14 +57,12 @@ public class Term : NonTerminatedSyntaxNode
    {
        if (Children.Count == 1)
        {
-            OnFactorGenerator?.Invoke(this, new FactorGeneratorEventArgs
-            {
-                Factor = Children[0].Convert<Factor>()
-            });
+            OnFactorGenerator?.Invoke(this, new FactorGeneratorEventArgs { Factor = Children[0].Convert<Factor>() });
        }
        else
        {
-            OnMultiplyGenerator?.Invoke(this, new MultiplyGeneratorEventArgs
+            OnMultiplyGenerator?.Invoke(this,
+                new MultiplyGeneratorEventArgs
                {
                    Left = Children[0].Convert<Term>(),
                    Operator = Children[1].Convert<MultiplyOperator>(),