PisaSimpleGenerator.C

Go to the documentation of this file.

#include "PisaSimpleGenerator.h"

#include <cassert>

#include "PHCompositeNode.h"
#include "phool.h"
#include "PisaHelper.h"
#include "PisaStack.h"

#include "TVirtualMCStack.h"

//_____________________________________________________________________________
PisaSimpleGenerator::PisaSimpleGenerator()
  : SubsysReco("PisaSimpleGenerator")
{
}

//_____________________________________________________________________________
PisaSimpleGenerator::~PisaSimpleGenerator()
{
}

//_____________________________________________________________________________
int
PisaSimpleGenerator::process_event(PHCompositeNode* topnode)
{
  PisaStack* stack = PisaHelper::getClass<PisaStack>(topnode,"Stack");
  assert(stack!=0);

  TVirtualMCStack* fStack = stack->stack();
  assert(fStack!=0);

  // Track ID (filled by stack)
  Int_t ntr;
  
  // Option: to be tracked
  Int_t toBeDone = 1; 
  
  // Geantino
  Int_t pdg  = 0;
  
  // Polarization
  Double_t polx = 0.; 
  Double_t poly = 0.; 
  Double_t polz = 0.; 
  
  // Position
  Double_t vx  = -200.; 
  Double_t vy  = 0.; 
  Double_t vz  = 0.; 
  Double_t tof = 0.;

  // Momentum 
  Double_t px, py, pz, e;
  px = 0.2; 
  py = 0.2; 
  pz = 1.0; 
  e  = sqrt(px*px+py*py+pz*pz);
  
  // Add particle to stack 
  fStack->PushTrack(toBeDone, -1, pdg, px, py, pz, e, vx, vy, vz, tof, polx, poly, polz, 
                    kPPrimary, ntr, 1., 0);
  
  return 0;
}

//_____________________________________________________________________________
int
PisaSimpleGenerator::ResetEvent(PHCompositeNode* topnode)
{
  PisaStack* stack = PisaHelper::getClass<PisaStack>(topnode,"Stack");
  assert(stack!=0);
  stack->Reset();
  return 0;
}