Import Geant4 0.0.0 source tree
This commit is contained in:
@@ -0,0 +1,191 @@
|
||||
// This code implementation is the intellectual property of
|
||||
// the RD44 GEANT4 collaboration.
|
||||
//
|
||||
// By copying, distributing or modifying the Program (or any work
|
||||
// based on the Program) you indicate your acceptance of this statement,
|
||||
// and all its terms.
|
||||
//
|
||||
// $Id: G4DigiManager.cc,v 2.2 1998/07/13 17:28:17 urbi Exp $
|
||||
// GEANT4 tag $Name: geant4-00 $
|
||||
//
|
||||
|
||||
#include "G4DigiManager.hh"
|
||||
#include "G4Event.hh"
|
||||
#include "G4VHitsCollection.hh"
|
||||
#include "G4VDigiCollection.hh"
|
||||
#include "G4DMmessenger.hh"
|
||||
#include "G4DCofThisEvent.hh"
|
||||
#include "G4RunManager.hh"
|
||||
#include "G4SDManager.hh"
|
||||
#include "G4ios.hh"
|
||||
|
||||
|
||||
G4DigiManager* G4DigiManager::fDManager = 0;
|
||||
|
||||
G4DigiManager* G4DigiManager::GetDMpointer()
|
||||
{
|
||||
if(!fDManager)
|
||||
{
|
||||
fDManager = new G4DigiManager;
|
||||
}
|
||||
return fDManager;
|
||||
}
|
||||
|
||||
G4DigiManager* G4DigiManager::GetDMpointerIfExist()
|
||||
{ return fDManager; }
|
||||
|
||||
G4DigiManager::G4DigiManager():verboseLevel(0)
|
||||
{
|
||||
theMessenger = new G4DMmessenger(this);
|
||||
runManager = G4RunManager::GetRunManager();
|
||||
SDManager = G4SDManager::GetSDMpointer();
|
||||
DCtable = new G4DCtable;
|
||||
}
|
||||
|
||||
G4DigiManager::~G4DigiManager()
|
||||
{
|
||||
DMtable.clearAndDestroy();
|
||||
delete DCtable;
|
||||
delete theMessenger;
|
||||
}
|
||||
|
||||
void G4DigiManager::AddNewModule(G4VDigitizerModule* DM)
|
||||
{
|
||||
G4String DMname = DM->GetName();
|
||||
if(DMtable.index(DM) != RW_NPOS)
|
||||
{
|
||||
G4cout << "<" << DMname << "> has already been registored." << endl;
|
||||
return;
|
||||
}
|
||||
else if( verboseLevel > 0 )
|
||||
{
|
||||
G4cout << "New DigitizerModule <" << DMname
|
||||
<< "> is registored." << endl;
|
||||
}
|
||||
DMtable.insert(DM);
|
||||
|
||||
G4int numberOfCollections = DM->GetNumberOfCollections();
|
||||
for(int i=0;i<numberOfCollections;i++)
|
||||
{
|
||||
G4String DCname = DM->GetCollectionName(i);
|
||||
if( DCtable->Registor(DMname,DCname) < 0 )
|
||||
{
|
||||
G4cout << "DigiCollection <" << DCname
|
||||
<< "> has already been registored with "
|
||||
<< DMname << " DigitizerModule." << endl;
|
||||
}
|
||||
else if( verboseLevel > 0 )
|
||||
{
|
||||
G4cout << "DigiCollection " << DCname
|
||||
<< " is registored. " << endl;
|
||||
}
|
||||
}
|
||||
|
||||
runManager->SetDCtable(DCtable);
|
||||
}
|
||||
|
||||
void G4DigiManager::Digitize(G4String mName)
|
||||
{
|
||||
G4VDigitizerModule* aDM = FindDigitizerModule(mName);
|
||||
if(aDM)
|
||||
{ aDM->Digitize(); }
|
||||
else
|
||||
{ G4cout << "Unknown digitizer module <" << mName << ">. Digitize() ignored." << endl; }
|
||||
}
|
||||
|
||||
G4VDigitizerModule* G4DigiManager::FindDigitizerModule(G4String mName)
|
||||
{
|
||||
for(G4int i=0;i<DMtable.entries();i++)
|
||||
{
|
||||
if(DMtable[i]->GetName() == mName) return DMtable[i];
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
const G4VHitsCollection* G4DigiManager::GetHitsCollection(G4int HCID,G4int eventID)
|
||||
{
|
||||
const G4Event* evt = NULL;
|
||||
if(eventID==0)
|
||||
{ evt = runManager->GetCurrentEvent(); }
|
||||
else
|
||||
{ evt = runManager->GetPreviousEvent(eventID); }
|
||||
if(evt==NULL) return NULL;
|
||||
|
||||
G4HCofThisEvent* HCE = evt->GetHCofThisEvent();
|
||||
if(HCE==NULL) return NULL;
|
||||
|
||||
return HCE->GetHC(HCID);
|
||||
}
|
||||
|
||||
const G4VDigiCollection* G4DigiManager::GetDigiCollection(G4int DCID,G4int eventID)
|
||||
{
|
||||
const G4Event* evt = NULL;
|
||||
if(eventID==0)
|
||||
{ evt = runManager->GetCurrentEvent(); }
|
||||
else
|
||||
{ evt = runManager->GetPreviousEvent(eventID); }
|
||||
if(evt==NULL) return NULL;
|
||||
|
||||
G4DCofThisEvent* DCE = evt->GetDCofThisEvent();
|
||||
if(DCE==NULL) return NULL;
|
||||
|
||||
return DCE->GetDC(DCID);
|
||||
}
|
||||
|
||||
G4int G4DigiManager::GetHitsCollectionID(G4String HCname)
|
||||
{
|
||||
return SDManager->GetCollectionID(HCname);
|
||||
}
|
||||
|
||||
G4int G4DigiManager::GetDigiCollectionID(G4String DCname)
|
||||
{
|
||||
G4int i = DCtable->GetCollectionID(DCname);
|
||||
if(i==-2)
|
||||
{ G4cout << "< " << DCname << "> is ambegious." << endl; }
|
||||
return i;
|
||||
}
|
||||
|
||||
void G4DigiManager::SetDigiCollection(G4int DCID,G4VDigiCollection* aDC)
|
||||
{
|
||||
const G4Event* consEvt = runManager->GetCurrentEvent();
|
||||
if(consEvt==NULL)
|
||||
{
|
||||
G4cout << "G4DigiManager::SetDigiCollection --- "
|
||||
<< "Event object is not available." << endl;
|
||||
return;
|
||||
}
|
||||
|
||||
G4Event* evt = (G4Event*)consEvt;
|
||||
G4DCofThisEvent* DCE = evt->GetDCofThisEvent();
|
||||
if(DCE==NULL)
|
||||
{
|
||||
DCE = new G4DCofThisEvent(DCtable->entries());
|
||||
evt->SetDCofThisEvent(DCE);
|
||||
if(verboseLevel>0)
|
||||
{ G4cout << "DCofThisEvent object is added to current G4Event." << endl; }
|
||||
}
|
||||
|
||||
DCE->AddDigiCollection(DCID,aDC);
|
||||
|
||||
if(verboseLevel>0)
|
||||
{
|
||||
G4cout << aDC->GetName() << " is stored at " << DCID
|
||||
<< "-th slot of G4DCofThisEvent." << endl;
|
||||
}
|
||||
}
|
||||
|
||||
void G4DigiManager::SetVerboseLevel(G4int val)
|
||||
{
|
||||
verboseLevel = val;
|
||||
for(G4int i=0;i<DMtable.entries();i++)
|
||||
{ DMtable[i]->SetVerboseLevel(val); }
|
||||
}
|
||||
|
||||
void G4DigiManager::List() const
|
||||
{
|
||||
for(G4int i=0;i<DMtable.entries();i++)
|
||||
{ G4cout << " " << i << " : " << DMtable[i]->GetName() << endl; }
|
||||
}
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user