#include <RecoHI/ParticleTowerProducer/src/ParticleTowerProducer.cc>

Inheritance diagram for ParticleTowerProducer:

Public Member Functions
	ParticleTowerProducer (const edm::ParameterSet &)

	~ParticleTowerProducer ()

Public Member Functions inherited from edm::EDProducer
	EDProducer ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducer ()

Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

std::vector < edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const

	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

void	resolvePutIndicies (BranchType iBranchType, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)

virtual	~ProducerBase () noexcept(false)

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

std::vector < ProductResolverIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

EDConsumerBase &	operator= (EDConsumerBase &&)=default

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

virtual	~EDConsumerBase () noexcept(false)

Private Member Functions
virtual void	beginJob ()

virtual void	endJob ()

int	eta2ieta (double eta) const

DetId	getNearestTower (const reco::PFCandidate &in) const

DetId	getNearestTower (double eta, double phi) const

int	phi2iphi (double phi, int ieta) const

virtual void	produce (edm::Event &, const edm::EventSetup &)

void	resetTowers (edm::Event &iEvent, const edm::EventSetup &iSetup)

Private Attributes
double	etaedge [42]

CaloGeometry const *	geo_

double	PI

TRandom *	random_

edm::EDGetTokenT < reco::PFCandidateCollection >	src_

std::map< DetId, double >	towers_

bool	useHF_

Static Private Attributes
static const double	etacent []

static const double	etatow []

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Description: [one line class summary]

Implementation: [Notes on implementation]

Definition at line 27 of file ParticleTowerProducer.h.

Constructor & Destructor Documentation

ParticleTowerProducer::ParticleTowerProducer ( const edm::ParameterSet & iConfig )

explicit

Definition at line 65 of file ParticleTowerProducer.cc.

References edm::ParameterSet::getParameter(), Pi, PI, random_, src_, and useHF_.

                                                                           :
    geo_(0)
 {
    //register your products  
   src_ = consumes<reco::PFCandidateCollection>(iConfig.getParameter<edm::InputTag>("src"));
   useHF_ = iConfig.getParameter<bool>("useHF");
   
   produces<CaloTowerCollection>();
   
   //now do what ever other initialization is needed
   random_ = new TRandom();
   PI = TMath::Pi();
   
 
 
 }

ParticleTowerProducer::~ParticleTowerProducer ( )

Definition at line 83 of file ParticleTowerProducer.cc.

References random_.

 {
  
    // do anything here that needs to be done at desctruction time
    // (e.g. close files, deallocate resources etc.)
   delete random_;
 }

Member Function Documentation

void ParticleTowerProducer::beginJob ( void )

privatevirtual

Reimplemented from edm::EDProducer.

Definition at line 212 of file ParticleTowerProducer.cc.

References etacent, etaedge, etatow, i, and useHF_.

 {
   // tower edges from fast sim, used starting at index 30 for the HF
   const double etatow[42] = {0.000, 0.087, 0.174, 0.261, 0.348, 0.435, 0.522, 0.609, 0.696, 0.783, 0.870, 0.957, 1.044, 1.131, 1.218, 1.305, 1.392, 1.479, 1.566, 1.653, 1.740, 1.830, 1.930, 2.043, 2.172, 2.322, 2.500, 2.650, 2.853, 3.000, 3.139, 3.314, 3.489, 3.664, 3.839, 4.013, 4.191, 4.363, 4.538, 4.716, 4.889, 5.191};
   
   // tower centers derived directly from by looping over HCAL towers and printing out the values, used up until eta = 3 where the HF/Endcap interface becomes problematic
   const double etacent[40] = {
     0.0435, // 1
     0.1305, // 2
     0.2175, // 3
     0.3045, // 4
     0.3915, // 5
     0.4785, // 6
     0.5655, // 7
     0.6525, // 8
     0.7395, // 9
     0.8265, // 10
     0.9135, // 11
     1.0005, // 12
     1.0875, // 13
     1.1745, // 14
     1.2615, // 15
     1.3485, // 16
     1.4355, // 17
     1.5225, // 18
     1.6095, // 19
     1.6965, // 20
     1.785 , // 21
     1.88  , // 22
     1.9865, // 23
     2.1075, // 24
     2.247 , // 25
     2.411 , // 26
     2.575 , // 27
     2.759 , // 28
     2.934 , // 29
     2.90138,// 29
     3.04448,// 30 
     3.21932,// 31
     3.39462,// 32
     3.56966,// 33 
     3.74485,// 34 
     3.91956,// 35 
     4.09497,// 36 
     4.27007,// 37 
     4.44417,// 38 
     4.62046 // 39 
       };
 
   // Use the real towers centrers for the barrel and endcap up to eta=3
   etaedge[0] = 0.;
   for(int i=1;i<30;i++){
     etaedge[i] = (etacent[i-1]-etaedge[i-1])*2.0 + etaedge[i-1];
     //std::cout<<" i "<<i<<" etaedge "<<etaedge[i]<<std::endl;  
   }
   // beyond eta = 3 just use the fast sim values
   if(useHF_){
     for(int i=30;i<42;i++){
       etaedge[i]=etatow[i];
       //std::cout<<" i "<<i<<" etaedge "<<etaedge[i]<<std::endl;  
     }
   }
 }

