Calibration/Tools/src/CalibElectron.cc - master - githubmirror/cmssw

Calibration/Tools/src/CalibElectron.cc
144 строки5 KB
Ivan Razumov
[LLVM Analyzer][AlCa] Cleanup clang analyzer warnings
10 окт 2024, 23:17
10 окт 2024, 23:17938fd49
#include "Calibration/Tools/interface/CalibElectron.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHit.h"
#include "DataFormats/EcalDetId/interface/EBDetId.h"
#include "DataFormats/EcalDetId/interface/EcalSubdetector.h"
#include "DataFormats/EgammaCandidates/interface/Electron.h"
#include "DataFormats/EgammaReco/interface/SuperCluster.h"
#include "Calibration/Tools/interface/EcalRingCalibrationTools.h"
#include "Calibration/Tools/interface/EcalIndexingTools.h"
#include <iostream>

using namespace calib;
using namespace std;

CalibElectron::CalibElectron() : theElectron_(nullptr), theHits_(nullptr), theEEHits_(nullptr) {}

std::vector<std::pair<int, float> > CalibElectron::getCalibModulesWeights(TString calibtype) {
  std::vector<std::pair<int, float> > theWeights;

  if (calibtype == "RING") {
    float w_ring[EcalRingCalibrationTools::N_RING_TOTAL];

    for (int i = 0; i < EcalRingCalibrationTools::N_RING_TOTAL; ++i)
      w_ring[i] = 0.;

    std::vector<std::pair<DetId, float> > scDetIds = theElectron_->superCluster()->hitsAndFractions();

    for (std::vector<std::pair<DetId, float> >::const_iterator idIt = scDetIds.begin(); idIt != scDetIds.end();
         ++idIt) {
      const EcalRecHit* rh = nullptr;
      if ((*idIt).first.subdetId() == EcalBarrel)
        rh = &*(theHits_->find((*idIt).first));
      else if ((*idIt).first.subdetId() == EcalEndcap)
        rh = &*(theEEHits_->find((*idIt).first));
      if (!rh)
        std::cout << "CalibElectron::BIG ERROR::RecHit NOT FOUND" << std::endl;
      w_ring[EcalRingCalibrationTools::getRingIndex((*idIt).first)] += rh->energy();
    }

    for (int i = 0; i < EcalRingCalibrationTools::N_RING_TOTAL; ++i)
      if (w_ring[i] != 0.)
        theWeights.push_back(std::pair<int, float>(i, w_ring[i]));
    // std::cout << " ring " << i << " - energy sum " << w_ring[i] << std::endl;
  }

  else if (calibtype == "MODULE") {
    float w_ring[EcalRingCalibrationTools::N_MODULES_BARREL];

    for (int i = 0; i < EcalRingCalibrationTools::N_MODULES_BARREL; ++i)
      w_ring[i] = 0.;

    std::vector<std::pair<DetId, float> > scDetIds = theElectron_->superCluster()->hitsAndFractions();

    for (std::vector<std::pair<DetId, float> >::const_iterator idIt = scDetIds.begin(); idIt != scDetIds.end();
         ++idIt) {
      const EcalRecHit* rh = nullptr;
      if ((*idIt).first.subdetId() == EcalBarrel)
        rh = &*(theHits_->find((*idIt).first));
      else if ((*idIt).first.subdetId() == EcalEndcap)
        rh = &*(theEEHits_->find((*idIt).first));
      if (!rh)
        std::cout << "CalibElectron::BIG ERROR::RecHit NOT FOUND" << std::endl;
      w_ring[EcalRingCalibrationTools::getModuleIndex((*idIt).first)] += rh->energy();
    }

    for (int i = 0; i < EcalRingCalibrationTools::N_MODULES_BARREL; ++i)
      if (w_ring[i] != 0.)
        theWeights.push_back(std::pair<int, float>(i, w_ring[i]));
    // std::cout << " ring " << i << " - energy sum " << w_ring[i] << std::endl;

  }

  else if (calibtype == "ABS_SCALE") {
    std::cout << "ENTERING CalibElectron, ABS SCALE mode" << std::endl;

    float w_ring(0.);

    std::vector<std::pair<DetId, float> > scDetIds = theElectron_->superCluster()->hitsAndFractions();

    for (std::vector<std::pair<DetId, float> >::const_iterator idIt = scDetIds.begin(); idIt != scDetIds.end();
         ++idIt) {
      const EcalRecHit* rh = nullptr;
      if ((*idIt).first.subdetId() == EcalBarrel)
        rh = &*(theHits_->find((*idIt).first));
      else if ((*idIt).first.subdetId() == EcalEndcap)
        rh = &*(theEEHits_->find((*idIt).first));
      if (!rh)
        std::cout << "CalibElectron::BIG ERROR::RecHit NOT FOUND" << std::endl;

      w_ring += rh->energy();
    }

    if (w_ring != 0.)
      theWeights.push_back(std::pair<int, float>(0, w_ring));
    std::cout << " ABS SCALE  - energy sum " << w_ring << std::endl;

  }

  else if (calibtype == "ETA_ET_MODE") {
    float w_ring[200];

    for (int i = 0; i < EcalIndexingTools::getInstance()->getNumberOfChannels(); ++i)
      w_ring[i] = 0.;

    std::vector<std::pair<DetId, float> > scDetIds = theElectron_->superCluster()->hitsAndFractions();

    for (std::vector<std::pair<DetId, float> >::const_iterator idIt = scDetIds.begin(); idIt != scDetIds.end();
         ++idIt) {
      const EcalRecHit* rh = nullptr;
      if ((*idIt).first.subdetId() == EcalBarrel)
        rh = &*(theHits_->find((*idIt).first));
      else if ((*idIt).first.subdetId() == EcalEndcap)
        rh = &*(theEEHits_->find((*idIt).first));
      if (!rh)
        std::cout << "CalibElectron::BIG ERROR::RecHit NOT FOUND" << std::endl;

      float eta = fabs(theElectron_->eta());
      float theta = 2. * atan(exp(-eta));
      float et = theElectron_->superCluster()->energy() * sin(theta);

      int in = EcalIndexingTools::getInstance()->getProgressiveIndex(eta, et);

      w_ring[in] += rh->energy();
      //w_ring[in]+=theElectron_->superCluster()->energy();

      std::cout << "CalibElectron::filling channel " << in << " with value " << theElectron_->superCluster()->energy()
                << std::endl;
    }

    for (int i = 0; i < EcalIndexingTools::getInstance()->getNumberOfChannels(); ++i) {
      [[clang::suppress]]
      if (w_ring[i] != 0.) {
        theWeights.push_back(std::pair<int, float>(i, w_ring[i]));
        std::cout << " ring " << i << " - energy sum " << w_ring[i] << std::endl;
      }
    }

  }

  else {
    std::cout << "CalibType not yet implemented" << std::endl;
  }

  return theWeights;
}