/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_5_3_13_patch3/src/DataFormats/CaloTowers/src/CaloTower.cc

Go to the documentation of this file.

#include "DataFormats/CaloTowers/interface/CaloTower.h"


CaloTower::CaloTower() {
  emE_=0;
  hadE_=0;
  outerE_=0;
  emLvl1_=0;
  hadLvl1_=0;
}

CaloTower::CaloTower(const CaloTowerDetId& id,
                     double emE, double hadE, double outerE,
                     int ecal_tp, int hcal_tp,
                     const PolarLorentzVector p4,
                     GlobalPoint emPos, GlobalPoint hadPos) : 
  LeafCandidate(0, p4, Point(0,0,0)),  
  id_(id),
  emPosition_(emPos), hadPosition_(hadPos), 
  emE_(emE), hadE_(hadE), outerE_(outerE),
  emLvl1_(ecal_tp), hadLvl1_(hcal_tp) {}


CaloTower::CaloTower(const CaloTowerDetId& id,
                     double emE, double hadE, double outerE,
                     int ecal_tp, int hcal_tp,
                     const LorentzVector p4,
                     GlobalPoint emPos, GlobalPoint hadPos) : 
  LeafCandidate(0, p4, Point(0,0,0)),  
  id_(id),
  emPosition_(emPos), hadPosition_(hadPos),
  emE_(emE), hadE_(hadE), outerE_(outerE),
  emLvl1_(ecal_tp), hadLvl1_(hcal_tp) {}


// recalculated momentum-related quantities wrt user provided vertex Z position


math::PtEtaPhiMLorentzVector CaloTower::hadP4(double vtxZ) const {


  // note: for now we use the same position for HO as for the other detectors

  double hcalTot;
  if (abs(ieta())<16) hcalTot = (energy() - emE_);
  else hcalTot = hadE_;

  if (hcalTot>0) {
    double ctgTheta = (hadPosition_.z() - vtxZ)/hadPosition_.perp();
    double newEta = asinh(ctgTheta);  
    double pf = 1.0/cosh(newEta);

    return PolarLorentzVector(hcalTot * pf, newEta, hadPosition_.phi(), 0.0);   
  }
  
  return math::PtEtaPhiMLorentzVector(0,0,0,0);
}

math::PtEtaPhiMLorentzVector CaloTower::emP4(double vtxZ) const {

  if (emE_>0) {
    double ctgTheta = (emPosition_.z() - vtxZ)/emPosition_.perp();
    double newEta = asinh(ctgTheta);  
    double pf = 1.0/cosh(newEta);
  
    return math::PtEtaPhiMLorentzVector(emE_ * pf, newEta, emPosition_.phi(), 0.0);   
  }
  
  return math::PtEtaPhiMLorentzVector(0,0,0,0);
}


// recalculated momentum-related quantities wrt user provided 3D vertex 


math::PtEtaPhiMLorentzVector CaloTower::hadP4(Point v) const {

  // note: for now we use the same position for HO as for the other detectors

  double hcalTot;
  if (abs(ieta())<16) hcalTot = (energy() - emE_);
  else hcalTot = hadE_;

  if (hcalTot>0) {
    GlobalPoint p(v.x(), v.y(), v.z());
    math::XYZVector dir = math::XYZVector(hadPosition_ - p);
    return math::PtEtaPhiMLorentzVector(hcalTot * sin(dir.theta()), dir.eta(), dir.phi(), 0.0);  
  }

  return   math::PtEtaPhiMLorentzVector(0,0,0,0);
}

math::PtEtaPhiMLorentzVector CaloTower::emP4(Point v) const {

  if (emE_>0) {
    GlobalPoint p(v.x(), v.y(), v.z());
    math::XYZVector dir = math::XYZVector(emPosition_ - p);
    return math::PtEtaPhiMLorentzVector(emE_ * sin(dir.theta()), dir.eta(), dir.phi(), 0.0);   
  }
  
  return   math::PtEtaPhiMLorentzVector(0,0,0,0);
}


math::PtEtaPhiMLorentzVector CaloTower::p4(double vtxZ) const {

  if (abs(ieta())<=29) {
    return (emP4(vtxZ)+hadP4(vtxZ));
  }
  // em and had energy in HF are defined in a special way
  double ctgTheta = (emPosition_.z() - vtxZ)/emPosition_.perp(); // em and had positions in HF are forced to be the same
  double newEta = asinh(ctgTheta);  
  double pf = 1.0/cosh(newEta);
  return math::PtEtaPhiMLorentzVector(p4().energy() * pf, newEta, emPosition_.phi(), 0.0);   
}


