L1TrackJetProducer Class Reference

Inheritance diagram for L1TrackJetProducer:

Public Types
typedef vector< L1TTTrackType >	L1TTTrackCollectionType
typedef edm::RefVector< L1TTTrackCollectionType >	L1TTTrackRefCollectionType
typedef TTTrack< Ref_Phase2TrackerDigi_ >	L1TTTrackType
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

Public Member Functions
	L1TrackJetProducer (const ParameterSet &)
	~L1TrackJetProducer () 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 (ConfigurationDescriptions &descriptions)

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

Private Attributes
const float	d0CutNStubs4_
const float	d0CutNStubs5_
const bool	displaced_
const int	etaBins_
float	etaStep_
const int	highpTJetMinTrackMultiplicity_
const float	highpTJetThreshold_
vector< Ptr< L1TTTrackType > >	L1TrkPtrs_
const int	lowpTJetMinTrackMultiplicity_
const float	lowpTJetThreshold_
const double	minTrkJetpT_
const int	nDisplacedTracks_
const int	phiBins_
float	phiStep_
const EDGetTokenT< L1TTTrackRefCollectionType >	trackToken_
const float	trkEtaMax_
const float	trkPtMax_
const float	trkZMax_
const int	zBins_
float	zStep_

Detailed Description

Definition at line 46 of file L1TrackJetProducer.cc.

Member Typedef Documentation

L1TTTrackCollectionType

typedef vector<L1TTTrackType> L1TrackJetProducer::L1TTTrackCollectionType

Definition at line 51 of file L1TrackJetProducer.cc.

L1TTTrackRefCollectionType

typedef edm::RefVector<L1TTTrackCollectionType> L1TrackJetProducer::L1TTTrackRefCollectionType

Definition at line 52 of file L1TrackJetProducer.cc.

L1TTTrackType

typedef TTTrack<Ref_Phase2TrackerDigi_> L1TrackJetProducer::L1TTTrackType

Definition at line 50 of file L1TrackJetProducer.cc.

Constructor & Destructor Documentation

L1TrackJetProducer()

L1TrackJetProducer::L1TrackJetProducer ( const ParameterSet &  iConfig )

explicit

Definition at line 83 of file L1TrackJetProducer.cc.

References displaced_, etaBins_, etaStep_, M_PI, phiBins_, phiStep_, trkEtaMax_, trkZMax_, zBins_, and zStep_.

Referenced by produce().

    : trkZMax_(iConfig.getParameter<double>("trk_zMax")),
      trkPtMax_(iConfig.getParameter<double>("trk_ptMax")),
      trkEtaMax_(iConfig.getParameter<double>("trk_etaMax")),
      lowpTJetMinTrackMultiplicity_(iConfig.getParameter<int>("lowpTJetMinTrackMultiplicity")),
      lowpTJetThreshold_(iConfig.getParameter<double>("lowpTJetThreshold")),
      highpTJetMinTrackMultiplicity_(iConfig.getParameter<int>("highpTJetMinTrackMultiplicity")),
      highpTJetThreshold_(iConfig.getParameter<double>("highpTJetThreshold")),
      zBins_(iConfig.getParameter<int>("zBins")),
      etaBins_(iConfig.getParameter<int>("etaBins")),
      phiBins_(iConfig.getParameter<int>("phiBins")),
      minTrkJetpT_(iConfig.getParameter<double>("minTrkJetpT")),
      displaced_(iConfig.getParameter<bool>("displaced")),
      d0CutNStubs4_(iConfig.getParameter<double>("d0_cutNStubs4")),
      d0CutNStubs5_(iConfig.getParameter<double>("d0_cutNStubs5")),
      nDisplacedTracks_(iConfig.getParameter<int>("nDisplacedTracks")),
      trackToken_(consumes<L1TTTrackRefCollectionType>(iConfig.getParameter<InputTag>("L1TrackInputTag"))) {
  zStep_ = 2.0 * trkZMax_ / (zBins_ + 1);  // added +1 in denom
  etaStep_ = 2.0 * trkEtaMax_ / etaBins_;  //etaStep is the width of an etabin
  phiStep_ = 2 * M_PI / phiBins_;          

  if (displaced_)
    produces<TkJetCollection>("L1TrackJetsExtended");
  else
    produces<TkJetCollection>("L1TrackJets");
}

