CbmStack.h

Go to the documentation of this file.

/* Copyright (C) 2009-2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Denis Bertini, Florian Uhlig [committer], Volker Friese */
 
// -------------------------------------------------------------------------
// -----                       CbmStack header file                    -----
// -----           Created 10/08/04  by D. Bertini / V. Friese         -----
// -------------------------------------------------------------------------
 
 
#ifndef CBMSTACK_H
#define CBMSTACK_H
 
 
#include "CbmDefs.h"
#include "CbmStackFilter.h"
 
#include "FairGenericStack.h"
 
#include <map>
#include <memory>
#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 process, Int_t& ntr, Double_t weight, Int_t status);
 
 
  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 generatorParentID);
 
 
  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(Option_t* = "") 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 SetFilter(std::unique_ptr<CbmStackFilter>& filter)
  {
    fFilter.reset();
    fFilter = std::move(filter);
  }
 
 
  void AddPoint(ECbmModuleId iDet);
 
 
  void AddPoint(ECbmModuleId iDet, Int_t iTrack);
 
 
  TParticle* GetParticle(Int_t trackId) const;
  TClonesArray* GetListOfParticles() { return fParticles; }
 
  void DoTracking(Bool_t doTracking = kTRUE) { fdoTracking = doTracking; }
 
 
private:
  std::stack<TParticle*> fStack;  
 
 
  std::unique_ptr<CbmStackFilter> fFilter;  
 
 
  TClonesArray* fParticles;  
 
 
  TClonesArray* fTracks;
 
 
  std::map<Int_t, Int_t> fIndexMap;             
  std::map<Int_t, Int_t>::iterator fIndexIter;  
 
 
  std::map<std::pair<Int_t, ECbmModuleId>, 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 fdoTracking;  
 
  CbmStack(const CbmStack&);
  CbmStack& operator=(const CbmStack&);
 
  ClassDef(CbmStack, 3)
};
 
 
#endif