ClusterUtils.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2017-2019 - Belle II Collaboration                        *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Torben Ferber (torben.ferber@desy.de)                    *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
// BEAMSPOT AND DATABASE
#include <mdst/dbobjects/BeamSpot.h>
#include <framework/database/DBObjPtr.h>
 
// MDST
#include <mdst/dataobjects/ECLCluster.h>
 
// ROOT
#include <TLorentzVector.h>
#include <TMatrixD.h>
 
namespace Belle2 {
  class ClusterUtils {
 
  public:
 
    ClusterUtils();
 
    const TLorentzVector GetCluster4MomentumFromCluster(const ECLCluster* cluster, ECLCluster::EHypothesisBit hypo);
 
    const TLorentzVector Get4MomentumFromCluster(const ECLCluster* cluster, ECLCluster::EHypothesisBit hypo);
 
    const TLorentzVector Get4MomentumFromCluster(const ECLCluster* cluster, const TVector3& vertex, ECLCluster::EHypothesisBit hypo);
 
    const TMatrixDSym GetCovarianceMatrix4x4FromCluster(const ECLCluster* cluster, ECLCluster::EHypothesisBit hypo);
 
    const TMatrixDSym GetCovarianceMatrix4x4FromCluster(const ECLCluster* cluster, const TVector3& vertex,
                                                        const TMatrixDSym& covmatvertex, ECLCluster::EHypothesisBit hypo);
 
    const TMatrixDSym GetCovarianceMatrix7x7FromCluster(const ECLCluster* cluster, ECLCluster::EHypothesisBit hypo);
 
    const TMatrixDSym GetCovarianceMatrix7x7FromCluster(const ECLCluster* cluster, const TVector3& vertex,
                                                        const TMatrixDSym& covmatvertex, ECLCluster::EHypothesisBit hypo);
 
    const TVector3 GetIPPosition();
 
    const TMatrixDSym GetIPPositionCovarianceMatrix();
 
  private:
 
    DBObjPtr<BeamSpot> m_beamSpotDB; 
  };
 
} // Belle2 namespace