ClusterUtils.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
 
// DATABASE objects
#include <mdst/dbobjects/BeamSpot.h>
#include <analysis/dbobjects/ECLPhotonEnergyResolution.h>
#include <framework/database/DBObjPtr.h>
 
// MDST
#include <mdst/dataobjects/ECLCluster.h>
 
// ROOT
#include <Math/Vector3D.h>
#include <Math/Vector4D.h>
#include <TMatrixD.h>
 
#include <framework/gearbox/Const.h>
 
namespace Belle2 {
  class ClusterUtils {
 
  public:
    ClusterUtils();
 
    const ROOT::Math::PxPyPzEVector GetCluster4MomentumFromCluster(const ECLCluster* cluster, ECLCluster::EHypothesisBit hypo);
 
    const ROOT::Math::PxPyPzEVector Get4MomentumFromCluster(const ECLCluster* cluster, ECLCluster::EHypothesisBit hypo);
 
    const ROOT::Math::PxPyPzEVector Get4MomentumFromCluster(const ECLCluster* cluster, const ROOT::Math::XYZVector& vertex,
                                                            ECLCluster::EHypothesisBit hypo);
    const TMatrixD GetJacobiMatrix4x6FromCluster(const ECLCluster* cluster, ECLCluster::EHypothesisBit hypo);
 
    const TMatrixD GetJacobiMatrix4x6FromCluster(const ECLCluster* cluster, const ROOT::Math::XYZVector& vertex,
                                                 ECLCluster::EHypothesisBit hypo);
 
    const TMatrixDSym GetCovarianceMatrix3x3FromCluster(const ECLCluster* cluster, int particleHypo = Const::photon.getPDGCode());
 
    const TMatrixDSym GetCovarianceMatrix4x4FromCluster(const ECLCluster* cluster, const TMatrixD& jacobiMatrix);
 
    const TMatrixDSym GetCovarianceMatrix4x4FromCluster(const ECLCluster* cluster,
                                                        const TMatrixDSym& covmatvertex,
                                                        const TMatrixD& jacobiMatrix);
 
    const TMatrixDSym GetCovarianceMatrix7x7FromCluster(const ECLCluster* cluster, const TMatrixD& jacobiMatrix);
 
    const TMatrixDSym GetCovarianceMatrix7x7FromCluster(const ECLCluster* cluster,
                                                        const TMatrixDSym& covmatvertex,
                                                        const TMatrixD& jacobiMatrix);
 
    const ROOT::Math::XYZVector GetIPPosition();
 
    const TMatrixDSym GetIPPositionCovarianceMatrix();
 
  private:
    DBObjPtr<BeamSpot> m_beamSpotDB;
 
    OptionalDBObjPtr<ECLPhotonEnergyResolution> m_photonEnergyResolutionDB;
 
    OptionalDBObjPtr<ECLPhotonEnergyResolution> m_neutronPositionResolutionDB{"neutronPositionResolutionDB"};
 
    OptionalDBObjPtr<ECLPhotonEnergyResolution> m_antiNeutronPositionResolutionDB{"antiNeutronPositionResolutionDB"};
 
    OptionalDBObjPtr<ECLPhotonEnergyResolution> m_kaonPositionResolutionDB{"kaonPositionResolutionDB"};
 
  };
 
} // Belle2 namespace