CMS 3D CMS Logo

EleIsoDetIdCollectionProducer.cc
Go to the documentation of this file.
2 
7 
9 
19 
22 
24 
27 
30 
32  : recHitsToken_(consumes<EcalRecHitCollection>(iConfig.getParameter<edm::InputTag>("recHitsLabel"))),
33  emObjectToken_(consumes<reco::GsfElectronCollection>(iConfig.getParameter<edm::InputTag>("emObjectLabel"))),
34  recHitsLabel_(iConfig.getParameter<edm::InputTag>("recHitsLabel")),
35  emObjectLabel_(iConfig.getParameter<edm::InputTag>("emObjectLabel")),
36  energyCut_(iConfig.getParameter<double>("energyCut")),
37  etCut_(iConfig.getParameter<double>("etCut")),
38  etCandCut_(iConfig.getParameter<double>("etCandCut")),
39  outerRadius_(iConfig.getParameter<double>("outerRadius")),
40  innerRadius_(iConfig.getParameter<double>("innerRadius")),
41  interestingDetIdCollection_(iConfig.getParameter<std::string>("interestingDetIdCollection")) {
42  const std::vector<std::string> flagnamesEB =
43  iConfig.getParameter<std::vector<std::string> >("RecHitFlagToBeExcludedEB");
44 
45  const std::vector<std::string> flagnamesEE =
46  iConfig.getParameter<std::vector<std::string> >("RecHitFlagToBeExcludedEE");
47 
48  flagsexclEB_ = StringToEnumValue<EcalRecHit::Flags>(flagnamesEB);
49 
50  flagsexclEE_ = StringToEnumValue<EcalRecHit::Flags>(flagnamesEE);
51 
52  const std::vector<std::string> severitynamesEB =
53  iConfig.getParameter<std::vector<std::string> >("RecHitSeverityToBeExcludedEB");
54 
55  severitiesexclEB_ = StringToEnumValue<EcalSeverityLevel::SeverityLevel>(severitynamesEB);
56 
57  const std::vector<std::string> severitynamesEE =
58  iConfig.getParameter<std::vector<std::string> >("RecHitSeverityToBeExcludedEE");
59 
60  severitiesexclEE_ = StringToEnumValue<EcalSeverityLevel::SeverityLevel>(severitynamesEE);
61 
62  //register your products
63  produces<DetIdCollection>(interestingDetIdCollection_);
64 }
65 
67 
69 
70 // ------------ method called to produce the data ------------
72  using namespace edm;
73  using namespace std;
74 
75  //Get EM Object
77  iEvent.getByToken(emObjectToken_, emObjectH);
78 
79  // take EcalRecHits
81  iEvent.getByToken(recHitsToken_, recHitsH);
82 
84  iSetup.get<CaloGeometryRecord>().get(pG);
85  const CaloGeometry* caloGeom = pG.product();
86 
87  //Get the channel status from the db
88  //edm::ESHandle<EcalChannelStatus> chStatus;
89  //iSetup.get<EcalChannelStatusRcd>().get(chStatus);
90 
92  iSetup.get<EcalSeverityLevelAlgoRcd>().get(sevlv);
93  const EcalSeverityLevelAlgo* sevLevel = sevlv.product();
94 
95  CaloDualConeSelector<EcalRecHit>* doubleConeSel_ = nullptr;
96  if (recHitsLabel_.instance() == "EcalRecHitsEB")
98  else if (recHitsLabel_.instance() == "EcalRecHitsEE")
100 
101  //Create empty output collections
102  auto detIdCollection = std::make_unique<DetIdCollection>();
103 
104  reco::GsfElectronCollection::const_iterator emObj;
105  if (doubleConeSel_) { //if cone selector was created
106  for (emObj = emObjectH->begin(); emObj != emObjectH->end(); emObj++) { //Loop over candidates
107 
108  if (emObj->et() < etCandCut_)
109  continue; //don't calculate if object hasn't enough energy
110 
111  GlobalPoint pclu(emObj->caloPosition().x(), emObj->caloPosition().y(), emObj->caloPosition().z());
112  doubleConeSel_->selectCallback(pclu, *recHitsH, [&](const EcalRecHit& recIt) {
113  if (recIt.energy() < energyCut_)
114  return; //dont fill if below E noise value
115 
116  double et =
117  recIt.energy() * caloGeom->getPosition(recIt.detid()).perp() / caloGeom->getPosition(recIt.detid()).mag();
118 
119  bool isBarrel = false;
120  if (fabs(caloGeom->getPosition(recIt.detid()).eta() < 1.479))
121  isBarrel = true;
122 
123  if (et < etCut_)
124  return; //dont fill if below ET noise value
125 
126  std::vector<int>::const_iterator sit;
127  int severityFlag = sevLevel->severityLevel(recIt.detid(), *recHitsH);
128  if (isBarrel) {
129  sit = std::find(severitiesexclEB_.begin(), severitiesexclEB_.end(), severityFlag);
130  if (sit != severitiesexclEB_.end())
131  return;
132  } else {
133  sit = std::find(severitiesexclEE_.begin(), severitiesexclEE_.end(), severityFlag);
134  if (sit != severitiesexclEE_.end())
135  return;
136  }
137 
138  if (isBarrel) {
139  // new rechit flag checks
140  if (!(recIt.checkFlag(EcalRecHit::kGood))) {
141  if (recIt.checkFlags(flagsexclEB_)) {
142  return;
143  }
144  }
145  } else {
146  // new rechit flag checks
147  if (!(recIt.checkFlag(EcalRecHit::kGood))) {
148  if (recIt.checkFlags(flagsexclEE_)) {
149  return;
150  }
151  }
152  }
153 
154  if (std::find(detIdCollection->begin(), detIdCollection->end(), recIt.detid()) == detIdCollection->end())
155  detIdCollection->push_back(recIt.detid());
156  }); //end rechits
157 
158  } //end candidates
159 
160  delete doubleConeSel_;
161  } //end if cone selector was created
162 
163  iEvent.put(std::move(detIdCollection), interestingDetIdCollection_);
164 }
T getParameter(std::string const &) const
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:131
EcalSeverityLevel::SeverityLevel severityLevel(const DetId &id) const
Evaluate status from id use channelStatus from DB.
bool checkFlags(const std::vector< int > &flagsvec) const
check if one of the flags in a set is true
Definition: EcalRecHit.h:190
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:525
T mag() const
The vector magnitude. Equivalent to sqrt(vec.mag2())
const DetId & detid() const
Definition: EcalRecHit.h:72
EleIsoDetIdCollectionProducer(const edm::ParameterSet &)
ctor
void find(edm::Handle< EcalRecHitCollection > &hits, DetId thisDet, std::vector< EcalRecHitCollection::const_iterator > &hit, bool debug=false)
Definition: FindCaloHit.cc:19
std::vector< GsfElectron > GsfElectronCollection
collection of GsfElectron objects
int iEvent
Definition: GenABIO.cc:224
bool checkFlag(int flag) const
check if the flag is true
Definition: EcalRecHit.h:187
void produce(edm::Event &, const edm::EventSetup &) override
producer
GlobalPoint getPosition(const DetId &id) const
Get the position of a given detector id.
Definition: CaloGeometry.cc:50
float energy() const
Definition: EcalRecHit.h:68
void selectCallback(double eta, double phi, const edm::SortedCollection< T > &inputCollection, std::function< void(const T &)> callback)
T perp() const
Magnitude of transverse component.
fixed size matrix
HLT enums.
T get() const
Definition: EventSetup.h:73
T const * product() const
Definition: ESHandle.h:86
std::string const & instance() const
Definition: InputTag.h:37
def move(src, dest)
Definition: eostools.py:511