PndStack.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                       PndStack header file                    -----
// -----           Created 10/08/04  by D. Bertini / V. Friese         -----
// -------------------------------------------------------------------------

#ifndef PNDSTACK_H
#define PNDSTACK_H

#include "PndDetectorList.h"
#include "FairGenericStack.h"

#include "TClonesArray.h"
#include "TVirtualMCStack.h"

#include <map>
#include <stack>

enum { kDoneBit = 1 };

class PndStack : public FairGenericStack {

 public:
  PndStack(Int_t size = 100);

  virtual ~PndStack();

  virtual void PushTrack(Int_t toBeDone, Int_t parentID, Int_t pdgCode, Double_t px, Double_t py, Double_t pz, Double_t e, Double_t vx, Double_t vy, Double_t vz, Double_t time,
                         Double_t polx, Double_t poly, Double_t polz, TMCProcess proc, Int_t &ntr, Double_t weight, Int_t is);

  virtual void PushTrack(Int_t toBeDone, Int_t parentID, Int_t pdgCode, Double_t px, Double_t py, Double_t pz, Double_t e, Double_t vx, Double_t vy, Double_t vz, Double_t time,
                         Double_t polx, Double_t poly, Double_t polz, TMCProcess proc, Int_t &ntr, Double_t weight, Int_t is, Int_t secondParentId);

  virtual TParticle *PopNextTrack(Int_t &iTrack);

  virtual TParticle *PopPrimaryForTracking(Int_t iPrim);

  virtual void SetCurrentTrack(Int_t iTrack) { fCurrentTrack = iTrack; }

  virtual Int_t GetNtrack() const { return fNParticles; }

  virtual Int_t GetNprimary() const { return fNPrimaries; }

  virtual TParticle *GetCurrentTrack() const;

  virtual Int_t GetCurrentTrackNumber() const { return fCurrentTrack; }

  virtual Int_t GetCurrentParentTrackNumber() const;

  virtual void AddParticle(TParticle *part);

  virtual void FillTrackArray();

  virtual void UpdateTrackIndex(TRefArray *detArray);

  virtual void Reset();

  virtual void Register();

  virtual void Print(Int_t iVerbose = 0) const;

  void StoreSecondaries(Bool_t choice = kTRUE) { fStoreSecondaries = choice; }
  void SetMinPoints(Int_t min) { fMinPoints = min; }
  void SetEnergyCut(Double_t eMin) { fEnergyCut = eMin; }
  void StoreMothers(Bool_t choice = kTRUE) { fStoreMothers = choice; }
  void SetPersistence(Bool_t choice) { fPersistence = choice; }

  void AddPoint(DetectorId iDet);

  void AddPoint(DetectorId iDet, Int_t iTrack);

  TParticle *GetParticle(Int_t trackId) const;
  TClonesArray *GetListOfParticles() { return fParticles; }

 private:
  std::stack<TParticle *> fStack; 

  TClonesArray *fParticles; 

  TClonesArray *fTracks;

  std::map<Int_t, Bool_t> fStoreMap;            
  std::map<Int_t, Bool_t>::iterator fStoreIter; 

  std::map<Int_t, Int_t> fIndexMap;            
  std::map<Int_t, Int_t>::iterator fIndexIter; 

  std::map<std::pair<Int_t, Int_t>, Int_t> fPointsMap; 

  Int_t fCurrentTrack; 
  Int_t fNPrimaries;   
  Int_t fNParticles;   
  Int_t fNTracks;      
  Int_t fIndex;        

  Bool_t fStoreSecondaries;
  Int_t fMinPoints;
  Double32_t fEnergyCut;
  Bool_t fStoreMothers;
  Bool_t fPersistence;

  void SelectTracks();
  void SetGeneratorFlags(Int_t myid);

  PndStack(const PndStack &L);
  PndStack &operator=(const PndStack &) { return *this; };

  ClassDef(PndStack, 1)
};

#endif