ERStack.h

// -------------------------------------------------------------------------
// -----                       ERStack header file                     -----
// -----           Created 11/12/15  by V.Schetinin                    -----
// -------------------------------------------------------------------------


#ifndef ERStack_H
#define ERStack_H


#include "ERDetectorList.h"
#include "FairGenericStack.h"

#include <map>
#include <stack>

class TClonesArray;

class ERStack : public FairGenericStack
{

 public:

  ERStack(Int_t size = 100);


  virtual ~ERStack();

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

  ERStack(const ERStack&);
  ERStack& operator=(const ERStack&);

  ClassDef(ERStack,1)


};

#endif