BPHOniaToMuMuBuilder.cc

Go to the documentation of this file.

/*
 *  See header file for a description of this class.
 *
 *  \author Paolo Ronchese INFN Padova
 *
 */

//-----------------------
// This Class' Header --
//-----------------------
#include "HeavyFlavorAnalysis/SpecificDecay/interface/BPHOniaToMuMuBuilder.h"

//-------------------------------
// Collaborating Class Headers --
//-------------------------------
#include "HeavyFlavorAnalysis/RecoDecay/interface/BPHRecoBuilder.h"
#include "HeavyFlavorAnalysis/RecoDecay/interface/BPHRecoSelect.h"
#include "HeavyFlavorAnalysis/RecoDecay/interface/BPHPlusMinusCandidate.h"
#include "HeavyFlavorAnalysis/SpecificDecay/interface/BPHMuonPtSelect.h"
#include "HeavyFlavorAnalysis/SpecificDecay/interface/BPHMuonEtaSelect.h"
#include "HeavyFlavorAnalysis/SpecificDecay/interface/BPHMassSelect.h"
#include "HeavyFlavorAnalysis/SpecificDecay/interface/BPHChi2Select.h"
#include "HeavyFlavorAnalysis/RecoDecay/interface/BPHMultiSelect.h"
#include "HeavyFlavorAnalysis/SpecificDecay/interface/BPHParticleMasses.h"

//---------------
// C++ Headers --
//---------------
using namespace std;

//-------------------
// Initializations --
//-------------------

//----------------
// Constructors --
//----------------
BPHOniaToMuMuBuilder::BPHOniaToMuMuBuilder(const edm::EventSetup& es,
                                           const BPHRecoBuilder::BPHGenericCollection* muPosCollection,
                                           const BPHRecoBuilder::BPHGenericCollection* muNegCollection)
    : muPosName("MuPos"),
      muNegName("MuNeg"),
      evSetup(&es),
      posCollection(muPosCollection),
      negCollection(muNegCollection) {
  setParameters(Phi, 3.0, 10.0, 0.50, 1.50, 0.0, BPHParticleMasses::phiMass, BPHParticleMasses::phiMWidth);
  setParameters(Psi1, 3.0, 10.0, 2.00, 3.40, 0.0, BPHParticleMasses::jPsiMass, BPHParticleMasses::jPsiMWidth);
  setParameters(Psi2, 3.0, 10.0, 3.40, 6.00, 0.0, BPHParticleMasses::psi2Mass, BPHParticleMasses::psi2MWidth);
  setParameters(Ups, 3.0, 10.0, 6.00, 12.00, 0.0, -1.0, 0.0);
  setParameters(Ups1, 3.0, 10.0, 6.00, 9.75, 0.0, BPHParticleMasses::ups1Mass, BPHParticleMasses::ups1MWidth);
  setParameters(Ups2, 3.0, 10.0, 9.75, 10.20, 0.0, BPHParticleMasses::ups2Mass, BPHParticleMasses::ups2MWidth);
  setParameters(Ups3, 3.0, 10.0, 10.20, 12.00, 0.0, BPHParticleMasses::ups3Mass, BPHParticleMasses::ups3MWidth);
  updated = false;
}

//--------------
// Destructor --
//--------------
BPHOniaToMuMuBuilder::~BPHOniaToMuMuBuilder() {
  map<oniaType, OniaParameters>::iterator iter = oniaPar.begin();
  map<oniaType, OniaParameters>::iterator iend = oniaPar.end();
  while (iter != iend) {
    OniaParameters& par = iter++->second;
    delete par.ptSel;
    delete par.etaSel;
    delete par.massSel;
    delete par.chi2Sel;
  }
}

