BesGenModule Class Reference

#include <BesGenModule.h>

Inheritance diagram for BesGenModule:

List of all members.

Public Member Functions
	BesGenModule (const std::string &name, ISvcLocator *pSvcLocator)
virtual	~BesGenModule ()
virtual StatusCode	genInitialize ()
virtual StatusCode	callGenerator ()
virtual StatusCode	genFinalize ()
virtual StatusCode	fillEvt (GenEvent *evt)
StatusCode	initialize ()
StatusCode	execute ()
StatusCode	finalize ()
virtual StatusCode	genuserInitialize ()
Public Attributes
IBesRndmGenSvc *	p_BesRndmGenSvc
Static Public Attributes
static std::string	besMcGen_stream = "PHOTOS_INIT"
Protected Member Functions
void	StripPartonsInit (void)
void	StripPartons (GenEvent *evt)
Protected Attributes
int	m_fortout
std::vector< long int >	m_seeds
const char *	cardname
std::string	runname
int	Nr
int	mvec [10]
int	nvec [10]
std::vector< int >	m_par
int	m_events
int	n
int	n1
int	m_GenIndex
int	m_SuperIndex
int	m_SubIndex
bool	m_fixedMode
double	m_meanInteractions
int	m_randomSeed
std::vector< int >	m_AllPartons
std::vector< int >	m_StripVector
bool	m_StripPartonSwitch
CLHEP::HepRandomEngine *	m_pRandomEngine
CLHEP::RandPoisson *	m_pPoissonGenerator
HepPDT::ParticleDataTable *	m_particleTable

---

Detailed Description

Definition at line 21 of file BesGenModule.h.

---

Constructor & Destructor Documentation

BesGenModule::BesGenModule	(	const std::string &	name,
		ISvcLocator *	pSvcLocator
	)

Definition at line 73 of file BesGenModule.cxx.

References besMcGen_stream, Nr, and runname.

                               : GenModule(name,pSvcLocator)
{
//--------------------------------------------------------------------------  
  besMcGen_stream       =       "PHOTOS_INIT";
//  declareProperty("BesMcGenCommand", m_BesMcGenCommandVector);

  declareProperty("Runcards",  runname = "run.cards");
  declareProperty("Ratio",     Nr = 1);  

}

BesGenModule::~BesGenModule

(

)

[virtual]

Definition at line 85 of file BesGenModule.cxx.

{}

---

Member Function Documentation

StatusCode BesGenModule::callGenerator

(

)

[virtual]

Reimplemented from GenModule.

Definition at line 143 of file BesGenModule.cxx.

References Bes_Common::INFO, m_events, mainexe_(), mvec, Nr, and DecayMode::putData().

                                       {
//---------------------------------------------------------------------------
  MsgStream log(messageService(), name());
  log << MSG::INFO << " BesGenModule generating.  \n"  << endreq;

  StatusCode status = StatusCode::SUCCESS;
  int iq;  

//  n=1;


  mainexe_(&Nr,mvec);

/*
    for (int iq = 1; iq < 11; iq++) {
       nvec[iq-1]=mvec[iq];
    }
*/

    McEvent* aMcEvent = new McEvent;
    StatusCode sc = eventSvc()->registerObject("/Event/MC",aMcEvent);


    SmartDataPtr<DecayMode> anMcCol(eventSvc(), "/Event/MC/DecayMode");
    if (anMcCol!=0) {
//     log << MSG::INFO << "Add McEvent to existing collection" << endreq;
    }
    else {
//     log << MSG::INFO << "No decay mode now" << endreq;
        DecayMode *decayMode = new DecayMode;
        decayMode->putData(mvec, 10);
    StatusCode sc = eventSvc()->registerObject("/Event/MC/DecayMode",decayMode);   
    }


    SmartDataPtr<DecayMode> aMcCol(eventSvc(), "/Event/MC/DecayMode");
    if (aMcCol!=0) {   
//     log << MSG::INFO << "Haha signal exist!!!!" << endreq;
    }




//  HEPEVT_Wrapper::print_hepevt();

  ++m_events;
  


  log << MSG::INFO << " BESGenModule generating done.  \n"  << endreq;
  return status;

}

StatusCode GenModule::execute

(

)

[inherited]

Definition at line 125 of file GenModule.cxx.

