#include <TauJetSelectorForHLTTrackSeeding.h>

Inheritance diagram for TauJetSelectorForHLTTrackSeeding:

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

	~TauJetSelectorForHLTTrackSeeding ()

Public Member Functions inherited from edm::global::EDProducer<>
	EDProducer ()=default

Public Member Functions inherited from edm::global::EDProducerBase
	EDProducerBase ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducerBase ()

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

void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...

	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 &	itemsToGetFrom (BranchType iType) 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)

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

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

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Private Member Functions
virtual void	beginJob () override

virtual void	endJob () override

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

Private Attributes
const double	etaMaxCaloJet_

const double	etaMinCaloJet_

const double	fractionMaxChargedPUInCaloCone_

const double	fractionMinCaloInTauCone_

edm::EDGetTokenT< reco::CaloJetCollection >	inputCaloJetToken_

edm::EDGetTokenT< reco::TrackJetCollection >	inputTrackJetToken_

edm::EDGetTokenT< reco::TrackCollection >	inputTrackToken_

const double	isolationConeSize_

const int	nTrkMaxInCaloCone_

const double	ptMinCaloJet_

const double	ptTrkMaxInCaloCone_

const double	tauConeSize_

Additional Inherited Members
Public Types inherited from edm::global::EDProducerBase
typedef EDProducerBase	ModuleType

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

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

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

Definition at line 27 of file TauJetSelectorForHLTTrackSeeding.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 6 of file TauJetSelectorForHLTTrackSeeding.cc.

References edm::ParameterSet::getParameter(), inputCaloJetToken_, inputTrackJetToken_, and inputTrackToken_.

                                                                                                 :
   ptMinCaloJet_(iConfig.getParameter< double > ("ptMinCaloJet")),
   etaMinCaloJet_(iConfig.getParameter< double > ("etaMinCaloJet")),
   etaMaxCaloJet_(iConfig.getParameter< double > ("etaMaxCaloJet")),
   tauConeSize_(iConfig.getParameter< double > ("tauConeSize")),
   isolationConeSize_(iConfig.getParameter< double > ("isolationConeSize")),
   fractionMinCaloInTauCone_(iConfig.getParameter< double > ("fractionMinCaloInTauCone")),
   fractionMaxChargedPUInCaloCone_(iConfig.getParameter< double > ("fractionMaxChargedPUInCaloCone")),
   ptTrkMaxInCaloCone_(iConfig.getParameter< double > ("ptTrkMaxInCaloCone")),
   nTrkMaxInCaloCone_(iConfig.getParameter< int > ("nTrkMaxInCaloCone"))
 {
    //now do what ever initialization is needed
   inputTrackJetToken_ = consumes< reco::TrackJetCollection >(iConfig.getParameter< edm::InputTag > ("inputTrackJetTag"));
   inputCaloJetToken_  = consumes< reco::CaloJetCollection >(iConfig.getParameter< edm::InputTag > ("inputCaloJetTag"));
   inputTrackToken_    = consumes< reco::TrackCollection >(iConfig.getParameter< edm::InputTag > ("inputTrackTag"));
 
   produces<reco::TrackJetCollection>();
 }

TauJetSelectorForHLTTrackSeeding::~TauJetSelectorForHLTTrackSeeding ( )

Definition at line 26 of file TauJetSelectorForHLTTrackSeeding.cc.

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

Member Function Documentation

void TauJetSelectorForHLTTrackSeeding::beginJob ( void )

overrideprivatevirtual

Reimplemented from edm::global::EDProducerBase.

Definition at line 144 of file TauJetSelectorForHLTTrackSeeding.cc.

144 {}

void TauJetSelectorForHLTTrackSeeding::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::global::EDProducerBase.

Definition at line 148 of file TauJetSelectorForHLTTrackSeeding.cc.

148 {}

void TauJetSelectorForHLTTrackSeeding::fillDescriptions ( edm::ConfigurationDescriptions & descriptions )

