EvalToMemRefVisitor.hpp

Go to the documentation of this file.

#ifndef EvalToMemRefVisitor_H
#define EvalToMemRefVisitor_H

#include <OpenAnalysis/ExprTree/ExprTreeVisitor.hpp>

namespace OA {

//--------------------------------------------------------------------
class EvalToMemRefVisitor : public ExprTreeVisitor {
public:
  EvalToMemRefVisitor() : mIsMemRef(false) {}
  ~EvalToMemRefVisitor() {}

  void visitExprTreeBefore(ExprTree&) { }
  void visitExprTreeAfter(ExprTree&) { }

  //---------------------------------------
  // method for each ExprTree::Node subclass
  //---------------------------------------
  // default base class so that visitors can handle unknown
  // node sub-classes in a generic fashion
  void visitNode(ExprTree::Node&) { mIsMemRef = false; }

  void visitOpNode(ExprTree::OpNode& n) { mIsMemRef = false; }
  void visitCallNode(ExprTree::CallNode& n)  { mIsMemRef = false; }
  void visitMemRefNode(ExprTree::MemRefNode& n) 
    { mIsMemRef = true; mMemRef = n.getHandle(); }
  void visitConstSymNode(ExprTree::ConstSymNode& n) { mIsMemRef = false; }
  void visitConstValNode(ExprTree::ConstValNode& n) { mIsMemRef = false; }

  bool isMemRef() { return mIsMemRef; }
  MemRefHandle getMemRef() { return mMemRef; }

private:
  bool mIsMemRef;
  MemRefHandle mMemRef;
};


} // end of OA namespace

#endif