UDDUChainsXAIF.cpp

Go to the documentation of this file.

#include <OpenAnalysis/XAIF/UDDUChainsXAIF.hpp>

namespace OA {
  namespace XAIF {

    UDDUChainsXAIF::UDDUChainsXAIF() {
      // initialize the extra map
      mMemRefToIdMap = new std::map<MemRefHandle,int>;
    }
 
    OA_ptr<UDDUChainsXAIF::ChainIterator> 
    UDDUChainsXAIF::getUDChainIterator(MemRefHandle h) {
      OA_ptr<ChainIterator> retval;
      // if haven't seen this MemRef before then just create an empty set
      // and return an iterator over that
      std::map<MemRefHandle,int>::iterator pos = mMemRefToIdMap->find(h);
      if (pos == mMemRefToIdMap->end()) {
        OA_ptr<StmtSet> emptySet;  emptySet = new StmtSet;
        retval = new ChainIterator(emptySet);
        // otherwise return an iterator over the set with which
        // this mem ref is associated
      } else {
        retval = new ChainIterator((*mIdToStmtSetMap)[pos->second]);
      } 
      return retval; 
    }

    OA_ptr<UDDUChainsXAIF::ChainIterator> 
    UDDUChainsXAIF::getDUChainIterator(MemRefHandle h) {
      // DU and UD chains are stored in the same fashion
      return getUDChainIterator(h);
    }

    int UDDUChainsXAIF::getUDDUChainId(MemRefHandle ref) {
      int retval;
      // if have seen this MemRef before then return setId
      std::map<MemRefHandle,int>::iterator pos = mMemRefToIdMap->find(ref);
      if (pos != mMemRefToIdMap->end()) {
        retval = pos->second;
      } else {
        // we don't have a chaing for this memory reference so indicate that
        // it has an empty chain by returning the zeroth chain
        retval = 0;
      }
      return retval;
    }

    OA_ptr<UDDUChainsXAIF::MemRefsWithChainIterator> 
    UDDUChainsXAIF::getMemRefsWithChainIterator() {
      OA_ptr<MemRefsWithChainIterator> retval;
      retval = new MemRefsWithChainIterator(mMemRefToIdMap);
      return retval;
    }
    
    void UDDUChainsXAIF::insertInto(MemRefHandle ref, int setId) {
      (*mMemRefToIdMap)[ref] = setId;
    }

    void UDDUChainsXAIF::dump(std::ostream& os, OA_ptr<IRHandlesIRInterface> ir) {
      ChainsXAIF::dump(os,ir);
      
      // print all memrefs and their mapping
      os << "MemRef mapping to UDDUChains sets:" << std::endl;
      OA_ptr<MemRefsWithChainIterator> memRefIterPtr
        = getMemRefsWithChainIterator();
      for ( ; memRefIterPtr->isValid(); (*memRefIterPtr)++ ) {
        MemRefHandle memref = memRefIterPtr->current();
        int setId = getUDDUChainId(memref);
        os << "(" << memref.hval() << ") " << ir->toString(memref);
        os << " --> " << setId << std::endl;
      }
    }

  } 
}