math::PtEtaPhiMLorentzVector CaloTower::p4(Point v) const {

  if (abs(ieta())<=29) {
    return emP4(v)+hadP4(v);
  }
  // em and had energy in HF are defined in a special way
  GlobalPoint p(v.x(), v.y(), v.z());
  math::XYZVector dir = math::XYZVector(emPosition_ - p); // em and had positions in HF are forced to be the same
  return math::PtEtaPhiMLorentzVector(p4().energy() * sin(dir.theta()), dir.eta(), dir.phi(), 0.0);   
}


// p4 contribution from HO alone (note: direction is always taken to be the same as used for HB.)

math::PtEtaPhiMLorentzVector CaloTower::p4_HO(Point v) const {

  if (ietaAbs()>15 || outerE_<0) return   math::PtEtaPhiMLorentzVector(0,0,0,0);
  
  GlobalPoint p(v.x(), v.y(), v.z());
  math::XYZVector dir = math::XYZVector(hadPosition_ - p);
  return math::PtEtaPhiMLorentzVector(outerE_ * sin(dir.theta()), dir.eta(), dir.phi(), 0.0);  
}

math::PtEtaPhiMLorentzVector CaloTower::p4_HO(double vtxZ) const {
    Point p(0, 0, vtxZ);
    return p4_HO(p);
}

math::PtEtaPhiMLorentzVector CaloTower::p4_HO() const {

  if (ietaAbs()>15 || outerE_<0) return   math::PtEtaPhiMLorentzVector(0.0,0.0,0.0,0.0);
  return math::PtEtaPhiMLorentzVector(outerE_ * sin(hadPosition_.theta()), hadPosition_.eta(), hadPosition_.phi(), 0.0);  
}


void CaloTower::addConstituents( const std::vector<DetId>& ids ) {
  constituents_.reserve(constituents_.size()+ids.size());
  constituents_.insert(constituents_.end(),ids.begin(),ids.end());
}

int CaloTower::numCrystals() const {
  if (id_.ietaAbs()>29) return 0;
  
  int nC = 0;
  std::vector<DetId>::const_iterator it = constituents_.begin();
  for (; it!=constituents_.end(); ++it) {
    if (it->det()==DetId::Ecal) ++nC;
  }

  return nC;
}



// Set the CaloTower status word from the number of bad/recovered/problematic
// cells in HCAL and ECAL.

void CaloTower::setCaloTowerStatus(unsigned int numBadHcalChan,unsigned int numBadEcalChan, 
                                   unsigned int numRecHcalChan,unsigned int numRecEcalChan,
                                   unsigned int numProbHcalChan,unsigned int numProbEcalChan) {

  twrStatusWord_ = 0x0;

  twrStatusWord_ |= (  numBadEcalChan  & 0x1F);
  twrStatusWord_ |= ( (numRecEcalChan  & 0x1F) << 5);
  twrStatusWord_ |= ( (numProbEcalChan & 0x1F) << 10); 
  twrStatusWord_ |= ( (numBadHcalChan  & 0x7)  << 15);
  twrStatusWord_ |= ( (numRecHcalChan  & 0x7)  << 18);
  twrStatusWord_ |= ( (numProbHcalChan & 0x7)  << 21);

  return;
}




// energy in the tower by HCAL subdetector
// This is trivia except for tower 16
// needed by JetMET cleanup in AOD.

double CaloTower::energyInHB() const  { 
  if (id_.ietaAbs()<16) return hadE_;
  else if (id_.ietaAbs()==16) return hadE_-outerE_;
  else return 0.0;
}

double CaloTower::energyInHE() const { 
  if (id_.ietaAbs()>16 && id_.ietaAbs()<30) return hadE_;
  else if (id_.ietaAbs()==16) return outerE_;
  else return 0.0;
}
 
double CaloTower::energyInHF() const {
  if (id_.ietaAbs()>29) return energy();
  else return 0.0;
}

// this is actual energy contributed to the tower
// (outerEnergy() returns HO energy regardless if it is used or not)
// Note: rounding error may lead to values not identically equal to zero
// when HO was not used 

double CaloTower::energyInHO() const {
  if (id_.ietaAbs()>15) return 0.0;
  else return (energy() - hadE_ -emE_);
}





std::ostream& operator<<(std::ostream& s, const CaloTower& ct) {
  return s << ct.id() << ":"  << ct.et()
           << " GeV ET (EM=" << ct.emEt() <<
    " HAD=" << ct.hadEt() << " OUTER=" << ct.outerEt() << ") (" << ct.eta() << "," << ct.phi() << ")";    
}