[u/mrichter/AliRoot.git] / AliGeant4 / AliTrackingAction.h

// $Id$
// Category: event
//
// Class that takes care of storing kinematics.

#ifndef ALI_TRACKING_ACTION_H
#define ALI_TRACKING_ACTION_H

#include <G4UserTrackingAction.hh>
#include <globals.hh>

#include <TClonesArray.h>

class AliTrackingActionMessenger;

class G4Track;

class AliTrackingAction : public G4UserTrackingAction 
{
  public:
    AliTrackingAction();
    // --> protected
    // AliTrackingAction(const AliTrackingAction& right);
    virtual ~AliTrackingAction();
   
    // static get method
    static AliTrackingAction* Instance();

    // methods
    void PrepareNewEvent();
    virtual void PreUserTrackingAction(const G4Track* aTrack);
    virtual void PostUserTrackingAction(const G4Track* aTrack);
    void SaveParticle(const G4Track* track, G4String processName);
    void SaveAndDestroyTrack();

    // set methods
    void SetVerboseLevel(G4int level);
    void SetSavePrimaries(G4bool savePrimaries);

    // get methods
    G4int GetVerboseLevel() const;
    G4bool GetSavePrimaries() const;
    G4int GetNofPrimaryTracks() const;
    G4int GetNofSavedTracks() const;

  protected:
    AliTrackingAction(const AliTrackingAction& right);

    // operators
    AliTrackingAction& operator=(const AliTrackingAction& right);

  private:
    // methods
    G4int GetParticleIndex(G4int trackID);
  
    // static data members
    static AliTrackingAction*    fgInstance; //this instance

    // data members
    TClonesArray*  fParticles;         //AliRun::fParticles
    G4int          fPrimaryTrackID;    //primary track ID 
    G4bool         fSavePrimaries;     //control of saving primaries
    G4int          fVerboseLevel;      //verbose level
    G4int          fPrimariesCounter;  //primary particles counter
    G4int          fParticlesCounter;  //particles counter
    G4int          fLastParticleIndex; //index of the last particle in fParticles
    AliTrackingActionMessenger*  fMessenger; //messenger
};

// inline methods

inline void AliTrackingAction::SetVerboseLevel(G4int level)
{ fVerboseLevel = level; }

inline void AliTrackingAction::SetSavePrimaries(G4bool savePrimaries)
{ fSavePrimaries = savePrimaries; }

inline G4int AliTrackingAction::GetVerboseLevel() const
{ return fVerboseLevel; }

inline G4bool AliTrackingAction::GetSavePrimaries() const
{ return fSavePrimaries; }

inline G4int AliTrackingAction::GetNofPrimaryTracks() const
{ return fPrimariesCounter; }

inline G4int AliTrackingAction::GetNofSavedTracks() const
{ return fParticlesCounter; }

#endif //ALI_TRACKING_ACTION_H
Commit	Line	Data
676fb573	1	// $Id$
	2	// Category: event
	3	//
	4	// Class that takes care of storing kinematics.
	5
	6	#ifndef ALI_TRACKING_ACTION_H
	7	#define ALI_TRACKING_ACTION_H
	8
	9	#include <G4UserTrackingAction.hh>
	10	#include <globals.hh>
	11
	12	#include <TClonesArray.h>
	13
	14	class AliTrackingActionMessenger;
	15
	16	class G4Track;
	17
	18	class AliTrackingAction : public G4UserTrackingAction
	19	{
	20	public:
	21	AliTrackingAction();
	22	// --> protected
	23	// AliTrackingAction(const AliTrackingAction& right);
fdfe18c5	24	virtual ~AliTrackingAction();
676fb573	25
	26	// static get method
	27	static AliTrackingAction* Instance();
	28
	29	// methods
	30	void PrepareNewEvent();
	31	virtual void PreUserTrackingAction(const G4Track* aTrack);
	32	virtual void PostUserTrackingAction(const G4Track* aTrack);
	33	void SaveParticle(const G4Track* track, G4String processName);
	34	void SaveAndDestroyTrack();
	35
	36	// set methods
	37	void SetVerboseLevel(G4int level);
	38	void SetSavePrimaries(G4bool savePrimaries);
	39
	40	// get methods
	41	G4int GetVerboseLevel() const;
	42	G4bool GetSavePrimaries() const;
	43	G4int GetNofPrimaryTracks() const;
	44	G4int GetNofSavedTracks() const;
	45
	46	protected:
	47	AliTrackingAction(const AliTrackingAction& right);
	48
	49	// operators
	50	AliTrackingAction& operator=(const AliTrackingAction& right);
	51
	52	private:
	53	// methods
	54	G4int GetParticleIndex(G4int trackID);
	55
	56	// static data members
	57	static AliTrackingAction* fgInstance; //this instance
	58
	59	// data members
	60	TClonesArray* fParticles; //AliRun::fParticles
	61	G4int fPrimaryTrackID; //primary track ID
	62	G4bool fSavePrimaries; //control of saving primaries
	63	G4int fVerboseLevel; //verbose level
	64	G4int fPrimariesCounter; //primary particles counter
	65	G4int fParticlesCounter; //particles counter
	66	G4int fLastParticleIndex; //index of the last particle in fParticles
	67	AliTrackingActionMessenger* fMessenger; //messenger
	68	};
	69
	70	// inline methods
	71
	72	inline void AliTrackingAction::SetVerboseLevel(G4int level)
	73	{ fVerboseLevel = level; }
	74
	75	inline void AliTrackingAction::SetSavePrimaries(G4bool savePrimaries)
	76	{ fSavePrimaries = savePrimaries; }
	77
	78	inline G4int AliTrackingAction::GetVerboseLevel() const
	79	{ return fVerboseLevel; }
	80
	81	inline G4bool AliTrackingAction::GetSavePrimaries() const
	82	{ return fSavePrimaries; }
	83
	84	inline G4int AliTrackingAction::GetNofPrimaryTracks() const
	85	{ return fPrimariesCounter; }
	86
	87	inline G4int AliTrackingAction::GetNofSavedTracks() const
	88	{ return fParticlesCounter; }
89
90	#endif //ALI_TRACKING_ACTION_H