//--------------
// Operations --
//--------------
vector<BPHPlusMinusConstCandPtr> BPHOniaToMuMuBuilder::build() {
  if (updated)
    return fullList;

  BPHMultiSelect<BPHRecoSelect> ptSel(BPHSelectOperation::or_mode);
  BPHMultiSelect<BPHRecoSelect> etaSel(BPHSelectOperation::or_mode);
  BPHMultiSelect<BPHMomentumSelect> mSel(BPHSelectOperation::or_mode);
  BPHMultiSelect<BPHVertexSelect> vSel(BPHSelectOperation::or_mode);

  map<oniaType, OniaParameters>::iterator iter = oniaPar.begin();
  map<oniaType, OniaParameters>::iterator iend = oniaPar.end();
  while (iter != iend) {
    OniaParameters& par = iter++->second;
    ptSel.include(*par.ptSel);
    etaSel.include(*par.etaSel);
    mSel.include(*par.massSel);
    vSel.include(*par.chi2Sel);
  }

  BPHRecoBuilder bOnia(*evSetup);
  bOnia.add(muPosName, posCollection, BPHParticleMasses::muonMass, BPHParticleMasses::muonMSigma);
  bOnia.add(muNegName, negCollection, BPHParticleMasses::muonMass, BPHParticleMasses::muonMSigma);
  bOnia.filter(muPosName, ptSel);
  bOnia.filter(muNegName, ptSel);
  bOnia.filter(muPosName, etaSel);
  bOnia.filter(muNegName, etaSel);
  bOnia.filter(mSel);
  bOnia.filter(vSel);

  fullList = BPHPlusMinusCandidate::build(bOnia, muPosName, muNegName);
  updated = true;
  return fullList;
}

vector<BPHPlusMinusConstCandPtr> BPHOniaToMuMuBuilder::getList(
    oniaType type, BPHRecoSelect* dSel, BPHMomentumSelect* mSel, BPHVertexSelect* vSel, BPHFitSelect* kSel) {
  extractList(type);
  vector<BPHPlusMinusConstCandPtr>& list = oniaList[type];
  int i;
  int n = list.size();
  vector<BPHPlusMinusConstCandPtr> lsub;
  lsub.reserve(n);
  for (i = 0; i < n; ++i) {
    BPHPlusMinusConstCandPtr ptr = list[i];
    const reco::Candidate* muPos = ptr->originalReco(ptr->getDaug(muPosName));
    const reco::Candidate* muNeg = ptr->originalReco(ptr->getDaug(muNegName));
    if ((dSel != nullptr) && (!dSel->accept(*muPos)))
      continue;
    if ((dSel != nullptr) && (!dSel->accept(*muNeg)))
      continue;
    if ((mSel != nullptr) && (!mSel->accept(*ptr)))
      continue;
    if ((vSel != nullptr) && (!vSel->accept(*ptr)))
      continue;
    if ((kSel != nullptr) && (!kSel->accept(*ptr)))
      continue;
    lsub.push_back(list[i]);
  }
  return lsub;
}

BPHPlusMinusConstCandPtr BPHOniaToMuMuBuilder::getOriginalCandidate(const BPHRecoCandidate& cand) {
  const reco::Candidate* mp = cand.originalReco(cand.getDaug(muPosName));
  const reco::Candidate* mn = cand.originalReco(cand.getDaug(muNegName));
  int nc = fullList.size();
  int ic;
  for (ic = 0; ic < nc; ++ic) {
    BPHPlusMinusConstCandPtr pmp = fullList[ic];
    const BPHPlusMinusCandidate* pmc = pmp.get();
    if (pmc->originalReco(pmc->getDaug(muPosName)) != mp)
      continue;
    if (pmc->originalReco(pmc->getDaug(muNegName)) != mn)
      continue;
    return pmp;
  }
  return BPHPlusMinusConstCandPtr(nullptr);
}

void BPHOniaToMuMuBuilder::setPtMin(oniaType type, double pt) {
  setNotUpdated();
  OniaParameters& par = oniaPar[type];
  par.ptSel->setPtMin(pt);
  return;
}

void BPHOniaToMuMuBuilder::setEtaMax(oniaType type, double eta) {
  setNotUpdated();
  OniaParameters& par = oniaPar[type];
  par.etaSel->setEtaMax(eta);
  return;
}

void BPHOniaToMuMuBuilder::setMassMin(oniaType type, double m) {
  setNotUpdated();
  OniaParameters& par = oniaPar[type];
  par.massSel->setMassMin(m);
  return;
}

void BPHOniaToMuMuBuilder::setMassMax(oniaType type, double m) {
  setNotUpdated();
  OniaParameters& par = oniaPar[type];
  par.massSel->setMassMax(m);
  return;
}