static

Definition at line 151 of file TauJetSelectorForHLTTrackSeeding.cc.

References edm::ConfigurationDescriptions::add(), edm::ParameterSetDescription::add(), edm::ConfigurationDescriptions::setComment(), and edm::ParameterDescriptionNode::setComment().

 {
   edm::ParameterSetDescription desc;
   desc.add<edm::InputTag>("inputTrackJetTag",edm::InputTag("hltAntiKT5TrackJetsIter0"))->setComment("Oryginal TrackJet collection");
   desc.add<edm::InputTag>("inputCaloJetTag",edm::InputTag("hltAntiKT5CaloJetsPFEt5"))->setComment("CaloJet collection");
   desc.add<edm::InputTag>("inputTrackTag",edm::InputTag("hltPFlowTrackSelectionHighPurity"))->setComment("Track collection for track isolation");
   desc.add<double>("ptMinCaloJet",5.0)->setComment("Min. pT of CaloJet");
   desc.add<double>("etaMinCaloJet",-2.7)->setComment("Min. eta of CaloJet");
   desc.add<double>("etaMaxCaloJet",+2.7)->setComment("Max. eta of CaloJet");
   desc.add<double>("tauConeSize",0.2)->setComment("Radius of tau signal cone");
   desc.add<double>("isolationConeSize",0.5)->setComment("Radius of isolation cone");
   desc.add<double>("fractionMinCaloInTauCone",0.8)->setComment("Min. fraction of calo energy in tau signal cone");
   desc.add<double>("fractionMaxChargedPUInCaloCone",0.2)->setComment("Max. fraction of PU charged energy Sum(Pt_trks)/Pt_jet in signal cone");
   desc.add<double>("ptTrkMaxInCaloCone",1.0)->setComment("Max. sum of pT of tracks in isolation cone");
   desc.add<int>("nTrkMaxInCaloCone",0)->setComment("Max. no. of tracks in isolation cone");
   descriptions.setComment("This module produces a collection of TrackJets to seed iterative tracking.\nIt is done by enriching the input TrackJet collection with CaloJets passing isolation critera - tau candidates"); 
   descriptions.add("tauJetSelectorForHLTTrackSeeding",desc);
 }

void TauJetSelectorForHLTTrackSeeding::produce	(	edm::StreamID	iStreamID,
		edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)		const

overrideprivatevirtual

Definition at line 41 of file TauJetSelectorForHLTTrackSeeding.cc.

