SteppingAction.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 STEPPINGACTION_H_
#define STEPPINGACTION_H_
 
#include <globals.hh>
#include <G4UserSteppingAction.hh>
 
#include <vector>
 
class G4Track;
 
namespace Belle2 {
  namespace Simulation {
 
    class SteppingAction : public G4UserSteppingAction {
 
    public:
 
      SteppingAction();
 
      virtual ~SteppingAction();
 
      void setMaxNumberSteps(int maxSteps) { m_maxNumberSteps = maxSteps; };
 
      void setStoreTrajectories(bool store) { m_storeTrajectories = store; }
 
      void setAbsorbersR(const std::vector<float>& vec) { m_absorbers = vec; };
 
      virtual void UserSteppingAction(const G4Step* step);
 
 
    protected:
 
      int m_maxNumberSteps; 
      bool m_storeTrajectories = false;
      std::vector<float> m_absorbers; 
    private:
 
      void writeVREventStep(const G4Step*, const G4Track*);
 
      bool m_writeSimSteps{false};
 
    };
 
  } // end namespace Simulation
 
} // end namespace Belle2
 
#endif /* STEPPINGACTION_H_ */