~L1TrackJetProducer()

L1TrackJetProducer::~L1TrackJetProducer ( )

overridedefault

Member Function Documentation

fillDescriptions()

void L1TrackJetProducer::fillDescriptions ( ConfigurationDescriptions &  descriptions )

static

Definition at line 283 of file L1TrackJetProducer.cc.

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

                                                                                 {
  ParameterSetDescription desc;
  desc.add<edm::InputTag>(
      "L1TrackInputTag", edm::InputTag("l1tTrackVertexAssociationProducerForJets", "Level1TTTracksSelectedAssociated"));
  desc.add<double>("trk_zMax", 15.0);
  desc.add<double>("trk_ptMax", 200.0);
  desc.add<double>("trk_etaMax", 2.4);
  desc.add<double>("minTrkJetpT", -1.0);
  desc.add<int>("etaBins", 24);
  desc.add<int>("phiBins", 27);
  desc.add<int>("zBins", 1);
  desc.add<double>("d0_cutNStubs4", -1);
  desc.add<double>("d0_cutNStubs5", -1);
  desc.add<int>("lowpTJetMinTrackMultiplicity", 2);
  desc.add<double>("lowpTJetThreshold", 50.0);
  desc.add<int>("highpTJetMinTrackMultiplicity", 3);
  desc.add<double>("highpTJetThreshold", 100.0);
  desc.add<bool>("displaced", false);
  desc.add<int>("nDisplacedTracks", 2);
  descriptions.add("l1tTrackJets", desc);
}

produce()

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

overrideprivate

Definition at line 110 of file L1TrackJetProducer.cc.

