extended/parallel/TBB/B2b/exampleB2b.cc File Reference

Main program of the B2b example. More...

#include "B2bDetectorConstruction.hh"
#include "B2ActionInitialization.hh"
#include "tbbUserWorkerInitialization.hh"
#include "tbbMasterRunManager.hh"
#include "G4Threading.hh"
#include "G4UImanager.hh"
#include "FTFP_BERT.hh"
#include "G4StepLimiterPhysics.hh"
#include "Randomize.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 extended/parallel/TBB/B2b/exampleB2b.cc.

Function Documentation

int main	(	int	argc,
		char **	argv
	)

Definition at line 81 of file extended/parallel/TBB/B2b/exampleB2b.cc.

References G4UImanager::ApplyCommand(), G4Threading::G4GetNumberOfCores(), G4THREADCREATE, G4THREADJOIN, G4UImanager::GetUIpointer(), G4MTRunManager::Initialize(), G4VisManager::Initialize(), G4UIExecutive::IsGUI(), G4VModularPhysicsList::RegisterPhysics(), G4UIExecutive::SessionStart(), tbbMasterRunManager::SetNumberEventsPerTask(), tbbMasterRunManager::SetTaskList(), G4MTRunManager::SetUserInitialization(), and startWork().

{
  // Choose the Random engine

  G4Random::setTheEngine(new CLHEP::RanecuEngine);
    
    
  //=== TBB engine initialization
  tbb::task_scheduler_init init( G4Threading::G4GetNumberOfCores() );
  tbb::task_list tasks;
  
   
  tbbMasterRunManager* runManager = new tbbMasterRunManager;
    
  //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();
  
#ifdef G4VIS_USE
  // Initialize visualization
  G4VisManager* visManager = new G4VisExecutive;
  // G4VisExecutive can take a verbosity argument - see /vis/verbose guidance.
  // G4VisManager* visManager = new G4VisExecutive("Quiet");
  visManager->Initialize();
#endif

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

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

    delete runManager;

    return 0;
}

G4ThreadFunReturnType startWork

(

G4ThreadFunArgType

arg

)

Definition at line 65 of file extended/parallel/TBB/B2b/exampleB2b.cc.

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 spawn work and waiting"<<std::endl;
        tbb::task::spawn_root_and_wait( *tasks );
    } catch(std::exception& e) {
        std::cerr<<"Error occurred. Error test is:\""<<e.what()<<"\""<<std::endl;
    }
    return static_cast<G4ThreadFunReturnType>(0);
}