ChainsXAIF.hpp

Go to the documentation of this file.

#ifndef ChainsXAIF_hpp
#define ChainsXAIF_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>

namespace OA {

  namespace XAIF {

    typedef std::set<StmtHandle> StmtSet;

    class ChainsXAIF {
    public:

      ChainsXAIF();
 
     ~ChainsXAIF() {}

      static const int CHAIN_ID_NONE = -1; 

      class ChainIterator;

      class ChainsIterator;

      OA_ptr<ChainsIterator> getChainsIterator();
    
      OA_ptr<ChainIterator> getChainIterator(int id);

      int makeEmptyChain();
 
      void addStmt(StmtHandle stmt, int id);

      void addStmtSet(StmtSet& stmtSet, int id);

      int findChain(StmtSet& stmtSet);

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

    protected:

      OA_ptr<std::map<int,OA_ptr<StmtSet> > > mIdToStmtSetMap;
      
    public:

      class ChainIterator : public SetIterator<StmtHandle>
      {
      public:
        ChainIterator(OA_ptr<std::set<StmtHandle> > pSet) 
            : SetIterator<StmtHandle>(pSet) {}
        ~ChainIterator() {}
      }; 

      class ChainsIterator {
      public:
        ChainsIterator(OA_ptr<std::map<int,OA_ptr<StmtSet> > > pChainMap) 
          : mChainMap(pChainMap) { reset(); }
        ~ChainsIterator() { }

        OA_ptr<ChainIterator> currentChainIterator()
          { OA_ptr<ChainIterator> retval;
            retval = new ChainIterator((*mChainMap)[mIter->first]);
            return retval;
          }
        
        int currentId() { return mIter->first; }
        bool isValid() { return (mIter != mChainMap->end()); }
        
        void operator++() { if (isValid()) mIter++; }
        void operator++(int) { ++*this; }
        
        void reset() { mIter = mChainMap->begin(); }

      private:
        OA_ptr<std::map<int,OA_ptr<StmtSet> > > mChainMap;
        std::map<int,OA_ptr<StmtSet> >::iterator mIter;
      };

    };

  } 
}


#endif