SensorTraversal.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.                  *
 **************************************************************************/
 
#ifndef SENSORTRAVERSAL_H
#define SENSORTRAVERSAL_H
 
#include <G4ThreeVector.hh>
 
namespace Belle2 {
  namespace fangs {
    class SensorTraversal {
    public:
      void add(const G4ThreeVector& position, double depEnergy, double length)
      {
        m_depEnergy += depEnergy;
        m_length += length;
        m_exitPosition = position;
      }
 
      int getTrackID() const { return m_trackID; }
      int getLadderID() const { return m_ladderID; }
      int getSensorID() const { return m_sensorID; }
      int getPDGCode() const { return m_pdgCode; }
      double getDepEnergy() const { return m_depEnergy; }
      double getLength() const { return m_length; }
      double getEntryTime() const { return m_entryTime; }
      const G4ThreeVector& getEntryMomentum() const { return m_entryMomentum; }
      const G4ThreeVector& getEntryPosition() const { return m_entryPosition; }
      const G4ThreeVector& getLocalEntryPosition() const { return m_localentryPosition; }
      const G4ThreeVector& getExitPosition() const { return m_exitPosition; }
 
 
 
      bool isContained() const { return m_contained; }
      bool isPrimary() const { return m_primary; }
 
      void hasEntered() { m_contained = false; }
      void hasLeft() { m_contained = false; }
 
      void setInitial(int trackID, int ladderID, int sensorID, int pdgCode, bool primary, const G4ThreeVector& position,
                      const G4ThreeVector& localposition, const G4ThreeVector& momentum, double time)
      {
        m_trackID = trackID;
        m_ladderID = ladderID;
        m_sensorID = sensorID;
        m_pdgCode = pdgCode;
        m_primary = primary;
        m_entryPosition = position;
        m_localentryPosition = localposition;
        m_entryMomentum = momentum;
        m_entryTime = time;
      }
 
      void reset()
      {
        m_trackID = 0;
        m_ladderID = 0;
        m_sensorID = 0;
        m_pdgCode = 0;
        m_depEnergy = 0;
        m_length = 0;
        m_contained = true;
        m_primary = false;
        m_entryTime = 0;
      }
 
      bool check(int trackID, int ladderID, int sensorID)
      {
        // object not set yet so we can use it
        if (m_trackID == 0) return true;
        return (trackID == m_trackID && ladderID == m_ladderID && sensorID == m_sensorID);
      }
 
    private:
      G4ThreeVector m_entryMomentum;
      G4ThreeVector m_entryPosition;
      G4ThreeVector m_localentryPosition;
      G4ThreeVector m_exitPosition;
      int m_trackID {0};
      int m_ladderID {0};
      int m_sensorID {0};
      int m_pdgCode {0};
      double m_depEnergy {0};
      double m_length {0};
      double m_entryTime {0};
      bool m_contained {true};
      bool m_primary {false};
    };
 
  }
} //Belle2 namespace
#endif // SENSORTRAVERSAL_H