CMergeBookKeeper.h

Go to the documentation of this file.

#ifndef RECOTOOL_CMERGEBOOKKEEPER_H
#define RECOTOOL_CMERGEBOOKKEEPER_H

#include <iostream>
#include <vector>
#include <TString.h>
#include "CMTException.h"

namespace cmtool {

  class CMergeBookKeeper : public std::vector<unsigned short>{
    
  public:
    
    CMergeBookKeeper(unsigned short nclusters=0);
    
    virtual ~CMergeBookKeeper(){};

    void Reset(unsigned short nclusters=0);

    void ProhibitMerge(unsigned short index1, unsigned short index2);

    bool MergeAllowed(unsigned short index1, unsigned short index2);

    void Merge(unsigned short index1, unsigned short index2);

    std::vector<unsigned short> GetMergedSet(unsigned short index1) const;

    bool IsMerged(unsigned short index1, unsigned short index2) const;

    void PassResult(std::vector<std::vector<unsigned short> > &result) const;


    std::vector<std::vector<unsigned short> > GetResult() const
    { 
      std::vector<std::vector<unsigned short> > result; 
      PassResult(result);
      return result;
    }

    void Combine(const CMergeBookKeeper &another);

    void Report() const;

  protected:

    std::vector<std::vector<bool> > _prohibit_merge;

    size_t _out_cluster_count;
    
  };
  
}

#endif
 // end of doxygen group