Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #27175 from nancymarinelli/deadChannelRecoveryBDT_V1
Ecal DPG - Deploy new software (based on BDT) to recover isolated dead channels…
- Loading branch information
Showing
10 changed files
with
268 additions
and
26 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
52 changes: 52 additions & 0 deletions
52
RecoLocalCalo/EcalDeadChannelRecoveryAlgos/interface/EcalDeadChannelRecoveryBDTG.h
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,52 @@ | ||
| #ifndef RecoLocalCalo_EcalDeadChannelRecoveryAlgos_EcalDeadChannelRecoveryBDTG_H | ||
| #define RecoLocalCalo_EcalDeadChannelRecoveryAlgos_EcalDeadChannelRecoveryBDTG_H | ||
|
|
||
| #include "DataFormats/EcalRecHit/interface/EcalRecHit.h" | ||
| #include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h" | ||
|
|
||
| #include "DataFormats/EcalDetId/interface/EBDetId.h" | ||
| #include "DataFormats/EcalDetId/interface/EEDetId.h" | ||
|
|
||
| #include "Geometry/CaloTopology/interface/CaloTopology.h" | ||
| #include "Geometry/CaloTopology/interface/CaloSubdetectorTopology.h" | ||
|
|
||
| #include "CommonTools/MVAUtils/interface/TMVAZipReader.h" | ||
| #include "FWCore/ParameterSet/interface/ParameterSet.h" | ||
|
|
||
| #include "TMVA/Reader.h" | ||
|
|
||
| #include <string> | ||
| #include <memory> | ||
|
|
||
| template <typename DetIdT> | ||
| class EcalDeadChannelRecoveryBDTG { | ||
| public: | ||
| EcalDeadChannelRecoveryBDTG(); | ||
| ~EcalDeadChannelRecoveryBDTG(); | ||
|
|
||
| void setParameters(const edm::ParameterSet &ps); | ||
| void setCaloTopology(const CaloTopology *topo) { topology_ = topo; } | ||
|
|
||
| double recover( | ||
| const DetIdT id, const EcalRecHitCollection &hit_collection, double single8Cut, double sum8Cut, bool *acceptFlag); | ||
|
|
||
| void loadFile(); | ||
| void addVariables(TMVA::Reader *reader); | ||
|
|
||
| private: | ||
| const CaloTopology *topology_; | ||
| struct XtalMatrix { | ||
| std::array<float, 9> rEn, ieta, iphi; | ||
| float sumE8; | ||
| }; | ||
|
|
||
| XtalMatrix mx_; | ||
|
|
||
| edm::FileInPath bdtWeightFileNoCracks_; | ||
| edm::FileInPath bdtWeightFileCracks_; | ||
|
|
||
| std::unique_ptr<TMVA::Reader> readerNoCrack; | ||
| std::unique_ptr<TMVA::Reader> readerCrack; | ||
| }; | ||
|
|
||
| #endif |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
149 changes: 149 additions & 0 deletions
149
RecoLocalCalo/EcalDeadChannelRecoveryAlgos/src/EcalDeadChannelRecoveryBDTG.cc
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,149 @@ | ||
| // Original Authors: S. Taroni, N. Marinelli | ||
| // University of Notre Dame - US | ||
| // Created: | ||
| // | ||
| // | ||
| // | ||
|
|
||
| #include "RecoLocalCalo/EcalDeadChannelRecoveryAlgos/interface/EcalDeadChannelRecoveryBDTG.h" | ||
| #include "RecoEcal/EgammaCoreTools/interface/EcalClusterTools.h" // can I use a egammatools here? | ||
| #include "FWCore/ParameterSet/interface/FileInPath.h" | ||
| #include <iostream> | ||
|
|
||
| #include <iostream> | ||
| #include <ostream> | ||
| #include <string> | ||
| #include <fstream> | ||
|
|
||
| template <> | ||
| void EcalDeadChannelRecoveryBDTG<EBDetId>::addVariables(TMVA::Reader *reader) { | ||
| for (int i = 0; i < 9; ++i) { | ||
| if (i == 4) | ||
| continue; | ||
| reader->AddVariable("E" + std::to_string(i + 1) + "/(E1+E2+E3+E4+E6+E7+E8+E9)", &(mx_.rEn[i])); | ||
| } | ||
| reader->AddVariable("E1+E2+E3+E4+E6+E7+E8+E9", &(mx_.sumE8)); | ||
| for (int i = 0; i < 9; ++i) | ||
| reader->AddVariable("iEta" + std::to_string(i + 1), &(mx_.ieta[i])); | ||
| for (int i = 0; i < 9; ++i) | ||
| reader->AddVariable("iPhi" + std::to_string(i + 1), &(mx_.iphi[i])); | ||
| } | ||
| template <> | ||
| void EcalDeadChannelRecoveryBDTG<EBDetId>::loadFile() { | ||
| readerNoCrack = std::unique_ptr<TMVA::Reader>(new TMVA::Reader("!Color:!Silent")); | ||
| readerCrack = std::unique_ptr<TMVA::Reader>(new TMVA::Reader("!Color:!Silent")); | ||
|
|
||
| addVariables(readerNoCrack.get()); | ||
| addVariables(readerCrack.get()); | ||
|
|
||
| reco::details::loadTMVAWeights(readerNoCrack.get(), "BDTG", bdtWeightFileNoCracks_.fullPath()); | ||
| reco::details::loadTMVAWeights(readerCrack.get(), "BDTG", bdtWeightFileCracks_.fullPath()); | ||
| } | ||
|
|
||
| template <typename T> | ||
| EcalDeadChannelRecoveryBDTG<T>::EcalDeadChannelRecoveryBDTG() {} | ||
|
|
||
| template <typename T> | ||
| EcalDeadChannelRecoveryBDTG<T>::~EcalDeadChannelRecoveryBDTG() {} | ||
|
|
||
| template <> | ||
| void EcalDeadChannelRecoveryBDTG<EBDetId>::setParameters(const edm::ParameterSet &ps) { | ||
| bdtWeightFileNoCracks_ = ps.getParameter<edm::FileInPath>("bdtWeightFileNoCracks"); | ||
| bdtWeightFileCracks_ = ps.getParameter<edm::FileInPath>("bdtWeightFileCracks"); | ||
|
|
||
| loadFile(); | ||
| } | ||
|
|
||
| template <> | ||
| void EcalDeadChannelRecoveryBDTG<EEDetId>::setParameters(const edm::ParameterSet &ps) {} | ||
|
|
||
| template <> | ||
| double EcalDeadChannelRecoveryBDTG<EEDetId>::recover( | ||
| const EEDetId id, const EcalRecHitCollection &hit_collection, double single8Cut, double sum8Cut, bool *acceptFlag) { | ||
| return 0; | ||
| } | ||
|
|
||
| template <> | ||
| double EcalDeadChannelRecoveryBDTG<EBDetId>::recover( | ||
| const EBDetId id, const EcalRecHitCollection &hit_collection, double single8Cut, double sum8Cut, bool *acceptFlag) { | ||
| bool isCrack = false; | ||
| int cellIndex = 0.; | ||
| double neighTotEn = 0.; | ||
| float val = 0.; | ||
|
|
||
| //find the matrix around id | ||
| std::vector<DetId> m3x3aroundDC = EcalClusterTools::matrixDetId(topology_, id, 1); | ||
| if (m3x3aroundDC.size() < 9) { | ||
| *acceptFlag = false; | ||
| return 0; | ||
| } | ||
|
|
||
| // Loop over all cells in the vector "NxNaroundDC", and for each cell find it's energy | ||
| // (from the EcalRecHits collection). | ||
| for (auto const &theCells : m3x3aroundDC) { | ||
| EBDetId cell = EBDetId(theCells); | ||
| if (cell == id) { | ||
| int iEtaCentral = std::abs(cell.ieta()); | ||
| int iPhiCentral = cell.iphi(); | ||
|
|
||
| if (iEtaCentral < 2 || std::abs(iEtaCentral - 25) < 2 || std::abs(iEtaCentral - 45) < 2 || | ||
| std::abs(iEtaCentral - 65) < 2 || iEtaCentral > 83 || (int(iPhiCentral + 0.5) % 20 == 0)) | ||
| isCrack = true; | ||
| } | ||
| if (!cell.null()) { | ||
| EcalRecHitCollection::const_iterator goS_it = hit_collection.find(cell); | ||
| if (goS_it != hit_collection.end() && cell != id) { | ||
| if (goS_it->energy() < single8Cut) { | ||
| *acceptFlag = false; | ||
| return 0.; | ||
| } else { | ||
| neighTotEn += goS_it->energy(); | ||
| mx_.rEn[cellIndex] = goS_it->energy(); | ||
| mx_.iphi[cellIndex] = cell.iphi(); | ||
| mx_.ieta[cellIndex] = cell.ieta(); | ||
| cellIndex++; | ||
| } | ||
| } else if (cell == id) { // the cell is the central one | ||
| mx_.rEn[cellIndex] = 0; | ||
| cellIndex++; | ||
| } else { //goS_it is not in the rechitcollection | ||
| *acceptFlag = false; | ||
| return 0.; | ||
| } | ||
| } else { //cell is null | ||
| *acceptFlag = false; | ||
| return 0.; | ||
| } | ||
| } | ||
| if (cellIndex > 0 && neighTotEn >= single8Cut * 8. && neighTotEn >= sum8Cut) { | ||
| bool allneighs = true; | ||
| mx_.sumE8 = neighTotEn; | ||
| for (unsigned int icell = 0; icell < 9; icell++) { | ||
| if (mx_.rEn[icell] < single8Cut && icell != 4) { | ||
| allneighs = false; | ||
| } | ||
| mx_.rEn[icell] = mx_.rEn[icell] / neighTotEn; | ||
| } | ||
| if (allneighs == true) { | ||
| // evaluate the regression | ||
| if (isCrack) { | ||
| val = exp((readerCrack->EvaluateRegression("BDTG"))[0]); | ||
| } else { | ||
| val = exp((readerNoCrack->EvaluateRegression("BDTG"))[0]); | ||
| } | ||
|
|
||
| *acceptFlag = true; | ||
| //return the estimated energy | ||
| return val; | ||
| } else { | ||
| *acceptFlag = false; | ||
| return 0; | ||
| } | ||
| } else { | ||
| *acceptFlag = false; | ||
| return 0; | ||
| } | ||
| } | ||
|
|
||
| template class EcalDeadChannelRecoveryBDTG<EBDetId>; | ||
| template class EcalDeadChannelRecoveryBDTG<EEDetId>; //not used. |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.