References a, funct::abs(), b, l1ttrackjet::MaxZBin::clusters, filterCSVwithJSON::copy, funct::cos(), d0CutNStubs4_, d0CutNStubs5_, displaced_, l1ttrackjet::EtaPhiBin::eta, PVValHelper::eta, egammaIdentification::eta_max, egammaIdentification::eta_min, etaBins_, etaStep_, highpTJetMinTrackMultiplicity_, highpTJetThreshold_, l1ttrackjet::MaxZBin::ht, mps_fire::i, iEvent, createfilelist::int, dqmiolumiharvest::j, reco::btau::jetEta, reco::btau::jetPhi, reco::btau::jetPt, isotrackApplyRegressor::k, L1TrackJetProducer(), L1TrkPtrs_, lowpTJetMinTrackMultiplicity_, lowpTJetThreshold_, M_PI, minTrkJetpT_, eostools::move(), l1ttrackjet::MaxZBin::nclust, nDisplacedTracks_, vertices_cff::ntracks, l1ttrackjet::EtaPhiBin::ntracks, l1ttrackjet::EtaPhiBin::nxtracks, phi, l1ttrackjet::EtaPhiBin::phi, phiBins_, phiStep_, l1ttrackjet::EtaPhiBin::pTtot, funct::sin(), edm::RefVector< C, T, F >::size(), jetUpdater_cfi::sort, l1ttrackjet::EtaPhiBin::trackidx, trackToken_, trkEtaMax_, trkPtMax_, trkZMax_, l1ttrackjet::EtaPhiBin::used, l1ttrackjet::MaxZBin::zbincenter, zBins_, SiStripMonitorCluster_cfi::zmax, SiStripMonitorCluster_cfi::zmin, l1ttrackjet::MaxZBin::znum, and zStep_.

                                                                        {
  unique_ptr<TkJetCollection> L1TrackJetProducer(new TkJetCollection);

  // L1 tracks
  edm::Handle<L1TTTrackRefCollectionType> TTTrackHandle;
  iEvent.getByToken(trackToken_, TTTrackHandle);

  L1TrkPtrs_.clear();

  // track selection
  for (unsigned int this_l1track = 0; this_l1track < TTTrackHandle->size(); this_l1track++) {
    edm::Ptr<L1TTTrackType> trkPtr(TTTrackHandle, this_l1track);
    L1TrkPtrs_.push_back(trkPtr);
  }

  // if no tracks pass selection return empty containers
  if (L1TrkPtrs_.empty()) {
    if (displaced_)
      iEvent.put(std::move(L1TrackJetProducer), "L1TrackJetsExtended");
    else
      iEvent.put(std::move(L1TrackJetProducer), "L1TrackJets");
    return;
  }

  MaxZBin mzb;
  mzb.znum = -1;
  mzb.nclust = -1;
  mzb.ht = -1;

  // create 2D grid of eta phi bins
  EtaPhiBin epbins_default[phiBins_][etaBins_];
  float phi = -1.0 * M_PI;
  for (int i = 0; i < phiBins_; ++i) {
    float eta = -1.0 * trkEtaMax_;
    for (int j = 0; j < etaBins_; ++j) {
      epbins_default[i][j].phi = (phi + (phi + phiStep_)) / 2.0;
      epbins_default[i][j].eta = (eta + (eta + etaStep_)) / 2.0;
      eta += etaStep_;
    }  // for each etabin
    phi += phiStep_;
  }  // for each phibin (finished creating bins)

  // create z-bins (might be useful for displaced if we run w/o dz cut)
  std::vector<float> zmins, zmaxs;
  for (int zbin = 0; zbin < zBins_; zbin++) {
    zmins.push_back(-1.0 * trkZMax_ + zStep_ * zbin);
    zmaxs.push_back(-1.0 * trkZMax_ + zStep_ * zbin + zStep_ * 2);
  }

  // create vectors of clusters
  std::vector<std::vector<EtaPhiBin>> L1clusters;
  L1clusters.reserve(phiBins_);
  std::vector<EtaPhiBin> L2clusters;

  // default (empty) grid
  for (int i = 0; i < phiBins_; ++i) {
    for (int j = 0; j < etaBins_; ++j) {
      epbins_default[i][j].pTtot = 0;
      epbins_default[i][j].used = false;
      epbins_default[i][j].ntracks = 0;
      epbins_default[i][j].nxtracks = 0;
      epbins_default[i][j].trackidx.clear();
    }
  }

  for (unsigned int zbin = 0; zbin < zmins.size(); ++zbin) {
    // initialize grid
    float zmin = zmins[zbin];
    float zmax = zmaxs[zbin];
    EtaPhiBin epbins[phiBins_][etaBins_];
    std::copy(&epbins_default[0][0], &epbins_default[0][0] + phiBins_ * etaBins_, &epbins[0][0]);

    //clear cluster containers
    L1clusters.clear();
    L2clusters.clear();

    // fill grid with tracks
    for (unsigned int k = 0; k < L1TrkPtrs_.size(); ++k) {
      float trkZ = L1TrkPtrs_[k]->z0();
      if (zmax < trkZ)
        continue;
      if (zmin > trkZ)
        continue;
      if (zbin == 0 && zmin == trkZ)
        continue;
      float trkpt = L1TrkPtrs_[k]->momentum().perp();
      float trketa = L1TrkPtrs_[k]->momentum().eta();
      float trkphi = L1TrkPtrs_[k]->momentum().phi();
      float trkd0 = L1TrkPtrs_[k]->d0();
      int trknstubs = (int)L1TrkPtrs_[k]->getStubRefs().size();
      for (int i = 0; i < phiBins_; ++i) {
        for (int j = 0; j < etaBins_; ++j) {
          float eta_min = epbins[i][j].eta - etaStep_ / 2.0;  //eta min
          float eta_max = epbins[i][j].eta + etaStep_ / 2.0;  //eta max
          float phi_min = epbins[i][j].phi - phiStep_ / 2.0;  //phi min
          float phi_max = epbins[i][j].phi + phiStep_ / 2.0;  //phi max

          if ((trketa < eta_min) || (trketa > eta_max) || (trkphi < phi_min) || (trkphi > phi_max))
            continue;

          if (trkpt < trkPtMax_)
            epbins[i][j].pTtot += trkpt;
          else
            epbins[i][j].pTtot += trkPtMax_;
          if ((std::abs(trkd0) > d0CutNStubs5_ && trknstubs >= 5 && d0CutNStubs5_ >= 0) ||
              (trknstubs == 4 && std::abs(trkd0) > d0CutNStubs4_ && d0CutNStubs4_ >= 0))
            epbins[i][j].nxtracks += 1;
          epbins[i][j].trackidx.push_back(k);
          ++epbins[i][j].ntracks;
        }  // for each etabin
      }  // for each phibin
    }  //end loop over tracks

    // cluster tracks in eta (first layer) using grid
    for (int phibin = 0; phibin < phiBins_; ++phibin) {
      L1clusters.push_back(L1_clustering<EtaPhiBin, float, float, float>(epbins[phibin], etaBins_, etaStep_));
    }

    // second layer clustering in phi for using eta clusters
    L2clusters = L2_clustering<EtaPhiBin, float, float, float>(L1clusters, phiBins_, phiStep_, etaStep_);

    // sum all cluster pTs in this zbin to find max
    float sum_pt = 0;
    for (unsigned int k = 0; k < L2clusters.size(); ++k) {
      if (L2clusters[k].pTtot > lowpTJetThreshold_ && L2clusters[k].ntracks < lowpTJetMinTrackMultiplicity_)
        continue;
      if (L2clusters[k].pTtot > highpTJetThreshold_ && L2clusters[k].ntracks < highpTJetMinTrackMultiplicity_)
        continue;
      if (L2clusters[k].pTtot > minTrkJetpT_)
        sum_pt += L2clusters[k].pTtot;
    }

    if (sum_pt < mzb.ht)
      continue;

    // if ht is larger than previous max, this is the new vertex zbin
    mzb.ht = sum_pt;
    mzb.znum = zbin;
    mzb.clusters = L2clusters;
    mzb.nclust = L2clusters.size();
    mzb.zbincenter = (zmin + zmax) / 2.0;
  }  //zbin loop

  // output
  vector<Ptr<L1TTTrackType>> L1TrackAssocJet;
  for (unsigned int j = 0; j < mzb.clusters.size(); ++j) {
    float jetEta = mzb.clusters[j].eta;
    float jetPhi = mzb.clusters[j].phi;
    float jetPt = mzb.clusters[j].pTtot;
    float jetPx = jetPt * cos(jetPhi);
    float jetPy = jetPt * sin(jetPhi);
    float jetPz = jetPt * sinh(jetEta);
    float jetP = jetPt * cosh(jetEta);
    int totalDisptrk = mzb.clusters[j].nxtracks;
    bool isDispJet = (totalDisptrk >= nDisplacedTracks_);

    math::XYZTLorentzVector jetP4(jetPx, jetPy, jetPz, jetP);
    L1TrackAssocJet.clear();
    for (unsigned int itrk = 0; itrk < mzb.clusters[j].trackidx.size(); itrk++)
      L1TrackAssocJet.push_back(L1TrkPtrs_[mzb.clusters[j].trackidx[itrk]]);

    TkJet trkJet(jetP4, L1TrackAssocJet, mzb.zbincenter, mzb.clusters[j].ntracks, 0, totalDisptrk, 0, isDispJet);

    L1TrackJetProducer->push_back(trkJet);
  }

  std::sort(L1TrackJetProducer->begin(), L1TrackJetProducer->end(), [](auto &a, auto &b) { return a.pt() > b.pt(); });
  if (displaced_)
    iEvent.put(std::move(L1TrackJetProducer), "L1TrackJetsExtended");
  else
    iEvent.put(std::move(L1TrackJetProducer), "L1TrackJets");
}

