Pythia6Generator.cxx

Go to the documentation of this file.

/********************************************************************************
 *    Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *              GNU Lesser General Public Licence (LGPL) version 3,             *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/
// -------------------------------------------------------------------------
// -----                  M. Al-Turany   June 2014                     -----
// -------------------------------------------------------------------------
// -------------------------------------------------------------------------
// -----             Pythia6Generator source file                      -----
// -----          Created 08/08/08  by S. Spataro                      -----
// -------------------------------------------------------------------------
#include "Pythia6Generator.h"

#include "FairLogger.h"
#include "FairPrimaryGenerator.h"

#include <cstdio>
#include <iostream>

using std::cout;
using std::endl;
using std::max;

// -----   Default constructor   ------------------------------------------
Pythia6Generator::Pythia6Generator() {}
// ------------------------------------------------------------------------

// -----   Standard constructor   -----------------------------------------
Pythia6Generator::Pythia6Generator(const char* fileName)
{
    fFileName = fileName;
    fVerbose = 0;
    cout << "-I Pythia6Generator: Opening input file " << fileName << endl;
    if ((fInputFile = fopen(fFileName, "r")) == NULL)
        //  fInputFile = new ifstream(fFileName);
        //  if ( ! fInputFile->is_open() )
        LOG(fatal) << "Pythia6Generator: Cannot open input file.";

    // fPDG=TDatabasePDG::Instance();
}
// ------------------------------------------------------------------------

// -----   Destructor   ---------------------------------------------------
Pythia6Generator::~Pythia6Generator() { CloseInput(); }
// ------------------------------------------------------------------------

// -----   Public method ReadEvent   --------------------------------------
Bool_t Pythia6Generator::ReadEvent(FairPrimaryGenerator* primGen)
{

    // Check for input file
    if (!fInputFile) {
        // if ( ! fInputFile->is_open() ) {
        cout << "-E Pythia6Generator: Input file not open!" << endl;
        return kFALSE;
    }

    // Define event variable to be read from file
    Int_t ntracks = 0, eventID = 0, ncols = 0;

    // Define track variables to be read from file
    Int_t nLev = 0, pdgID = 0, nM1 = -1, nM2 = -1, nDF = -1, nDL = -1;
    Float_t fPx = 0., fPy = 0., fPz = 0., fM = 0., fE = 0.;
    Float_t fVx = 0., fVy = 0., fVz = 0., fT = 0.;

    // Read event header line from input file

    Int_t max_nr = 0;

    Text_t buffer[200];
    ncols = fscanf(fInputFile, "%d\t%d", &eventID, &ntracks);

    if (ncols && ntracks > 0) {

        if (fVerbose > 0)
            cout << "Event number: " << eventID << "\tNtracks: " << ntracks << endl;

        for (Int_t ll = 0; ll < ntracks; ll++) {
            ncols = fscanf(fInputFile,
                           "%d %d %d %d %d %d %f %f %f %f %f %f %f %f %f",
                           &nLev,
                           &pdgID,
                           &nM1,
                           &nM2,
                           &nDF,
                           &nDL,
                           &fPx,
                           &fPy,
                           &fPz,
                           &fE,
                           &fM,
                           &fVx,
                           &fVy,
                           &fVz,
                           &fT);
            if (fVerbose > 0)
                cout << nLev << "\t" << pdgID << "\t" << nM1 << "\t" << nM2 << "\t" << nDF << "\t" << nDL << "\t" << fPx
                     << "\t" << fPy << "\t" << fPz << "\t" << fE << "\t" << fM << "\t" << fVx << "\t" << fVy << "\t"
                     << fVz << "\t" << fT << endl;
            if (nLev == 1)
                primGen->AddTrack(pdgID, fPx, fPy, fPz, fVx, fVy, fVz);
        }
    } else {
        cout << "-I Pythia6Generator: End of input file reached " << endl;
        CloseInput();
        return kFALSE;
    }

    // If end of input file is reached : close it and abort run
    if (feof(fInputFile)) {
        cout << "-I Pythia6Generator: End of input file reached " << endl;
        CloseInput();
        return kFALSE;
    }

    /*
    cout << "-I Pythia6Generator: Event " << eventID << ",  vertex = ("
    << vx << "," << vy << "," << vz << ") cm,  multiplicity "
    << ntracks << endl;
  */

    return kTRUE;
}
// ------------------------------------------------------------------------

// -----   Private method CloseInput   ------------------------------------
void Pythia6Generator::CloseInput()
{
    if (fInputFile) {
        // if ( fInputFile->is_open() ) {
        {
            cout << "-I Pythia6Generator: Closing input file " << fFileName << endl;
            //  fInputFile->close();

            fclose(fInputFile);
        }
        delete fInputFile;
        fInputFile = NULL;
    }
}
// ------------------------------------------------------------------------

ClassImp(Pythia6Generator);