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