CMergeManager.h

Go to the documentation of this file.

#ifndef RECOTOOL_CMERGEMANAGER_H
#define RECOTOOL_CMERGEMANAGER_H

#include <iostream>

#include "CMManagerBase.h"
#include "CMergeBookKeeper.h"
#include "CBoolAlgoBase.h"

namespace cmtool {

  class CMergeManager : public CMManagerBase {

  public:
    
    CMergeManager();
    
    virtual ~CMergeManager(){}

    virtual void Reset();

    void AddMergeAlgo(CBoolAlgoBase* algo) {_merge_algo = algo;}

    void AddSeparateAlgo(CBoolAlgoBase* algo) {_separate_algo = algo;}

    const std::vector<cluster::ClusterParamsAlg>& GetClusters() const { return _out_clusters; }

    const CMergeBookKeeper& GetBookKeeper() const { return _book_keeper; }

  protected:
    
    //
    // FMWK functions override
    //

    virtual void EventBegin();

    virtual void IterationBegin();

    virtual bool IterationProcess();

    virtual void IterationEnd();
    
    virtual void EventEnd();

  protected:

    void RunMerge(const std::vector<cluster::ClusterParamsAlg > &in_clusters,
                  CMergeBookKeeper &book_keeper) const;

    void RunMerge(const std::vector<cluster::ClusterParamsAlg > &in_clusters,
                  const std::vector<bool> &merge_flag,
                  CMergeBookKeeper &book_keeper) const;

    void RunSeparate(const std::vector<cluster::ClusterParamsAlg > &in_clusters,
                     CMergeBookKeeper &book_keeper) const;

  protected:

    std::vector<cluster::ClusterParamsAlg> _out_clusters;

    CMergeBookKeeper _book_keeper;

    ::cmtool::CBoolAlgoBase* _merge_algo;

    ::cmtool::CBoolAlgoBase* _separate_algo;

    size_t _iter_ctr;

    std::vector<CMergeBookKeeper> _book_keeper_v;

    std::vector<std::vector<unsigned short> > _tmp_merged_indexes;

    std::vector<cluster::ClusterParamsAlg> _tmp_merged_clusters;

  };
}

#endif
 // end of doxygen group