TPFastJetProducer Class Reference

Inheritance diagram for TPFastJetProducer:

Public Member Functions
	TPFastJetProducer (const edm::ParameterSet &)
	~TPFastJetProducer () override=default
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

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

Private Member Functions
void	produce (edm::Event &, const edm::EventSetup &) override

Private Attributes
const float	coneSize_
const float	tpEtaMax_
const int	tpNStubLayerMin_
const int	tpNStubMin_
const float	tpPtMin_
const float	tpZMax_
edm::EDGetTokenT< std::vector< TrackingParticle > >	trackingParticleToken_
edm::ESGetToken< TrackerTopology, TrackerTopologyRcd >	tTopoToken_
edm::EDGetTokenT< TTStubAssociationMap< Ref_Phase2TrackerDigi_ > >	ttStubMCTruthToken_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
using	CacheTypes = CacheContexts< T... >
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T... >
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Detailed Description

Definition at line 58 of file TPFastJetProducer.cc.

Constructor & Destructor Documentation

TPFastJetProducer()

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

explicit

Definition at line 82 of file TPFastJetProducer.cc.

    : tpPtMin_((float)iConfig.getParameter<double>("tp_ptMin")),
      tpEtaMax_((float)iConfig.getParameter<double>("tp_etaMax")),
      tpZMax_((float)iConfig.getParameter<double>("tp_zMax")),
      tpNStubMin_((int)iConfig.getParameter<int>("tp_nStubMin")),
      tpNStubLayerMin_((int)iConfig.getParameter<int>("tp_nStubLayerMin")),
      coneSize_((float)iConfig.getParameter<double>("coneSize")),
      trackingParticleToken_(
          consumes<std::vector<TrackingParticle>>(iConfig.getParameter<edm::InputTag>("TrackingParticleInputTag"))),
      ttStubMCTruthToken_(consumes<TTStubAssociationMap<Ref_Phase2TrackerDigi_>>(
          iConfig.getParameter<edm::InputTag>("MCTruthStubInputTag"))),
      tTopoToken_(esConsumes<TrackerTopology, TrackerTopologyRcd>(edm::ESInputTag("", ""))) {
  produces<TkJetCollection>("TPFastJets");
}

~TPFastJetProducer()

TPFastJetProducer::~TPFastJetProducer ( )

overridedefault

Member Function Documentation

fillDescriptions()

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

static

Definition at line 201 of file TPFastJetProducer.cc.

References edm::ConfigurationDescriptions::add(), submitPVResolutionJobs::desc, and ProducerED_cfi::InputTag.

                                                                                   {
  // The following says we do not know what parameters are allowed so do no validation
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("TrackingParticleInputTag", edm::InputTag("mix", "MergedTrackTruth"));
  desc.add<edm::InputTag>("MCTruthStubInputTag", edm::InputTag("TTStubAssociatorFromPixelDigis", "StubAccepted"));
  desc.add<double>("tp_ptMin", 2.0);
  desc.add<double>("tp_etaMax", 2.4);
  desc.add<double>("tp_zMax", 15.);
  desc.add<int>("tp_nStubMin", 4);
  desc.add<int>("tp_nStubLayerMin", 4);
  desc.add<double>("coneSize", 0.4);
  descriptions.add("tpFastJets", desc);
}

produce()

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

overrideprivate

Definition at line 98 of file TPFastJetProducer.cc.

