TTClusterAlgorithm_official.h

Go to the documentation of this file.

#ifndef L1_TRACK_TRIGGER_CLUSTER_ALGO_official_H
#define L1_TRACK_TRIGGER_CLUSTER_ALGO_official_H

#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/ModuleFactory.h"
#include "FWCore/Framework/interface/ESProducer.h"

#include "L1Trigger/TrackTrigger/interface/TTClusterAlgorithm.h"
#include "L1Trigger/TrackTrigger/interface/TTClusterAlgorithmRecord.h"

#include <memory>
#include <string>
#include <map>

template <typename T>
class TTClusterAlgorithm_official : public TTClusterAlgorithm<T> {
private:
  int mWidthCut;  

  static bool CompareClusters(const T& a, const T& b);

public:

  TTClusterAlgorithm_official(int aWidthCut) : TTClusterAlgorithm<T>(__func__) { mWidthCut = aWidthCut; }

  ~TTClusterAlgorithm_official() override {}

  void Cluster(std::vector<std::vector<T> >& output, const std::vector<T>& input, bool isPS) const override;

};  

template <>
bool TTClusterAlgorithm_official<Ref_Phase2TrackerDigi_>::CompareClusters(const Ref_Phase2TrackerDigi_& a,
                                                                          const Ref_Phase2TrackerDigi_& b);

template <>
void TTClusterAlgorithm_official<Ref_Phase2TrackerDigi_>::Cluster(
    std::vector<std::vector<Ref_Phase2TrackerDigi_> >& output,
    const std::vector<Ref_Phase2TrackerDigi_>& input,
    bool isPS) const;

template <typename T>
class ES_TTClusterAlgorithm_official : public edm::ESProducer {
private:
  int mWidthCut;

public:
  ES_TTClusterAlgorithm_official(const edm::ParameterSet& p) : mWidthCut(p.getParameter<int>("WidthCut")) {
    setWhatProduced(this);
  }

  ~ES_TTClusterAlgorithm_official() override {}

  std::unique_ptr<TTClusterAlgorithm<T> > produce(const TTClusterAlgorithmRecord& record) {
    TTClusterAlgorithm<T>* TTClusterAlgo = new TTClusterAlgorithm_official<T>(mWidthCut);

    return std::unique_ptr<TTClusterAlgorithm<T> >(TTClusterAlgo);
  }

};  

#endif