lars/geant4
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Import Geant4 11.0.0.beta source tree

Browse Source
This commit is contained in:
Gabriele Cosmo
2021-06-25 16:12:29 +02:00
parent c968e26a39
commit 6399a014b6
4200 changed files with 207479 additions and 237366 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/digits_hits/utils/include/G4TScoreNtupleWriter.icc
+102 -71
View File
@@ -39,19 +39,20 @@
//_____________________________________________________________________________
template <typename T>
G4TScoreNtupleWriter<T>::G4TScoreNtupleWriter()
: G4VScoreNtupleWriter(),
fMessenger(nullptr),
fHCIds(),
fAnalysisManager(nullptr),
fFileName("scoring"),
fVerboseLevel(1),
fHasAnalysisManager(false),
fHasAnalysisFile(false),
fIsBooked(false),
fIsInitialized(false),
fFirstNtupleId(0)
: G4VScoreNtupleWriter()
, fMessenger(nullptr)
, fHCIds()
, fAnalysisManager(nullptr)
, fFileName("scoring")
, fVerboseLevel(1)
, fHasAnalysisManager(false)
, fHasAnalysisFile(false)
, fIsBooked(false)
, fIsInitialized(false)
, fFirstNtupleId(0)
{
if ( G4Threading::IsMasterThread() ) {
if(G4Threading::IsMasterThread())
{
fMessenger = new G4TScoreNtupleWriterMessenger<T>(this);
}
}
@@ -60,7 +61,8 @@ G4TScoreNtupleWriter<T>::G4TScoreNtupleWriter()
template <typename T>
G4TScoreNtupleWriter<T>::~G4TScoreNtupleWriter()
{
if ( fHasAnalysisManager ) {
if(fHasAnalysisManager)
{
delete fAnalysisManager;
}
@@ -75,12 +77,14 @@ G4TScoreNtupleWriter<T>::~G4TScoreNtupleWriter()
template <typename T>
void G4TScoreNtupleWriter<T>::CreateAnalysisManager()
{
if ( fAnalysisManager ) return;
if(fAnalysisManager)
return;
// create analysis manager
if ( ! T::IsInstance() ) fHasAnalysisManager = true;
if(!T::IsInstance())
fHasAnalysisManager = true;
fAnalysisManager = T::Instance();
fAnalysisManager->SetVerboseLevel(fVerboseLevel);
}
@@ -108,63 +112,73 @@ template <typename T>
G4bool G4TScoreNtupleWriter<T>::Book(G4HCofThisEvent* hce)
{
#ifdef G4VERBOSE
if ( fVerboseLevel > 1 ) {
if(fVerboseLevel > 1)
{
G4cout << "--- G4TScoreNtupleWriter<T>::Book" << G4endl;
}
#endif
if ( ! hce) return false;
if(!hce)
return false;
// book collection ID if DoubleHitsMap
if (fHCIds.size() == 0) {
if(fHCIds.size() == 0)
{
#ifdef G4VERBOSE
if ( fVerboseLevel > 1 ) {
G4cout << "--- going to fill fHCIds " << G4endl;
if(fVerboseLevel > 1)
{
G4cout << "--- going to fill fHCIds " << G4endl;
}
#endif
for (G4int i=0; i<hce->GetNumberOfCollections(); ++i) {
auto hitsMap
= dynamic_cast<G4THitsMap<G4double>*>(hce->GetHC(i));
for(G4int i = 0; i < hce->GetNumberOfCollections(); ++i)
{
auto hitsMap = dynamic_cast<G4THitsMap<G4double>*>(hce->GetHC(i));
if ( hitsMap ) {
if(hitsMap)
{
#ifdef G4VERBOSE
if ( fVerboseLevel > 1 ) {
G4cout << "--- adding hcID = " << i << G4endl;
}
if(fVerboseLevel > 1)
{
G4cout << "--- adding hcID = " << i << G4endl;
}
#endif
fHCIds.push_back(i);
fHCIds.push_back(i);
}
}
}
// Create analysis manager
if ( fHCIds.size() ) {
if(fHCIds.size())
{
CreateAnalysisManager();
}
// Create ntuples (only once)
if ( ! fIsInitialized ) {
if(!fIsInitialized)
{
#ifdef G4VERBOSE
if ( fVerboseLevel > 1 ) {
if(fVerboseLevel > 1)
{
G4cout << "-- going to create ntuples " << G4endl;
}
#endif
auto first = true;
for (auto id : fHCIds) {
auto hitsMap
= static_cast<G4THitsMap<G4double>*>(hce->GetHC(id));
auto first = true;
for(auto id : fHCIds)
{
auto hitsMap = static_cast<G4THitsMap<G4double>*>(hce->GetHC(id));
// Create ntuple for this primitive
G4String ntupleName(hitsMap->GetSDname());
ntupleName.append("_");
ntupleName.append(hitsMap->GetName());
G4int ntupleId
= fAnalysisManager->CreateNtuple(ntupleName, ntupleName);
if ( first ) {
G4int ntupleId = fAnalysisManager->CreateNtuple(ntupleName, ntupleName);
if(first)
{
fFirstNtupleId = ntupleId;
#ifdef G4VERBOSE
if ( fVerboseLevel > 1 ) {
G4cout << "-- set first ntuple Id " << fFirstNtupleId << G4endl;
if(fVerboseLevel > 1)
{
G4cout << "-- set first ntuple Id " << fFirstNtupleId << G4endl;
}
#endif
first = false;
@@ -180,7 +194,7 @@ G4bool G4TScoreNtupleWriter<T>::Book(G4HCofThisEvent* hce)
colName.append("_score");
fAnalysisManager->CreateNtupleDColumn(colName);
fAnalysisManager->FinishNtuple();
fIsBooked = true;
fIsBooked = true;
}
fIsInitialized = true;
}
@@ -193,26 +207,31 @@ template <typename T>
void G4TScoreNtupleWriter<T>::OpenFile()
{
#ifdef G4VERBOSE
if ( fVerboseLevel > 1 ) {
G4cout << "--- G4TScoreNtupleWriter<T>::OpenFile" << G4endl;
if(fVerboseLevel > 1)
{
G4cout << "--- G4TScoreNtupleWriter<T>::OpenFile" << G4endl;
}
#endif
if ( fAnalysisManager->IsOpenFile() ) return;
if(fAnalysisManager->IsOpenFile())
return;
#ifdef G4VERBOSE
if ( fVerboseLevel > 1 ) {
G4cout << "--- G4TScoreNtupleWriter<T>::OpenFile executing" << G4endl;
if(fVerboseLevel > 1)
{
G4cout << "--- G4TScoreNtupleWriter<T>::OpenFile executing" << G4endl;
}
#endif
if ( fAnalysisManager->GetFileName() == "" ) {
if(fAnalysisManager->GetFileName() == "")
{
fAnalysisManager->SetFileName(fFileName);
}
fAnalysisManager->OpenFile();
#ifdef G4VERBOSE
if ( fVerboseLevel > 1 ) {
if(fVerboseLevel > 1)
{
G4cout << "--- G4TScoreNtupleWriter<T>::OpenFile isOpenFile: "
<< fAnalysisManager->IsOpenFile() << G4endl;
}
@@ -225,38 +244,45 @@ void G4TScoreNtupleWriter<T>::OpenFile()
template <typename T>
void G4TScoreNtupleWriter<T>::Fill(G4HCofThisEvent* hce, G4int eventNumber)
{
#ifdef G4VERBOSE
if ( fVerboseLevel > 1 ) {
if(fVerboseLevel > 1)
{
G4cout << "--- start G4TScoreNtupleWriter<T>::Fill" << G4endl;
}
#endif
auto counter = 0;
for (auto id : fHCIds) {
for(auto id : fHCIds)
{
#ifdef G4VERBOSE
if ( fVerboseLevel > 1 ) {
if(fVerboseLevel > 1)
{
G4cout << "in loop over fHCIds, counter " << counter << G4endl;
}
}
#endif
auto hitsMap
= static_cast<G4THitsMap<G4double>*>(hce->GetHC(id));
auto hitsMap = static_cast<G4THitsMap<G4double>*>(hce->GetHC(id));
//G4cout << eventNumber << ".. go to fill ntuple " << counter + fFirstNtupleId << G4endl;
// G4cout << eventNumber << ".. go to fill ntuple " << counter +
// fFirstNtupleId << G4endl;
// fill hits in ntuple
std::map<G4int, G4double*>::iterator it;
for ( it = hitsMap->GetMap()->begin(); it != hitsMap->GetMap()->end(); it++ ) {
fAnalysisManager->FillNtupleIColumn(counter + fFirstNtupleId, 0, eventNumber);
fAnalysisManager->FillNtupleIColumn(counter + fFirstNtupleId, 1, it->first);
fAnalysisManager->FillNtupleDColumn(counter + fFirstNtupleId, 2, *(it->second));
for(it = hitsMap->GetMap()->begin(); it != hitsMap->GetMap()->end(); it++)
{
fAnalysisManager->FillNtupleIColumn(counter + fFirstNtupleId, 0,
eventNumber);
fAnalysisManager->FillNtupleIColumn(counter + fFirstNtupleId, 1,
it->first);
fAnalysisManager->FillNtupleDColumn(counter + fFirstNtupleId, 2,
*(it->second));
fAnalysisManager->AddNtupleRow(counter + fFirstNtupleId);
}
counter++;
}
#ifdef G4VERBOSE
if ( fVerboseLevel > 1 ) {
if(fVerboseLevel > 1)
{
G4cout << "--- done G4TScoreNtupleWriter<T>::Fill" << G4endl;
}
#endif
@@ -267,14 +293,17 @@ template <typename T>
void G4TScoreNtupleWriter<T>::Write()
{
#ifdef G4VERBOSE
if ( fVerboseLevel > 1 ) {
if(fVerboseLevel > 1)
{
G4cout << "--- start G4TScoreNtupleWriter<T>::Write" << G4endl;
}
#endif
if ( fHasAnalysisFile ) {
if(fHasAnalysisFile)
{
#ifdef G4VERBOSE
if ( fVerboseLevel > 1 ) {
if(fVerboseLevel > 1)
{
G4cout << "--- G4TScoreNtupleWriter<T>::Write - has file" << G4endl;
}
#endif
@@ -284,7 +313,8 @@ void G4TScoreNtupleWriter<T>::Write()
}
#ifdef G4VERBOSE
if ( fVerboseLevel > 1 ) {
if(fVerboseLevel > 1)
{
G4cout << "--- done G4TScoreNtupleWriter<T>::Write" << G4endl;
}
#endif
@@ -293,9 +323,10 @@ void G4TScoreNtupleWriter<T>::Write()
//_____________________________________________________________________________
template <typename T>
void G4TScoreNtupleWriter<T>::SetVerboseLevel(G4int value)
{
fVerboseLevel = value;
if ( fAnalysisManager ) {
{
fVerboseLevel = value;
if(fAnalysisManager)
{
fAnalysisManager->SetVerboseLevel(value);
}
}
@@ -303,6 +334,6 @@ void G4TScoreNtupleWriter<T>::SetVerboseLevel(G4int value)
//_____________________________________________________________________________
template <typename T>
void G4TScoreNtupleWriter<T>::SetFileName(const G4String& fileName)
{
fFileName = fileName;
{
fFileName = fileName;
}