void ParticleTowerProducer::endJob ( void )

privatevirtual

Reimplemented from edm::EDProducer.

Definition at line 278 of file ParticleTowerProducer.cc.

278 {

279 }

int ParticleTowerProducer::eta2ieta ( double eta ) const

private

Definition at line 468 of file ParticleTowerProducer.cc.

References gather_cfg::cout, etaedge, findQualityFiles::size, useHF_, and x.

Referenced by produce().

                                                     {
   // binary search in the array of towers eta edges
 
   int size = 42;
   if(!useHF_) size = 30;
 
   if(fabs(eta)>etaedge[size-1]) return -1;
 
   double x = fabs(eta);
   int curr = size / 2;
   int step = size / 4;
   int iter;
   int prevdir = 0; 
   int actudir = 0; 
 
   for (iter = 0; iter < size ; iter++) {
 
     if( curr >= size || curr < 1 )
       std::cout <<  " ParticleTowerProducer::eta2ieta - wrong current index = "
         << curr << " !!!" << std::endl;
 
     if ((x <= etaedge[curr]) && (x > etaedge[curr-1])) break;
     prevdir = actudir;
     if(x > etaedge[curr]) {actudir =  1;}
     else {actudir = -1;}
     if(prevdir * actudir < 0) { if(step > 1) step /= 2;}
     curr += actudir * step;
     if(curr > size) curr = size;
     else { if(curr < 1) {curr = 1;}}
 
     /*
     std::cout << " HCALProperties::eta2ieta  end of iter." << iter 
           << " curr, etaedge[curr-1], etaedge[curr] = "
             << curr << " " << etaedge[curr-1] << " " << etaedge[curr] << std::endl;
     */
     
   }
 
   /*
   std::cout << " HCALProperties::eta2ieta  for input x = " << x 
       << "  found index = " << curr-1
           << std::endl;
   */
   
   return curr;
 }

DetId ParticleTowerProducer::getNearestTower ( const reco::PFCandidate & in ) const

private

Definition at line 324 of file ParticleTowerProducer.cc.

