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