StarPythia8Decayer.cxx

#include "StarPythia8Decayer.h"
#include "TLorentzVector.h"
#include <assert.h>
#include "TParticle.h"
#include "StarGenerator/UTIL/StarRandom.h"
#include "TSystem.h"
#include "StMessMgr.h"

#include "ParticleData.h"

#ifndef Pythia8_version 
#error "Pythia8_version is not defined"
#endif

using namespace Pythia8;

// ----------------------------------------------------------------------------
// Remap pythia's random number generator to StarRandom
class PyRand : public Pythia8::RndmEngine {
public:
  double flat() { return StarRandom::Instance().flat(); }
};

//..................................................................................................
StarPythia8Decayer::StarPythia8Decayer() : 
  mPythia(0), 
  mOwner(false), 
  mDebug(0),
  mRootS(510.0)
{

  TString path = "StRoot/StarGenerator/"; path+= Pythia8_version; path+="/xmldoc/"; 
  { 
    ifstream in(path); 
    if (!in.good()) { path = "$(STAR)/"+path; }
    path = gSystem->ExpandPathName(path.Data());
  }
  
  Info(GetName(),Form("MC version is %s data at %s",Pythia8_version,path.Data()));
  Info(GetName(),Form("Configuration files at %s",path.Data()));

  mPythia = new Pythia8::Pythia( path.Data() );
  mPythia -> setRndmEnginePtr( new PyRand() );

  mPythia -> readString("ProcessLevel:all = off"); // switch off initial state machinery
  mPythia -> readString("Check:event      = off"); // no consistency checks against IS
  mPythia -> init();
    
}
//..................................................................................................
void StarPythia8Decayer::Init()
{
#define Set(x) mPythia -> readString(x)
  Set("Beams:idA=2112");
  Set("Beams:idB=2112");
  Set("Beams:frameType=1");
  Set(Form("Beams:eCM=%f",mRootS));
  mPythia->init();
#undef Set      
}
//..................................................................................................
void StarPythia8Decayer::Decay( int pdgid, TLorentzVector *_p )
{

  LOG_INFO << "Decay pdgid=" << pdgid << endm;

  // Clear the event from the last run
  ClearEvent();
  // Add the particle to the pythia stack
  AppendParticle( pdgid, _p );
  // Get the particle ID and allow it to decay
  //  int id = mPythia -> event[0].id();
  //  mPythia->particleData.mayDecay( id, true );
  mPythia->next();

  // Print list of pythia lines on debug
  if ( mDebug ) mPythia->event.list();
}
//..................................................................................................
int StarPythia8Decayer::ImportParticles( TClonesArray *_array )
{
  // Save the decay products
  assert(_array);
  TClonesArray &array = *_array;
  array.Clear();

  int nparts = 0;

  for ( int i=0; i<mPythia->event.size();i++ )
    {
      //if ( mPythia->event[i].id() == 90 ) continue; //??
      new(array[nparts++]) TParticle (            
          mPythia->event[i].id(),
          mPythia->event[i].status(),
          mPythia->event[i].mother1(),
          mPythia->event[i].mother2(),
          mPythia->event[i].daughter1(),
          mPythia->event[i].daughter2(),
          mPythia->event[i].px(),       // [GeV/c]
          mPythia->event[i].py(),       // [GeV/c]
          mPythia->event[i].pz(),       // [GeV/c]
          mPythia->event[i].e(),        // [GeV]
          mPythia->event[i].xProd(),    // [mm]
          mPythia->event[i].yProd(),    // [mm]
          mPythia->event[i].zProd(),    // [mm]
          mPythia->event[i].tProd());   // [mm/c]                 
    };
  return nparts;
}
//..................................................................................................
// not implemented.  complain about it.  loudly.
void StarPythia8Decayer::SetForceDecay( int type ){ assert(0); }
void StarPythia8Decayer::ForceDecay(){ assert(0); }
float StarPythia8Decayer::GetPartialBranchingRatio( int ipdg ){ assert(0); }
void StarPythia8Decayer::ReadDecayTable(){ assert(0); }
//..................................................................................................
float StarPythia8Decayer::GetLifetime(int pdg) 
{
  // return lifetime in seconds of teh particle with PDG number pdg
  return (mPythia->particleData.tau0(pdg) * 3.3333e-12) ;
}
//..................................................................................................
void StarPythia8Decayer::AppendParticle(int pdg, TLorentzVector* p)
{
  // Append a particle to the stack to be decayed  
  const int status = 23;

  double pt2 = p->Perp2();   // pt squared
  double px  = p->Px();
  double py  = p->Py();
  double pz  = p->Pz();  // z-component
  double M2  = p->M2();   // mass squared

  // Lookup particle entry in pythia8
  const auto* pde = mPythia->particleData.particleDataEntryPtr( pdg );
  if ( pde ) {
    M2 = pde->m0() * pde->m0();
  }
 
  double E = TMath::Sqrt( pt2 + M2 + pz*pz );
 
  mPythia->event.append(pdg, status, 0, 0, px, py, pz, E, TMath::Sqrt(M2) );
}
//..................................................................................................
void StarPythia8Decayer::ClearEvent()
{
  mPythia->event.reset(); // Reset the event
}
//..................................................................................................
StarPythia8Decayer::~StarPythia8Decayer()
{
  if (mPythia) delete mPythia; // delete 
}