KLMCluster.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
 
/* External headers. */
#include <Math/Vector3D.h>
#include <Math/Vector4D.h>
#include <TMatrixDSym.h>
 
/* Basf2 headers. */
#include <framework/datastore/RelationsObject.h>
#include <framework/gearbox/Const.h>
 
namespace Belle2 {

  class KLMCluster: public RelationsObject {
 
  public:

    KLMCluster();

    KLMCluster(float x, float y, float z, float time, int nLayers,
               int nInnermostLayer, float p);

    ~KLMCluster();

    inline float getTime() const
    {return m_time;}

    inline int getLayers() const
    {return m_layers;}

    inline int getInnermostLayer() const
    {return m_innermostLayer;}

    inline ROOT::Math::XYZVector getClusterPosition() const
    {return ROOT::Math::XYZVector(m_globalX, m_globalY, m_globalZ);}

    inline ROOT::Math::XYZVector getPosition() const
    {return ROOT::Math::XYZVector(0, 0, 0);}

    float getMomentumMag() const;

    float getEnergy() const;

    ROOT::Math::PxPyPzEVector getMomentum() const;

    TMatrixDSym getError4x4() const;

    TMatrixDSym getError7x7() const;

    bool getAssociatedEclClusterFlag() const;

    bool getAssociatedTrackFlag() const;

    inline float getClusterTrackRotationAngle() const { return m_clusterTrackRotationAngle; }

    inline float getClusterTrackSeparationAngle() const { return m_clusterTrackSeparationAngle; }

    inline float getClusterTrackSeparation() const { return m_clusterTrackSeparation; }

    inline float getShapeStdDev1() const { return m_shapeStdDev1; }

    inline float getShapeStdDev2() const { return m_shapeStdDev2; }

    inline float getShapeStdDev3() const { return m_shapeStdDev3; }

    inline int getKLMnDigits() const
    { return m_KLMnDigits; }

    void setClusterTrackRotationAngle(float rotation)
    {
      m_clusterTrackRotationAngle = rotation;
    }

    void setClusterTrackSeparationAngle(float separation)
    {
      m_clusterTrackSeparationAngle = separation;
    }

    void setClusterTrackSeparation(float dist)
    {
      m_clusterTrackSeparation = dist;
    }

    void setShapeStdDev1(float std1)
    {
      m_shapeStdDev1 = std1;
    }

    void setShapeStdDev2(float std2)
    {
      m_shapeStdDev2 = std2;
    }

    void setShapeStdDev3(float std3)
    {
      m_shapeStdDev3 = std3;
    }

    void setTime(float time)
    {
      m_time = time;
    }

    void setLayers(int layers)
    {
      m_layers = layers;
    }

    void setInnermostLayer(int innermostLayer)
    {
      m_innermostLayer = innermostLayer;
    }

    void setClusterPosition(float globalX, float globalY, float globalZ)
    {
      m_globalX = globalX;
      m_globalY = globalY;
      m_globalZ = globalZ;
    }

    void setKLMnDigits(int nDigits)
    { m_KLMnDigits = nDigits; }

    void setMomentumMag(float momentumMag)
    {
      m_p = momentumMag;
    }
 
  private:

    float m_time;

    int m_layers;

    int m_innermostLayer;

    float m_globalX;

    float m_globalY;

    float m_globalZ;

    float m_p;

    float m_clusterTrackRotationAngle = Const::floatNaN;

    float m_clusterTrackSeparationAngle = Const::floatNaN;

    float m_clusterTrackSeparation = Const::floatNaN;

    float m_shapeStdDev1 = 0;

    float m_shapeStdDev2 = 0;

    float m_shapeStdDev3 = 0;

    int m_KLMnDigits = 0;

    ClassDef(Belle2::KLMCluster, 5);
 
  };

}