KlongCalculatorUtils.h

/**************************************************************************
 * 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.                  *
 **************************************************************************/
 
#pragma once
 
// dataobjects
#include <analysis/dataobjects/Particle.h>
#include <mdst/dataobjects/KLMCluster.h>
#include <mdst/dataobjects/ECLCluster.h>
#include <analysis/ClusterUtility/ClusterUtils.h>
 
// ROOT
#include <Math/Vector4D.h>
 
#include <cmath>
#include <vector>
 
namespace Belle2 {
  struct KlongCalculatorUtils {
    static bool calculateBtoKlongX(ROOT::Math::PxPyPzEVector& BMomentum,
                                   ROOT::Math::PxPyPzEVector& KMomentum,
                                   const std::vector<Particle*> daughters,
                                   const double m_b,
                                   int& idx)
    {
 
      bool k_check = false;
      ROOT::Math::PxPyPzEVector klDaughters; // 4-vector of K_L
      ROOT::Math::PxPyPzEVector pDaughters; // 4-vector of other daughters
      double m_j = 0;
      for (auto daughter : daughters) {
        if (daughter->getPDGCode() == Const::Klong.getPDGCode()) {
          if (k_check)
            B2FATAL("More than one K_L is detected! This tool accepts only one K_L in the final state.");
 
          const Belle2::KLMCluster* klm_cluster = daughter->getKLMCluster();
          if (klm_cluster)
            klDaughters = klm_cluster->getMomentum();
          else {
            const Belle2::ECLCluster* ecl_cluster = daughter->getECLCluster();
            if (ecl_cluster) {
              ClusterUtils clutls;
              klDaughters = clutls.Get4MomentumFromCluster(ecl_cluster, ECLCluster::EHypothesisBit::c_neutralHadron);
            } else {
              B2ERROR("K_L candidate must have related KLM or ECL cluster!");
            }
          }
          k_check = true;
        } else {
          pDaughters += daughter->get4Vector();
 
          // if there is only two daughter (e.g. K_L J/psi), use the PDG mass of the other daughter.
          m_j = daughter->getPDGMass();
          idx = daughter->getArrayIndex() + idx * 100;
        }
      }
 
      if (!k_check)
        B2FATAL("This tool is meant to reconstruct decays with a K_L0 in the final state. There is no K_L0 in this decay!");
 
      // if there are more than two daughters, use the invariant mass of sum of the other daughters.
      if (daughters.size() == 3) {
        m_j = pDaughters.M();
      }
 
      const double m_k2 = Const::Klong.getMass() * Const::Klong.getMass();
 
      const double s_p = (klDaughters.Vect().Unit()).Dot(pDaughters.Vect());
      const double m_sum = (m_b * m_b) - (m_j * m_j) - m_k2;
 
      const double s_p2 = s_p * s_p;
      const double m_sum2 = m_sum * m_sum;
      const double s_pm = s_p * m_sum;
      const double e_j2 = pDaughters.E() * pDaughters.E();
 
      const double k_mag1 = (s_pm + std::sqrt(s_p2 * m_sum2 - 4 * (e_j2 - s_p2) * (e_j2 * m_k2 - m_sum2 / 4))) / (2 *
                            (e_j2 - s_p2));
      const double k_mag2 = (s_pm - std::sqrt(s_p2 * m_sum2 - 4 * (e_j2 - s_p2) * (e_j2 * m_k2 - m_sum2 / 4))) / (2 *
                            (e_j2 - s_p2));
 
 
      ROOT::Math::PxPyPzEVector missDaughters;
 
      if (k_mag1 > 0)
        missDaughters = k_mag1 * klDaughters / klDaughters.P();
      else
        missDaughters = k_mag2 * klDaughters / klDaughters.P();
 
      missDaughters.SetE(std::sqrt(m_k2 + missDaughters.P2()));
 
      if (std::isnan(missDaughters.P()))
        return false;
 
      ROOT::Math::PxPyPzEVector mom = pDaughters + missDaughters;
      mom.SetE(std::sqrt(m_b * m_b + mom.P2()));
      if (std::isnan(mom.P()))
        return false;
 
      KMomentum = missDaughters;
      BMomentum = mom;
 
      return true;
    };
 
  };
}