References Bes_Common::DEBUG, calibUtil::ERROR, GenModule::fillEvt(), genRecEmupikp::i, Bes_Common::INFO, GenModule::m_fixedMode, GenModule::m_meanInteractions, GenModule::m_pPoissonGenerator, GenModule::m_StripPartonSwitch, and GenModule::StripPartons().

                              {
  //---------------------------------------------------------------------------
    
  int numToGenerate;
  StatusCode status;

  MsgStream log(messageService(), name());

  log << MSG::DEBUG << "GenModule::execute()" << endreq;

  // Decide how many interactions to generate
  if(m_fixedMode)  {
    numToGenerate = (int) m_meanInteractions;
  }
  else {
    numToGenerate = m_pPoissonGenerator->fire();
  }

  // Generate as many interactions as requested
  for(int i=0; i<numToGenerate; i++) {

    // Call the code that generates an event
    status = this->callGenerator();

    // Create the MC event and send the GeneratorEvent stored in it to fillEvt
    GenEvent* evt = new GenEvent(1,1);
    status = this->fillEvt(evt);

    // Strip out the requested partons here.
    if (m_StripPartonSwitch)    StripPartons(evt);
    
    // Check if the McCollection already exists
    SmartDataPtr<McGenEventCol> anMcCol(eventSvc(), "/Event/Gen");
    if (anMcCol!=0) {
      // Add event to existing collection
      MsgStream log(messageService(), name());
      log << MSG::INFO << "Add McGenEvent to existing collection" << endreq;
      McGenEvent* mcEvent = new McGenEvent(evt);
      anMcCol->push_back(mcEvent);
    }
    else {
      // Create Collection and add  to the transient store
      McGenEventCol *mcColl = new McGenEventCol;
      McGenEvent* mcEvent = new McGenEvent(evt);
      mcColl->push_back(mcEvent);

      StatusCode sc = eventSvc()->registerObject("/Event/Gen",mcColl);

      if (sc != StatusCode::SUCCESS) {
        MsgStream log(messageService(), name());
        log << MSG::ERROR << "Could not register McGenEvent" << endreq;
        delete mcColl;
        delete evt;
        delete mcEvent;
        return StatusCode::FAILURE;
      }
      else {
//        std::cout << " McGenEventCol made and stored" << std::endl;
      }
      
    }
  }

  // now call the incident service and set the signal.
  IIncidentSvc *incSvc;
  service("IncidentSvc",incSvc);
  incSvc->fireIncident( Incident( name(), "McGenEvent Generated") );

  return status;
}

StatusCode BesGenModule::fillEvt

(

GenEvent *

evt

)

[virtual]

Reimplemented from GenModule.

Definition at line 206 of file BesGenModule.cxx.

References Bes_Common::INFO, and m_events.

                                              {
  //---------------------------------------------------------------------------
  MsgStream log(messageService(), name());
  log << MSG:: INFO << " BESGenModule Filling...  \n"  << endreq;

      evt->set_event_number(m_events);
      HepMC::IO_HEPEVT hepio;
      hepio.fill_next_event(evt);
//   cmTomm(evt);
//   GeVToMeV(evt);


 return StatusCode::SUCCESS;
}

StatusCode GenModule::finalize

(

)

[inherited]

Definition at line 197 of file GenModule.cxx.

References GenModule::genFinalize().

                               {
  //---------------------------------------------------------------------------

  StatusCode status = this->genFinalize();

  return status;
}

StatusCode BesGenModule::genFinalize

(

)

[virtual]

Reimplemented from GenModule.

Definition at line 198 of file BesGenModule.cxx.

References Bes_Common::INFO, mainfin_(), and n.

                                     {
  //---------------------------------------------------------------------------
  MsgStream log(messageService(), name());
 mainfin_(&n);
  log << MSG:: INFO << " BESGenModule Ending...  \n"  << endreq;
  return StatusCode::SUCCESS;
}

StatusCode BesGenModule::genInitialize

(

)

[virtual]

Reimplemented from GenModule.

Definition at line 104 of file BesGenModule.cxx.

