ShipStack.h

Go to the documentation of this file.

// SPDX-License-Identifier: LGPL-3.0-or-later
// SPDX-FileCopyrightText: Copyright CERN for the benefit of the SHiP
// Collaboration
 
// -------------------------------------------------------------------------
// -----                       ShipStack header file                   -----
// -------------------------------------------------------------------------
 
#ifndef SHIPDATA_SHIPSTACK_H_
#define SHIPDATA_SHIPSTACK_H_
 
#include <map>      // for map, map<>::iterator
#include <stack>    // for stack
#include <utility>  // for pair
#include <vector>   // for vector
 
#include "FairGenericStack.h"  // for FairGenericStack
#include "Rtypes.h"            // for Int_t, Double_t, Bool_t, etc
#include "ShipDetectorList.h"  // for DetectorId
#include "TMCProcess.h"        // for TMCProcess
 
class TClonesArray;
class ShipMCTrack;
class TParticle;
class TRefArray;
class FairLogger;
 
enum { kDoneBit = 1 };
 
class ShipStack : public FairGenericStack {
 public:
  explicit ShipStack(Int_t size = 100);
 
  ~ShipStack() override;
 
  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) override;
 
  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) override;
 
  TParticle* PopNextTrack(Int_t& iTrack) override;
 
  TParticle* PopPrimaryForTracking(Int_t iPrim) override;
 
  void SetCurrentTrack(Int_t iTrack) override { fCurrentTrack = iTrack; }
 
  Int_t GetNtrack() const override { return fNParticles; }
 
  Int_t GetNprimary() const override { return fNPrimaries; }
 
  TParticle* GetCurrentTrack() const override;
 
  Int_t GetCurrentTrackNumber() const override { return fCurrentTrack; }
 
  Int_t GetCurrentParentTrackNumber() const override;
 
  void AddParticle(TParticle* part);
 
  void FillTrackArray() override;
 
  void UpdateTrackIndex(TRefArray* detArray = 0) override;
 
  void Reset() override;
 
  void Register() override;
 
  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;  
 
  std::vector<ShipMCTrack>* 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&) = delete;
  ShipStack& operator=(const ShipStack&) = delete;
 
  ClassDefOverride(ShipStack, 1)
};
 
#endif  // SHIPDATA_SHIPSTACK_H_