EvtModel.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 "EvtGenBase/EvtModel.hh"
 
#include "EvtGenBase/EvtDecayBase.hh"
#include "EvtGenBase/EvtPDL.hh"
#include "EvtGenBase/EvtParser.hh"
#include "EvtGenBase/EvtParticle.hh"
#include "EvtGenBase/EvtParticleDecayList.hh"
#include "EvtGenBase/EvtRandom.hh"
#include "EvtGenBase/EvtReport.hh"
 
#include <assert.h>
#include <ctype.h>
#include <fstream>
#include <iomanip>
#include <iostream>
#include <stdlib.h>
#include <string>
using std::fstream;
 
thread_local EvtModel* EvtModel::m_instance = nullptr;
 
EvtModel::EvtModel()
{
}
 
EvtDecayBase* EvtModel::getFcn( std::string model_name )
{
    EvtDecayBase* model = nullptr;
    if ( m_modelNameHash.find( model_name ) != m_modelNameHash.end() ) {
        model = m_modelNameHash[model_name];
    }
 
    if ( !model ) {
        EvtGenReport( EVTGEN_ERROR, "EvtGen" )
            << "Did not find the right model:" << model_name.c_str() << "\n";
        return nullptr;
    }
 
    return model->clone();
}
 
void EvtModel::registerModel( EvtDecayBase* prototype )
{
    std::string modelName = prototype->getName();
 
    m_modelNameHash[modelName] = prototype;
 
    std::string commandName = prototype->commandName();
 
    if ( commandName != "" ) {
        m_commandNameHash[commandName] = prototype;
    }
}
 
int EvtModel::isModel( std::string model_name )
{
    if ( m_modelNameHash.find( model_name ) != m_modelNameHash.end() ) {
        return 1;
    }
    return 0;
}
 
int EvtModel::isCommand( std::string cmd )
{
    if ( m_commandNameHash.find( cmd ) != m_commandNameHash.end() ) {
        return 1;
    }
    return 0;
}
 
void EvtModel::storeCommand( std::string cmd, std::string cnfgstr )
{
    EvtDecayBase* model = nullptr;
    if ( m_commandNameHash.find( cmd ) != m_commandNameHash.end() ) {
        model = m_commandNameHash[cmd];
    }
 
    assert( model );
 
    model->command( cnfgstr );
}