MyProjStack.h

Go to the documentation of this file.

/********************************************************************************
 *    Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *              GNU Lesser General Public Licence (LGPL) version 3,             *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/

// -------------------------------------------------------------------------
// -----                       MyProjStack header file                 -----
// -----                  M. Al-Turany   June 2014                     -----
// -------------------------------------------------------------------------

#ifndef MyProjStack_H
#define MyProjStack_H

#include "FairGenericStack.h"     // for FairGenericStack
#include "MyProjDetectorList.h"   // for DetectorId

#include <Rtypes.h>       // for Int_t, Double_t, Bool_t, etc
#include <TMCProcess.h>   // for TMCProcess
#include <map>            // for map, map<>::iterator
#include <stack>          // for stack
#include <utility>        // for pair

class TClonesArray;
class TParticle;
class TRefArray;

class MyProjStack : public FairGenericStack
{

  public:
    MyProjStack(Int_t size = 100);

    virtual ~MyProjStack();

    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 = 0);

    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; }

    virtual FairGenericStack* CloneStack() const { return new MyProjStack(*this); }

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

    MyProjStack(const MyProjStack&);
    MyProjStack& operator=(const MyProjStack&);

    ClassDef(MyProjStack, 1);
};

#endif