KLMHit2d.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
 
/* KLM headers. */
#include <klm/dataobjects/bklm/BKLMHit1d.h>
#include <klm/dataobjects/KLMElementNumbers.h>
 
/* Basf2 headers. */
#include <framework/datastore/RelationsObject.h>
 
/* ROOT headers. */
#include <Math/Vector3D.h>
 
/* CLHEP headers. */
#include <CLHEP/Vector/ThreeVector.h>
 
namespace Belle2 {
  class KLMHit2d : public RelationsObject {
 
  public:
 
    KLMHit2d();
 
    explicit KLMHit2d(KLMDigit* digit1, KLMDigit* digit2);
 
    KLMHit2d(const BKLMHit1d* hitPhi, const BKLMHit1d* hitZ,
             const CLHEP::Hep3Vector& globalPos, double time);
 
    ~KLMHit2d()
    {
    }
 
    int getSubdetector() const
    {
      return m_Subdetector;
    }
 
    void setSubdetector(int subdetector)
    {
      m_Subdetector = subdetector;
    }
 
    int getSection() const
    {
      return m_Section;
    }
 
    void setSection(int section)
    {
      m_Section = section;
    }
 
    int getSector() const
    {
      return m_Sector;
    }
 
    void setSector(int sector)
    {
      m_Sector = sector;
    }
 
    int getLayer() const
    {
      return m_Layer;
    }
 
    void setLayer(int layer)
    {
      m_Layer = layer;
    }
 
    void setXStripMin(int strip)
    {
      m_Strip[0] = strip;
    }
 
    void setXStripMax(int strip)
    {
      m_LastStrip[0] = strip;
    }
 
    void setYStripMin(int strip)
    {
      m_Strip[1] = strip;
    }
 
    void setYStripMax(int strip)
    {
      m_LastStrip[1] = strip;
    }
 
    bool inRPC() const
    {
      return (m_Subdetector == KLMElementNumbers::c_BKLM) &&
             (m_Layer >= BKLMElementNumbers::c_FirstRPCLayer);
    }
 
    int getZStripMin() const
    {
      return m_Strip[BKLMElementNumbers::c_ZPlane];
    }
 
    int getZStripMax() const
    {
      return m_LastStrip[BKLMElementNumbers::c_ZPlane];
    }
 
    double getZStripAve() const
    {
      return 0.5 * (getZStripMin() + getZStripMax());
    }
 
    int getPhiStripMin() const
    {
      return m_Strip[BKLMElementNumbers::c_PhiPlane];
    }
 
    int getPhiStripMax() const
    {
      return m_LastStrip[BKLMElementNumbers::c_PhiPlane];
    }
 
    double getPhiStripAve() const
    {
      return 0.5 * (getPhiStripMin() + getPhiStripMax());
    }
 
    int getXStripMin() const
    {
      return m_Strip[0];
    }
 
    int getXStripMax() const
    {
      return m_LastStrip[0];
    }
 
    int getYStripMin() const
    {
      return m_Strip[1];
    }
 
    int getYStripMax() const
    {
      return m_LastStrip[1];
    }
 
    void setPosition(float x, float y, float z)
    {
      m_GlobalX = x;
      m_GlobalY = y;
      m_GlobalZ = z;
    }
 
    //void setPosition(const ROOT::Math::XYZVector& pos);
 
    float getPositionX() const
    {
      return m_GlobalX;
    }
 
    float getPositionY() const
    {
      return m_GlobalY;
    }
 
    float getPositionZ() const
    {
      return m_GlobalZ;
    }
 
    ROOT::Math::XYZVector getPosition() const
    {
      return ROOT::Math::XYZVector(m_GlobalX, m_GlobalY, m_GlobalZ);
    }
 
    float getTime() const
    {
      return m_Time;
    }
 
    void setTime(float time)
    {
      m_Time = time;
    }
 
    void setMCTime(float t)
    {
      m_MCTime = t;
    }
 
    float getMCTime() const
    {
      return m_MCTime;
    }
 
    float getEnergyDeposit() const
    {
      return m_EnergyDeposit;
    }
 
    void setEnergyDeposit(float energyDeposit)
    {
      m_EnergyDeposit = energyDeposit;
    }
 
    float getChiSq() const
    {
      return m_ChiSq;
    }
 
    void setChiSq(float chisq)
    {
      m_ChiSq = chisq;
    }
 
    bool isOutOfTime() const
    {
      return m_IsOutOfTime;
    }
 
    void isOutOfTime(bool outOfTime)
    {
      m_IsOutOfTime = outOfTime;
    }
 
    bool isOnTrack() const
    {
      return m_IsOnTrack;
    }
 
    void isOnTrack(bool onTrack)
    {
      m_IsOnTrack = onTrack;
    }
 
    bool isOnStaTrack() const
    {
      return m_IsOnStaTrack;
    }
 
    void isOnStaTrack(bool onStaTrack)
    {
      m_IsOnStaTrack = onStaTrack;
    }
 
  private:
 
    int m_Subdetector = 0;
 
    int m_Section = 0;
 
    int m_Sector = 0;
 
    int m_Layer = 0;
 
    int m_Strip[KLMElementNumbers::getMaximalPlaneNumber()] = {0};
 
    int m_LastStrip[KLMElementNumbers::getMaximalPlaneNumber()] = {0};
 
    float m_GlobalX = 0.0;
 
    float m_GlobalY = 0.0;
 
    float m_GlobalZ = 0.0;
 
    float m_Time = 0.0;
 
    float m_MCTime = 0.0;
 
    float m_EnergyDeposit = 0.0;
 
    /* From EKLMHit2d. */
 
    float m_ChiSq = 0.0;
 
    /* From BKLMHit2d. */
 
    bool m_IsOutOfTime = false;
 
    bool m_IsOnTrack = false;
 
    bool m_IsOnStaTrack = false;
 
    ClassDef(Belle2::KLMHit2d, 1);
 
  };
 
}