IslandClusterProducer.h

Go to the documentation of this file.

#ifndef RecoEcal_EgammaClusterProducers_IslandClusterProducer_h_
#define RecoEcal_EgammaClusterProducers_IslandClusterProducer_h_

#include <memory>
#include <ctime>

#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() override;
  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
  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