ASTIfStmt.h

Go to the documentation of this file.

#pragma once
 
#include "ASTExpr.h"
#include "ASTStmt.h"
 
class ASTIfStmt : public ASTStmt {
  std::shared_ptr<ASTExpr> COND;
  std::shared_ptr<ASTStmt> THEN, ELSE;
 
public:
  std::vector<std::shared_ptr<ASTNode>> getChildren() override;
  ASTIfStmt(std::shared_ptr<ASTExpr> COND, std::shared_ptr<ASTStmt> THEN,
            std::shared_ptr<ASTStmt> ELSE)
      : COND(COND), THEN(THEN), ELSE(ELSE) {}
  ASTExpr *getCondition() const { return COND.get(); }
  ASTStmt *getThen() const { return THEN.get(); }
 
  ASTStmt *getElse() const { return ELSE.get(); }
  void accept(ASTVisitor *visitor) override;
  llvm::Value *codegen() override;
 
protected:
  std::ostream &print(std::ostream &out) const override;
};