CMS 3D CMS Logo

PFEcalBarrelRecHitCreator.h
Go to the documentation of this file.
1 #ifndef RecoParticleFlow_PFClusterProducer_PFEcalBarrelRecHitCreator_h
2 #define RecoParticleFlow_PFClusterProducer_PFEcalBarrelRecHitCreator_h
3 
9 
13 
19 
27 
29 
30  public:
32  PFRecHitCreatorBase(iConfig,iC)
33  {
35  auto srF = iConfig.getParameter<edm::InputTag>("srFlags");
36  if (not srF.label().empty())
38  triggerTowerMap_ = nullptr;
39  }
40 
41  void importRecHits(std::unique_ptr<reco::PFRecHitCollection>&out,std::unique_ptr<reco::PFRecHitCollection>& cleaned ,const edm::Event& iEvent,const edm::EventSetup& iSetup) override {
42 
43  beginEvent(iEvent,iSetup);
44 
46 
48  iSetup.get<CaloGeometryRecord>().get(geoHandle);
49 
50  bool useSrF = false;
51  if (not srFlagToken_.isUninitialized()){
53  useSrF = true;
54  }
55 
56  // get the ecal geometry
57  const CaloSubdetectorGeometry *gTmp =
59 
60  const EcalBarrelGeometry *ecalGeo =dynamic_cast< const EcalBarrelGeometry* > (gTmp);
61 
62  iEvent.getByToken(recHitToken_,recHitHandle);
63  for(const auto& erh : *recHitHandle ) {
64  const DetId& detid = erh.detid();
65  auto energy = erh.energy();
66  auto time = erh.time();
67  bool hi = (useSrF ? isHighInterest(detid) : true);
68 
69  const CaloCellGeometry * thisCell= ecalGeo->getGeometry(detid);
70 
71  // find rechit geometry
72  if(!thisCell) {
73  throw cms::Exception("PFEcalBarrelRecHitCreator")
74  << "detid " << detid.rawId() << "not found in geometry";
75  }
76 
77  out->emplace_back(thisCell, detid.rawId(), PFLayer::ECAL_BARREL, energy);
78 
79  auto & rh = out->back();
80 
81  bool rcleaned = false;
82  bool keep=true;
83 
84  //Apply Q tests
85  for( const auto& qtest : qualityTests_ ) {
86  if (!qtest->test(rh,erh,rcleaned,hi)) {
87  keep = false;
88  }
89  }
90 
91  if(keep) {
92  rh.setTime(time);
93  rh.setDepth(1);
94  }
95  else {
96  if (rcleaned)
97  cleaned->push_back(std::move(out->back()));
98  out->pop_back();
99  }
100  }
101  }
102 
103  void init(const edm::EventSetup &es) override {
104 
106  es.get<IdealGeometryRecord>().get(hTriggerTowerMap);
107  triggerTowerMap_ = hTriggerTowerMap.product();
108 
109  }
110 
111 
112  protected:
113 
114  bool isHighInterest(const EBDetId& detid) {
115  bool result=false;
116  auto srf = srFlagHandle_->find(readOutUnitOf(detid));
117  if(srf==srFlagHandle_->end()) return false;
118  else result = ((srf->value() & ~EcalSrFlag::SRF_FORCED_MASK) == EcalSrFlag::SRF_FULL);
119  return result;
120  }
121 
123  return triggerTowerMap_->towerOf(detid);
124  }
125 
128 
129  // ECAL trigger tower mapping
131  // selective readout flags collection
133 
134 };
135 
136 #endif
edm::EDGetTokenT< EBSrFlagCollection > srFlagToken_
EDGetTokenT< ProductType > consumes(edm::InputTag const &tag)
T getParameter(std::string const &) const
const CaloSubdetectorGeometry * getSubdetectorGeometry(const DetId &id) const
access the subdetector geometry for the given subdetector directly
Definition: CaloGeometry.cc:45
std::vector< std::unique_ptr< PFRecHitQTestBase > > qualityTests_
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:460
void importRecHits(std::unique_ptr< reco::PFRecHitCollection > &out, std::unique_ptr< reco::PFRecHitCollection > &cleaned, const edm::Event &iEvent, const edm::EventSetup &iSetup) override
PFEcalBarrelRecHitCreator(const edm::ParameterSet &iConfig, edm::ConsumesCollector &iC)
edm::EDGetTokenT< EcalRecHitCollection > recHitToken_
EcalTrigTowerDetId towerOf(const DetId &id) const
Get the tower id for this det id (or null if not known)
virtual const CaloCellGeometry * getGeometry(const DetId &id) const
Get the cell geometry of a given detector id. Should return false if not found.
uint32_t rawId() const
get the raw id
Definition: DetId.h:43
const int keep
int iEvent
Definition: GenABIO.cc:230
static const int SRF_FULL
Definition: EcalSrFlag.h:24
void beginEvent(const edm::Event &event, const edm::EventSetup &setup)
void init(const edm::EventSetup &es) override
Definition: DetId.h:18
edm::Handle< EBSrFlagCollection > srFlagHandle_
const T & get() const
Definition: EventSetup.h:55
EcalTrigTowerDetId readOutUnitOf(const EBDetId &detid) const
const EcalTrigTowerConstituentsMap * triggerTowerMap_
bool isHighInterest(const EBDetId &detid)
T const * product() const
Definition: ESHandle.h:86
def move(src, dest)
Definition: eostools.py:510