References besMcGen_stream, cardname, calibUtil::ERROR, Bes_Common::INFO, m_events, mainini_(), n, p_BesRndmGenSvc, runi_seed_(), runname, and IBesRndmGenSvc::setGenseed().

                                       {
//---------------------------------------------------------------------------

  MsgStream log(messageService(), name());
  log << MSG::INFO << " BES PHOTOS INITIALISING.  \n"  << endreq;
//  log << MSG::INFO << " BES PHOTOS 2.  \n"  << endreq;

  //caogf add for random seed of genbes
  StatusCode RndmStatus = service("BesRndmGenSvc", p_BesRndmGenSvc, true);
  if (!RndmStatus.isSuccess() || 0 == p_BesRndmGenSvc)
  {
     log << MSG::ERROR << " Could not initialize Random Number Service" << endreq;
     return RndmStatus;
  }

  BesGenModule::besMcGen_stream =       "PHOTOS_INIT";

  BesGenModule::besMcGen_stream =       "PHOTOS111";

  m_events = 0;
  n=0;
  cardname = runname.c_str();


  mainini_(&n,cardname);

  int theseed;
  runi_seed_(&theseed);
  p_BesRndmGenSvc->setGenseed(theseed+1);

  HepMC::HEPEVT_Wrapper::set_sizeof_int(4);
  HepMC::HEPEVT_Wrapper::set_sizeof_real(4);
  HepMC::HEPEVT_Wrapper::set_max_number_entries(2000);

 
  return StatusCode::SUCCESS;
}

StatusCode GenModule::genuserInitialize

(

)

[virtual, inherited]

Definition at line 212 of file GenModule.cxx.

Referenced by GenModule::initialize().

                                        { 
  //---------------------------------------------------------------------------
  return StatusCode::SUCCESS;
}

StatusCode GenModule::initialize

(

)

[inherited]

Definition at line 63 of file GenModule.cxx.

References calibUtil::ERROR, GenModule::genInitialize(), GenModule::genuserInitialize(), Bes_Common::INFO, GenModule::m_fixedMode, GenModule::m_meanInteractions, GenModule::m_particleTable, GenModule::m_pPoissonGenerator, GenModule::m_pRandomEngine, GenModule::m_randomSeed, GenModule::m_StripPartonSwitch, GenModule::m_StripVector, and GenModule::StripPartonsInit().

                                {
  //---------------------------------------------------------------------------

  // Inform the user what the mode and conditions are:
  MsgStream log(messageService(), name());

  // Initialize StripPartons variables
  if (m_StripVector[0] > 0) {
    StripPartonsInit();
  } else {
    m_StripPartonSwitch =       false;
    m_StripVector.clear();
  }

   
  // Get the Particle Properties Service
  IPartPropSvc* p_PartPropSvc;
  //static const bool CREATEIFNOTTHERE(true);
  //StatusCode PartPropStatus = service("PartPropSvc", p_PartPropSvc, CREATEIFNOTTHERE);
  StatusCode PartPropStatus = service("PartPropSvc", p_PartPropSvc);
  if (!PartPropStatus.isSuccess() || 0 == p_PartPropSvc) {
    log << MSG::ERROR << " Could not initialize Particle Properties Service" << endreq;
    return PartPropStatus;
  }      

  m_particleTable = p_PartPropSvc->PDT();
  
  m_pRandomEngine = new CLHEP::Ranlux64Engine(m_randomSeed);

  if(m_fixedMode) {
    if(m_meanInteractions == 1.0) {
      log << MSG::INFO << "Standard Initialization:  Single Interaction Mode "
          << endreq;
    }
    else {
      log << MSG::INFO << "Fixed Number of Interactions per Event is: "
          << m_meanInteractions << endreq;
    }
  }
  else {
    m_pPoissonGenerator = new CLHEP::RandPoisson(*m_pRandomEngine, 
                                          m_meanInteractions);

    log << MSG::INFO << 
      "Poisson Distribution of Interactions per Event with Mean: "
        << m_meanInteractions << endreq;
  }
  // Initialize the generator itself
  StatusCode status = this->genInitialize();
  if (status.isFailure()) {
    log << MSG::ERROR << "Could not initialize Generator properly" << endreq;
    return status;
  }  
  StatusCode  status1 = this->genuserInitialize();
  if (status1.isFailure()) {
    log << MSG::ERROR << "Could not initialize user part properly" << endreq;
    return status1;
  } 
  return status;
}

