KLMClusterShape.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
 
/* Basf2 headers. */
#include <framework/datastore/RelationsObject.h>
 
/* ROOT headers. */
#include <Math/Vector3D.h>
#include <TMatrixT.h>
 
/* C++ headers. */
#include <cmath>
 
namespace Belle2 {
  class KLMClusterShape : public RelationsObject {
 
  public:
 
    KLMClusterShape();
 
    KLMClusterShape(const KLMClusterShape&);
 
    KLMClusterShape& operator=(const KLMClusterShape&);
 
    ~KLMClusterShape()
    {
    }
 
    //Define get functions to obtain private information
 
    double getVariance1() const
    {
      return std::abs(m_primaryVec[3]);
    }
 
    double getVariance2() const
    {
      return std::abs(m_secondaryVec[3]);
    }
 
    double getVariance3() const
    {
      return std::abs(m_tertiaryVec[3]);
    }
 
    ROOT::Math::XYZVector getPrimaryVector();
 
    ROOT::Math::XYZVector getSecondaryVector();
 
    ROOT::Math::XYZVector getTertiaryVector();
 
    int getNHits() const
    {
      return m_nHits;
    }
 
    void setEigen(TMatrixT<double> eigenList);
 
    void setNHits(int nHits)
    {
      m_nHits = nHits;
    }
 
  private:
 
    int m_nHits;
 
 
    //Normalization for eigenvectors should be unity.
    //Indexing: 0-2 is unit eigenvector, 3 is eigenvalue.
 
    std::array<double, 4> m_primaryVec;
 
    std::array<double, 4> m_secondaryVec;
 
    std::array<double, 4> m_tertiaryVec;
 
    ClassDef(KLMClusterShape, 1)
 
  }; // end of public RelationsObject
 
}// end of namespace Belle2