exampleB2b.cc File Reference

Main program of the B2b example. More...

#include "B2bDetectorConstruction.hh"
#include "B2ActionInitialization.hh"
#include "tbbUserWorkerInitialization.hh"
#include "SimpleTbbMasterRunManager.hh"
#include "G4Threading.hh"
#include "G4UImanager.hh"
#include "FTFP_BERT.hh"
#include "G4StepLimiterPhysics.hh"
#include "Randomize.hh"
#include "G4VisExecutive.hh"
#include "G4UIExecutive.hh"
#include <tbb/task_scheduler_init.h>
#include <tbb/task.h>

Go to the source code of this file.

Functions
G4ThreadFunReturnType	startWork (G4ThreadFunArgType arg)
int	main (int argc, char **argv)

Detailed Description

Main program of the B2b example.

Definition in file exampleB2b.cc.

Function Documentation

int main	(	int	argc,
		char **	argv
	)

Definition at line 75 of file exampleB2b.cc.

References startWork().

{
  // Instantiate G4UIExecutive if there are no arguments (interactive mode)
  G4UIExecutive* ui = 0;
  if ( argc == 1 ) {
    ui = new G4UIExecutive(argc, argv);
  }

  // Choose the Random engine

  G4Random::setTheEngine(new CLHEP::RanecuEngine);
  
  unsigned int numCoresAvailable=  G4Threading::G4GetNumberOfCores();
  unsigned int numberOfCoresToUse= (numCoresAvailable > 1 ) ? 2 : 1 ;
  //=== TBB engine initialization
  tbb::task_scheduler_init init( numberOfCoresToUse );
  tbb::task_list tasks;
   
  SimpleTbbMasterRunManager* runManager = new SimpleTbbMasterRunManager;
    
  //Set TBB specific data to run-manager, 1 event per tbb::task (e.g. Nevents == N tasks)
  //Note that a /run/beamOn command will just create tasks and add them to tasks
  runManager->SetNumberEventsPerTask(1); //Not needed since 1 is however default
  runManager->SetTaskList(&tasks);
  //Set user-initialization that specify threading model, in this case TBB.
  //This overwrites default that uses pthreads
  runManager->SetUserInitialization(new tbbUserWorkerInitialization );
    
  //==== Geant4 specific stuff, from now up to END-G4 comment is copy from MT example
  // Set mandatory initialization classes

  runManager->SetUserInitialization(new B2bDetectorConstruction());

  G4VModularPhysicsList* physicsList = new FTFP_BERT;
  physicsList->RegisterPhysics(new G4StepLimiterPhysics());
  runManager->SetUserInitialization(physicsList);
    
  // Set user action classes

  runManager->SetUserInitialization(new B2ActionInitialization());
  
  // Initialize G4 kernel

  runManager->Initialize();
  
  // Get the pointer to the User Interface manager
  G4UImanager* UImanager = G4UImanager::GetUIpointer();

  if (!ui)   // batch mode
    {
      G4String command = "/control/execute ";
      G4String fileName = argv[1];
      UImanager->ApplyCommand(command+fileName);
    }
  else
    {  // interactive mode : define UI session
#if 1
      G4int nEvents= 50; 
      runManager->BeamOn(nEvents); 
#else     
      UImanager->ApplyCommand("/control/execute init.mac");
      if (ui->IsGUI())
         UImanager->ApplyCommand("/control/execute gui.mac");
      ui->SessionStart();
      delete ui;
#endif
    }
 //END-G4
    G4Thread* aThread = new G4Thread;
    G4THREADCREATE(aThread, startWork, static_cast<G4ThreadFunArgType>(&tasks));
    
    //Wait for work to be finised
    if(aThread)
        aThread->join();
  
    delete runManager;

    return 0;
}

G4ThreadFunReturnType startWork

(

G4ThreadFunArgType

arg

)

Definition at line 59 of file exampleB2b.cc.

References e.

Referenced by main().

{
    tbb::task_list* tasks = static_cast<tbb::task_list*>(arg);
    //We assume at least one /run/beamOn was executed, thus the tasklist is now filled,
    //lets start TBB
    try {
        std::cout<<"Now calling 'tbb::task::spawn_work_and_wait' "<<std::endl;
        tbb::task::spawn_root_and_wait( *tasks );
    } catch(std::exception& e) {
        std::cerr<<"Error occurred. Error info is:\""<<e.what()<<"\""<<std::endl;
    }
    return static_cast<G4ThreadFunReturnType>(0);
}