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.                  *
 **************************************************************************/
 
#pragma once
#ifndef VXD_SIMULATION_SENSORTRAVERSAL_H
#define VXD_SIMULATION_SENSORTRAVERSAL_H
 
#include <G4ThreeVector.hh>
#include <vector>
 
namespace Belle2 {
  struct StepInformation {
    StepInformation(const G4ThreeVector& stepPosition, const G4ThreeVector& stepMomentum, double stepElectrons, double stepTime,
                    double stepLength):
      position(stepPosition), momentum(stepMomentum), electrons(stepElectrons), time(stepTime), length(stepLength) {}
    G4ThreeVector position;
    G4ThreeVector momentum;
    double electrons;
    double time {0};
    double length {0};
  };
 
  class SensorTraversal: public std::vector<StepInformation> {
  public:
    typedef std::pair<iterator, iterator> range;
 
    SensorTraversal() = default;
 
    void add(const G4ThreeVector& position, const G4ThreeVector& momentum, double electrons, double time, double length)
    {
      m_electrons += electrons;
      m_length += length;
      emplace_back(position, momentum, m_electrons, time, m_length);
    }
 
    int getTrackID() const { return m_trackID; }
    int getPDGCode() const { return m_pdgCode; }
    double getElectrons() const { return m_electrons; }
    double getLength() const { return m_length; }
    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 pdgCode, bool primary)
    {
      m_trackID = trackID;
      m_pdgCode = pdgCode;
      m_primary = primary;
    }
 
    void reset()
    {
      m_trackID = 0;
      m_pdgCode = 0;
      m_electrons = 0;
      m_length = 0;
      m_contained = true;
      m_primary = false;
      clear();
    }
 
  private:
    int m_trackID {0};
    int m_pdgCode {0};
    double m_electrons {0};
    double m_length {0};
    bool m_contained {true};
    bool m_primary {false};
  };
 
} //Belle2 namespace
#endif // VXD_SIMULATION_SENSORTRAVERSAL_H