text: /**
   * Copyright (c) 2012 Stephen Tu
   * All rights reserved.
   *
   * Redistribution and use in source and binary forms, with or without
   * modification, are permitted provided that the following conditions are met:
   *
   * 1. Redistributions of source code must retain the above copyright notice,
   * this list of conditions and the following disclaimer.
   * 2. Redistributions in binary form must reproduce the above copyright notice,
   * this list of conditions and the following disclaimer in the documentation
   * and/or other materials provided with the distribution.
   * 3. Neither the name of the author nor the names
   * of its contributors may be used to endorse or promote products derived
   * from this software without specific prior written permission.
   *
   * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
   * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
   * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
   * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
   * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
   * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
   * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
   * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
   * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
   * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
   * POSSIBILITY OF SUCH DAMAGE.
   */
  
  #ifndef VENOM_AST_BINOP_H
  #define VENOM_AST_BINOP_H
  
  #include <string>
  
  #include <ast/expression/node.h>
  #include <util/macros.h>
  
  namespace venom {
  namespace ast {
  
  class BinopNode : public ASTExpressionNode {
  public:
  
    enum Type {
      /* numeric ops */
      ADD,
      SUB,
      MULT,
      DIV,
      MOD,
  
      /* comparison */
      CMP_AND,
      CMP_OR,
      CMP_LT,
      CMP_LE,
      CMP_GT,
      CMP_GE,
      CMP_EQ,
      CMP_NEQ,
  
      /* bit ops */
      BIT_AND,
      BIT_OR,
      BIT_XOR,
      BIT_LSHIFT,
      BIT_RSHIFT,
    };
  
    static std::string StringifyType(Type type);
  
    BinopNode(ASTExpressionNode* left, ASTExpressionNode* right, Type type)
      : left(left), right(right), type(type) {}
  
    ~BinopNode() {
      delete left;
      delete right;
    }
  
    bool isShortCircuitOp() const;
  
    virtual size_t getNumKids() const { return 2; }
  
    virtual ASTNode* getNthKid(size_t kid) {
      ASTNode *kids[] = {left, right};
      VENOM_SAFE_RETURN(kids, kid);
    }
  
    virtual void setNthKid(size_t idx, ASTNode* kid) {
      VENOM_CHECK_RANGE(idx, 2);
      VENOM_SAFE_SET2(left, right, kid, idx);
    }
  
    virtual bool needsNewScope(size_t k) const {
      VENOM_CHECK_RANGE(k, 2);
      return false;
    }
  
    virtual ASTNode* rewriteLocal(analysis::SemanticContext* ctx,
                                  RewriteMode mode);
  
    VENOM_AST_TYPED_CLONE_WITH_IMPL_DECL_EXPR(BinopNode)
  
  protected:
    virtual analysis::InstantiatedType*
      typeCheckImpl(analysis::SemanticContext* ctx,
                    analysis::InstantiatedType* expected,
                    const analysis::InstantiatedTypeVec& typeParamArgs);
  
  public:
    virtual void codeGen(backend::CodeGenerator& cg);
  
    virtual void print(std::ostream& o, size_t indent = 0) {
      o << "(binop " << StringifyType(type) << " ";
      left->print(o, indent);
      o << " ";
      right->print(o, indent);
      o << ")";
    }
  
  private:
    ASTExpressionNode* left;
    ASTExpressionNode* right;
    Type type;
  };
  
  }
  }
  
  #endif /* VENOM_AST_BINOP_H */
  