/*
 * Copyright 2004-2019 Cray Inc.
 * Other additional copyright holders may be indicated within.
 *
 * The entirety of this work is licensed under the Apache License,
 * Version 2.0 (the "License"); you may not use this file except
 * in compliance with the License.
 *
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef _IF_EXPR_H_
#define _IF_EXPR_H_

#include "baseAST.h"
#include "expr.h"

#include <ostream>


class IfExpr : public Expr {
public:

  IfExpr(Expr* condition, Expr* thenExpr, Expr* elseExpr);

  Expr*      getCondition();
  BlockStmt* getThenStmt();
  BlockStmt* getElseStmt();

  virtual void verify();

  DECLARE_COPY(IfExpr);

  virtual Expr*         getFirstExpr();
  virtual void          replaceChild(Expr* old_ast, Expr* new_ast);
  virtual QualifiedType qualType();

  virtual void   accept(AstVisitor* visitor);
  virtual GenRet codegen();
  virtual void   prettyPrint(std::ostream* o);

private:
  Expr* condition;
  BlockStmt* thenStmt;
  BlockStmt* elseStmt;
};

#endif