References MuonCkfTrajectoryBuilder_cfi::deltaPhi, reco::deltaR2(), reco::TrackBase::eta(), etaMaxCaloJet_, etaMinCaloJet_, fractionMaxChargedPUInCaloCone_, fractionMinCaloInTauCone_, edm::Event::getByToken(), inputCaloJetToken_, inputTrackJetToken_, inputTrackToken_, isolationConeSize_, eostools::move(), nTrkMaxInCaloCone_, p4, phi, reco::TrackBase::phi(), reco::TrackBase::pt(), ptMinCaloJet_, ptTrkMaxInCaloCone_, edm::Event::put(), tauConeSize_, HiIsolationCommonParameters_cff::track, reco::btau::trackEta, reco::btau::trackPhi, and l1t::tracks.

 {
    std::unique_ptr< reco::TrackJetCollection > augmentedTrackJets (new reco::TrackJetCollection);
 
    edm::Handle<reco::TrackJetCollection> trackjets;
    iEvent.getByToken(inputTrackJetToken_, trackjets);
 
    for (reco::TrackJetCollection::const_iterator trackjet = trackjets->begin();
       trackjet != trackjets->end(); trackjet++) {
      augmentedTrackJets->push_back(*trackjet);
    }
 
    edm::Handle<reco::TrackCollection> tracks;
    iEvent.getByToken(inputTrackToken_,tracks);
 
    edm::Handle<reco::CaloJetCollection> calojets;
    iEvent.getByToken(inputCaloJetToken_,calojets);
 
    const double tauConeSize2       = tauConeSize_ * tauConeSize_;
    const double isolationConeSize2 = isolationConeSize_ * isolationConeSize_;
 
    for (reco::CaloJetCollection::const_iterator calojet = calojets->begin();
       calojet != calojets->end(); calojet++) {
 
      if ( calojet->pt() < ptMinCaloJet_ ) continue;
      double etaJet = calojet->eta();
      double phiJet = calojet->phi();
      if ( etaJet < etaMinCaloJet_ ) continue;
      if ( etaJet > etaMaxCaloJet_ ) continue;
 
      std::vector <CaloTowerPtr> const & theTowers = calojet->getCaloConstituents();
      double ptIn = 0.;
      double ptOut = 0.;
      for ( unsigned int itwr = 0; itwr < theTowers.size(); ++itwr ) { 
        double etaTwr = theTowers[itwr]->eta() - etaJet;
        double phiTwr = deltaPhi(theTowers[itwr]->phi(), phiJet);
        double deltaR2 = etaTwr*etaTwr + phiTwr*phiTwr;
        //std::cout << "Tower eta/phi/et : " << etaTwr << " " << phiTwr << " " << theTowers[itwr]->pt() << std::endl;
        if ( deltaR2 < tauConeSize2 ) { 
      ptIn += theTowers[itwr]->pt(); 
        } else if ( deltaR2 < isolationConeSize2 ) { 
      ptOut += theTowers[itwr]->pt(); 
        }
      }
      double ptTot = ptIn+ptOut;
      double fracIn = ptIn/ptTot;
 
      // We are looking for isolated tracks
      if ( fracIn < fractionMinCaloInTauCone_) continue;
 
      int ntrk = 0;
      double ptTrk = 0.;
 
      for (reco::TrackJetCollection::const_iterator trackjet = trackjets->begin();
       trackjet != trackjets->end(); trackjet++) {
        for (unsigned itr=0; itr<trackjet->numberOfTracks(); ++itr) { 
      edm::Ptr<reco::Track> track = trackjet->track(itr);
      double trackEta = track->eta() - etaJet;
      double trackPhi = deltaPhi(track->phi(), phiJet);
      double deltaR2 = trackEta*trackEta + trackPhi*trackPhi;
      if ( deltaR2 < isolationConeSize2 ) { 
        ntrk++; 
        ptTrk += track->pt();
      }
        }
      }
      // We are looking for calojets without signal tracks already in
      if ( ntrk > nTrkMaxInCaloCone_ ) continue;
      if ( ptTrk > ptTrkMaxInCaloCone_ ) continue;
 
      int ntrk2 = 0;
      double ptTrk2 = 0.;
 
      for (reco::TrackCollection::const_iterator track = tracks->begin();
       track != tracks->end(); track++) {
        double trackEta = track->eta() - etaJet;
        double trackPhi = deltaPhi(track->phi(), phiJet);
        double deltaR2 = trackEta*trackEta + trackPhi*trackPhi;
        if ( deltaR2 < isolationConeSize2 ) { 
      ntrk2++; 
      ptTrk2 += track->pt();
        }
      }
      // We are looking for signal jets, not PU jets
      double fractionChargedPU = ptTrk2/calojet->pt(); 
      if ( fractionChargedPU > fractionMaxChargedPUInCaloCone_ ) continue;
      /*
      std::cout << "Calo Jet " << calojet->pt() << " " << calojet->eta() 
            << " " << ptIn << " " << ptOut << " " << fracIn
            << " " << ptTrk << " " << ntrk 
            << " " << fractionChargedPU
            << std::endl;
      */
      math::XYZTLorentzVector p4(calojet->p4());
      math::XYZPoint vertex(calojet->vertex());
      augmentedTrackJets->push_back(reco::TrackJet(p4,vertex));
    }
 
    iEvent.put(std::move(augmentedTrackJets));
 }