References funct::abs(), deltaR(), reco::deltaR(), HcalDetId::depth(), eta, PV3DBase< T, PVType, FrameType >::eta(), reco::LeafCandidate::eta(), geo_, CaloGeometry::getGeometry(), CaloCellGeometry::getPosition(), CaloGeometry::getValidDetIds(), DetId::Hcal, HLT_25ns10e33_v2_cff::minDeltaR, phi, PV3DBase< T, PVType, FrameType >::phi(), and reco::LeafCandidate::phi().

                                                                              {
 
   double eta = in.eta();
   double phi  = in.phi();
 
   double minDeltaR = 9999;
 
   std::vector<DetId> alldid =  geo_->getValidDetIds();
 
   DetId returnId;
   
   //int nclosetowers=0;
 
   for(std::vector<DetId>::const_iterator did=alldid.begin(); did != alldid.end(); did++){
     if( (*did).det() == DetId::Hcal ){
 
       HcalDetId hid(*did);
       
       // which layer is irrelevant for an eta-phi map, no?
 
       if( hid.depth() != 1 ) continue;
 
       GlobalPoint pos =geo_->getGeometry(hid)->getPosition();
       
       double hcalEta = pos.eta();
       double hcalPhi = pos.phi();
 
       //std::cout<<" transverse distance = "<<sqrt(pos.x()*pos.x()+pos.y()*pos.y())<<std::endl;
 
       //std::cout<<" ieta "<<(hid).ieta()<<" iphi "<<(hid).iphi()<<" hcalEta "<<hcalEta<<" hcalPhi "<<hcalPhi<<std::endl;
 
       double deltaR = reco::deltaR(eta,phi,hcalEta,hcalPhi);
       
       // need to factor in the size of the tower
       double towersize = 0.087;
      
       int ieta  = (hid).ieta();
       
       if(abs(ieta)>21){
     if(abs(ieta)>29) towersize=0.175;
     else{
       if(abs(ieta)==22) towersize=0.1;
       else if(abs(ieta)==23) towersize=0.113;
       else if(abs(ieta)==24) towersize=0.129;
       else if(abs(ieta)==25) towersize=0.16;
       else if(abs(ieta)==26) towersize=0.168;
       else if(abs(ieta)==27) towersize=0.15;
       else if(abs(ieta)==28) towersize=0.218;
       else if(abs(ieta)==29) towersize=0.132;
     }
       }
 
       deltaR /= towersize;
       //if(deltaR<1/3.) nclosetowers++;
 
       if(deltaR<minDeltaR){
      returnId = DetId(*did);
      minDeltaR = deltaR;
       }
       
       //if(abs(eta-hcalEta)<towersize/2. && abs(phi-hcalPhi)<towersize/2.) break;
     }
   }
   //if(nclosetowers>1)std::cout<<"eta "<<eta<<" phi "<<phi<<" minDeltaR "<<minDeltaR<<" nclosetowers "<<nclosetowers<<std::endl;
   return returnId;
 
 
 }

DetId ParticleTowerProducer::getNearestTower	(	double	eta,
		double	phi
	)		const

private

Definition at line 394 of file ParticleTowerProducer.cc.

References funct::abs(), deltaR(), reco::deltaR(), HcalDetId::depth(), PV3DBase< T, PVType, FrameType >::eta(), geo_, CaloGeometry::getGeometry(), CaloCellGeometry::getPosition(), CaloGeometry::getValidDetIds(), DetId::Hcal, HLT_25ns10e33_v2_cff::minDeltaR, and PV3DBase< T, PVType, FrameType >::phi().

                                                                          {
 
   // get closest tower by delta R matching
   // Now obsolute, instead use faster premade eta-phi grid
 
   double minDeltaR = 9999;
 
   std::vector<DetId> alldid =  geo_->getValidDetIds();
 
   DetId returnId;
   
   //int nclosetowers=0;
 
   for(std::vector<DetId>::const_iterator did=alldid.begin(); did != alldid.end(); did++){
     if( (*did).det() == DetId::Hcal ){
 
       HcalDetId hid(*did);
       
       // which layer is irrelevant for an eta-phi map, no?
 
       if( hid.depth() != 1 ) continue;
 
       GlobalPoint pos =geo_->getGeometry(hid)->getPosition();
       
       double hcalEta = pos.eta();
       double hcalPhi = pos.phi();
 
       //std::cout<<" transverse distance = "<<sqrt(pos.x()*pos.x()+pos.y()*pos.y())<<std::endl;
 
       //std::cout<<" ieta "<<(hid).ieta()<<" iphi "<<(hid).iphi()<<" hcalEta "<<hcalEta<<" hcalPhi "<<hcalPhi<<std::endl;
 
       double deltaR = reco::deltaR(eta,phi,hcalEta,hcalPhi);
       
       // need to factor in the size of the tower
       double towersize = 0.087;
      
       int ieta  = (hid).ieta();
       
       if(abs(ieta)>21){
     if(abs(ieta)>29) towersize=0.175;
     else{
       if(abs(ieta)==22) towersize=0.1;
       else if(abs(ieta)==23) towersize=0.113;
       else if(abs(ieta)==24) towersize=0.129;
       else if(abs(ieta)==25) towersize=0.16;
       else if(abs(ieta)==26) towersize=0.168;
       else if(abs(ieta)==27) towersize=0.15;
       else if(abs(ieta)==28) towersize=0.218;
       else if(abs(ieta)==29) towersize=0.132;
     }
       }
 
       deltaR /= towersize;
       //if(deltaR<1/3.) nclosetowers++;
 
       if(deltaR<minDeltaR){
      returnId = DetId(*did);
      minDeltaR = deltaR;
       }
       
       //if(abs(eta-hcalEta)<towersize/2. && abs(phi-hcalPhi)<towersize/2.) break;
     }
   }
   //if(nclosetowers>1)std::cout<<"eta "<<eta<<" phi "<<phi<<" minDeltaR "<<minDeltaR<<" nclosetowers "<<nclosetowers<<std::endl;
   return returnId;
 
 
 }

