UDDUChainsXAIF.hpp

Go to the documentation of this file.

#ifndef UDDUChainsXAIF_hpp
#define UDDUChainsXAIF_hpp

#include <cassert>
#include <iostream>
#include <map>
#include <set>
#include <vector>
#include "OpenAnalysis/Utils/OA_ptr.hpp"
#include "OpenAnalysis/Utils/SetIterator.hpp"
#include "OpenAnalysis/IRInterface/IRHandles.hpp"
#include "OpenAnalysis/UDDUChains/Interface.hpp"
#include "OpenAnalysis/XAIF/ChainsXAIF.hpp"

namespace OA {

  namespace XAIF {


    class UDDUChainsXAIF : public ChainsXAIF {

    public:

      UDDUChainsXAIF();

      ~UDDUChainsXAIF() {}

      class MemRefsWithChainIterator; 

      OA_ptr<ChainIterator> getUDChainIterator(MemRefHandle h);

      OA_ptr<ChainIterator> getDUChainIterator(MemRefHandle h);

      OA_ptr<MemRefsWithChainIterator> getMemRefsWithChainIterator();
      
      int getUDDUChainId(MemRefHandle ref);
      
      void insertInto(MemRefHandle ref, int Id);

      void dump(std::ostream& os, OA_ptr<IRHandlesIRInterface> ir);

    private:

      OA_ptr<std::map<MemRefHandle,int> > mMemRefToIdMap;

    public:

      class MemRefsWithChainIterator {
      public:
        MemRefsWithChainIterator(OA_ptr<std::map<MemRefHandle,int> > pMap) 
          : mMap(pMap), mIter(pMap->begin()) {}
        ~MemRefsWithChainIterator() {}
        
        MemRefHandle current() { return mIter->first; }
        bool isValid() { return (mIter != mMap->end()); }
        void operator++() { if (isValid()) mIter++; }
        void operator++(int) { ++*this; }
        void reset() { mIter = mMap->begin(); }

      private:
        OA_ptr<std::map<MemRefHandle,int> > mMap;
        std::map<MemRefHandle,int>::iterator mIter;
      };

    };

  } 
} 

#endif