AtStack.h

Go to the documentation of this file.

/********************************************************************************
 *    Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *         GNU Lesser General Public Licence version 3 (LGPL) version 3,        *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/
 
// -------------------------------------------------------------------------
// -----                       AtStack header file                 -----
// -----                  M. Al-Turany   June 2014                     -----
// -------------------------------------------------------------------------
 
#ifndef AtStack_H
#define AtStack_H
 
#include "AtDetectorList.h" // for DetectorId
 
#include <FairGenericStack.h> // for FairGenericStack
 
#include <Rtypes.h>     // for Int_t, Double_t, Bool_t, etc
#include <TMCProcess.h> // for TMCProcess
 
#include <map>     // for map, map<>::iterator
#include <stack>   // for stack
#include <utility> // for pair
 
class TClonesArray;
class TParticle;
class TRefArray;
class FairLogger;
class TBuffer;
class TClass;
class TMemberInspector;
 
class AtStack : public FairGenericStack {
 
public:
   AtStack(Int_t size = 100);
 
   virtual ~AtStack();
 
   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 = 0);
 
   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 AddPoint(DetectorId iDet);
 
   void AddPoint(DetectorId iDet, Int_t iTrack);
 
   TParticle *GetParticle(Int_t trackId) const;
   TClonesArray *GetListOfParticles() { return fParticles; }
 
private:
   FairLogger *fLogger;
 
   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;
 
   void SelectTracks();
 
   AtStack(const AtStack &);
   AtStack &operator=(const AtStack &);
 
   ClassDef(AtStack, 1)
};
 
#endif