FairPlutoReactionGenerator.cxx

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----             FairPlutoReactionGenerator header file            -----
// -----          Created 05/07/08  by I. Froehlich                    -----
// -------------------------------------------------------------------------
 
 
 
#include "FairPlutoReactionGenerator.h"
 
#include "FairPrimaryGenerator.h"
 
#include "TClonesArray.h"
#include "TDatabasePDG.h"
#include "TFile.h"
#include "TLorentzVector.h"
#include "TTree.h"
#include "TVector3.h"
 
#include <iostream>
 
 
// -----   Default constructor   ------------------------------------------
FairPlutoReactionGenerator::FairPlutoReactionGenerator()
{
  iEvent     = 0;
  fReaction  = NULL;
#ifdef PLUTOVERSION_FOR_FAIR
  fPlutoFairGenerator = NULL;
#endif
}
// ------------------------------------------------------------------------
 
 
 
// -----   Standard constructor   -----------------------------------------
FairPlutoReactionGenerator::FairPlutoReactionGenerator(PReaction* reaction)
{
  iEvent     = 0;
  fReaction  = reaction;
#ifdef PLUTOVERSION_FOR_FAIR
  fPlutoFairGenerator = new PFairGenerator();
  if (!reaction) {
    std::cout << "-E FairPlutoReactionGenerator: No PReaction!" << std::endl;
  } else {
    reaction->AddBulk(fPlutoFairGenerator);
    reaction->IsInline();
  }
#else
  std::cout << "-E FairPlutoReactionGenerator: You are using the wrong Pluto version" << std::endl;
#endif
}
// ------------------------------------------------------------------------
 
 
 
// -----   Destructor   ---------------------------------------------------
FairPlutoReactionGenerator::~FairPlutoReactionGenerator()
{
#ifdef PLUTOVERSION_FOR_FAIR
  if (fPlutoFairGenerator) { delete fPlutoFairGenerator; }
#endif
}
// ------------------------------------------------------------------------
 
 
 
// -----   Public method ReadEvent   --------------------------------------
Bool_t FairPlutoReactionGenerator::ReadEvent(FairPrimaryGenerator* primGen)
{
#ifdef PLUTOVERSION_FOR_FAIR
 
  // Check for input file
  if ( ! fReaction ) {
    cout << "-E FairPlutoReactionGenerator: No PReaction!" << endl;
    return kFALSE;
  }
 
  fReaction->InitChannels();
  fReaction->DisableWeightReset();
  if (!fReaction->loop(1,0,0)) { return kFALSE; }
 
 
  Double_t px, py, pz, vx, vy, vz;
  Int_t    pdgType;
 
  while (fPlutoFairGenerator->GetNextParticle(&pdgType, &px, &py, &pz, &vx, &vy, &vz)) {
 
    if (pdgType) {
      primGen->AddTrack(pdgType, px, py, pz, vx, vy, vz);
    }
  }
 
  return kTRUE;
#else
 
  return kFALSE;
#endif
}
// ------------------------------------------------------------------------