Import Geant4 10.7.0.beta source tree
This commit is contained in:
Gabriele Cosmo
+151
-160
@@ -24,219 +24,210 @@
|
||||
// ********************************************************************
|
||||
//
|
||||
#include "G4WorkerThread.hh"
|
||||
#include "G4WorkerRunManager.hh"
|
||||
#include "G4MTRunManager.hh"
|
||||
#include "G4WorkerRunManager.hh"
|
||||
|
||||
#include "G4GeometryWorkspace.hh"
|
||||
#include "G4SolidsWorkspace.hh"
|
||||
#include "G4ParticlesWorkspace.hh"
|
||||
#include "G4PhysicsListWorkspace.hh"
|
||||
#include "G4SolidsWorkspace.hh"
|
||||
|
||||
#include "G4LogicalVolume.hh"
|
||||
#include "G4LogicalVolumeStore.hh"
|
||||
#include "G4PhysicalVolumeStore.hh"
|
||||
#include "G4Region.hh"
|
||||
#include "G4RegionStore.hh"
|
||||
#include "G4LogicalVolume.hh"
|
||||
#include "G4Region.hh"
|
||||
#include "G4PhysicalVolumeStore.hh"
|
||||
#include "G4LogicalVolumeStore.hh"
|
||||
|
||||
void G4WorkerThread::SetThreadId(G4int tid)
|
||||
{
|
||||
threadId = tid;
|
||||
}
|
||||
void G4WorkerThread::SetThreadId(G4int tid) { threadId = tid; }
|
||||
|
||||
G4int G4WorkerThread::GetThreadId() const
|
||||
{
|
||||
return threadId;
|
||||
}
|
||||
G4int G4WorkerThread::GetThreadId() const { return threadId; }
|
||||
|
||||
void G4WorkerThread::SetNumberThreads(G4int nw)
|
||||
{
|
||||
numThreads = nw;
|
||||
}
|
||||
void G4WorkerThread::SetNumberThreads(G4int nw) { numThreads = nw; }
|
||||
|
||||
G4int G4WorkerThread::GetNumberThreads() const
|
||||
{
|
||||
return numThreads;
|
||||
}
|
||||
G4int G4WorkerThread::GetNumberThreads() const { return numThreads; }
|
||||
|
||||
void G4WorkerThread::BuildGeometryAndPhysicsVector()
|
||||
{
|
||||
// Initialise all split classes
|
||||
// with copy of data from master thread
|
||||
// Initialise all split classes
|
||||
// with copy of data from master thread
|
||||
|
||||
G4GeometryWorkspace::GetPool()->CreateAndUseWorkspace();
|
||||
G4SolidsWorkspace::GetPool()->CreateAndUseWorkspace();
|
||||
G4ParticlesWorkspace::GetPool()->CreateAndUseWorkspace();
|
||||
G4PhysicsListWorkspace::GetPool()->CreateAndUseWorkspace();
|
||||
G4GeometryWorkspace::GetPool()->CreateAndUseWorkspace();
|
||||
G4SolidsWorkspace::GetPool()->CreateAndUseWorkspace();
|
||||
G4ParticlesWorkspace::GetPool()->CreateAndUseWorkspace();
|
||||
G4PhysicsListWorkspace::GetPool()->CreateAndUseWorkspace();
|
||||
}
|
||||
|
||||
void G4WorkerThread::DestroyGeometryAndPhysicsVector()
|
||||
{
|
||||
// Clear all split classes
|
||||
// Clear all split classes
|
||||
|
||||
G4GeometryWorkspace::GetPool()->CleanUpAndDestroyAllWorkspaces();
|
||||
G4SolidsWorkspace::GetPool()->CleanUpAndDestroyAllWorkspaces();
|
||||
G4ParticlesWorkspace::GetPool()->CleanUpAndDestroyAllWorkspaces();
|
||||
G4PhysicsListWorkspace::GetPool()->CleanUpAndDestroyAllWorkspaces();
|
||||
G4GeometryWorkspace::GetPool()->CleanUpAndDestroyAllWorkspaces();
|
||||
G4SolidsWorkspace::GetPool()->CleanUpAndDestroyAllWorkspaces();
|
||||
G4ParticlesWorkspace::GetPool()->CleanUpAndDestroyAllWorkspaces();
|
||||
G4PhysicsListWorkspace::GetPool()->CleanUpAndDestroyAllWorkspaces();
|
||||
}
|
||||
|
||||
void G4WorkerThread::UpdateGeometryAndPhysicsVectorFromMaster()
|
||||
{
|
||||
// =================================================
|
||||
// Step-0: keep sensitive detector and field manager
|
||||
// =================================================
|
||||
// First remember SD and Filed Associated with worker
|
||||
// in order to re-use it
|
||||
// (note that all the stuff after this will reset SD and Field)
|
||||
typedef std::map<G4LogicalVolume*,
|
||||
std::pair<G4VSensitiveDetector*,G4FieldManager*> > LV2SDFM;
|
||||
LV2SDFM lvmap;
|
||||
// =================================================
|
||||
// Step-0: keep sensitive detector and field manager
|
||||
// =================================================
|
||||
// First remember SD and Filed Associated with worker
|
||||
// in order to re-use it
|
||||
// (note that all the stuff after this will reset SD and Field)
|
||||
typedef std::map<G4LogicalVolume*,
|
||||
std::pair<G4VSensitiveDetector*, G4FieldManager*>>
|
||||
LV2SDFM;
|
||||
LV2SDFM lvmap;
|
||||
|
||||
typedef std::map<G4Region*,
|
||||
std::pair<G4FastSimulationManager*,G4UserSteppingAction*> > R2FSM;
|
||||
R2FSM rgnmap;
|
||||
typedef std::map<G4Region*,
|
||||
std::pair<G4FastSimulationManager*, G4UserSteppingAction*>>
|
||||
R2FSM;
|
||||
R2FSM rgnmap;
|
||||
|
||||
G4LogicalVolumeStore* mLogVolStore = G4LogicalVolumeStore::GetInstance();
|
||||
for(size_t ip=0; ip<mLogVolStore->size(); ip++)
|
||||
{
|
||||
G4LogicalVolume *lv = (*mLogVolStore)[ip];
|
||||
G4LogicalVolumeStore* mLogVolStore = G4LogicalVolumeStore::GetInstance();
|
||||
for(size_t ip = 0; ip < mLogVolStore->size(); ip++)
|
||||
{
|
||||
G4LogicalVolume* lv = (*mLogVolStore)[ip];
|
||||
|
||||
// The following needs an explanation.
|
||||
// Consider the case in which the user adds one LogVolume between
|
||||
// the runs. The problem is that the thread-local part (split class)
|
||||
// of the G4LogicalVolume object is not initialized for workers
|
||||
// because the initialization is done once when the thread starts
|
||||
// (see G4MTRunManagerKernel::StartThread Step-2 that calls
|
||||
// G4WorkerThread::BuildGeometryAndPhysicsVector in this class).
|
||||
// The problem is that pointers of SD and FM for these newly added LV
|
||||
// may be invalid pointers (because never initialized, we have seen
|
||||
// this behavior in our testing). If now we remember them and re-use
|
||||
// them in Step-4 below we set invalid pointers to LV for this thread.
|
||||
// Thus we need a way to know if for a given LV we need to remember
|
||||
// or not the SD and FM pointers.
|
||||
// To solve this problem: We assume that the ConstructSDandField() is
|
||||
// called also by Master thread, thus for newly added LV the shadow
|
||||
// pointers of SD and Fields are correct.
|
||||
// (LIMITATION: this assumption may be too stringent, a user to save
|
||||
// memory could instantiate SD only for workers, but we require this
|
||||
// not to happen!).
|
||||
// Thus if a SD and FieldMgr are needed for this particular LV, and
|
||||
// shadow are !=0 it means that user wants an SD and FM to be
|
||||
// associated with LV, we get the values and we remember them.
|
||||
//
|
||||
G4VSensitiveDetector* sd = 0;
|
||||
G4FieldManager* fmgr = 0;
|
||||
if ( lv->GetMasterSensitiveDetector() != 0 )
|
||||
{
|
||||
sd = lv->GetSensitiveDetector();
|
||||
}
|
||||
if ( lv->GetMasterFieldManager() != 0 )
|
||||
{
|
||||
fmgr = lv->GetFieldManager();
|
||||
}
|
||||
if ( sd || fmgr )
|
||||
{
|
||||
lvmap[lv] = std::make_pair(sd,fmgr);
|
||||
}
|
||||
}
|
||||
G4RegionStore* mRegStore = G4RegionStore::GetInstance();
|
||||
for(size_t ir=0; ir<mRegStore->size(); ir++)
|
||||
// The following needs an explanation.
|
||||
// Consider the case in which the user adds one LogVolume between
|
||||
// the runs. The problem is that the thread-local part (split class)
|
||||
// of the G4LogicalVolume object is not initialized for workers
|
||||
// because the initialization is done once when the thread starts
|
||||
// (see G4MTRunManagerKernel::StartThread Step-2 that calls
|
||||
// G4WorkerThread::BuildGeometryAndPhysicsVector in this class).
|
||||
// The problem is that pointers of SD and FM for these newly added LV
|
||||
// may be invalid pointers (because never initialized, we have seen
|
||||
// this behavior in our testing). If now we remember them and re-use
|
||||
// them in Step-4 below we set invalid pointers to LV for this thread.
|
||||
// Thus we need a way to know if for a given LV we need to remember
|
||||
// or not the SD and FM pointers.
|
||||
// To solve this problem: We assume that the ConstructSDandField() is
|
||||
// called also by Master thread, thus for newly added LV the shadow
|
||||
// pointers of SD and Fields are correct.
|
||||
// (LIMITATION: this assumption may be too stringent, a user to save
|
||||
// memory could instantiate SD only for workers, but we require this
|
||||
// not to happen!).
|
||||
// Thus if a SD and FieldMgr are needed for this particular LV, and
|
||||
// shadow are !=0 it means that user wants an SD and FM to be
|
||||
// associated with LV, we get the values and we remember them.
|
||||
//
|
||||
G4VSensitiveDetector* sd = 0;
|
||||
G4FieldManager* fmgr = 0;
|
||||
if(lv->GetMasterSensitiveDetector() != 0)
|
||||
{
|
||||
G4Region* reg = (*mRegStore)[ir];
|
||||
G4FastSimulationManager* fsm = reg->GetFastSimulationManager();
|
||||
G4UserSteppingAction* usa = reg->GetRegionalSteppingAction();
|
||||
if ( reg || usa )
|
||||
{
|
||||
rgnmap[reg] = std::make_pair(fsm,usa);
|
||||
}
|
||||
sd = lv->GetSensitiveDetector();
|
||||
}
|
||||
if(lv->GetMasterFieldManager() != 0)
|
||||
{
|
||||
fmgr = lv->GetFieldManager();
|
||||
}
|
||||
if(sd || fmgr)
|
||||
{
|
||||
lvmap[lv] = std::make_pair(sd, fmgr);
|
||||
}
|
||||
}
|
||||
G4RegionStore* mRegStore = G4RegionStore::GetInstance();
|
||||
for(size_t ir = 0; ir < mRegStore->size(); ir++)
|
||||
{
|
||||
G4Region* reg = (*mRegStore)[ir];
|
||||
G4FastSimulationManager* fsm = reg->GetFastSimulationManager();
|
||||
G4UserSteppingAction* usa = reg->GetRegionalSteppingAction();
|
||||
if(reg || usa)
|
||||
{
|
||||
rgnmap[reg] = std::make_pair(fsm, usa);
|
||||
}
|
||||
}
|
||||
|
||||
//===========================
|
||||
// Step-1: Clean the workspace
|
||||
//===========================
|
||||
G4GeometryWorkspace* geomWorkspace =
|
||||
G4GeometryWorkspace::GetPool()->GetWorkspace();
|
||||
geomWorkspace->DestroyWorkspace();
|
||||
G4SolidsWorkspace* solidWorkspace =
|
||||
G4SolidsWorkspace::GetPool()->GetWorkspace();
|
||||
solidWorkspace->DestroyWorkspace();
|
||||
|
||||
//===========================
|
||||
// Step-2: Re-create and initialize workspace
|
||||
//===========================
|
||||
geomWorkspace->InitialiseWorkspace();
|
||||
solidWorkspace->InitialiseWorkspace();
|
||||
|
||||
//===================================================
|
||||
// Step-4: Restore sensitive detector and field manaer
|
||||
//===================================================
|
||||
for ( LV2SDFM::const_iterator it = lvmap.begin() ;
|
||||
it != lvmap.end() ; ++it )
|
||||
{
|
||||
G4LogicalVolume* lv = it->first;
|
||||
G4VSensitiveDetector* sd = (it->second).first;
|
||||
G4FieldManager* fmgr = (it->second).second;
|
||||
if (fmgr) // What should be the second parameter?
|
||||
{ // We use always false for MT mode
|
||||
lv->SetFieldManager(fmgr, false);
|
||||
}
|
||||
if (sd)
|
||||
{
|
||||
lv->SetSensitiveDetector(sd);
|
||||
}
|
||||
//===========================
|
||||
// Step-1: Clean the workspace
|
||||
//===========================
|
||||
G4GeometryWorkspace* geomWorkspace =
|
||||
G4GeometryWorkspace::GetPool()->GetWorkspace();
|
||||
geomWorkspace->DestroyWorkspace();
|
||||
G4SolidsWorkspace* solidWorkspace =
|
||||
G4SolidsWorkspace::GetPool()->GetWorkspace();
|
||||
solidWorkspace->DestroyWorkspace();
|
||||
|
||||
//===========================
|
||||
// Step-2: Re-create and initialize workspace
|
||||
//===========================
|
||||
geomWorkspace->InitialiseWorkspace();
|
||||
solidWorkspace->InitialiseWorkspace();
|
||||
|
||||
//===================================================
|
||||
// Step-4: Restore sensitive detector and field manaer
|
||||
//===================================================
|
||||
for(LV2SDFM::const_iterator it = lvmap.begin(); it != lvmap.end(); ++it)
|
||||
{
|
||||
G4LogicalVolume* lv = it->first;
|
||||
G4VSensitiveDetector* sd = (it->second).first;
|
||||
G4FieldManager* fmgr = (it->second).second;
|
||||
if(fmgr) // What should be the second parameter?
|
||||
{ // We use always false for MT mode
|
||||
lv->SetFieldManager(fmgr, false);
|
||||
}
|
||||
for ( R2FSM::const_iterator it3 = rgnmap.begin() ;
|
||||
it3 != rgnmap.end() ; it3++ )
|
||||
if(sd)
|
||||
{
|
||||
G4Region* reg = it3->first;
|
||||
G4FastSimulationManager* fsm = (it3->second).first;
|
||||
if(fsm) reg->SetFastSimulationManager(fsm);
|
||||
G4UserSteppingAction* usa = (it3->second).second;
|
||||
if(usa) reg->SetRegionalSteppingAction(usa);
|
||||
lv->SetSensitiveDetector(sd);
|
||||
}
|
||||
}
|
||||
for(R2FSM::const_iterator it3 = rgnmap.begin(); it3 != rgnmap.end(); it3++)
|
||||
{
|
||||
G4Region* reg = it3->first;
|
||||
G4FastSimulationManager* fsm = (it3->second).first;
|
||||
if(fsm)
|
||||
reg->SetFastSimulationManager(fsm);
|
||||
G4UserSteppingAction* usa = (it3->second).second;
|
||||
if(usa)
|
||||
reg->SetRegionalSteppingAction(usa);
|
||||
}
|
||||
}
|
||||
|
||||
void G4WorkerThread::SetPinAffinity(G4int affinity) const
|
||||
{
|
||||
if ( affinity == 0 ) return;
|
||||
if(affinity == 0)
|
||||
return;
|
||||
|
||||
#if !defined(WIN32)
|
||||
G4cout << "AFFINITY SET" << G4endl;
|
||||
// Assign this thread to cpus in a round robin way
|
||||
G4int offset = affinity;
|
||||
G4int offset = affinity;
|
||||
G4int cpuindex = 0;
|
||||
if ( std::abs(offset)>G4Threading::G4GetNumberOfCores() )
|
||||
if(std::abs(offset) > G4Threading::G4GetNumberOfCores())
|
||||
{
|
||||
G4Exception("G4WorkerThread::SetPinAffinity()","Run0100", JustWarning,
|
||||
"Cannot set thread affinity, affinity parameter larger than number of cores");
|
||||
return;
|
||||
G4Exception("G4WorkerThread::SetPinAffinity()", "Run0100", JustWarning,
|
||||
"Cannot set thread affinity, affinity parameter larger than "
|
||||
"number of cores");
|
||||
return;
|
||||
}
|
||||
if (offset>0) // Start assigning affinity to given CPU
|
||||
if(offset > 0) // Start assigning affinity to given CPU
|
||||
{
|
||||
--offset;
|
||||
cpuindex = (GetThreadId()+offset) % G4Threading::G4GetNumberOfCores();
|
||||
// Round robin
|
||||
--offset;
|
||||
cpuindex = (GetThreadId() + offset) % G4Threading::G4GetNumberOfCores();
|
||||
// Round robin
|
||||
}
|
||||
else // Exclude the given CPU
|
||||
{
|
||||
offset *= -1;
|
||||
--offset;
|
||||
G4int myidx = GetThreadId()%(G4Threading::G4GetNumberOfCores()-1);
|
||||
cpuindex = myidx + (myidx>=offset);
|
||||
offset *= -1;
|
||||
--offset;
|
||||
G4int myidx = GetThreadId() % (G4Threading::G4GetNumberOfCores() - 1);
|
||||
cpuindex = myidx + (myidx >= offset);
|
||||
}
|
||||
G4cout << "Setting affinity to:" << cpuindex << G4endl;
|
||||
|
||||
#if defined(G4MULTITHREADED)
|
||||
// Avoid compilation warning in C90 standard w/o MT
|
||||
G4NativeThread t = pthread_self();
|
||||
#else
|
||||
G4NativeThread t;
|
||||
#endif
|
||||
G4bool success = G4Threading::G4SetPinAffinity(cpuindex,t);
|
||||
if ( ! success )
|
||||
# if defined(G4MULTITHREADED)
|
||||
// Avoid compilation warning in C90 standard w/o MT
|
||||
G4NativeThread t = pthread_self();
|
||||
# else
|
||||
G4NativeThread t;
|
||||
# endif
|
||||
G4bool success = G4Threading::G4SetPinAffinity(cpuindex, t);
|
||||
if(!success)
|
||||
{
|
||||
G4Exception("G4MTRunManagerKernel::StarThread()", "Run0101",
|
||||
JustWarning, "Cannot set thread affinity.");
|
||||
G4Exception("G4MTRunManagerKernel::StarThread()", "Run0101", JustWarning,
|
||||
"Cannot set thread affinity.");
|
||||
}
|
||||
#endif
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user