CPAlgoArray.h

Go to the documentation of this file.

#ifndef RECOTOOL_CPALGOARRAY_H
#define RECOTOOL_CPALGOARRAY_H

#include "larreco/RecoAlg/CMTool/CMToolBase/CPriorityAlgoBase.h"

namespace cmtool {
  class CPAlgoArray : public CPriorityAlgoBase {

  public:
    enum EvaluationMode_t {
      kPositiveAddition,
      kSimpleAddition,
      kMultiplication,
      kLastAlgo,
    };

    CPAlgoArray();

    virtual ~CPAlgoArray(){};

    void AddAlgo(CPriorityAlgoBase* algo) { _algo_array.push_back(algo); }

    void SetMode(EvaluationMode_t mode) { _mode = mode; }

    virtual float Priority(const ::cluster::ClusterParamsAlg& cluster);

    virtual void Report()
    {
      for (auto const& algo : _algo_array)
        algo->Report();
    }

    virtual void Reset()
    {
      for (auto const& algo : _algo_array)
        algo->Reset();
    }

    virtual void EventBegin(const std::vector<cluster::ClusterParamsAlg>& clusters)
    {
      for (auto const& algo : _algo_array)
        algo->EventBegin(clusters);
    }

    virtual void EventEnd()
    {
      for (auto const& algo : _algo_array)
        algo->EventEnd();
    }

    virtual void IterationBegin(const std::vector<cluster::ClusterParamsAlg>& clusters)
    {
      for (auto const& algo : _algo_array)
        algo->IterationBegin(clusters);
    }

    virtual void IterationEnd()
    {
      for (auto const& algo : _algo_array)
        algo->IterationEnd();
    }

  protected:
    std::vector<CPriorityAlgoBase*> _algo_array;

    EvaluationMode_t _mode;
  };
}
#endif
 // end of doxygen group