#include <TauJetSelectorForHLTTrackSeeding.h>

Inheritance diagram for TauJetSelectorForHLTTrackSeeding:

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

	~TauJetSelectorForHLTTrackSeeding () override

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

bool	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () const final

bool	wantsGlobalLuminosityBlocks () const final

bool	wantsGlobalRuns () const final

bool	wantsStreamLuminosityBlocks () const final

bool	wantsStreamRuns () const final

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

ModuleDescription const &	moduleDescription () const

	~EDProducerBase () override

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 ()

std::vector< edm::ProductResolverIndex > const &	putTokenIndexToProductResolverIndex () const

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, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)

	~ProducerBase () override

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

ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) 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
void	beginJob () override

void	endJob () override

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
using	ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const , const char , edm::ProductResolverIndex >>

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 ( )

override

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

overrideprivate

Definition at line 41 of file TauJetSelectorForHLTTrackSeeding.cc.

References hiPixelPairStep_cff::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));
 }