PisaVMCStackv1.h

Go to the documentation of this file.

#ifndef __PISAVMCSTACKV1_H__
#define __PISAVMCSTACKV1_H__

#include "TVirtualMCStack.h"
#include <stack>

class TParticle;
class TClonesArray;

class PisaVMCStackv1 : public TVirtualMCStack
{
 public:
  PisaVMCStackv1();
  PisaVMCStackv1(size_t thesize);
  virtual ~PisaVMCStackv1();
  //
  // Methods for stacking
  //

  // Create a new particle and push into stack;
  // Arguments:
  // toBeDone   - 1 if particles should go to tracking, 0 otherwise
  // parent     - number of the parent track, -1 if track is primary
  // pdg        - PDG encoding
  // px, py, pz - particle momentum [GeV/c]
  // e          - total energy [GeV]
  // vx, vy, vx - position [cm]
  // tof        - time of flight [s]
  // polx, poly, polz - polarization
  // mech       - creator process VMC code
  // ntr        - track number (is filled by the stack
  // weight     - particle weight
  // is         - generation status code
  //
  void PushTrack(Int_t toBeDone, Int_t parent, Int_t pdg,
                 Double_t px, Double_t py, Double_t pz, Double_t e,
                 Double_t vx, Double_t vy, Double_t vz, Double_t tof,
                 Double_t polx, Double_t poly, Double_t polz,
                 TMCProcess mech, Int_t& ntr, Double_t weight,
                 Int_t is);

  // The stack has to provide two pop mechanisms:
  // PopNextTrack() - pops all particles with toBeDone = 1,
  //                  both primaries and seconadies
  // PopPrimaryForTracking() - pops only primary particles with toBeDone = 1,
  //                  stacking of secondaries is done by MC
  //
  TParticle* PopNextTrack(Int_t& itrack);
  TParticle* PopPrimaryForTracking(Int_t i);

  //
  // Set methods
  //

  // Set current track number
  void SetCurrentTrack(Int_t trackNumber);

  //
  // Get methods
  //

  // Total number of tracks
  Int_t GetNtrack() const;

  // Total number of primary tracks
  Int_t GetNprimary() const;

  // Current track particle
  TParticle* GetCurrentTrack() const;

  // Current track number
  Int_t GetCurrentTrackNumber() const;

  // Number of the parent of the current track
  Int_t GetCurrentParentTrackNumber() const;

  void Reset();

  void Print(Option_t*) const;

 private:
  TParticle* GetParticle(Int_t id) const;

 private:

  std::stack<TParticle*> fStack;
  TClonesArray* fParticles;
  Int_t fCurrentTrack;
  Int_t fNPrimary;

  ClassDef(PisaVMCStackv1,1)
};

#endif