void GenModule::StripPartons

(

GenEvent *

evt

)

[protected, inherited]

Definition at line 274 of file GenModule.cxx.

References GenModule::m_StripVector.

Referenced by GenModule::execute().

{
  //    for ( HepMC::GenEvent::particle_iterator p = evt->particles_begin();
  //              p != evt->particles_end(); ++p )
  //    {
  //  //            std::cout << " PDG, BAR " << (*p)->pdg_id() << " " << (*p)->barcode() << std::endl;
  //        if ( std::find(m_StripVector.begin(),
  //                       m_StripVector.end(),
  //                       (*p)->pdg_id()) != m_StripVector.end() )
  //        {
  //  //                std::cout << " REMOVING " << (*p)->pdg_id() << " " << (*p)->barcode() << std::endl;
  //            HepMC::GenVertex*       pvtx    =       (*p)->production_vertex();
  //            HepMC::GenVertex*       evtx    =       (*p)->end_vertex();
  //            if (pvtx) pvtx->remove_particle(*p);
  //            if (evtx) evtx->remove_particle(*p);
  //            delete  *p;
  //        }
            
  //    }
        
  // Loop on all vertices of the event and remove the particle from the vertex
  for ( HepMC::GenEvent::vertex_iterator vtx = evt->vertices_begin();
        vtx != evt->vertices_end(); ++vtx ) {
    // Loop on all children particles and remove the ones that should be stripped out
    for ( HepMC::GenVertex::particle_iterator p = (*vtx)-> particles_begin(HepMC::children);
          p != (*vtx)->particles_end(HepMC::children);
          ++p ){
      if ( std::find(m_StripVector.begin(),
                     m_StripVector.end(),
                     (*p)->pdg_id()) != m_StripVector.end() ) {
        if ( (*p)->end_vertex() ) (*p)->end_vertex()->remove_particle(*p);
        if ( (*p)->production_vertex() ) (*p)->production_vertex()->remove_particle(*p);
        delete *p;
      }
    }
    // Loop on all parents particles and remove the ones that should be stripped out
    for ( HepMC::GenVertex::particle_iterator p = (*vtx)-> particles_begin(HepMC::parents);
          p != (*vtx)->particles_end(HepMC::parents);
          ++p ) {
      if ( std::find(m_StripVector.begin(),
                     m_StripVector.end(),
                     (*p)->pdg_id()) != m_StripVector.end() ) {
        if ( (*p)->end_vertex() ) (*p)->end_vertex()->remove_particle(*p);
        if ( (*p)->production_vertex() ) (*p)->production_vertex()->remove_particle(*p);
        delete *p;
      }
    }
  }
}

void GenModule::StripPartonsInit

(

void

)

[protected, inherited]

Definition at line 236 of file GenModule.cxx.

References genRecEmupikp::i, Bes_Common::INFO, GenModule::m_AllPartons, GenModule::m_StripPartonSwitch, and GenModule::m_StripVector.

Referenced by GenModule::initialize().

{
  MsgStream log(messageService(), name());

  for (int i = 1; i < 9; ++i) { m_AllPartons.push_back(i); m_AllPartons.push_back(-i); } // Quarks
  m_AllPartons.push_back(21); // gluon

  m_StripPartonSwitch = true;
  m_StripVector.erase(m_StripVector.begin(), m_StripVector.begin()+1);
  // When the user specifies only the StripPartonSwitch or gives a Particle Code
  // which is NOT a parton then strip ALL partons
  if (m_StripVector.size() == 0) {
    log << MSG::INFO << " !!!! NO SPECIFIC PARTON FOR STRIPOUT WAS REQUESTED => STRIPOUT ALL "
        << endreq;
    m_StripVector = m_AllPartons;
  } else {
    bool        value_ok        =       true;
    std::vector<int>::const_iterator    i =     m_StripVector.begin();
    do {
      if ( std::find(m_AllPartons.begin(),
                     m_AllPartons.end(),
                     *i) == m_AllPartons.end() ) value_ok = false;
      ++i;
    } while (i != m_StripVector.end() && value_ok);
    if (!value_ok) {
      log << MSG::INFO << " !!!! ILEGAL PDG-ID FOR STRIPOUT WAS REQUESTED => STRIPOUT ALL "
          << endreq;
      m_StripVector = m_AllPartons;
    }
  }
  log << MSG::INFO << " THE FOLLOWING PARTON(S) WILL BE STRIPED OUT ";
  for (std::vector<int>::const_iterator ip = m_StripVector.begin(); ip != m_StripVector.end(); ++ip)
    log << *ip << "  ";
  log << endreq;
  
}

