NotationGenerator.hpp

Go to the documentation of this file.

#ifndef NOTATIONGENERATOR_HPP_
#define NOTATIONGENERATOR_HPP_

#include <OpenAnalysis/Utils/OA_ptr.hpp>
#include <OpenAnalysis/IRInterface/IRHandles.hpp>
#include <OpenAnalysis/Location/Locations.hpp>
#include <OpenAnalysis/MemRefExpr/MemRefExpr.hpp>
#include <OpenAnalysis/MemRefExpr/MemRefExprVisitor.hpp>
#include <OpenAnalysis/IRInterface/AliasIRInterface.hpp>
#include <iostream>

std::ostream &indt(std::ostream &os);

std::ostream &pushIndt(std::ostream &os);

std::ostream &popIndt(std::ostream &os);

class NGOutputVisitor;

class NotationGenerator
{
    friend class NGOutputVisitor;

    public:
    NotationGenerator(OA::OA_ptr<OA::Alias::AliasIRInterface> ir,
                      std::ostream &codeStream);
 
    void generate(OA::ProcHandle proc);

    private:
    void outputProcedure(OA::ProcHandle proc);

    void outputFormals(OA::ProcHandle proc);

    void outputStmt(OA::StmtHandle stmt);
    
    void outputMemRefExprs(OA::StmtHandle stmt);

    void outputPtrAssignPairs(OA::StmtHandle stmt);

    void outputCallSites(OA::StmtHandle stmt);

    void outputParamBindPtrAssignPairs(OA::StmtHandle stmt);

    void outputParamBindPtrAssignPairs(OA::StmtHandle stmt,
         OA::CallHandle call);

    void outputMemRefExpr(OA::OA_ptr<OA::MemRefExpr> expr);
    void outputMemRefExpr(OA::NamedRef& expr);
    void outputMemRefExpr(OA::UnnamedRef& expr);
    void outputMemRefExpr(OA::UnknownRef& expr);
    void outputMemRefExpr(OA::AddressOf& expr);
    void outputMemRefExpr(OA::Deref& expr);
    void outputMemRefExpr(OA::FieldAccess& expr);
    void outputMemRefExpr(OA::IdxAccess& expr);
    void outputMemRefExpr(OA::IdxExprAccess& expr);
    void outputMemRefExpr(OA::SubSetRef& ref);


    void outputHandle(OA::StmtHandle h);
    void outputHandle(OA::ExprHandle h);
    void outputHandle(OA::MemRefHandle h);
    void outputHandle(OA::ProcHandle h);
    void outputHandle(OA::SymHandle h);
    void outputHandle(OA::CallHandle h);

    void outputMemRefType(OA::MemRefExpr &expr);

    void outputTF(bool value);
    
    void outputAccuracy(bool hasFullAccuracy);

    void outputLocations(OA::ProcHandle proc);

    void outputLocation(OA::ProcHandle proc, OA::SymHandle sym);

    //vars
    OA::OA_ptr<OA::Alias::AliasIRInterface> mIR;
    std::ostream &mCodeStream;

    std::set<OA::SymHandle> mSymHandles;
};

class NGOutputVisitor : public virtual OA::MemRefExprVisitor {
  public:
  NGOutputVisitor(NotationGenerator& ng) : mNG(ng) {}
  ~NGOutputVisitor() {}

  void visitNamedRef(OA::NamedRef& ref) { mNG.outputMemRefExpr(ref); }
  void visitUnnamedRef(OA::UnnamedRef& ref){ mNG.outputMemRefExpr(ref); }
  void visitUnknownRef(OA::UnknownRef& ref){ mNG.outputMemRefExpr(ref); }
  void visitAddressOf(OA::AddressOf& ref){ mNG.outputMemRefExpr(ref); }
  void visitDeref(OA::Deref& ref){ mNG.outputMemRefExpr(ref); }

  void visitSubSetRef(OA::SubSetRef& ref) 
    { 
        /* PLM 1/23/07 there is SubSetRef in the ouptut
        std::cerr << __FILE__ << ":" << __LINE__
             << "Unknown SubSetRef type.\n";
        exit(-1);
        */

        mNG.outputMemRefExpr(ref);
    }

  void visitFieldAccess(OA::FieldAccess& ref) { mNG.outputMemRefExpr(ref); }
  void visitIdxExprAccess(OA::IdxExprAccess& ref) { mNG.outputMemRefExpr(ref); }
  void visitIdxAccess(OA::IdxAccess& ref) { mNG.outputMemRefExpr(ref); }

  private:
  NotationGenerator& mNG;
};


#endif