HiSuperClusterProducer.h

Go to the documentation of this file.

#ifndef RecoHi_HiEgammaAlgos_HiSuperClusterProducer_h_
#define RecoHi_HiEgammaAlgos_HiSuperClusterProducer_h_

#include <memory>

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/EgammaReco/interface/BasicClusterFwd.h"

#include "RecoHI/HiEgammaAlgos/interface/HiBremRecoveryClusterAlgo.h"

//


class HiSuperClusterProducer : public edm::stream::EDProducer<> 
{
  
  public:

      HiSuperClusterProducer(const edm::ParameterSet& ps);

      ~HiSuperClusterProducer() override;

      void produce(edm::Event&, const edm::EventSetup&) override;
      virtual void endJob();

   private:

      int nMaxPrintout_; // max # of printouts
      int nEvt_;         // internal counter of events
 
      HiBremRecoveryClusterAlgo::VerbosityLevel verbosity;

      std::string endcapSuperclusterCollection_;
      std::string barrelSuperclusterCollection_;

      edm::EDGetTokenT<reco::BasicClusterCollection>  eeClustersToken_;
      edm::EDGetTokenT<reco::BasicClusterCollection>  ebClustersToken_;

      float barrelEtaSearchRoad_;
      float barrelPhiSearchRoad_;
      float endcapEtaSearchRoad_; 
      float endcapPhiSearchRoad_;
      float seedTransverseEnergyThreshold_;
      float barrelBCEnergyThreshold_;
      float endcapBCEnergyThreshold_;

      bool doBarrel_;
      bool doEndcaps_;

      HiBremRecoveryClusterAlgo * bremAlgo_p;

      double totalE;
      int noSuperClusters;

      
      void getClusterPtrVector(edm::Event& evt, const edm::EDGetTokenT<reco::BasicClusterCollection>& clustersToken, reco::CaloClusterPtrVector *);
  
      void produceSuperclustersForECALPart(edm::Event& evt, 
                       const edm::EDGetTokenT<reco::BasicClusterCollection>& clustersToken,
                       std::string superclusterColection);

      void outputValidationInfo(reco::SuperClusterCollection &superclusterCollection);
    
      bool counterExceeded() const { return ((nEvt_ > nMaxPrintout_) || (nMaxPrintout_ < 0));}
};


#endif