IslandClusterProducer.h

Go to the documentation of this file.

#ifndef RecoEcal_EgammaClusterProducers_IslandClusterProducer_h_
#define RecoEcal_EgammaClusterProducers_IslandClusterProducer_h_

#include <memory>
#include <time.h>

#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/EcalRecHit/interface/EcalRecHitCollections.h"
#include "RecoEcal/EgammaClusterAlgos/interface/IslandClusterAlgo.h"
#include "RecoEcal/EgammaCoreTools/interface/PositionCalc.h"
#include "RecoEcal/EgammaCoreTools/interface/ClusterShapeAlgo.h"
#include "Geometry/CaloTopology/interface/CaloSubdetectorTopology.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"

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

//


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

      IslandClusterProducer(const edm::ParameterSet& ps);

      ~IslandClusterProducer();

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

   private:

      int nMaxPrintout_; // max # of printouts
      int nEvt_;         // internal counter of events

      IslandClusterAlgo::VerbosityLevel verbosity;


      edm::EDGetTokenT<EcalRecHitCollection> barrelRecHits_;
      edm::EDGetTokenT<EcalRecHitCollection> endcapRecHits_;

      std::string barrelClusterCollection_;
      std::string endcapClusterCollection_;

      std::string clustershapecollectionEB_;
      std::string clustershapecollectionEE_;

      //BasicClusterShape AssociationMap
      std::string barrelClusterShapeAssociation_;
      std::string endcapClusterShapeAssociation_; 

      PositionCalc posCalculator_; // position calculation algorithm
      ClusterShapeAlgo shapeAlgo_; // cluster shape algorithm
      IslandClusterAlgo * island_p;

      bool counterExceeded() const { return ((nEvt_ > nMaxPrintout_) || (nMaxPrintout_ < 0)); }

      const EcalRecHitCollection * getCollection(edm::Event& evt,
                                   const edm::EDGetTokenT<EcalRecHitCollection>& token);


      void clusterizeECALPart(edm::Event &evt, const edm::EventSetup &es,
                              const edm::EDGetTokenT<EcalRecHitCollection>& token,
                              const std::string& clusterCollection,
                  const std::string& clusterShapeAssociation,
                              const IslandClusterAlgo::EcalPart& ecalPart);

      void outputValidationInfo(reco::CaloClusterPtrVector &clusterPtrVector);
};


#endif