//===- AsmExpr.h - Assembly file expressions --------------------*- C++ -*-===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//

#ifndef ASMEXPR_H
#define ASMEXPR_H

#include "llvm/Support/Casting.h"
#include "llvm/Support/DataTypes.h"

namespace llvm {
class MCContext;
class MCSymbol;

class AsmExpr {
public:
  enum AsmExprKind {
    Binary,    /// Binary expressions.
    Constant,  /// Constant expressions.
    SymbolRef, /// References to labels and assigned expressions.
    Unary      /// Unary expressions.
  };
  
private:
  AsmExprKind Kind;
  
protected:
  AsmExpr(AsmExprKind _Kind) : Kind(_Kind) {}
  
public:
  virtual ~AsmExpr();

  AsmExprKind getKind() const { return Kind; }

  /// EvaluateAsAbsolute - Try to evaluate the expression to an absolute value.
  ///
  /// @param Res - The absolute value if evaluation succeeds.
  /// @result - True on success.
  bool EvaluateAsAbsolute(MCContext &Ctx, int64_t &Res) const;

  static bool classof(const AsmExpr *) { return true; }
};

class AsmConstantExpr : public AsmExpr {
  int64_t Value;

public:
  AsmConstantExpr(int64_t _Value) 
    : AsmExpr(AsmExpr::Constant), Value(_Value) {}
  
  int64_t getValue() const { return Value; }

  static bool classof(const AsmExpr *E) { 
    return E->getKind() == AsmExpr::Constant; 
  }
  static bool classof(const AsmConstantExpr *) { return true; }
};

class AsmSymbolRefExpr : public AsmExpr {
  MCSymbol *Symbol;

public:
  AsmSymbolRefExpr(MCSymbol *_Symbol) 
    : AsmExpr(AsmExpr::SymbolRef), Symbol(_Symbol) {}
  
  MCSymbol *getSymbol() const { return Symbol; }

  static bool classof(const AsmExpr *E) { 
    return E->getKind() == AsmExpr::SymbolRef; 
  }
  static bool classof(const AsmSymbolRefExpr *) { return true; }
};

class AsmUnaryExpr : public AsmExpr {
public:
  enum Opcode {
    LNot,  /// Logical negation.
    Minus, /// Unary minus.
    Not,   /// Bit-wise negation.
    Plus   /// Unary plus.
  };

private:
  Opcode Op;
  AsmExpr *Expr;

public:
  AsmUnaryExpr(Opcode _Op, AsmExpr *_Expr)
    : AsmExpr(AsmExpr::Unary), Op(_Op), Expr(_Expr) {}
  ~AsmUnaryExpr() {
    delete Expr;
  }

  Opcode getOpcode() const { return Op; }

  AsmExpr *getSubExpr() const { return Expr; }

  static bool classof(const AsmExpr *E) { 
    return E->getKind() == AsmExpr::Unary; 
  }
  static bool classof(const AsmUnaryExpr *) { return true; }
};

class AsmBinaryExpr : public AsmExpr {
public:
  enum Opcode {
    Add,  /// Addition.
    And,  /// Bitwise and.
    Div,  /// Division.
    EQ,   /// Equality comparison.
    GT,   /// Greater than comparison.
    GTE,  /// Greater than or equal comparison.
    LAnd, /// Logical and.
    LOr,  /// Logical or.
    LT,   /// Less than comparison.
    LTE,  /// Less than or equal comparison.
    Mod,  /// Modulus.
    Mul,  /// Multiplication.
    NE,   /// Inequality comparison.
    Or,   /// Bitwise or.
    Shl,  /// Bitwise shift left.
    Shr,  /// Bitwise shift right.
    Sub,  /// Subtraction.
    Xor   /// Bitwise exclusive or.
  };

private:
  Opcode Op;
  AsmExpr *LHS, *RHS;

public:
  AsmBinaryExpr(Opcode _Op, AsmExpr *_LHS, AsmExpr *_RHS)
    : AsmExpr(AsmExpr::Binary), Op(_Op), LHS(_LHS), RHS(_RHS) {}
  ~AsmBinaryExpr() {
    delete LHS;
    delete RHS;
  }

  Opcode getOpcode() const { return Op; }

  AsmExpr *getLHS() const { return LHS; }
  AsmExpr *getRHS() const { return RHS; }

  static bool classof(const AsmExpr *E) { 
    return E->getKind() == AsmExpr::Binary; 
  }
  static bool classof(const AsmBinaryExpr *) { return true; }
};

} // end namespace llvm

#endif