Geant4  v4-10.4-release
 모두 클래스 네임스페이스들 파일들 함수 변수 타입정의 열거형 타입 열거형 멤버 Friends 매크로 그룹들 페이지들
extended/parallel/TBB/B2b/exampleB2b.cc
이 파일의 문서화 페이지로 가기
1 //
2 // ********************************************************************
3 // * License and Disclaimer *
4 // * *
5 // * The Geant4 software is copyright of the Copyright Holders of *
6 // * the Geant4 Collaboration. It is provided under the terms and *
7 // * conditions of the Geant4 Software License, included in the file *
8 // * LICENSE and available at http://cern.ch/geant4/license . These *
9 // * include a list of copyright holders. *
10 // * *
11 // * Neither the authors of this software system, nor their employing *
12 // * institutes,nor the agencies providing financial support for this *
13 // * work make any representation or warranty, express or implied, *
14 // * regarding this software system or assume any liability for its *
15 // * use. Please see the license in the file LICENSE and URL above *
16 // * for the full disclaimer and the limitation of liability. *
17 // * *
18 // * This code implementation is the result of the scientific and *
19 // * technical work of the GEANT4 collaboration. *
20 // * By using, copying, modifying or distributing the software (or *
21 // * any work based on the software) you agree to acknowledge its *
22 // * use in resulting scientific publications, and indicate your *
23 // * acceptance of all terms of the Geant4 Software license. *
24 // ********************************************************************
25 //
26 // $Id$
27 //
30 
31 #include "B2bDetectorConstruction.hh"
32 #include "B2ActionInitialization.hh"
33 
34 //G4-TBB interfaces
37 #include "G4Threading.hh"
38 
39 #include "G4UImanager.hh"
40 #include "FTFP_BERT.hh"
41 #include "G4StepLimiterPhysics.hh"
42 
43 #include "Randomize.hh"
44 
45 #ifdef G4VIS_USE
46 #include "G4VisExecutive.hh"
47 #endif
48 
49 #ifdef G4UI_USE
50 #include "G4UIExecutive.hh"
51 #endif
52 
53 //TBB includes
54 #include <tbb/task_scheduler_init.h>
55 #include <tbb/task.h>
56 
57 
58 //This function is very simple: it just start tbb work.
59 //This is done in a seperate thread, because for G4 the
60 //master cannot live in the same thread where workers are
61 //Starting tbb work in a thread guarantees that no workers
62 //are created where the master lives (the main thread)
63 //Clearly a separate solution is to create and configure master
64 //in a separate thread. But this is much simpler.
66 {
67  tbb::task_list* tasks = static_cast<tbb::task_list*>(arg);
68  //We assume at least one /run/beamOn was executed, thus the tasklist is now filled,
69  //lets start TBB
70  try {
71  std::cout<<"Now calling 'tbb::task::spawn_work_and_wait' "<<std::endl;
72  tbb::task::spawn_root_and_wait( *tasks );
73  } catch(std::exception& e) {
74  std::cerr<<"Error occurred. Error info is:\""<<e.what()<<"\""<<std::endl;
75  }
76  return static_cast<G4ThreadFunReturnType>(0);
77 }
78 
79 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
80 
81 int main(int argc,char** argv)
82 {
83  // Choose the Random engine
84 
85  G4Random::setTheEngine(new CLHEP::RanecuEngine);
86 
87  unsigned int numCoresAvailable= G4Threading::G4GetNumberOfCores();
88  unsigned int numberOfCoresToUse= (numCoresAvailable > 1 ) ? 2 : 1 ;
89  //=== TBB engine initialization
90  tbb::task_scheduler_init init( numberOfCoresToUse );
91  tbb::task_list tasks;
92 
94 
95  //Set TBB specific data to run-manager, 1 event per tbb::task (e.g. Nevents == N tasks)
96  //Note that a /run/beamOn command will just create tasks and add them to tasks
97  runManager->SetNumberEventsPerTask(1); //Not needed since 1 is however default
98  runManager->SetTaskList(&tasks);
99  //Set user-initialization that specify threading model, in this case TBB.
100  //This overwrites default that uses pthreads
102 
103  //==== Geant4 specific stuff, from now up to END-G4 comment is copy from MT example
104  // Set mandatory initialization classes
105 
107 
108  G4VModularPhysicsList* physicsList = new FTFP_BERT;
109  physicsList->RegisterPhysics(new G4StepLimiterPhysics());
110  runManager->SetUserInitialization(physicsList);
111 
112  // Set user action classes
113 
115 
116  // Initialize G4 kernel
117 
118  runManager->Initialize();
119 
120  // Get the pointer to the User Interface manager
121  G4UImanager* UImanager = G4UImanager::GetUIpointer();
122 
123  if (argc!=1) // batch mode
124  {
125  G4String command = "/control/execute ";
126  G4String fileName = argv[1];
127  UImanager->ApplyCommand(command+fileName);
128  }
129  else
130  { // interactive mode : define UI session
131 #if 1
132  G4int nEvents= 50;
133  runManager->BeamOn(nEvents);
134 #else
135  #ifdef G4UI_USE
136  G4UIExecutive* ui = new G4UIExecutive(argc, argv);
137  UImanager->ApplyCommand("/control/execute init.mac");
138  if (ui->IsGUI())
139  UImanager->ApplyCommand("/control/execute gui.mac");
140  ui->SessionStart();
141  delete ui;
142  #endif
143 #endif
144  }
145  //END-G4
146  G4Thread aThread;
147  G4THREADCREATE(&aThread,startWork,static_cast<G4ThreadFunArgType>(&tasks));
148 
149  //Wait for work to be finised
150  G4THREADJOIN(aThread);
151 
152  delete runManager;
153 
154  return 0;
155 }
156 
157 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
G4int ApplyCommand(const char *aCommand)
Definition: G4UImanager.cc:466
G4int G4GetNumberOfCores()
Definition: G4Threading.cc:127
static G4UImanager * GetUIpointer()
Definition: G4UImanager.cc:73
virtual void BeamOn(G4int n_event, const char *macroFile=0, G4int n_select=-1)
void SetTaskList(tbb::task_list *tl)
void RegisterPhysics(G4VPhysicsConstructor *)
virtual void SetUserInitialization(G4VUserPhysicsList *userPL)
int main(int argc, char **argv)
Definition: genwindef.cc:359
TFTFP_BERT< G4VModularPhysicsList > FTFP_BERT
Definition: FTFP_BERT.hh:63
G4ThreadFunReturnType startWork(G4ThreadFunArgType arg)
void * G4ThreadFunReturnType
Definition: G4Threading.hh:102
int G4int
Definition: G4Types.hh:78
void * G4ThreadFunArgType
Definition: G4Threading.hh:103
#define G4THREADJOIN(worker)
Definition: G4Threading.hh:237
virtual void Initialize()
void G4THREADCREATE(_Worker *&worker, _Func func, _Args...args)
Definition: G4Threading.hh:244
G4bool IsGUI() const