CDCDedxHit.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
 
#include <framework/datastore/RelationsObject.h>
#include <cdc/dataobjects/WireID.h>
#include <Math/Vector3D.h>
#include <cmath>
 
namespace Belle2 {
  class CDCDedxHit : public RelationsObject {
 
  public:
 
    CDCDedxHit()
    {}
 
    CDCDedxHit(WireID wireID, short TDC, unsigned short ADC,
               const ROOT::Math::XYZVector& pocaMomentum,
               const ROOT::Math::XYZVector& pocaOnTrack,
               const ROOT::Math::XYZVector& pocaOnWire,
               int foundByTrackFinder, double weightPionHypo, double weightKaonHypo, double weightProtHypo):
      m_wireID(wireID), m_tdcCount(TDC), m_adcCount(ADC),
      m_px(pocaMomentum.X()), m_py(pocaMomentum.Y()), m_pz(pocaMomentum.Z()),
      m_x(pocaOnTrack.X()), m_y(pocaOnTrack.Y()), m_z(pocaOnTrack.Z()),
      m_dx(pocaOnWire.X() - pocaOnTrack.X()), m_dy(pocaOnWire.Y() - pocaOnTrack.Y()), m_dz(pocaOnWire.Z() - pocaOnTrack.Z()),
      m_foundByTrackFinder(foundByTrackFinder),
      m_weightPionHypo(weightPionHypo), m_weightKaonHypo(weightKaonHypo), m_weightProtHypo(weightProtHypo)
    {}
 
    const WireID& getWireID() const {return m_wireID;}
 
    short getTDCCount() const {return m_tdcCount;}
 
    unsigned short getADCCount() const {return m_adcCount;}
 
    ROOT::Math::XYZVector getPOCAMomentum() const {return ROOT::Math::XYZVector(m_px, m_py, m_pz);}
 
    ROOT::Math::XYZVector getPOCAOnTrack() const {return ROOT::Math::XYZVector(m_x, m_y, m_z);}
 
    ROOT::Math::XYZVector getPOCAOnWire() const {return ROOT::Math::XYZVector(m_x + m_dx, m_y + m_dy, m_z + m_dz);}
 
    ROOT::Math::XYZVector getDOCAVector() const {return ROOT::Math::XYZVector(m_dx, m_dy, m_dz);}
 
    double getDOCA() const {return std::sqrt(m_dx * m_dx + m_dy * m_dy + m_dz * m_dz);}
 
 
    double getSignedDOCAXY() const;
 
    double getEntranceAngle() const;
 
    int getFoundByTrackFinder() const {return m_foundByTrackFinder;}
 
    double getWeightPionHypo() const {return m_weightPionHypo;}
 
    double getWeightKaonHypo() const {return m_weightKaonHypo;}
 
    double getWeightProtonHypo() const {return m_weightProtHypo;}
 
  private:
 
    WireID m_wireID; 
    short m_tdcCount = 0; 
    unsigned short m_adcCount = 0; 
    float m_px = 0; 
    float m_py = 0; 
    float m_pz = 0; 
    float m_x = 0; 
    float m_y = 0; 
    float m_z = 0; 
    float m_dx = 0; 
    float m_dy = 0; 
    float m_dz = 0; 
    int m_foundByTrackFinder = 0; 
    float m_weightPionHypo = 0; 
    float m_weightKaonHypo = 0; 
    float m_weightProtHypo = 0; 
    ClassDef(CDCDedxHit, 1); 
  };
 
} // end namespace Belle2