EcalIsolatedParticleCandidateProducer.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    EcalIsolatedParticleCandidateProducer
// Class:      EcalIsolatedParticleCandidateProducer
// 
//
// Original Author:  Grigory Safronov
//         Created:  Thu Jun  7 17:21:58 MSD 2007
//
//


// system include files
#include <cmath>
#include <memory>

// user include files

#include "FWCore/Framework/interface/EventSetup.h"


#include "DataFormats/Common/interface/Ref.h"
#include "DataFormats/DetId/interface/DetId.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "Geometry/CaloTopology/interface/EcalBarrelTopology.h"
#include "Geometry/Records/interface/CaloGeometryRecord.h"

#include "Calibration/HcalIsolatedTrackReco/interface/EcalIsolatedParticleCandidateProducer.h"

EcalIsolatedParticleCandidateProducer::EcalIsolatedParticleCandidateProducer(const edm::ParameterSet& conf)
{
  InConeSize_ = conf.getParameter<double>("EcalInnerConeSize");
  OutConeSize_= conf.getParameter<double>("EcalOuterConeSize");
  hitCountEthr_= conf.getParameter<double>("ECHitCountEnergyThreshold");
  hitEthr_=conf.getParameter<double>("ECHitEnergyThreshold");
  tok_l1tau_ = consumes<l1extra::L1JetParticleCollection>(conf.getParameter<edm::InputTag>("L1eTauJetsSource"));
  tok_hlt_ = consumes<trigger::TriggerFilterObjectWithRefs>(conf.getParameter<edm::InputTag>("L1GTSeedLabel"));
  tok_EB_ = consumes<EcalRecHitCollection>(conf.getParameter<edm::InputTag>("EBrecHitCollectionLabel"));
  tok_EE_ = consumes<EcalRecHitCollection>(conf.getParameter<edm::InputTag>("EErecHitCollectionLabel"));

   //register your products
  produces< reco::IsolatedPixelTrackCandidateCollection >();

}


EcalIsolatedParticleCandidateProducer::~EcalIsolatedParticleCandidateProducer()
{
 
   // do anything here that needs to be done at desctruction time
   // (e.g. close files, deallocate resources etc.)

}


//
// member functions
//

