GeneratorMessenger.cc

Go to the documentation of this file.

// $Header: /cvs/lcd/slic/src/GeneratorMessenger.cc,v 1.14 2010/01/08 23:18:56 jeremy Exp $
#include "GeneratorMessenger.hh"

// slic
#include "EventSourceManager.hh"

// lcdd
#include "StringUtil.hh"

// geant4
#include "G4UIcommand.hh"
#include "G4UIdirectory.hh"
#include "G4UIcmdWithAnInteger.hh"
#include "G4UIcmdWithADoubleAndUnit.hh"

// clhep
#include "CLHEP/Random/Random.h"
#include "CLHEP/Random/Randomize.h"

namespace CLHEP {}
using namespace CLHEP;

namespace slic
{

    GeneratorMessenger::GeneratorMessenger()
    {
        defineCommands();
    }

    GeneratorMessenger::~GeneratorMessenger()
    {}

    void GeneratorMessenger::SetNewValue(G4UIcommand* cmd, G4String newVals)
    {
        std::istringstream is ( ( const char* ) newVals );

        std::string s;
        is >> s;

        EventSourceManager* mgr = EventSourceManager::instance();

        // select
        if ( cmd == m_selectCmd ) 
        {
            mgr->setupEventSource( s );
        }
        // reset
        else if ( cmd == m_resetCmd ) 
        {
            mgr->resetCurrentEventSource();
        }
        // skipEvents
        else if ( cmd == m_skipEventsCmd ) 
        {
            mgr->setSkipEvents( StringUtil::toInt( s ) );
        }
        // filename
        else if ( cmd == m_filenameCmd ) 
        {
            mgr->setFilename( s );
        }
        // dumpCurrentEvent
        else if ( cmd == m_dumpCurrentEventCmd ) 
        {
            mgr->dumpCurrentEvent();
        }
        // printNumEventsGenerated
        else if ( cmd == m_printNumEventsGeneratedCmd ) 
        {
            mgr->printNumEventsGenerated();
        }
        // set random seed
        else if ( cmd == m_randomSeedCmd ) 
        {
            G4int seed = 0;

            // seed from user arg
            if ( s != std::string("") ) {
                seed = StringUtil::toInt( s );
            }
            // seed from time
            // @todo Get MS instead of just seconds, which is not very good quality for a seed.
            else {
                seed = ((unsigned)time(NULL));
            }

            // seed it
            HepRandom::setTheSeed( seed );

            mgr->log() << LOG::okay << "set random seed: " << seed << LOG::done;
        }
        // Lorentz transformation angle
        else if ( cmd == m_setLorentzTransformationAngleCmd ) 
        {
            EventSourceManager::instance()->setLorentzTransformationAngle(G4UIcmdWithADoubleAndUnit::GetNewDoubleValue(newVals));
        }
        else {
            G4Exception("Unknown cmd for this messenger.");
        }
    }

    void GeneratorMessenger::defineCommands()
    {
        G4UIparameter *p;

        // generator dir
        m_generatorDir = new G4UIdirectory( "/generator/" );
        m_generatorDir->SetGuidance( "Event generation commands. [SLIC]" );

        // select
        m_selectCmd = new G4UIcommand( "/generator/select", this );
        m_selectCmd->SetGuidance( "Set Generator to use: lcio, stdhep, gps or gun." );
        p = new G4UIparameter( "generator", 's', false );
        m_selectCmd->SetParameter(p);

        // reset
        m_resetCmd = new G4UIcommand( "/generator/reset", this);
        m_resetCmd->SetGuidance( "Reset the generator using the current info." );

        // skipEvents
        m_skipEventsCmd = new G4UIcmdWithAnInteger( "/generator/skipEvents", this);
        m_skipEventsCmd->SetGuidance( "Skip n events of current generator." );
        m_skipEventsCmd->SetParameterName( "NumEvents", false);

        // filename
        m_filenameCmd = new G4UIcommand( "/generator/filename", this);
        m_filenameCmd->SetGuidance( "Set generator input filename.");
        p = new G4UIparameter( "filename", 's', false);
        m_filenameCmd->SetParameter(p);

        // dumpCurrentEvent
        m_dumpCurrentEventCmd = new G4UIcommand( "/generator/dumpEvent", this);
        m_dumpCurrentEventCmd->SetGuidance( "Dump information about the current generator event." );

        // printNumEventsGenerated
        m_printNumEventsGeneratedCmd = new G4UIcommand( "/generator/printNumEventsGenerated", this);
        m_printNumEventsGeneratedCmd->SetGuidance( "Print out number of events generated with current generator.");

        // random seed cmd
        m_randomSeedCmd = new G4UIcommand( "/random/seed", this );
        m_randomSeedCmd->SetGuidance( "Set random seed; no argument seeds with current time. [SLIC]" );
        p = new G4UIparameter( "seedValue", 'i', true );
        m_randomSeedCmd->SetParameter( p );

        // Lorentz transformation
        m_setLorentzTransformationAngleCmd = new G4UIcmdWithADoubleAndUnit("/generator/setLorentzTransformationAngle",this);
        m_setLorentzTransformationAngleCmd->SetGuidance( "Set the Lorentz transformation angle to boost generated events." );
    }
}

---