PhysicsObjectsMiraBelleEcmsBBModule.cc

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
#include <dqm/modules/PhysicsObjectsMiraBelle/PhysicsObjectsMiraBelleEcmsBBModule.h>
 
#include <analysis/dataobjects/ParticleList.h>
#include <analysis/variables/ContinuumSuppressionVariables.h>
#include <analysis/utility/PCmsLabTransform.h>
#include <mdst/dataobjects/PIDLikelihood.h>
#include <framework/particledb/EvtGenDatabasePDG.h>
#include <mdst/dataobjects/SoftwareTriggerResult.h>
#include <framework/datastore/StoreObjPtr.h>
#include <framework/datastore/StoreArray.h>
#include <analysis/dataobjects/ParticleList.h>
 
#include <TTree.h>
#include <TH1D.h>
 
 
using namespace Belle2;
 
 
 
//-----------------------------------------------------------------
//                 Register the Module
//-----------------------------------------------------------------
REG_MODULE(PhysicsObjectsMiraBelleEcmsBB);
 
//-----------------------------------------------------------------
//                 Implementation
//-----------------------------------------------------------------
 
PhysicsObjectsMiraBelleEcmsBBModule::PhysicsObjectsMiraBelleEcmsBBModule() : HistoModule()
{
  //Set module properties
  setDescription("Monitor of the CMS collision energy based on hadronic B decays");
  setPropertyFlags(c_ParallelProcessingCertified);
 
  addParam("TriggerIdentifier", m_triggerIdentifier,
           "Trigger identifier string used to select events for the histograms", std::string("software_trigger_cut&skim&accept_btocharm"));
  addParam("BmListName", m_BmListName, "Name of the B- particle list", std::string("B-:combined"));
  addParam("B0ListName", m_B0ListName, "Name of the B0 particle list", std::string("B0:combined"));
}
 
 
void PhysicsObjectsMiraBelleEcmsBBModule::initialize()
{
  REG_HISTOGRAM
}
 
void PhysicsObjectsMiraBelleEcmsBBModule::defineHisto()
{
  TDirectory* oldDir = gDirectory;
  oldDir->mkdir("PhysicsObjectsMiraBelleEcmsBB");
  oldDir->cd("PhysicsObjectsMiraBelleEcmsBB");
 
  const double cMBp = EvtGenDatabasePDG::Instance()->GetParticle("B+")->Mass();
 
  // In the original ML-based analysis there are 40 bins for visualisation
  // Since binned fit is used here, 80 bins are used instead.
  // For simplicity, the same histogram lower bound (B+ mass) is used
  m_hB0 = new TH1D("hB0", "", 80, cMBp, 5.37);
  m_hBp = new TH1D("hBp", "", 80, cMBp, 5.37);
  oldDir->cd();
}
 
void PhysicsObjectsMiraBelleEcmsBBModule::beginRun()
{
  if (m_hBp) m_hBp->Reset();
  if (m_hB0) m_hB0->Reset();
}
 
 
 
 
 
void PhysicsObjectsMiraBelleEcmsBBModule::event()
{
  StoreObjPtr<SoftwareTriggerResult> result;
  if (!result.isValid()) {
    B2WARNING("SoftwareTriggerResult object not available but needed to select events for the histograms.");
    return;
  }
 
  const bool accepted = (result->getResult(m_triggerIdentifier) == SoftwareTriggerCutResult::c_accept);
  if (accepted == false) return;
 
  StoreObjPtr<ParticleList> B0(m_B0ListName);
  StoreObjPtr<ParticleList> Bm(m_BmListName);
 
  //put all the B candidates into the vector
  std::vector<const Particle*> Bparts;
 
  if (B0.isValid()) {
    for (unsigned i = 0; i < B0->getListSize(); ++i)
      if (B0->getParticle(i))
        Bparts.push_back(B0->getParticle(i));
  }
  if (Bm.isValid()) {
    for (unsigned i = 0; i < Bm->getListSize(); ++i)
      if (Bm->getParticle(i))
        Bparts.push_back(Bm->getParticle(i));
  }
 
 
  if (Bparts.size() == 0) return;
 
  for (unsigned i = 0; i < Bparts.size(); ++i) {
    const Particle* Bpart = Bparts[i];
 
    const Particle* D    = nullptr;
    double dmDstar = std::numeric_limits<double>::quiet_NaN();
 
    static const int c_PdgD0    = abs(EvtGenDatabasePDG::Instance()->GetParticle("D0")->PdgCode());
    static const int c_PdgDplus = abs(EvtGenDatabasePDG::Instance()->GetParticle("D+")->PdgCode());
 
    static const int c_PdgDstar0    = abs(EvtGenDatabasePDG::Instance()->GetParticle("D*0")->PdgCode());
    static const int c_PdgDstarPlus = abs(EvtGenDatabasePDG::Instance()->GetParticle("D*+")->PdgCode());
 
    //if D0 or D+ meson
    if (abs(Bpart->getDaughter(0)->getPDGCode()) == c_PdgD0 || abs(Bpart->getDaughter(0)->getPDGCode()) == c_PdgDplus) {
      D = Bpart->getDaughter(0);
    } else if (abs(Bpart->getDaughter(0)->getPDGCode()) == c_PdgDstar0 || abs(Bpart->getDaughter(0)->getPDGCode()) == c_PdgDstarPlus) {
      const Particle* Dstar = Bpart->getDaughter(0);
      D = Dstar->getDaughter(0);
      dmDstar    = Dstar->getMass() - D->getMass();
    } else {
      B2INFO("No D meson found");
    }
    double mD = D ? D->getMass() : std::numeric_limits<double>::quiet_NaN();
 
    //Convert mBC and deltaE to the Y4S reference
    double pBcms  = PCmsLabTransform::labToCms(Bpart->get4Vector()).P();
    double mInv        = Bpart->getMass();
    double pdg       = Bpart->getPDGCode();
    double R2        = Variable::R2(Bpart);
 
    //get mass of B+- or B0
    double mB = EvtGenDatabasePDG::Instance()->GetParticle(abs(pdg))->Mass();
 
 
    // Filling the histograms
    if (c_mDmin < mD && mD < c_mDmax)
      if (abs(mInv - mB) < c_mBwindow)
        if (R2 < c_R2max)
          if (std::isnan(dmDstar) || (c_dmDstarMin < dmDstar && dmDstar < c_dmDstarMax)) {
            double eBC = sqrt(pBcms * pBcms + mB * mB); // beam constrained energy
            if (abs(pdg) == 511) {
              m_hB0->Fill(eBC);
            } else {
              m_hBp->Fill(eBC);
            }
          }
  }
}