---

Member Data Documentation

std::string BesGenModule::besMcGen_stream = "PHOTOS_INIT" [static]

Definition at line 33 of file BesGenModule.h.

Referenced by BesGenModule(), and genInitialize().

const char* BesGenModule::cardname [protected]

Definition at line 44 of file BesGenModule.h.

Referenced by genInitialize().

std::vector<int> GenModule::m_AllPartons [protected, inherited]

Definition at line 60 of file GenModule.h.

Referenced by GenModule::StripPartonsInit().

int BesGenModule::m_events [protected]

Definition at line 53 of file BesGenModule.h.

Referenced by callGenerator(), fillEvt(), and genInitialize().

bool GenModule::m_fixedMode [protected, inherited]

Definition at line 57 of file GenModule.h.

Referenced by GenModule::execute(), GenModule::GenModule(), and GenModule::initialize().

int BesGenModule::m_fortout [protected]

Definition at line 39 of file BesGenModule.h.

int BesGenModule::m_GenIndex [protected]

Definition at line 57 of file BesGenModule.h.

double GenModule::m_meanInteractions [protected, inherited]

Definition at line 58 of file GenModule.h.

Referenced by GenModule::execute(), GenModule::GenModule(), and GenModule::initialize().

std::vector<int> BesGenModule::m_par [protected]

Definition at line 50 of file BesGenModule.h.

HepPDT::ParticleDataTable* GenModule::m_particleTable [protected, inherited]

Definition at line 73 of file GenModule.h.

Referenced by SingleParticleGun::genInitialize(), and GenModule::initialize().

CLHEP::RandPoisson* GenModule::m_pPoissonGenerator [protected, inherited]

Definition at line 68 of file GenModule.h.

Referenced by GenModule::execute(), GenModule::initialize(), and GenModule::~GenModule().

CLHEP::HepRandomEngine* GenModule::m_pRandomEngine [protected, inherited]

Definition at line 67 of file GenModule.h.

Referenced by GenModule::initialize(), and GenModule::~GenModule().

int GenModule::m_randomSeed [protected, inherited]

Definition at line 59 of file GenModule.h.

Referenced by GenModule::GenModule(), and GenModule::initialize().

std::vector<long int> BesGenModule::m_seeds [protected]

Definition at line 42 of file BesGenModule.h.

bool GenModule::m_StripPartonSwitch [protected, inherited]

Definition at line 62 of file GenModule.h.

Referenced by GenModule::execute(), GenModule::initialize(), and GenModule::StripPartonsInit().

std::vector<int> GenModule::m_StripVector [protected, inherited]

Definition at line 61 of file GenModule.h.

Referenced by GenModule::GenModule(), GenModule::initialize(), GenModule::StripPartons(), and GenModule::StripPartonsInit().

int BesGenModule::m_SubIndex [protected]

Definition at line 61 of file BesGenModule.h.

int BesGenModule::m_SuperIndex [protected]

Definition at line 59 of file BesGenModule.h.

int BesGenModule::mvec[10] [protected]

Definition at line 47 of file BesGenModule.h.

Referenced by callGenerator().

int BesGenModule::n [protected]

Definition at line 54 of file BesGenModule.h.

Referenced by genFinalize(), and genInitialize().

int BesGenModule::n1 [protected]

Definition at line 55 of file BesGenModule.h.

int BesGenModule::Nr [protected]

Definition at line 46 of file BesGenModule.h.

Referenced by BesGenModule(), and callGenerator().

int BesGenModule::nvec[10] [protected]

Definition at line 48 of file BesGenModule.h.

IBesRndmGenSvc* BesGenModule::p_BesRndmGenSvc

Definition at line 32 of file BesGenModule.h.

Referenced by genInitialize().

std::string BesGenModule::runname [protected]

Definition at line 45 of file BesGenModule.h.

Referenced by BesGenModule(), and genInitialize().

---