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>
 
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;
 
    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 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;
 
    ClassDef(Belle2::KLMCluster, 3);
 
  };
 
}