EvtExternalGenFactory.cpp

Go to the documentation of this file.

/***********************************************************************
* Copyright 1998-2020 CERN for the benefit of the EvtGen authors       *
*                                                                      *
* This file is part of EvtGen.                                         *
*                                                                      *
* EvtGen is free software: you can redistribute it and/or modify       *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation, either version 3 of the License, or    *
* (at your option) any later version.                                  *
*                                                                      *
* EvtGen is distributed in the hope that it will be useful,            *
* but WITHOUT ANY WARRANTY; without even the implied warranty of       *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the        *
* GNU General Public License for more details.                         *
*                                                                      *
* You should have received a copy of the GNU General Public License    *
* along with EvtGen.  If not, see <https://www.gnu.org/licenses/>.     *
***********************************************************************/

#include "EvtGenExternal/EvtExternalGenFactory.hh"

#include "EvtGenBase/EvtPatches.hh"
#include "EvtGenBase/EvtReport.hh"

#ifdef EVTGEN_PYTHIA
#include "EvtGenExternal/EvtPythiaEngine.hh"
#endif

#ifdef EVTGEN_PHOTOS
#include "EvtGenExternal/EvtPhotosEngine.hh"
#endif

#ifdef EVTGEN_TAUOLA
#include "EvtGenExternal/EvtTauolaEngine.hh"
#endif

#include <iostream>
using std::endl;

EvtExternalGenFactory::EvtExternalGenFactory()
{
    _extGenMap.clear();
}

EvtExternalGenFactory::~EvtExternalGenFactory()
{
    ExtGenMap::iterator iter;
    for ( iter = _extGenMap.begin(); iter != _extGenMap.end(); ++iter ) {
        EvtAbsExternalGen* theGenerator = iter->second;
        delete theGenerator;
    }

    _extGenMap.clear();
}

EvtExternalGenFactory* EvtExternalGenFactory::getInstance()
{
    static EvtExternalGenFactory* theFactory = 0;

    if ( theFactory == 0 ) {
        theFactory = new EvtExternalGenFactory();
    }

    return theFactory;
}

void EvtExternalGenFactory::definePythiaGenerator( std::string xmlDir,
                                                   bool convertPhysCodes,
                                                   bool useEvtGenRandom )
{
    // Only define the generator if we have the external ifdef variable set
#ifdef EVTGEN_PYTHIA

    int genId = EvtExternalGenFactory::PythiaGenId;

    EvtGenReport( EVTGEN_INFO, "EvtGen" )
        << "Defining EvtPythiaEngine: data tables defined in " << xmlDir << endl;
    if ( convertPhysCodes == true ) {
        EvtGenReport( EVTGEN_INFO, "EvtGen" )
            << "Pythia 6 codes in decay files will be converted to Pythia 8 codes"
            << endl;
    } else {
        EvtGenReport( EVTGEN_INFO, "EvtGen" )
            << "Pythia 8 codes need to be used in decay files" << endl;
    }

    if ( useEvtGenRandom == true ) {
        EvtGenReport( EVTGEN_INFO, "EvtGen" )
            << "Using EvtGen random engine for Pythia 8 as well" << endl;
    }

    EvtAbsExternalGen* pythiaGenerator =
        new EvtPythiaEngine( xmlDir, convertPhysCodes, useEvtGenRandom );
    _extGenMap[genId] = pythiaGenerator;

#endif
}

void EvtExternalGenFactory::definePhotosGenerator( std::string photonType,
                                                   bool useEvtGenRandom )
{
#ifdef EVTGEN_PHOTOS

    int genId = EvtExternalGenFactory::PhotosGenId;

    EvtGenReport( EVTGEN_INFO, "EvtGen" )
        << "Defining EvtPhotosEngine using photonType = " << photonType << endl;

    EvtAbsExternalGen* photosGenerator = new EvtPhotosEngine( photonType,
                                                              useEvtGenRandom );
    _extGenMap[genId] = photosGenerator;

#endif
}

void EvtExternalGenFactory::defineTauolaGenerator( bool useEvtGenRandom )
{
#ifdef EVTGEN_TAUOLA

    int genId = EvtExternalGenFactory::TauolaGenId;

    EvtGenReport( EVTGEN_INFO, "EvtGen" ) << "Defining EvtTauolaEngine." << endl;

    EvtAbsExternalGen* tauolaGenerator = new EvtTauolaEngine( useEvtGenRandom );
    _extGenMap[genId] = tauolaGenerator;

#endif
}

EvtAbsExternalGen* EvtExternalGenFactory::getGenerator( int genId )
{
    EvtAbsExternalGen* theGenerator( 0 );

    ExtGenMap::iterator iter;

    if ( ( iter = _extGenMap.find( genId ) ) != _extGenMap.end() ) {
        // Retrieve the external generator engine
        theGenerator = iter->second;

    } else {
        EvtGenReport( EVTGEN_INFO, "EvtGen" )
            << "EvtAbsExternalGen::getGenerator: could not find generator for genId = "
            << genId << endl;
    }

    return theGenerator;
}

void EvtExternalGenFactory::initialiseAllGenerators()
{
    ExtGenMap::iterator iter;
    for ( iter = _extGenMap.begin(); iter != _extGenMap.end(); ++iter ) {
        EvtAbsExternalGen* theGenerator = iter->second;
        if ( theGenerator != 0 ) {
            theGenerator->initialise();
        }
    }
}