SteppingAction.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010-2011  Belle II Collaboration                         *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Guofu Cao, Andreas Moll                                  *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef STEPPINGACTION_H_
#define STEPPINGACTION_H_
 
#include <globals.hh>
#include <G4UserSteppingAction.hh>
 
#include <vector>
 
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; 
    };
 
  } // end namespace Simulation
 
} // end namespace Belle2
 
#endif /* STEPPINGACTION_H_ */