PisaVMCStackv1.C

Go to the documentation of this file.

#include "PisaVMCStackv1.h"

#include <cassert>
#include <iostream>

#include "TClonesArray.h"
#include "TParticle.h"

ClassImp(PisaVMCStackv1)

//_____________________________________________________________________________
PisaVMCStackv1::PisaVMCStackv1()
  : fParticles(0),
    fCurrentTrack(-1),
    fNPrimary(0)
{
}

//_____________________________________________________________________________
PisaVMCStackv1::PisaVMCStackv1(size_t thesize)
  : fParticles(new TClonesArray("TParticle",thesize)),
    fCurrentTrack(-1),
    fNPrimary(0)
{  
}

//_____________________________________________________________________________
PisaVMCStackv1::~PisaVMCStackv1()
{
  if ( fParticles )
    {
      fParticles->Delete();
    }
  delete fParticles;
}

//_____________________________________________________________________________
void
PisaVMCStackv1::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) 
{
// Creates a new particle with specified properties,
// adds it to the particles array (fParticles) and if not done to the 
// stack (fStack).
// Use TParticle::fMother[1] to store Track ID. 
// ---

  const Int_t kFirstDaughter=-1;
  const Int_t kLastDaughter=-1;
  
  TClonesArray& particlesRef = *fParticles;
  Int_t trackId = GetNtrack();
  assert(trackId>=0);
  TParticle* particle
    = new(particlesRef[trackId]) 
      TParticle(pdg, is, parent, trackId, kFirstDaughter, kLastDaughter,
                px, py, pz, e, vx, vy, vz, tof);
   
  particle->SetPolarisation(polx, poly, polz);
  particle->SetWeight(weight);
  particle->SetUniqueID(mech);

  if (parent<0) fNPrimary++;  
    
  if (toBeDone) fStack.push(particle);  
  ntr = trackId;
}                        

//_____________________________________________________________________________
TParticle* 
PisaVMCStackv1::PopNextTrack(Int_t& itrack)
{
// Gets next particle for tracking from the stack.
// ---

  itrack = -1;
  if  (fStack.empty()) return 0;
                      
  TParticle* particle = fStack.top();
  fStack.pop();

  if (!particle) return 0;  
  
  fCurrentTrack = particle->GetSecondMother();
  itrack = fCurrentTrack;
  
  return particle;
}    

//_____________________________________________________________________________
TParticle* 
PisaVMCStackv1::PopPrimaryForTracking(Int_t i)
{
// Returns i-th particle in fParticles.
// ---

  if (i < 0 || i >= fNPrimary)
    Fatal("GetPrimaryForTracking", "Index out of range"); 
  
  return (TParticle*)fParticles->At(i);
}     

//_____________________________________________________________________________
void 
PisaVMCStackv1::Print(Option_t*) const 
{
  //Prints info for all particles.

  std::cout << "PisaVMCStackv1 Info  " << std::endl;
  std::cout << "Total number of particles:   " <<  GetNtrack() << std::endl;
  std::cout << "Number of primary particles: " <<  GetNprimary() << std::endl;

  for (Int_t i=0; i<GetNtrack(); i++)
    {
      GetParticle(i)->Print();
    }
}

//_____________________________________________________________________________
void
PisaVMCStackv1::Reset()
{
  //Deletes contained particles, resets particles array and stack.

  fCurrentTrack = -1;
  fNPrimary = 0;
  fParticles->Clear();
}       

//_____________________________________________________________________________
void  
PisaVMCStackv1::SetCurrentTrack(Int_t track) 
{
// Sets the current track to a given value.
// ---

  fCurrentTrack = track;
}     

//_____________________________________________________________________________
Int_t  
PisaVMCStackv1::GetNtrack() const 
{
// Returns the number of all tracks.
// ---

  return fParticles->GetEntriesFast();
}  

//_____________________________________________________________________________
Int_t 
PisaVMCStackv1::GetNprimary() const 
{
// Returns the number of primary tracks.
// ---

  return fNPrimary;
}  

//_____________________________________________________________________________
TParticle*  
PisaVMCStackv1::GetCurrentTrack() const 
{
// Returns the current track parent ID.
// ---

  TParticle* current = GetParticle(fCurrentTrack);

  if (!current)    
    Warning("GetCurrentTrack", "Current track not found in the stack");

  return current;
}  

//_____________________________________________________________________________
Int_t  
PisaVMCStackv1::GetCurrentTrackNumber() const 
{
// Returns the current track ID.
// ---

  return fCurrentTrack;
}  

//_____________________________________________________________________________
Int_t  
PisaVMCStackv1::GetCurrentParentTrackNumber() const 
{
// Returns the current track parent ID.
// ---

  TParticle* current = GetCurrentTrack();

  if (current) 
    return current->GetFirstMother();
  else 
    return -1;
}  

//_____________________________________________________________________________
TParticle*  
PisaVMCStackv1::GetParticle(Int_t id) const
{
// Returns id-th particle in fParticles.
// ---

  if (id < 0 || id >= fParticles->GetEntriesFast())
    Fatal("GetParticle", "Index out of range"); 
   
  return (TParticle*)fParticles->At(id);
}