MdEdepSimulatorAG/SteppingAction.h

Go to the documentation of this file.

#ifndef _EdepSimulatorAG_SteppingAction_h_
#define _EdepSimulatorAG_SteppingAction_h_

#include <globals.hh>
#include <G4UserSteppingAction.hh>

#include <utl/Particle.h>

#include <vector>
#include <map>
#include <string>
using std::vector;
using std::string;
using std::map;


namespace EdepSimulatorAG {

class Data {
public:
  Data()
  {
    Sname = "unknown";
    Pname = "unknown";
    PDGCode = -1;
    ParentID = -1;
    TrackID =- 1;
    PParent = 0;
    x = 0;
    y = 0;
    z = 0;
    px = 0;
    py = 0;
    pz = 0;
    KinE = 0;
    Edep = 0;
    Time = 0;
  }

  string Sname;
  string Pname;
  int PDGCode;
  int ParentID;//parent of any G4 particle created inside scintillator
  int TrackID;
  utl::Particle *PParent;//Injected particle in scintillator
  //Position
  double x;
  double y;
  double z;
  //Direction
  double px;
  double py;
  double pz;
  //Energy & Time
  double KinE;
  double Edep;
  double Time;
};


class DataCollection : public std::vector<Data> {
public:
  DataCollection() : std::vector<Data>() { }
  DataCollection(const Data &pd) : std::vector<Data>() { push_back(pd); }
};


class SteppingAction : public G4UserSteppingAction {

public:

  typedef map<int, DataCollection> ScintMap;
  typedef ScintMap::iterator ScintMapIt;

  SteppingAction();
  virtual ~SteppingAction();

  virtual void UserSteppingAction(const G4Step*);

  void
  SetImpingingParticle(utl::Particle* p)
  {
    fPParent = p;
#ifdef AMIGA_DEBUG
    G4cerr << "Setting parent to "
           << fPParent->GetType() << '\n';
#endif
  }

  utl::Particle* GetImpingingParticle() { return fPParent; }

  //PartCollection fPartColl;
  //PartCollectionIt fPartCollIt;

  void SetScintillatorId(int id) { fScintID = id; }
  const ScintMap& GetScintMap() { return fScintMap; }
  void ClearMap() { fScintMap.clear(); }

private:
  ScintMap fScintMap;
  G4int fScintID;
  utl::Particle* fPParent;

};

}


#endif