BPHBuToJPsiKBuilder.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/BPHBuToJPsiKBuilder.h"

//-------------------------------
// Collaborating Class Headers --
//-------------------------------
#include "HeavyFlavorAnalysis/RecoDecay/interface/BPHRecoBuilder.h"
#include "HeavyFlavorAnalysis/RecoDecay/interface/BPHPlusMinusCandidate.h"
#include "HeavyFlavorAnalysis/RecoDecay/interface/BPHRecoCandidate.h"
#include "HeavyFlavorAnalysis/SpecificDecay/interface/BPHParticleNeutralVeto.h"
#include "HeavyFlavorAnalysis/SpecificDecay/interface/BPHParticlePtSelect.h"
#include "HeavyFlavorAnalysis/SpecificDecay/interface/BPHParticleEtaSelect.h"
#include "HeavyFlavorAnalysis/SpecificDecay/interface/BPHMassSelect.h"
#include "HeavyFlavorAnalysis/SpecificDecay/interface/BPHChi2Select.h"
#include "HeavyFlavorAnalysis/SpecificDecay/interface/BPHMassFitSelect.h"
#include "HeavyFlavorAnalysis/SpecificDecay/interface/BPHParticleMasses.h"

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

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

//----------------
// Constructors --
//----------------
BPHBuToJPsiKBuilder::BPHBuToJPsiKBuilder(const edm::EventSetup& es,
                                         const std::vector<BPHPlusMinusConstCandPtr>& jpsiCollection,
                                         const BPHRecoBuilder::BPHGenericCollection* kaonCollection)
    : jPsiName("JPsi"), kaonName("Kaon"), evSetup(&es), jCollection(&jpsiCollection), kCollection(kaonCollection) {
  jpsiSel = new BPHMassSelect(2.80, 3.40);
  knVeto = new BPHParticleNeutralVeto;
  ptSel = new BPHParticlePtSelect(0.7);
  etaSel = new BPHParticleEtaSelect(10.0);
  massSel = new BPHMassSelect(3.50, 8.00);
  chi2Sel = new BPHChi2Select(0.02);
  mFitSel = new BPHMassFitSelect(jPsiName, BPHParticleMasses::jPsiMass, BPHParticleMasses::jPsiMWidth, 5.00, 6.00);
  massConstr = true;
  minPDiff = 1.0e-4;
  updated = false;
}

//--------------
// Destructor --
//--------------
BPHBuToJPsiKBuilder::~BPHBuToJPsiKBuilder() {
  delete jpsiSel;
  delete knVeto;
  delete ptSel;
  delete etaSel;
  delete massSel;
  delete chi2Sel;
  delete mFitSel;
}

//--------------
// Operations --
//--------------
vector<BPHRecoConstCandPtr> BPHBuToJPsiKBuilder::build() {
  if (updated)
    return buList;

  BPHRecoBuilder bBu(*evSetup);
  bBu.setMinPDiffererence(minPDiff);
  bBu.add(jPsiName, *jCollection);
  bBu.add(kaonName, kCollection, BPHParticleMasses::kaonMass, BPHParticleMasses::kaonMSigma);
  bBu.filter(jPsiName, *jpsiSel);
  bBu.filter(kaonName, *knVeto);
  bBu.filter(kaonName, *ptSel);
  bBu.filter(kaonName, *etaSel);

  bBu.filter(*massSel);
  bBu.filter(*chi2Sel);
  if (massConstr)
    bBu.filter(*mFitSel);

  buList = BPHRecoCandidate::build(bBu);
  //
  //  Apply kinematic constraint on the JPsi mass.
  //  The operation is already performed when apply the mass selection,
  //  so it's not repeated. The following code is left as example
  //  for similar operations
  //
  //  int iBu;
  //  int nBu = ( massConstr ? buList.size() : 0 );
  //  for ( iBu = 0; iBu < nBu; ++iBu ) {
  //    BPHRecoCandidate* cptr( const_cast<BPHRecoCandidate*>(
  //                            buList[iBu].get() ) );
  //    BPHRecoConstCandPtr jpsi = cptr->getComp( jPsiName );
  //    double jMass = jpsi->constrMass();
  //    if ( jMass < 0 ) continue;
  //    double sigma = jpsi->constrSigma();
  //    cptr->kinematicTree( jPsiName, jMass, sigma );
  //  }
  updated = true;
  return buList;
}

void BPHBuToJPsiKBuilder::setJPsiMassMin(double m) {
  updated = false;
  jpsiSel->setMassMin(m);
  return;
}

void BPHBuToJPsiKBuilder::setJPsiMassMax(double m) {
  updated = false;
  jpsiSel->setMassMax(m);
  return;
}

void BPHBuToJPsiKBuilder::setKPtMin(double pt) {
  updated = false;
  ptSel->setPtMin(pt);
  return;
}

void BPHBuToJPsiKBuilder::setKEtaMax(double eta) {
  updated = false;
  etaSel->setEtaMax(eta);
  return;
}

void BPHBuToJPsiKBuilder::setMassMin(double m) {
  updated = false;
  massSel->setMassMin(m);
  return;
}

void BPHBuToJPsiKBuilder::setMassMax(double m) {
  updated = false;
  massSel->setMassMax(m);
  return;
}

void BPHBuToJPsiKBuilder::setProbMin(double p) {
  updated = false;
  chi2Sel->setProbMin(p);
  return;
}

void BPHBuToJPsiKBuilder::setMassFitMin(double m) {
  updated = false;
  mFitSel->setMassMin(m);
  return;
}

void BPHBuToJPsiKBuilder::setMassFitMax(double m) {
  updated = false;
  mFitSel->setMassMax(m);
  return;
}

void BPHBuToJPsiKBuilder::setConstr(bool flag) {
  updated = false;
  massConstr = flag;
  return;
}

double BPHBuToJPsiKBuilder::getJPsiMassMin() const { return jpsiSel->getMassMin(); }

double BPHBuToJPsiKBuilder::getJPsiMassMax() const { return jpsiSel->getMassMax(); }

double BPHBuToJPsiKBuilder::getKPtMin() const { return ptSel->getPtMin(); }

double BPHBuToJPsiKBuilder::getKEtaMax() const { return etaSel->getEtaMax(); }

double BPHBuToJPsiKBuilder::getMassMin() const { return massSel->getMassMin(); }

double BPHBuToJPsiKBuilder::getMassMax() const { return massSel->getMassMax(); }

double BPHBuToJPsiKBuilder::getProbMin() const { return chi2Sel->getProbMin(); }

double BPHBuToJPsiKBuilder::getMassFitMin() const { return mFitSel->getMassMin(); }

double BPHBuToJPsiKBuilder::getMassFitMax() const { return mFitSel->getMassMax(); }

bool BPHBuToJPsiKBuilder::getConstr() const { return massConstr; }