AST development

Src/AST/Myc.Ast.pas

@@ -22,11 +22,17 @@ type
             const ACondition: IExpressionNode;
             const AThenBranch, AElseBranch: IExpressionNode
         ): IIfExpressionNode; static;
+        // Added factory function for the new ternary expression node.
+        class function TernaryExpr(
+            const ACondition: IExpressionNode;
+            const AThenBranch, AElseBranch: IExpressionNode
+        ): ITernaryExpressionNode; static;
         class function LambdaExpr(const AParameters: TArray<IIdentifierNode>; const ABody: IExpressionNode): ILambdaExpressionNode; static;
         class function FunctionCall(const ACallee: IExpressionNode; const AArguments: array of IExpressionNode): IFunctionCallNode; static;
         class function Block(const AExpressions: array of IExpressionNode): IBlockExpressionNode; static;
         class function VarDecl(const AIdentifier: IIdentifierNode; AInitializer: IExpressionNode): IVariableDeclarationNode; static;
         class function Assign(const AIdentifier: IIdentifierNode; const AValue: IExpressionNode): IAssignmentNode; static;
+        class function AssignResult(const AValue: IExpressionNode): IAssignmentNode; static;
     end;
 
 implementation
@@ -103,6 +109,20 @@ type
         function Accept(const Visitor: IAstVisitor): TAstValue; override;
     end;
 
+    { TTernaryExpressionNode }
+    TTernaryExpressionNode = class(TAstNode, ITernaryExpressionNode)
+    private
+        FCondition: IExpressionNode;
+        FThenBranch: IExpressionNode;
+        FElseBranch: IExpressionNode;
+        function GetCondition: IExpressionNode;
+        function GetThenBranch: IExpressionNode;
+        function GetElseBranch: IExpressionNode;
+    public
+        constructor Create(const ACondition, AThenBranch, AElseBranch: IExpressionNode);
+        function Accept(const Visitor: IAstVisitor): TAstValue; override;
+    end;
+
     { TLambdaExpressionNode }
     TLambdaExpressionNode = class(TAstNode, ILambdaExpressionNode)
     private
@@ -283,6 +303,36 @@ begin
     Result := FThenBranch;
 end;
 
+{ TTernaryExpressionNode }
+
+constructor TTernaryExpressionNode.Create(const ACondition, AThenBranch, AElseBranch: IExpressionNode);
+begin
+    inherited Create;
+    FCondition := ACondition;
+    FThenBranch := AThenBranch;
+    FElseBranch := AElseBranch;
+end;
+
+function TTernaryExpressionNode.Accept(const Visitor: IAstVisitor): TAstValue;
+begin
+    Result := Visitor.VisitTernaryExpression(Self);
+end;
+
+function TTernaryExpressionNode.GetCondition: IExpressionNode;
+begin
+    Result := FCondition;
+end;
+
+function TTernaryExpressionNode.GetElseBranch: IExpressionNode;
+begin
+    Result := FElseBranch;
+end;
+
+function TTernaryExpressionNode.GetThenBranch: IExpressionNode;
+begin
+    Result := FThenBranch;
+end;
+
 { TLambdaExpressionNode }
 
 constructor TLambdaExpressionNode.Create(const AParameters: TArray<IIdentifierNode>; const ABody: IExpressionNode);
@@ -424,6 +474,11 @@ begin
     Result := TAssignmentNode.Create(AIdentifier, AValue);
 end;
 
+class function TAst.AssignResult(const AValue: IExpressionNode): IAssignmentNode;
+begin
+    Result := Assign(TAst.Identifier('Result'), AValue);
+end;
+
 class function TAst.Block(const AExpressions: array of IExpressionNode): IBlockExpressionNode;
 begin
     Result := TBlockExpressionNode.Create(AExpressions);
@@ -454,6 +509,11 @@ begin
     Result := TIfExpressionNode.Create(ACondition, AThenBranch, AElseBranch);
 end;
 
+class function TAst.TernaryExpr(const ACondition, AThenBranch, AElseBranch: IExpressionNode): ITernaryExpressionNode;
+begin
+    Result := TTernaryExpressionNode.Create(ACondition, AThenBranch, AElseBranch);
+end;
+
 class function TAst.LambdaExpr(const AParameters: TArray<IIdentifierNode>; const ABody: IExpressionNode): ILambdaExpressionNode;
 begin
     Result := TLambdaExpressionNode.Create(AParameters, ABody);