HSCParticleProducer.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    HSCParticleProducer
// Class:      HSCParticleProducer
//
//
// Original Author:  Loic Quertenmont
//         Created:  Wed Oct 10 12:01:28 CEST 2007
//
//

// user include files
#include "SUSYBSMAnalysis/HSCP/plugins/HSCParticleProducer.h"

using namespace susybsm;
using namespace reco;

HSCParticleProducer::HSCParticleProducer(const edm::ParameterSet& iConfig) {
  using namespace edm;
  using namespace std;

  // the Act as Event filter
  Filter_ = iConfig.getParameter<bool>("filter");

  // the input collections
  m_trackToken = consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("tracks"));
  m_muonsToken = consumes<reco::MuonCollection>(iConfig.getParameter<edm::InputTag>("muons"));
  m_MTmuonsToken = consumes<reco::MuonCollection>(iConfig.getParameter<edm::InputTag>("MTmuons"));
  m_trackIsoToken = consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("tracksIsolation"));

  useBetaFromTk = iConfig.getParameter<bool>("useBetaFromTk");
  useBetaFromMuon = iConfig.getParameter<bool>("useBetaFromMuon");
  useBetaFromRpc = iConfig.getParameter<bool>("useBetaFromRpc");
  useBetaFromEcal = iConfig.getParameter<bool>("useBetaFromEcal");

  // the parameters
  minTkP = iConfig.getParameter<double>("minTkP");              // 30
  maxTkChi2 = iConfig.getParameter<double>("maxTkChi2");        // 5
  minTkHits = iConfig.getParameter<uint32_t>("minTkHits");      // 9
  minMuP = iConfig.getParameter<double>("minMuP");              // 30
  minDR = iConfig.getParameter<double>("minDR");                // 0.1
  minSAMuPt = iConfig.getParameter<double>("minSAMuPt");        // 70
  minMTMuPt = iConfig.getParameter<double>("minMTMuPt");        // 70
  minMTDR = iConfig.getParameter<double>("minMTDR");            // 0.3
  maxInvPtDiff = iConfig.getParameter<double>("maxInvPtDiff");  // 0.005

  if (useBetaFromTk)
    beta_calculator_TK = new BetaCalculatorTK(iConfig, consumesCollector());
  if (useBetaFromMuon)
    beta_calculator_MUON = new BetaCalculatorMUON(iConfig, consumesCollector());
  if (useBetaFromRpc)
    beta_calculator_RPC = new BetaCalculatorRPC(iConfig, consumesCollector());
  if (useBetaFromEcal)
    beta_calculator_ECAL = new BetaCalculatorECAL(iConfig, consumesCollector());

  // Load all the selections
  std::vector<edm::ParameterSet> SelectionParameters =
      iConfig.getParameter<std::vector<edm::ParameterSet> >("SelectionParameters");
  for (unsigned int i = 0; i < SelectionParameters.size(); i++) {
    Selectors.push_back(new CandidateSelector(SelectionParameters[i]));
  }

  // what I produce
  produces<susybsm::HSCParticleCollection>();
  if (useBetaFromEcal)
    produces<susybsm::HSCPCaloInfoCollection>();
}