int ParticleTowerProducer::phi2iphi	(	double	phi,
		int	ieta
	)		const

private

Definition at line 515 of file ParticleTowerProducer.cc.

References funct::abs(), and PI.

Referenced by produce().

                                                               {
   
   if(phi<0) phi += 2.*PI;
   else if(phi> 2.*PI) phi -= 2.*PI;
   
   int iphi = (int) TMath::Ceil(phi/2.0/PI*72.);
   // take into account larger granularity in endcap (x2) and at the end of the HF (x4)
   if(abs(ieta)>20){
     if(abs(ieta)<40) iphi -= (iphi+1)%2;
     else {
       iphi -= (iphi+1)%4;
       if(iphi==-1) iphi=71;
       }
   }
   
   return iphi;
 
 }

void ParticleTowerProducer::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

privatevirtual

Implements edm::EDProducer.

Definition at line 98 of file ParticleTowerProducer.cc.

References funct::abs(), reco::LeafCandidate::et(), eta, PV3DBase< T, PVType, FrameType >::eta(), reco::LeafCandidate::eta(), eta2ieta(), geo_, edm::EventSetup::get(), edm::Event::getByToken(), CaloGeometry::getGeometry(), CaloCellGeometry::getPosition(), HcalBarrel, HcalEndcap, HcalForward, eostools::move(), p4, PV3DBase< T, PVType, FrameType >::phi(), reco::LeafCandidate::phi(), phi2iphi(), parseEventContent::prod, edm::ESHandle< class >::product(), edm::Event::put(), resetTowers(), sd, src_, towers_, and useHF_.

 {
    using namespace edm;
 
    if(!geo_){
       edm::ESHandle<CaloGeometry> pG;
       iSetup.get<CaloGeometryRecord>().get(pG);
       geo_ = pG.product();
    }
 
 
    resetTowers(iEvent, iSetup);
 
 
    edm::Handle<reco::PFCandidateCollection> inputsHandle;
    iEvent.getByToken(src_, inputsHandle);
    
    for(reco::PFCandidateCollection::const_iterator ci  = inputsHandle->begin(); ci!=inputsHandle->end(); ++ci)  {
 
     const reco::PFCandidate& particle = *ci;
     
     // put a cutoff if you want
     //if(particle.et() < 0.3) continue;      
     
     double eta = particle.eta();
 
     if(!useHF_ && fabs(eta) > 3. ) continue;
 
     int ieta = eta2ieta(eta);
     if(ieta==-1) continue;
     if(eta<0) ieta  *= -1;
 
     int iphi = phi2iphi(particle.phi(),ieta);
        
     HcalSubdetector sd;
     if(abs(ieta)<=16) sd = HcalBarrel;
     else if(fabs(eta)< 3.) sd = HcalEndcap;  // Use the endcap until eta =3
     else sd = HcalForward;
     
     HcalDetId hid = HcalDetId(sd,ieta,iphi,1); // assume depth=1
 
     // check against the old method (boundaries slightly shifted in the endcaps
     /*
     HcalDetId hid_orig = getNearestTower(particle);
     
     if(hid != hid_orig)
       { 
     if(abs((hid).ieta()-(hid_orig).ieta())>1 || abs((hid).iphi()-(hid_orig).iphi()) >1 ){
 
       int phi_diff = 1;
       if(abs((hid).ieta())>39 || abs((hid_orig).ieta())>39) phi_diff = 4;
       else if(abs((hid).ieta())>20 || abs((hid_orig).ieta())>20) phi_diff = 2;
 
       if(abs((hid).ieta()-(hid_orig).ieta()) > 1 || abs((hid).iphi()-(hid_orig).iphi()) > phi_diff ){
           std::cout<<" eta "<<eta<<std::endl;
           std::cout<<" hid "<<hid<<std::endl;
           std::cout<<" old method "<<hid_orig<<std::endl;
       }
     }
       }
     */
     towers_[hid] += particle.et();      
    }
    
    
    auto prod = std::make_unique<CaloTowerCollection>();
 
    for ( std::map< DetId, double >::const_iterator iter = towers_.begin();
      iter != towers_.end(); ++iter ){
      
      HcalDetId id = HcalDetId(iter->first);
      CaloTowerDetId newTowerId(id.ieta(),id.iphi());
      double et = iter->second;
 
      if(et>0){
 
        GlobalPoint pos =geo_->getGeometry(newTowerId)->getPosition();
        
        if(!useHF_ && fabs(pos.eta()) > 3. ) continue;
 
        // currently sets et =  pt, mass to zero
        // pt, eta , phi, mass
        reco::Particle::PolarLorentzVector p4(et,pos.eta(),pos.phi(),0.);
        
        CaloTower newTower(newTowerId,et,0,0,0,0,p4,pos,pos);
        prod->push_back(newTower);     
      }
    }
 
    
    //For reference, Calo Tower Constructors
 
    /*
    CaloTower(const CaloTowerDetId& id, 
              double emE, double hadE, double outerE,
              int ecal_tp, int hcal_tp,
              const PolarLorentzVector p4,
        GlobalPoint emPosition, GlobalPoint hadPosition);
  
    CaloTower(const CaloTowerDetId& id, 
              double emE, double hadE, double outerE,
              int ecal_tp, int hcal_tp,
              const LorentzVector p4,
        GlobalPoint emPosition, GlobalPoint hadPosition);
    */
 
 
    iEvent.put(std::move(prod));
 
 
 }

