CbmStack.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file CbmStack.h

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


#ifndef CBMSTACK_H
#define CBMSTACK_H


#include "CbmDetectorList.h"
#include "FairGenericStack.h"

#include <map>
#include <stack>

class TClonesArray;

class CbmStack : public FairGenericStack
{

 public:

  CbmStack(Int_t size = 100);


  virtual ~CbmStack();

  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 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();

  CbmStack(const CbmStack&);
  CbmStack& operator=(const CbmStack&);

  ClassDef(CbmStack,1)

      
};






#endif