Member Data Documentation

d0CutNStubs4_

const float L1TrackJetProducer::d0CutNStubs4_

private

Definition at line 76 of file L1TrackJetProducer.cc.

Referenced by produce().

d0CutNStubs5_

const float L1TrackJetProducer::d0CutNStubs5_

private

Definition at line 77 of file L1TrackJetProducer.cc.

Referenced by produce().

displaced_

const bool L1TrackJetProducer::displaced_

private

Definition at line 75 of file L1TrackJetProducer.cc.

Referenced by L1TrackJetProducer(), and produce().

etaBins_

const int L1TrackJetProducer::etaBins_

private

Definition at line 69 of file L1TrackJetProducer.cc.

Referenced by L1TrackJetProducer(), and produce().

etaStep_

float L1TrackJetProducer::etaStep_

private

Definition at line 73 of file L1TrackJetProducer.cc.

Referenced by L1TrackJetProducer(), and produce().

highpTJetMinTrackMultiplicity_

const int L1TrackJetProducer::highpTJetMinTrackMultiplicity_

private

Definition at line 66 of file L1TrackJetProducer.cc.

Referenced by produce().

highpTJetThreshold_

const float L1TrackJetProducer::highpTJetThreshold_

private

Definition at line 67 of file L1TrackJetProducer.cc.