HSCParticleProducer::~HSCParticleProducer() {
  // 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  ------------
bool HSCParticleProducer::filter(edm::Event& iEvent, const edm::EventSetup& iSetup) {
  using namespace edm;
  using namespace reco;
  using namespace std;
  using namespace susybsm;

  // information from the muons
  edm::Handle<reco::MuonCollection> muonCollectionHandle;
  iEvent.getByToken(m_muonsToken, muonCollectionHandle);

  //information from the mean timer muons
  edm::Handle<reco::MuonCollection> MTmuonCollectionHandle;
  iEvent.getByToken(m_MTmuonsToken, MTmuonCollectionHandle);

  // information from the tracks
  edm::Handle<reco::TrackCollection> trackCollectionHandle;
  iEvent.getByToken(m_trackToken, trackCollectionHandle);

  // information from the tracks iso
  edm::Handle<reco::TrackCollection> trackIsoCollectionHandle;
  iEvent.getByToken(m_trackIsoToken, trackIsoCollectionHandle);

  // creates the output collection
  susybsm::HSCParticleCollection* hscp = new susybsm::HSCParticleCollection;
  std::unique_ptr<susybsm::HSCParticleCollection> result(hscp);

  susybsm::HSCPCaloInfoCollection* caloInfoColl = new susybsm::HSCPCaloInfoCollection;
  std::unique_ptr<susybsm::HSCPCaloInfoCollection> caloInfoCollaptr(caloInfoColl);

  // Fill the output collection with HSCP Candidate (the candiate only contains ref to muon AND/OR track object)
  *hscp = getHSCPSeedCollection(trackCollectionHandle, muonCollectionHandle, MTmuonCollectionHandle);

  // find the track ref for isolation purposed (main track is supposed to be the Iso track after refitting)
  for (susybsm::HSCParticleCollection::iterator hscpcandidate = hscp->begin(); hscpcandidate != hscp->end();
       ++hscpcandidate) {
    // Matching is needed because input track collection and muon inner track may lightly differs due to track refit
    reco::TrackRef track = hscpcandidate->trackRef();
    if (track.isNull())
      continue;
    float dRMin = 1000;
    int found = -1;
    for (unsigned int t = 0; t < trackIsoCollectionHandle->size(); t++) {
      reco::TrackRef Isotrack = reco::TrackRef(trackIsoCollectionHandle, t);
      if (fabs((1.0 / track->pt()) - (1.0 / Isotrack->pt())) > maxInvPtDiff)
        continue;
      float dR = deltaR(track->momentum(), Isotrack->momentum());
      if (dR <= minDR && dR < dRMin) {
        dRMin = dR;
        found = t;
      }
    }
    if (found >= 0)
      hscpcandidate->setTrackIso(reco::TrackRef(trackIsoCollectionHandle, found));
  }

  // compute the TRACKER contribution
  if (useBetaFromTk) {
    for (susybsm::HSCParticleCollection::iterator hscpcandidate = hscp->begin(); hscpcandidate != hscp->end();
         ++hscpcandidate) {
      beta_calculator_TK->addInfoToCandidate(*hscpcandidate, iEvent, iSetup);
    }
  }

  // compute the MUON contribution
  if (useBetaFromMuon) {
    for (susybsm::HSCParticleCollection::iterator hscpcandidate = hscp->begin(); hscpcandidate != hscp->end();
         ++hscpcandidate) {
      beta_calculator_MUON->addInfoToCandidate(*hscpcandidate, iEvent, iSetup);
    }
  }

  // compute the RPC contribution
  if (useBetaFromRpc) {
    for (susybsm::HSCParticleCollection::iterator hscpcandidate = hscp->begin(); hscpcandidate != hscp->end();
         ++hscpcandidate) {
      beta_calculator_RPC->addInfoToCandidate(*hscpcandidate, iEvent, iSetup);
    }
  }

  // compute the ECAL contribution
  if (useBetaFromEcal) {
    int Index = 0;
    caloInfoColl->resize(hscp->size());
    for (susybsm::HSCParticleCollection::iterator hscpcandidate = hscp->begin(); hscpcandidate != hscp->end();
         ++hscpcandidate, Index++) {
      beta_calculator_ECAL->addInfoToCandidate(
          *hscpcandidate, trackCollectionHandle, iEvent, iSetup, (*caloInfoColl)[Index]);
    }
  }

  // cleanup the collection based on the input selection
  for (int i = 0; i < (int)hscp->size(); i++) {
    susybsm::HSCParticleCollection::iterator hscpcandidate = hscp->begin() + i;
    bool decision = false;
    for (unsigned int s = 0; s < Selectors.size(); s++) {
      decision |= Selectors[s]->isSelected(*hscpcandidate);
    }
    if (!decision) {
      hscp->erase(hscpcandidate);
      if (useBetaFromEcal)
        caloInfoColl->erase(caloInfoColl->begin() + i);
      i--;
    }
  }
  bool filterResult = !Filter_ || (Filter_ && !hscp->empty());

  // output result
  if (useBetaFromEcal) {
    edm::OrphanHandle<susybsm::HSCPCaloInfoCollection> caloInfoHandle = iEvent.put(std::move(caloInfoCollaptr));
    // adding the reftoCaloInfoObject to the HSCP Object
    for (int i = 0; i < (int)hscp->size(); i++) {
      susybsm::HSCParticleCollection::iterator hscpcandidate = hscp->begin() + i;
      hscpcandidate->setCaloInfo(HSCPCaloInfoRef(caloInfoHandle, i));
    }
  }

  // output result

  edm::OrphanHandle<susybsm::HSCParticleCollection> putHandle = iEvent.put(std::move(result));
  //  if(useBetaFromEcal){
  //      edm::RefProd<susybsm::HSCParticleCollection> hscpCollectionHandle = iEvent.getRefBeforePut<susybsm::HSCParticleCollection>();
  //    filler.insert(putHandle, CaloInfoColl.begin(), CaloInfoColl.end());
  //    filler.fill();
  //    iEvent.put(std::move(CaloInfoMap));
  //  }

  return filterResult;
}

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

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

std::vector<HSCParticle> HSCParticleProducer::getHSCPSeedCollection(
    edm::Handle<reco::TrackCollection>& trackCollectionHandle,
    edm::Handle<reco::MuonCollection>& muonCollectionHandle,
    edm::Handle<reco::MuonCollection>& MTmuonCollectionHandle) {
  std::vector<HSCParticle> HSCPCollection;

  // Store a local vector of track ref (that can be modified if matching)
  std::vector<reco::TrackRef> tracks;
  for (unsigned int i = 0; i < trackCollectionHandle->size(); i++) {
    TrackRef track = reco::TrackRef(trackCollectionHandle, i);

    //If track is from muon always keep it
    bool isMuon = false;
    for (unsigned int m = 0; m < muonCollectionHandle->size(); m++) {
      reco::MuonRef muon = reco::MuonRef(muonCollectionHandle, m);
      TrackRef innertrack = muon->innerTrack();
      if (innertrack.isNull())
        continue;
      if (fabs((1.0 / innertrack->pt()) - (1.0 / track->pt())) > maxInvPtDiff)
        continue;
      float dR = deltaR(innertrack->momentum(), track->momentum());
      if (dR <= minDR)
        isMuon = true;
    }

    if ((track->p() < minTkP || (track->chi2() / track->ndof()) > maxTkChi2 || track->found() < minTkHits) && !isMuon)
      continue;
    tracks.push_back(track);
  }

  // Loop on muons with inner track ref and create Muon HSCP Candidate
  for (unsigned int m = 0; m < muonCollectionHandle->size(); m++) {
    reco::MuonRef muon = reco::MuonRef(muonCollectionHandle, m);
    double SApt = -1;
    if (muon->isStandAloneMuon())
      SApt = muon->standAloneMuon()->pt();
    if (muon->p() < minMuP && SApt < minSAMuPt)
      continue;
    TrackRef innertrack = muon->innerTrack();
    if (innertrack.isNull())
      continue;

    // Check if the inner track match any track in order to create a Muon+Track HSCP Candidate
    // Matching is needed because input track collection and muon inner track may lightly differs due to track refit
    float dRMin = 1000;
    int found = -1;
    for (unsigned int t = 0; t < tracks.size(); t++) {
      reco::TrackRef track = tracks[t];
      if (fabs((1.0 / innertrack->pt()) - (1.0 / track->pt())) > maxInvPtDiff)
        continue;
      float dR = deltaR(innertrack->momentum(), track->momentum());
      if (dR <= minDR && dR < dRMin) {
        dRMin = dR;
        found = t;
      }
    }

    HSCParticle candidate;
    candidate.setMuon(muon);
    if (found >= 0) {
      //        printf("MUON with Inner Track Matching --> DR = %6.2f (%6.2f %+6.2f %+6.2f):(%6.2f %+6.2f %+6.2f) vs (%6.2f %+6.2f %+6.2f)\n",dRMin,muon->pt(), muon->eta(), muon->phi(), innertrack->pt(), innertrack->eta(), innertrack->phi(), tracks[found]->pt(), tracks[found]->eta(), tracks[found]->phi() );
      candidate.setTrack(tracks[found]);
      tracks.erase(tracks.begin() + found);
    }
    HSCPCollection.push_back(candidate);
  }

  // Loop on muons without inner tracks and create Muon HSCP Candidate
  for (unsigned int m = 0; m < muonCollectionHandle->size(); m++) {
    reco::MuonRef muon = reco::MuonRef(muonCollectionHandle, m);
    double SApt = -1;
    if (muon->isStandAloneMuon())
      SApt = muon->standAloneMuon()->pt();
    if (muon->p() < minMuP && SApt < minSAMuPt)
      continue;
    TrackRef innertrack = muon->innerTrack();
    if (innertrack.isNonnull())
      continue;

    // Check if the muon match any track in order to create a Muon+Track HSCP Candidate
    float dRMin = 1000;
    int found = -1;
    for (unsigned int t = 0; t < tracks.size(); t++) {
      reco::TrackRef track = tracks[t];
      if (fabs((1.0 / muon->pt()) - (1.0 / track->pt())) > maxInvPtDiff)
        continue;
      float dR = deltaR(muon->momentum(), track->momentum());
      if (dR <= minDR && dR < dRMin) {
        dRMin = dR;
        found = t;
      }
    }

    HSCParticle candidate;
    candidate.setMuon(muon);
    if (found >= 0) {
      //        printf("MUON without Inner Track Matching --> DR = %6.2f (%6.2f %+6.2f %+6.2f) vs (%6.2f %+6.2f %+6.2f)\n",dRMin,muon->pt(), muon->eta(), muon->phi(), tracks[found]->pt(), tracks[found]->eta(), tracks[found]->phi() );
      candidate.setTrack(tracks[found]);
      tracks.erase(tracks.begin() + found);
    }
    HSCPCollection.push_back(candidate);
  }

  //Loop on MT muons and add to collection
  for (unsigned int m = 0; m < MTmuonCollectionHandle->size(); m++) {
    reco::MuonRef MTmuon = reco::MuonRef(MTmuonCollectionHandle, m);
    if (MTmuon->pt() < minMTMuPt)
      continue;

    //Check if matches muon HSCP candidate and add reference
    float dRMin = 1000;
    int found = -1;
    for (unsigned int i = 0; i < HSCPCollection.size(); i++) {
      if (!HSCPCollection[i].hasMuonRef())
        continue;
      reco::MuonRef muon = HSCPCollection[i].muonRef();
      float dR = deltaR(muon->momentum(), MTmuon->momentum());
      if (dR <= minMTDR && dR < dRMin) {
        dRMin = dR;
        found = i;
      }
    }
    if (found > -1)
      HSCPCollection[found].setMTMuon(MTmuon);
    else {
      HSCParticle candidate;
      candidate.setMTMuon(MTmuon);
      HSCPCollection.push_back(candidate);
    }
  }

  // Loop on tracks not matching muon and create Track HSCP Candidate
  for (unsigned int i = 0; i < tracks.size(); i++) {
    HSCParticle candidate;
    candidate.setTrack(tracks[i]);
    HSCPCollection.push_back(candidate);
  }

  return HSCPCollection;
}

//define this as a plug-in
DEFINE_FWK_MODULE(HSCParticleProducer);