ShipStack.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                       ShipStack header file                   -----
// -------------------------------------------------------------------------
 
 
#ifndef ShipStack_H
#define ShipStack_H
 
#include "FairGenericStack.h"           // for FairGenericStack
 
#include "ShipDetectorList.h"           // for DetectorId
 
#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;
 
enum {kDoneBit=1};
 
class ShipStack : public FairGenericStack
{
 
  public:
 
    ShipStack(Int_t size = 100);
 
 
    virtual ~ShipStack();
 
 
    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:
 
    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();
 
    ShipStack(const ShipStack&);
    ShipStack& operator=(const ShipStack&);
 
    ClassDef(ShipStack,1)
 
 
};
 
 
 
 
 
 
#endif