Referenced by produce().

L1TrkPtrs_

vector<Ptr<L1TTTrackType> > L1TrackJetProducer::L1TrkPtrs_

private

Definition at line 60 of file L1TrackJetProducer.cc.

Referenced by produce().

lowpTJetMinTrackMultiplicity_

const int L1TrackJetProducer::lowpTJetMinTrackMultiplicity_

private

Definition at line 64 of file L1TrackJetProducer.cc.

Referenced by produce().

lowpTJetThreshold_

const float L1TrackJetProducer::lowpTJetThreshold_

private

Definition at line 65 of file L1TrackJetProducer.cc.

Referenced by produce().

minTrkJetpT_

const double L1TrackJetProducer::minTrkJetpT_

private

Definition at line 71 of file L1TrackJetProducer.cc.

Referenced by produce().

nDisplacedTracks_

const int L1TrackJetProducer::nDisplacedTracks_

private

Definition at line 78 of file L1TrackJetProducer.cc.

Referenced by produce().

phiBins_

const int L1TrackJetProducer::phiBins_

private

Definition at line 70 of file L1TrackJetProducer.cc.

Referenced by L1TrackJetProducer(), and produce().

phiStep_

float L1TrackJetProducer::phiStep_

private

Definition at line 74 of file L1TrackJetProducer.cc.

Referenced by L1TrackJetProducer(), and produce().

trackToken_

const EDGetTokenT<L1TTTrackRefCollectionType> L1TrackJetProducer::trackToken_

private

Definition at line 80 of file L1TrackJetProducer.cc.

Referenced by produce().

trkEtaMax_

const float L1TrackJetProducer::trkEtaMax_

private

Definition at line 63 of file L1TrackJetProducer.cc.

Referenced by L1TrackJetProducer(), and produce().

trkPtMax_

const float L1TrackJetProducer::trkPtMax_

private

Definition at line 62 of file L1TrackJetProducer.cc.

Referenced by produce().

trkZMax_

const float L1TrackJetProducer::trkZMax_

private

Definition at line 61 of file L1TrackJetProducer.cc.

Referenced by L1TrackJetProducer(), and produce().

zBins_

const int L1TrackJetProducer::zBins_

private

Definition at line 68 of file L1TrackJetProducer.cc.

Referenced by L1TrackJetProducer(), and produce().

zStep_

float L1TrackJetProducer::zStep_

private

Definition at line 72 of file L1TrackJetProducer.cc.

Referenced by L1TrackJetProducer(), and produce().