ECLCluster.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.                  *
 **************************************************************************/
 
/* Own header. */
#include <mdst/dataobjects/ECLCluster.h>
 
/* Basf2 headers. */
#include <framework/logging/Logger.h>
 
/* ROOT headers. */
#include <TMatrixD.h>
 
/* C++ headers. */
#include <bitset>
 
using namespace Belle2;
 
double ECLCluster::getEnergy(ECLCluster::EHypothesisBit hypothesis) const
{
  // check if cluster has the requested hypothesis
  if (!hasHypothesis(hypothesis)) {
    B2ERROR("This cluster does not support the requested hypothesis,"
            << LogVar(" it has the nPhotons hypothesis (y/n = 1/0): ", hasHypothesis(ECLCluster::EHypothesisBit::c_nPhotons))
            << LogVar(" and it has the neutralHadron hypothesis (y/n = 1/0): ", hasHypothesis(ECLCluster::EHypothesisBit::c_neutralHadron))
            << LogVar(" You requested the hypothesis bitmask: ", std::bitset<16>(static_cast<unsigned short>(hypothesis)).to_string()));
    return std::numeric_limits<double>::quiet_NaN();
  }
 
  // return the sensible answers
  if (hypothesis == ECLCluster::EHypothesisBit::c_nPhotons) return exp(m_logEnergy);
  else if (hypothesis == ECLCluster::EHypothesisBit::c_neutralHadron) return exp(m_logEnergyRaw);
  else {
    // throw error if the cluster is not supported
    B2ERROR("EHypothesisBit is not supported yet: " << static_cast<unsigned short>(hypothesis));
    return std::numeric_limits<double>::quiet_NaN();
  };
}
 
ROOT::Math::XYZVector ECLCluster::getClusterPosition() const
{
  const double cluster_x =  getR() * sin(getTheta()) * cos(getPhi());
  const double cluster_y =  getR() * sin(getTheta()) * sin(getPhi());
  const double cluster_z =  getR() * cos(getTheta());
  return ROOT::Math::XYZVector(cluster_x, cluster_y, cluster_z);
}
 
TMatrixDSym ECLCluster::getCovarianceMatrix3x3() const
{
  TMatrixDSym covmatecl(3);
  covmatecl(0, 0) = m_sqrtcovmat_00 * m_sqrtcovmat_00;
  covmatecl(1, 0) = m_covmat_10;
  covmatecl(1, 1) = m_sqrtcovmat_11 * m_sqrtcovmat_11;
  covmatecl(2, 0) = m_covmat_20;
  covmatecl(2, 1) = m_covmat_21;
  covmatecl(2, 2) = m_sqrtcovmat_22 * m_sqrtcovmat_22;
 
  //make symmetric
  for (int i = 0; i < 3; i++)
    for (int j = 0; j < i ; j++)
      covmatecl(j, i) = covmatecl(i, j);
  return covmatecl;
}
 
 
int ECLCluster::getDetectorRegion() const
{
  const double theta = getTheta();
 
  if (theta < 0.2164208) return 0;   // < 12.4deg
  if (theta < 0.5480334) return 1;   // < 31.4deg
  if (theta < 0.561996) return 11;   // < 32.2deg
  if (theta < 2.2462387) return 2;   // < 128.7
  if (theta < 2.2811453) return 13;   // < 130.7
  if (theta < 2.7070057) return 3;   // < 155.1
  else return 0;
}