void ParticleTowerProducer::resetTowers	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

private

Definition at line 282 of file ParticleTowerProducer.cc.

References HcalDetId::depth(), PV3DBase< T, PVType, FrameType >::eta(), geo_, CaloGeometry::getGeometry(), CaloCellGeometry::getPosition(), CaloGeometry::getValidDetIds(), DetId::Hcal, towers_, and useHF_.

Referenced by produce().

 {
   
   std::vector<DetId> alldid =  geo_->getValidDetIds();
 
   for(std::vector<DetId>::const_iterator did=alldid.begin(); did != alldid.end(); did++){
     if( (*did).det() == DetId::Hcal ){
        HcalDetId hid = HcalDetId(*did);
        if( hid.depth() == 1 ) {
      
      if(!useHF_){
        GlobalPoint pos =geo_->getGeometry(hid)->getPosition();   
        //if((hid).iphi()==1)std::cout<<" ieta "<<(hid).ieta()<<" eta "<<pos.eta()<<" iphi "<<(hid).iphi()<<" phi "<<pos.phi()<<std::endl;
        if(fabs(pos.eta())>3.) continue;
      }
       towers_[(*did)] = 0.;
        }
        
     }
   }
 
 
 
 
   //print out tower eta boundaries
   /*
   int current_ieta=0;
   float testphi =0.;
   for(double testeta=1.7; testeta<=5.4;testeta+=0.00001){
     HcalDetId hid = getNearestTower(testeta,testphi);
     if((hid).ieta()!=current_ieta) {
       std::cout<<" new ieta "<<current_ieta<<" testeta "<<testeta<<std::endl;
       current_ieta = (hid).ieta();
     }
   }
   */
 
 }