References coneSize_, callgraph::cs, ALCARECOPPSCalTrackBasedSel_cff::detid, edm::EventSetup::getData(), mps_fire::i, iEvent, createfilelist::int, isum, nano_mu_digi_cff::layer, TrackerTopology::layer(), eostools::move(), TrackingParticle::pt(), multPhiCorr_741_25nsDY_cfi::px, multPhiCorr_741_25nsDY_cfi::py, StripSubdetector::TID, StripSubdetector::TOB, tpEtaMax_, tpNStubLayerMin_, tpNStubMin_, tpPtMin_, tpZMax_, trackingParticleToken_, tTopoToken_, ttStubMCTruthToken_, and TrackingParticle::z0().

                                                                             {
  std::unique_ptr<TkJetCollection> TPFastJets(new TkJetCollection);

  // Tracking particles
  edm::Handle<std::vector<TrackingParticle>> TrackingParticleHandle;
  iEvent.getByToken(trackingParticleToken_, TrackingParticleHandle);
  std::vector<TrackingParticle>::const_iterator iterTP;

  // MC truth association maps
  edm::Handle<TTStubAssociationMap<Ref_Phase2TrackerDigi_>> MCTruthTTStubHandle;
  iEvent.getByToken(ttStubMCTruthToken_, MCTruthTTStubHandle);

  // Tracker Topology
  const TrackerTopology& tTopo = iSetup.getData(tTopoToken_);

  fastjet::JetDefinition jet_def(fastjet::antikt_algorithm, coneSize_);
  std::vector<fastjet::PseudoJet> JetInputs;

  // loop over tps
  unsigned int this_tp = 0;
  for (iterTP = TrackingParticleHandle->begin(); iterTP != TrackingParticleHandle->end(); iterTP++) {
    edm::Ptr<TrackingParticle> tp_ptr(TrackingParticleHandle, this_tp);
    this_tp++;

    std::vector<edm::Ref<edmNew::DetSetVector<TTStub<Ref_Phase2TrackerDigi_>>, TTStub<Ref_Phase2TrackerDigi_>>>
        theStubRefs = MCTruthTTStubHandle->findTTStubRefs(tp_ptr);
    int nStubTP = (int)theStubRefs.size();

    // how many layers/disks have stubs?
    int hasStubInLayer[11] = {0};
    for (auto& theStubRef : theStubRefs) {
      DetId detid(theStubRef->getDetId());

      int layer = -1;
      if (detid.subdetId() == StripSubdetector::TOB) {
        layer = static_cast<int>(tTopo.layer(detid)) - 1;  //fill in array as entries 0-5
      } else if (detid.subdetId() == StripSubdetector::TID) {
        layer = static_cast<int>(tTopo.layer(detid)) + 5;  //fill in array as entries 6-10
      }

      //treat genuine stubs separately (==2 is genuine, ==1 is not)
      if (MCTruthTTStubHandle->findTrackingParticlePtr(theStubRef).isNull() && hasStubInLayer[layer] < 2)
        hasStubInLayer[layer] = 1;
      else
        hasStubInLayer[layer] = 2;
    }

    int nStubLayerTP = 0;
    for (int isum : hasStubInLayer) {
      if (isum >= 1)
        nStubLayerTP += 1;
    }

    // tp quality cuts to match L1 tracks
    if (iterTP->pt() < tpPtMin_)
      continue;
    if (fabs(iterTP->eta()) > tpEtaMax_)
      continue;
    if (nStubTP < tpNStubMin_)
      continue;
    if (nStubLayerTP < tpNStubLayerMin_)
      continue;
    if (fabs(iterTP->z0()) > tpZMax_)
      continue;
    if (iterTP->charge() == 0.)  // extra check that all tps are charged
      continue;
    if (iterTP->eventId().event() > 0)  // only select hard interaction tps
      continue;

    fastjet::PseudoJet psuedoJet(iterTP->px(), iterTP->py(), iterTP->pz(), iterTP->energy());
    JetInputs.push_back(psuedoJet);                // input tps for clustering
    JetInputs.back().set_user_index(this_tp - 1);  // save tp index in the collection
  }  // end loop over tps

  fastjet::ClusterSequence cs(JetInputs, jet_def);  // define the output jet collection
  std::vector<fastjet::PseudoJet> JetOutputs =
      fastjet::sorted_by_pt(cs.inclusive_jets(0));  // output jet collection, pT-ordered

  for (unsigned int ijet = 0; ijet < JetOutputs.size(); ++ijet) {
    math::XYZTLorentzVector jetP4(
        JetOutputs[ijet].px(), JetOutputs[ijet].py(), JetOutputs[ijet].pz(), JetOutputs[ijet].modp());
    float sumpt = 0;
    float avgZ = 0;
    std::vector<edm::Ptr<TrackingParticle>> tpPtrs;
    std::vector<fastjet::PseudoJet> fjConstituents = fastjet::sorted_by_pt(cs.constituents(JetOutputs[ijet]));

    for (unsigned int i = 0; i < fjConstituents.size(); ++i) {
      auto index = fjConstituents[i].user_index();
      edm::Ptr<TrackingParticle> tpPtr(TrackingParticleHandle, index);
      tpPtrs.push_back(tpPtr);  // tracking particles in the jet
      sumpt = sumpt + tpPtr->pt();
      avgZ = avgZ + tpPtr->pt() * tpPtr->z0();
    }
    avgZ = avgZ / sumpt;
    edm::Ref<JetBxCollection> jetRef;
    std::vector<edm::Ptr<TTTrack<Ref_Phase2TrackerDigi_>>> dummyL1TrackPtrs;  // can't create TkJet with tp references
    TkJet tpJet(jetP4, dummyL1TrackPtrs, avgZ, fjConstituents.size(), 0, 0, 0, false);
    TPFastJets->push_back(tpJet);
  }  //end loop over Jet Outputs

  iEvent.put(std::move(TPFastJets), "TPFastJets");
}

Member Data Documentation

coneSize_

const float TPFastJetProducer::coneSize_

private

Definition at line 74 of file TPFastJetProducer.cc.

Referenced by produce().

tpEtaMax_

const float TPFastJetProducer::tpEtaMax_

private

Definition at line 70 of file TPFastJetProducer.cc.

Referenced by produce().

tpNStubLayerMin_

const int TPFastJetProducer::tpNStubLayerMin_

private

Definition at line 73 of file TPFastJetProducer.cc.

Referenced by produce().

tpNStubMin_

const int TPFastJetProducer::tpNStubMin_

private

Definition at line 72 of file TPFastJetProducer.cc.

Referenced by produce().

tpPtMin_

const float TPFastJetProducer::tpPtMin_

private

Definition at line 69 of file TPFastJetProducer.cc.

Referenced by produce().

tpZMax_

const float TPFastJetProducer::tpZMax_

private

Definition at line 71 of file TPFastJetProducer.cc.

Referenced by produce().

trackingParticleToken_

edm::EDGetTokenT<std::vector<TrackingParticle> > TPFastJetProducer::trackingParticleToken_

private

Definition at line 76 of file TPFastJetProducer.cc.

Referenced by produce().

tTopoToken_

edm::ESGetToken<TrackerTopology, TrackerTopologyRcd> TPFastJetProducer::tTopoToken_

private

Definition at line 78 of file TPFastJetProducer.cc.

Referenced by produce().

ttStubMCTruthToken_

edm::EDGetTokenT<TTStubAssociationMap<Ref_Phase2TrackerDigi_> > TPFastJetProducer::ttStubMCTruthToken_

private

Definition at line 77 of file TPFastJetProducer.cc.

Referenced by produce().