TMCManagerStack.h

Go to the documentation of this file.

// -----------------------------------------------------------------------
// Copyright (C) 2019 CERN and copyright holders of VMC Project.
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "LICENSE".
//
// See https://github.com/vmc-project/vmc for full licensing information.
// -----------------------------------------------------------------------
 
// Authors: Benedikt Volkel 07/03/2019
 
/*************************************************************************
 * Copyright (C) 2019, Rene Brun and Fons Rademakers.                    *
 * Copyright (C) 2019, ALICE Experiment at CERN.                         *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/
 
#ifndef ROOT_TMCManagerStack
#define ROOT_TMCManagerStack
 
// Class TMCManagerStack
// ---------------------
// stack used by the TMCManager when handling multiple engines
//
 
#include <vector>
#include <stack>
#include <memory>
 
#include "TMCtls.h"
#include "TLorentzVector.h"
#include "TMCProcess.h"
 
#include "TVirtualMCStack.h"
 
struct TMCParticleStatus;
class TGeoBranchArray;
class TGeoMCBranchArrayContainer;
 
class TMCManagerStack : public TVirtualMCStack {
 
public:
   TMCManagerStack();
   virtual ~TMCManagerStack() = default;
 
   //
   // Methods for stacking
   //
 
   void PushTrack(Int_t toBeDone, Int_t parent, Int_t pdg, Double_t px, Double_t py, Double_t pz, Double_t e,
                  Double_t vx, Double_t vy, Double_t vz, Double_t tof, Double_t polx, Double_t poly, Double_t polz,
                  TMCProcess mech, Int_t &ntr, Double_t weight, Int_t is) override final;
 
   //
   // Get methods
   //
 
   TParticle *PopNextTrack(Int_t &itrack) override final;
 
   TParticle *PopPrimaryForTracking(Int_t i) override final;
 
   TParticle *PopPrimaryForTracking(Int_t i, Int_t &itrack);
 
   Int_t GetNtrack() const override final;
 
   Int_t GetStackedNtrack() const;
 
   Int_t GetNprimary() const override final;
 
   Int_t GetStackedNprimary() const;
 
   TParticle *GetCurrentTrack() const override final;
 
   Int_t GetCurrentTrackNumber() const override final;
 
   Int_t GetCurrentParentTrackNumber() const override final;
 
   void SetCurrentTrack(Int_t trackId) override final;
 
   const TMCParticleStatus *GetParticleStatus(Int_t trackId) const;
 
   const TGeoBranchArray *GetGeoState(Int_t trackId) const;
 
   const TGeoBranchArray *GetCurrentGeoState() const;
 
private:
   friend class TMCManager;
   Bool_t HasTrackId(Int_t trackId) const;
   void SetUserStack(TVirtualMCStack *stack);
   void ConnectTrackContainers(std::vector<TParticle *> *particles,
                               std::vector<std::unique_ptr<TMCParticleStatus>> *tracksStatus,
                               TGeoMCBranchArrayContainer *branchArrayContainer, Int_t *totalNPrimaries,
                               Int_t *totalNTracks);
   void PushPrimaryTrackId(Int_t trackId);
   void PushSecondaryTrackId(Int_t trackId);
   void ResetInternals();
 
private:
   Int_t fCurrentTrackId;
   TVirtualMCStack *fUserStack;
   Int_t *fTotalNPrimaries;
   Int_t *fTotalNTracks;
   std::vector<TParticle *> *fParticles;
   std::vector<std::unique_ptr<TMCParticleStatus>> *fParticlesStatus;
   TGeoMCBranchArrayContainer *fBranchArrayContainer;
   std::stack<Int_t> fPrimariesStack;
   std::stack<Int_t> fSecondariesStack;
 
   ClassDefOverride(TMCManagerStack, 1)
};
 
#endif // ROOT_TMCManagerStack