void BPHOniaToMuMuBuilder::setProbMin(oniaType type, double p) {
  setNotUpdated();
  OniaParameters& par = oniaPar[type];
  par.chi2Sel->setProbMin(p);
  return;
}

void BPHOniaToMuMuBuilder::setConstr(oniaType type, double mass, double sigma) {
  setNotUpdated();
  OniaParameters& par = oniaPar[type];
  par.mass = mass;
  par.sigma = sigma;
  return;
}

double BPHOniaToMuMuBuilder::getPtMin(oniaType type) const {
  const OniaParameters& par = oniaPar.at(type);
  return par.ptSel->getPtMin();
}

double BPHOniaToMuMuBuilder::getEtaMax(oniaType type) const {
  const OniaParameters& par = oniaPar.at(type);
  return par.etaSel->getEtaMax();
}

double BPHOniaToMuMuBuilder::getMassMin(oniaType type) const {
  const OniaParameters& par = oniaPar.at(type);
  return par.massSel->getMassMin();
}

double BPHOniaToMuMuBuilder::getMassMax(oniaType type) const {
  const OniaParameters& par = oniaPar.at(type);
  return par.massSel->getMassMax();
}

double BPHOniaToMuMuBuilder::getProbMin(oniaType type) const {
  const OniaParameters& par = oniaPar.at(type);
  return par.chi2Sel->getProbMin();
}

double BPHOniaToMuMuBuilder::getConstrMass(oniaType type) const {
  const OniaParameters& par = oniaPar.at(type);
  return par.mass;
}

double BPHOniaToMuMuBuilder::getConstrSigma(oniaType type) const {
  const OniaParameters& par = oniaPar.at(type);
  return par.sigma;
}

void BPHOniaToMuMuBuilder::setNotUpdated() {
  map<oniaType, OniaParameters>::iterator iter = oniaPar.begin();
  map<oniaType, OniaParameters>::iterator iend = oniaPar.end();
  while (iter != iend)
    iter++->second.updated = false;
  updated = false;
  return;
}

void BPHOniaToMuMuBuilder::setParameters(oniaType type,
                                         double ptMin,
                                         double etaMax,
                                         double massMin,
                                         double massMax,
                                         double probMin,
                                         double mass,
                                         double sigma) {
  OniaParameters& par = oniaPar[type];
  par.ptSel = new BPHMuonPtSelect(ptMin);
  par.etaSel = new BPHMuonEtaSelect(etaMax);
  par.massSel = new BPHMassSelect(massMin, massMax);
  par.chi2Sel = new BPHChi2Select(probMin);
  par.mass = mass;
  par.sigma = sigma;
  par.updated = false;
  return;
}

void BPHOniaToMuMuBuilder::extractList(oniaType type) {
  if (!updated)
    build();
  OniaParameters& par = oniaPar[type];
  vector<BPHPlusMinusConstCandPtr>& list = oniaList[type];
  if (par.updated)
    return;
  int i;
  int n = fullList.size();
  list.clear();
  list.reserve(n);
  for (i = 0; i < n; ++i) {
    BPHPlusMinusConstCandPtr ptr = fullList[i];
    const reco::Candidate* mcPos = ptr->getDaug("MuPos");
    const reco::Candidate* mcNeg = ptr->getDaug("MuNeg");
    const reco::Candidate* muPos = ptr->originalReco(mcPos);
    const reco::Candidate* muNeg = ptr->originalReco(mcNeg);
    if (!par.massSel->accept(*ptr))
      continue;
    if (!par.ptSel->accept(*muPos))
      continue;
    if (!par.etaSel->accept(*muPos))
      continue;
    if (!par.ptSel->accept(*muNeg))
      continue;
    if (!par.etaSel->accept(*muNeg))
      continue;
    if (!par.chi2Sel->accept(*ptr))
      continue;
    BPHPlusMinusCandidate* np = new BPHPlusMinusCandidate(evSetup);
    np->add("MuPos", muPos, ptr->getTrackSearchList(mcPos), BPHParticleMasses::muonMass);
    np->add("MuNeg", muNeg, ptr->getTrackSearchList(mcNeg), BPHParticleMasses::muonMass);
    if (par.mass > 0.0)
      np->setConstraint(par.mass, par.sigma);
    list.push_back(BPHPlusMinusConstCandPtr(np));
  }
  par.updated = true;
  return;
}