ExtHit.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 <framework/gearbox/Const.h>
#include <Math/Vector3D.h>
#include <Math/Point3D.h>
#include <TMatrixDSym.h>
#include <TVector3.h>
#include <G4ThreeVector.hh>
#include <G4ErrorSymMatrix.hh>
 
namespace Belle2 {
  enum ExtHitStatus { EXT_FIRST = -1, EXT_ENTER, EXT_EXIT, EXT_STOP, EXT_ESCAPE,
                      EXT_ECLCROSS, EXT_ECLDL, EXT_ECLNEAR
                    };
 
  class ExtHit : public RelationsObject {
 
  public:
 
    ExtHit();
 
    ExtHit(int pdgCode, Const::EDetector detector, int element, ExtHitStatus status, bool backwardPropagation, double t,
           const ROOT::Math::XYZPoint& r, const ROOT::Math::XYZVector& p, const TMatrixDSym& e);
 
    ExtHit(int pdgCode, Const::EDetector detector, int element, ExtHitStatus status, bool backwardPropagation, double t,
           const G4ThreeVector& r, const G4ThreeVector& p, const G4ErrorSymMatrix& e);
 
    ExtHit(const ExtHit&);
 
    ExtHit& operator=(const ExtHit&);
 
    virtual ~ExtHit() {}
 
    void setPDGCode(int pdgCode) { m_PdgCode = pdgCode; }
 
    void setDetectorID(Const::EDetector detectorID) { m_DetectorID = detectorID; }
 
    void setCopyID(int copyID) { m_CopyID = copyID; }
 
    void setBackwardPropagation(bool backwardPropagation) { m_BackwardPropagation = backwardPropagation; }
 
    void setStatus(ExtHitStatus extHitStatus) { m_Status = extHitStatus; }
 
    void setTOF(double tof) { m_TOF = tof; }
 
    void setPosition(const ROOT::Math::XYZVector& position)
    {
      m_Position[0] = position.x();
      m_Position[1] = position.y();
      m_Position[2] = position.z();
    }
 
    void setMomentum(const ROOT::Math::XYZVector& momentum)
    {
      m_Momentum[0] = momentum.x();
      m_Momentum[1] = momentum.y();
      m_Momentum[2] = momentum.z();
    }
 
    void setCovariance(double covArray[21])
    {
      for (int k = 0; k < 21; ++k) {
        m_Covariance[k] = covArray[k];
      }
    }
 
    int getPdgCode() const { return m_PdgCode; }
 
    Const::EDetector getDetectorID() const { return m_DetectorID; }
 
    int getCopyID() const { return m_CopyID; }
 
    ExtHitStatus getStatus() const { return m_Status; }
 
    bool isBackwardPropagated() const { return m_BackwardPropagation; }
 
    double getTOF() const { return m_TOF; }
 
    double getLength() const { return m_TOF; }
 
    ROOT::Math::XYZVector getPosition() const
    {
      return ROOT::Math::XYZVector(m_Position[0], m_Position[1], m_Position[2]);
    }
 
    TVector3 getPositionTVector3() const
    {
      return TVector3(m_Position[0], m_Position[1], m_Position[2]);
    }
 
    ROOT::Math::XYZVector getMomentum() const
    {
      return ROOT::Math::XYZVector(m_Momentum[0], m_Momentum[1], m_Momentum[2]);
    }
 
    TVector3 getMomentumTVector3() const
    {
      return TVector3(m_Momentum[0], m_Momentum[1], m_Momentum[2]);
    }
 
    TMatrixDSym getCovariance() const;
 
    void update(ExtHitStatus status, double t,
                const G4ThreeVector& r, const G4ThreeVector& p, const G4ErrorSymMatrix& e);
 
    double getErrorTheta() const;
 
    double getErrorPhi() const;
 
  private:
 
    double getPolarCovariance(int i) const;
 
    int m_PdgCode;
 
    Const::EDetector m_DetectorID;
 
    int m_CopyID;
 
    ExtHitStatus m_Status;
 
    bool m_BackwardPropagation;
 
    float m_TOF;
 
    float m_Position[3];
 
    float m_Momentum[3];
 
    float m_Covariance[21];
 
    ClassDef(ExtHit, 8);
 
  };
}