// ------------ method called to produce the data  ------------
void 
EcalIsolatedParticleCandidateProducer::produce(edm::StreamID, edm::Event& iEvent, const edm::EventSetup& iSetup) const {

//  std::cout<<"get tau"<<std::endl;

  edm::Handle<l1extra::L1JetParticleCollection> l1Taus;
  iEvent.getByToken(tok_l1tau_,l1Taus);
  
//  std::cout<<"get geom"<<std::endl;

  edm::ESHandle<CaloGeometry> pG;
  iSetup.get<CaloGeometryRecord>().get(pG);
  const CaloGeometry* geo = pG.product();

//  std::cout<<" get ec rechit"<<std::endl;

  edm::Handle<EcalRecHitCollection> ecalEB;
  iEvent.getByToken(tok_EB_,ecalEB);

  edm::Handle<EcalRecHitCollection> ecalEE;
  iEvent.getByToken(tok_EE_,ecalEE);

//  std::cout<<"get l1 trig obj"<<std::endl;

  edm::Handle<trigger::TriggerFilterObjectWithRefs> l1trigobj;
  iEvent.getByToken(tok_hlt_, l1trigobj);

  std::vector< edm::Ref<l1t::TauBxCollection> > l1tauobjref;
  std::vector< edm::Ref<l1t::JetBxCollection> > l1jetobjref;
  
  l1trigobj->getObjects(trigger::TriggerL1Tau, l1tauobjref);
  l1trigobj->getObjects(trigger::TriggerL1Jet, l1jetobjref);
   
  double ptTriggered=-10;
  double etaTriggered=-100;
  double phiTriggered=-100;

//  std::cout<<"find highest pT triggered obj"<<std::endl;

  for (unsigned int p=0; p<l1tauobjref.size(); p++) {
    if (l1tauobjref[p]->pt()>ptTriggered) {
      ptTriggered=l1tauobjref[p]->pt();
      phiTriggered=l1tauobjref[p]->phi();
      etaTriggered=l1tauobjref[p]->eta();
    }
  }
  for (unsigned int p=0; p<l1jetobjref.size(); p++) {
    if (l1jetobjref[p]->pt()>ptTriggered) {
      ptTriggered=l1jetobjref[p]->pt();
      phiTriggered=l1jetobjref[p]->phi();
      etaTriggered=l1jetobjref[p]->eta();
    }
  }

  auto iptcCollection = std::make_unique<reco::IsolatedPixelTrackCandidateCollection>();

//  std::cout<<"loop over l1taus"<<std::endl;

  for (l1extra::L1JetParticleCollection::const_iterator tit=l1Taus->begin(); tit!=l1Taus->end(); tit++) {
    double dphi=fabs(tit->phi()-phiTriggered);
    if (dphi>M_PI) dphi=2*M_PI-dphi;
    double Rseed=sqrt(pow(etaTriggered-tit->eta(),2)+dphi*dphi);
    if (Rseed<1.2) continue;
    int nhitOut=0;
    int nhitIn=0;
    double OutEnergy=0;
    double InEnergy=0;
//  std::cout<<" loops over rechits"<<std::endl;
    for (EcalRecHitCollection::const_iterator eItr=ecalEB->begin(); eItr!=ecalEB->end(); eItr++) {
      double phiD, R;
      const GlobalPoint& pos = geo->getPosition(eItr->detid());
      double phihit = pos.phi();
      double etahit = pos.eta();
      phiD=fabs(phihit-tit->phi());
      if (phiD>M_PI) phiD=2*M_PI-phiD;
      R=sqrt(pow(etahit-tit->eta(),2)+phiD*phiD);
                
      if (R<OutConeSize_&&R>InConeSize_&&eItr->energy()>hitCountEthr_) {
    nhitOut++;
      }
      if (R<InConeSize_&&eItr->energy()>hitCountEthr_)                 {
    nhitIn++;
      }

      if (R<OutConeSize_&&R>InConeSize_&&eItr->energy()>hitEthr_)      {
    OutEnergy+=eItr->energy();
      }
      if (R<InConeSize_&&eItr->energy()>hitEthr_)                      {
    InEnergy+=eItr->energy();
      }

    }
    
    for (EcalRecHitCollection::const_iterator eItr=ecalEE->begin(); eItr!=ecalEE->end(); eItr++)  {
      double phiD, R;
      const GlobalPoint& pos = geo->getPosition(eItr->detid());
      double phihit = pos.phi();
      double etahit = pos.eta();
      phiD=fabs(phihit-tit->phi());
      if (phiD>M_PI) phiD=2*M_PI-phiD;
      R=sqrt(pow(etahit-tit->eta(),2)+phiD*phiD);
      if (R<OutConeSize_&&R>InConeSize_&&eItr->energy()>hitCountEthr_)  {
    nhitOut++;
      }
      if (R<InConeSize_&&eItr->energy()>hitCountEthr_)                  {
    nhitIn++;
      }
      if (R<OutConeSize_&&R>InConeSize_&&eItr->energy()>hitEthr_)       {
    OutEnergy+=eItr->energy();
      }
      if (R<InConeSize_&&eItr->energy()>hitEthr_)                       {
    InEnergy+=eItr->energy();
      }

    }
//  std::cout<<"create and push_back candidate"<<std::endl;
    reco::IsolatedPixelTrackCandidate newca(l1extra::L1JetParticleRef(l1Taus,tit-l1Taus->begin()), InEnergy, OutEnergy, nhitIn, nhitOut);
    iptcCollection->push_back(newca);   
  }


  
  //Use the ExampleData to create an ExampleData2 which 
  // is put into the Event
  
//  std::cout<<"put cand into event"<<std::endl;
  iEvent.put(std::move(iptcCollection));

}
// ------------ method called once each job just before starting event loop  ------------
void 
EcalIsolatedParticleCandidateProducer::beginJob() {
}

// ------------ method called once each job just after ending the event loop  ------------
void 
EcalIsolatedParticleCandidateProducer::endJob() {
}