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

Reviewed-on: PostGuard/Canon#78

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}";
@@ -774,6 +836,7 @@ public class CodeGeneratorVisitor : TypeCheckVisitor
             if (expression.VariableType == PascalBasicType.Real)
             {
                 // 这里需要按照输出调整
+                // 在输出real的前面需要添加一个空格
                 value += output ? "%.6lf" : "%lf";
             }
 
@@ -783,7 +846,7 @@ public class CodeGeneratorVisitor : TypeCheckVisitor
             }
         }
 
-        return value.Trim();
+        return value;
     }
 
     private static string GenerateBasicTypeString(PascalType pascalType)

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>

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>();
 

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
+            OnTermGenerator?.Invoke(this, new TermGeneratorEventArgs { Term = Children[0].Convert<Term>() });
-            {
-                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>(),

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
+            OnFactorGenerator?.Invoke(this, new FactorGeneratorEventArgs { Factor = Children[0].Convert<Factor>() });
-            {
-                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>(),