Import Geant4 11.1.0 source tree
This commit is contained in:
@@ -55,132 +55,19 @@ G4bool G4XmlAnalysisManager::IsInstance()
|
||||
G4XmlAnalysisManager::G4XmlAnalysisManager()
|
||||
: G4ToolsAnalysisManager("Xml")
|
||||
{
|
||||
if ( ! G4Threading::IsWorkerThread() ) fgMasterInstance = this;
|
||||
|
||||
// File Manager
|
||||
fFileManager = std::make_shared<G4XmlFileManager>(fState);
|
||||
SetFileManager(fFileManager);
|
||||
auto fileManager = std::make_shared<G4XmlFileManager>(fState);
|
||||
SetFileManager(fileManager);
|
||||
|
||||
// Ntuple file manager
|
||||
fNtupleFileManager = std::make_shared<G4XmlNtupleFileManager>(fState);
|
||||
fNtupleFileManager->SetFileManager(fFileManager);
|
||||
SetNtupleFileManager(fNtupleFileManager);
|
||||
fNtupleFileManager->SetFileManager(fileManager);
|
||||
fNtupleFileManager->SetBookingManager(fNtupleBookingManager);
|
||||
}
|
||||
|
||||
//_____________________________________________________________________________
|
||||
G4XmlAnalysisManager::~G4XmlAnalysisManager()
|
||||
{
|
||||
if ( fState.GetIsMaster() ) fgMasterInstance = nullptr;
|
||||
fgIsInstance = false;
|
||||
}
|
||||
|
||||
//
|
||||
// protected methods
|
||||
//
|
||||
|
||||
//_____________________________________________________________________________
|
||||
G4bool G4XmlAnalysisManager::OpenFileImpl(const G4String& fileName)
|
||||
{
|
||||
// Create ntuple manager(s)
|
||||
// and set it to base class which takes then their ownership
|
||||
SetNtupleManager(fNtupleFileManager->CreateNtupleManager());
|
||||
|
||||
auto result = true;
|
||||
|
||||
// Save file name in file manager
|
||||
result &= fFileManager->OpenFile(fileName);
|
||||
|
||||
// Open ntuple files and create ntuples from bookings
|
||||
result &= fNtupleFileManager->ActionAtOpenFile(fFileManager->GetFullFileName());
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
//_____________________________________________________________________________
|
||||
G4bool G4XmlAnalysisManager::WriteImpl()
|
||||
{
|
||||
auto result = true;
|
||||
|
||||
Message(kVL4, "write", "files");
|
||||
|
||||
// ntuples
|
||||
fNtupleFileManager->ActionAtWrite();
|
||||
|
||||
if ( G4Threading::IsWorkerThread() ) {
|
||||
result &= G4ToolsAnalysisManager::Merge();
|
||||
}
|
||||
else {
|
||||
// Open all files registered with objects
|
||||
fFileManager->OpenFiles();
|
||||
|
||||
// Write all histograms/profile on master
|
||||
result &= G4ToolsAnalysisManager::WriteImpl();
|
||||
}
|
||||
|
||||
// Write ASCII if activated
|
||||
if ( IsAscii() ) {
|
||||
result &= WriteAscii(fFileManager->GetFileName());
|
||||
}
|
||||
|
||||
// File
|
||||
result &= fFileManager->WriteFiles();
|
||||
|
||||
Message(kVL3, "write", "files", "", result);
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
//_____________________________________________________________________________
|
||||
G4bool G4XmlAnalysisManager::CloseFileImpl(G4bool reset)
|
||||
{
|
||||
auto result = true;
|
||||
|
||||
Message(kVL4, "close", "files");
|
||||
|
||||
// Close open files
|
||||
result &= fFileManager->CloseFiles();
|
||||
|
||||
// Close ntuple files
|
||||
result &= fNtupleFileManager->ActionAtCloseFile(reset);
|
||||
|
||||
// Reset data
|
||||
if ( reset ) {
|
||||
if ( ! Reset() ) {
|
||||
Warn("Resetting data failed", fkClass, "CloseFileImpl");
|
||||
result = false;
|
||||
}
|
||||
}
|
||||
|
||||
// delete files if empty
|
||||
// (ntuple files are created only if an ntuple is created)
|
||||
if ( fFileManager->GetFile() && G4ToolsAnalysisManager::IsEmpty() ) {
|
||||
if ( std::remove(fFileManager->GetFullFileName()) ) {
|
||||
// std::remove returns 0 when success
|
||||
Warn(G4String(
|
||||
"Removing file ") + fFileManager->GetFullFileName() + " failed",
|
||||
fkClass, "CloseFileImpl");
|
||||
result = false;
|
||||
}
|
||||
Message(kVL1, "delete", "empty file", fFileManager->GetFullFileName());
|
||||
}
|
||||
else {
|
||||
Message(kVL3, "close", "files");
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
//_____________________________________________________________________________
|
||||
G4bool G4XmlAnalysisManager::ResetImpl()
|
||||
{
|
||||
// Reset histograms and ntuple
|
||||
|
||||
auto result = true;
|
||||
|
||||
result &= G4ToolsAnalysisManager::ResetImpl();
|
||||
if ( fNtupleFileManager != nullptr ) {
|
||||
result &= fNtupleFileManager->Reset();
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
@@ -56,14 +56,18 @@ G4XmlFileManager::G4XmlFileManager(const G4AnalysisManagerState& state)
|
||||
G4String G4XmlFileManager::GetNtupleFileName(XmlNtupleDescription* ntupleDescription)
|
||||
{
|
||||
// get ntuple file name
|
||||
auto ntupleFileName = ntupleDescription->fFileName;
|
||||
if ( ntupleFileName.size() ) {
|
||||
auto ntupleFileName = ntupleDescription->GetFileName();
|
||||
if (ntupleFileName.size() != 0u) {
|
||||
// update filename per object per thread
|
||||
ntupleFileName = GetTnFileName(ntupleFileName, GetFileType());
|
||||
} else {
|
||||
// compose ntuple file name from the default file name
|
||||
ntupleFileName = GetNtupleFileName(ntupleDescription->fNtupleBooking.name());
|
||||
ntupleFileName =
|
||||
GetTnFileName(ntupleFileName, GetFileType()/*, ntupleDescription->GetCycle()*/);
|
||||
}
|
||||
else {
|
||||
// compose ntuple file name from the default file name
|
||||
ntupleFileName = GetNtupleFileName(
|
||||
ntupleDescription->GetNtupleBooking().name()/*, ntupleDescription->GetCycle()*/);
|
||||
}
|
||||
|
||||
return ntupleFileName;
|
||||
}
|
||||
|
||||
@@ -138,54 +142,31 @@ G4bool G4XmlFileManager::OpenFile(const G4String& fileName)
|
||||
G4bool G4XmlFileManager::CreateNtupleFile(
|
||||
XmlNtupleDescription* ntupleDescription)
|
||||
{
|
||||
// get ntuple file name per object (if defined)
|
||||
// Get ntuple file name per object (if defined)
|
||||
auto ntupleFileName = GetNtupleFileName(ntupleDescription);
|
||||
|
||||
Message(kVL4, "create", "ntuple file", ntupleFileName);
|
||||
|
||||
// update file name if it is already in use
|
||||
while ( GetTFile(ntupleFileName, false) ) {
|
||||
// the file is already in use
|
||||
auto oldName = ntupleDescription->fFileName;
|
||||
auto newName = GetBaseName(oldName) + "_bis." + GetExtension(oldName);
|
||||
ntupleDescription->fFileName = newName;
|
||||
|
||||
Warn(G4String( "Ntuple filename ") + oldName + " is already in use.\n" +
|
||||
"It will be replaced with : " + newName,
|
||||
fkClass, "CreateNtupleFile");
|
||||
|
||||
ntupleFileName = GetNtupleFileName(ntupleDescription);
|
||||
// Create ntuple file name if it does not yet exist
|
||||
auto ntupleFile = GetTFile(ntupleFileName, false);
|
||||
if ( ntupleFile == nullptr) {
|
||||
ntupleFile = CreateTFile(ntupleFileName);
|
||||
}
|
||||
|
||||
ntupleDescription->fFile = CreateTFile(ntupleFileName);
|
||||
ntupleDescription->SetFile(ntupleFile);
|
||||
|
||||
Message(kVL2, "create", "ntuple file", ntupleFileName);
|
||||
|
||||
return (ntupleDescription->fFile != nullptr);
|
||||
return (ntupleDescription->GetFile() != nullptr);
|
||||
}
|
||||
|
||||
//_____________________________________________________________________________
|
||||
G4bool G4XmlFileManager::CloseNtupleFile(
|
||||
XmlNtupleDescription* ntupleDescription)
|
||||
{
|
||||
// Do nothing if there is no file
|
||||
if ( ! ntupleDescription->fFile ) return true;
|
||||
|
||||
auto result = true;
|
||||
|
||||
auto ntupleFileName = GetNtupleFileName(ntupleDescription);
|
||||
|
||||
Message(kVL4, "close", "ntuple file", ntupleFileName);
|
||||
|
||||
// Close file
|
||||
result &= CloseTFile(ntupleFileName);
|
||||
// Notify not empty file
|
||||
result &=SetIsEmpty(ntupleFileName, ! ntupleDescription->fHasFill);
|
||||
auto ntupleFileName = GetNtupleFileName(ntupleDescription);
|
||||
auto result = SetIsEmpty(ntupleFileName, ! ntupleDescription->GetHasFill());
|
||||
|
||||
// Reset file info in ntuple description
|
||||
ntupleDescription->fFile.reset();
|
||||
|
||||
Message(kVL2, "close", "ntuple file", ntupleFileName);
|
||||
// Ntuple files are registered in file manager map.
|
||||
// they will be closed with CloseFiles() calls
|
||||
ntupleDescription->GetFile().reset();
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
@@ -41,24 +41,6 @@ G4XmlNtupleFileManager::G4XmlNtupleFileManager(const G4AnalysisManagerState& sta
|
||||
: G4VNtupleFileManager(state, "csv")
|
||||
{}
|
||||
|
||||
//
|
||||
// private methods
|
||||
//
|
||||
|
||||
//_____________________________________________________________________________
|
||||
G4bool G4XmlNtupleFileManager::CloseNtupleFiles()
|
||||
{
|
||||
// Close ntuple files
|
||||
|
||||
auto result = true;
|
||||
auto ntupleVector = fNtupleManager->GetNtupleDescriptionVector();
|
||||
for ( auto ntupleDescription : ntupleVector) {
|
||||
result &= fFileManager->CloseNtupleFile(ntupleDescription);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
//
|
||||
// public methods
|
||||
//
|
||||
@@ -88,27 +70,24 @@ G4bool G4XmlNtupleFileManager::ActionAtWrite()
|
||||
{
|
||||
auto ntupleVector = fNtupleManager->GetNtupleDescriptionVector();
|
||||
|
||||
for ( auto ntuple : ntupleVector ) {
|
||||
if ( ntuple->fNtuple ) ntuple->fNtuple->write_trailer();
|
||||
for ( auto ntupleDescription : ntupleVector ) {
|
||||
if (ntupleDescription->GetNtuple() != nullptr) {
|
||||
ntupleDescription->GetNtuple()->write_trailer();
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
//_____________________________________________________________________________
|
||||
G4bool G4XmlNtupleFileManager::ActionAtCloseFile(G4bool reset)
|
||||
G4bool G4XmlNtupleFileManager::ActionAtCloseFile()
|
||||
{
|
||||
auto result = true;
|
||||
|
||||
// Close ntuple files
|
||||
result &= CloseNtupleFiles();
|
||||
|
||||
// Reset data
|
||||
if ( ! reset ) {
|
||||
result = Reset();
|
||||
if ( ! result ) {
|
||||
Warn("Resetting data failed", fkClass, "ActionAtCloseFile");
|
||||
}
|
||||
auto ntupleVector = fNtupleManager->GetNtupleDescriptionVector();
|
||||
for ( auto ntupleDescription : ntupleVector) {
|
||||
result &= fFileManager->CloseNtupleFile(ntupleDescription);
|
||||
}
|
||||
|
||||
return result;
|
||||
|
||||
@@ -53,10 +53,11 @@ void G4XmlNtupleManager::CreateTNtupleFromBooking(
|
||||
if ( ! fFileManager->CreateNtupleFile(ntupleDescription) ) return;
|
||||
|
||||
// create ntuple
|
||||
ntupleDescription->fNtuple
|
||||
= new tools::waxml::ntuple(
|
||||
*(ntupleDescription->fFile), G4cerr, ntupleDescription->fNtupleBooking);
|
||||
fNtupleVector.push_back(ntupleDescription->fNtuple);
|
||||
ntupleDescription->SetNtuple(
|
||||
new tools::waxml::ntuple(
|
||||
*(ntupleDescription->GetFile()), G4cerr,
|
||||
ntupleDescription->GetNtupleBooking()));
|
||||
fNtupleVector.push_back(ntupleDescription->GetNtuple());
|
||||
}
|
||||
|
||||
//_____________________________________________________________________________
|
||||
@@ -66,24 +67,32 @@ void G4XmlNtupleManager::FinishTNtuple(
|
||||
|
||||
{
|
||||
// Do nothing if the base file name was not yet defined
|
||||
if ( ! fFileManager->GetFileName().size() ) return;
|
||||
if (fFileManager->GetFileName().size() == 0u) return;
|
||||
|
||||
// Create ntuple from booking
|
||||
if ( ! ntupleDescription->fNtuple ) {
|
||||
if (ntupleDescription->GetNtuple() == nullptr) {
|
||||
CreateTNtupleFromBooking(ntupleDescription);
|
||||
}
|
||||
|
||||
// Return if creating ntuple failed
|
||||
if ( ! ntupleDescription->fNtuple ) {
|
||||
if (ntupleDescription->GetNtuple() == nullptr) {
|
||||
Warn("Creating ntuple has failed. ", fkClass, "FinishTNtuple");
|
||||
return;
|
||||
}
|
||||
|
||||
// Update ntuple name if cycle >0
|
||||
auto ntupleName = ntupleDescription->GetNtupleBooking().name();
|
||||
// if (ntupleDescription->GetCycle() > 0) {
|
||||
if (GetCycle() > 0) {
|
||||
ntupleName.append("_v");
|
||||
ntupleName.append(std::to_string(GetCycle()));
|
||||
}
|
||||
|
||||
// Write header
|
||||
G4String path = "/";
|
||||
path.append(fFileManager->GetNtupleDirectoryName());
|
||||
ntupleDescription->fNtuple
|
||||
->write_header(path, ntupleDescription->fNtupleBooking.name(),
|
||||
ntupleDescription->fNtupleBooking.title());
|
||||
ntupleDescription->GetNtuple()
|
||||
->write_header(path, ntupleName,
|
||||
ntupleDescription->GetNtupleBooking().title());
|
||||
fFileManager->LockDirectoryNames();
|
||||
}
|
||||
|
||||
@@ -74,7 +74,7 @@ G4bool G4XmlRFileManager::OpenRFile(const G4String& fileName)
|
||||
G4bool verbose = false;
|
||||
|
||||
// create factory (if it does not yet exist)
|
||||
if ( ! fReadFactory ) {
|
||||
if (fReadFactory == nullptr) {
|
||||
fReadFactory = new tools::xml::default_factory();
|
||||
}
|
||||
|
||||
@@ -116,9 +116,8 @@ toolx::raxml* G4XmlRFileManager::GetRFile(const G4String& fileName) const
|
||||
G4String name = GetFullFileName(fileName, isPerThread);
|
||||
|
||||
auto it = fRFiles.find(name);
|
||||
if ( it != fRFiles.end() )
|
||||
if (it != fRFiles.end()) {
|
||||
return it->second;
|
||||
else {
|
||||
return nullptr;
|
||||
}
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
@@ -88,7 +88,7 @@ G4int G4XmlRNtupleManager::ReadNtupleImpl(const G4String& ntupleName,
|
||||
|
||||
auto handler = fFileManager->GetHandler<tools::aida::ntuple>(
|
||||
fullFileName, ntupleName, "ReadNtupleImpl");
|
||||
if ( ! handler ) return kInvalidId;
|
||||
if (handler == nullptr) return kInvalidId;
|
||||
|
||||
auto rntuple = static_cast<tools::aida::ntuple*>(handler->object());
|
||||
auto id = SetNtuple(new G4TRNtupleDescription<tools::aida::ntuple>(rntuple));
|
||||
@@ -109,7 +109,7 @@ G4bool G4XmlRNtupleManager::SetNtupleIColumn(G4int ntupleId,
|
||||
" ntupleId " + to_string(ntupleId) + " " + columnName);
|
||||
|
||||
auto ntupleDescription = GetNtupleDescriptionInFunction(ntupleId, "SetNtupleIColumn");
|
||||
if ( ! ntupleDescription ) return false;
|
||||
if (ntupleDescription == nullptr) return false;
|
||||
|
||||
// not supported
|
||||
//tools::ntuple_binding* ntupleBinding = ntupleDescription->fNtupleBinding;
|
||||
@@ -138,7 +138,7 @@ G4bool G4XmlRNtupleManager::SetNtupleFColumn(G4int ntupleId,
|
||||
|
||||
|
||||
auto ntupleDescription = GetNtupleDescriptionInFunction(ntupleId, "SetNtupleFColumn");
|
||||
if ( ! ntupleDescription ) return false;
|
||||
if (ntupleDescription == nullptr) return false;
|
||||
|
||||
// not supported
|
||||
//tools::ntuple_binding* ntupleBinding = ntupleDescription->fNtupleBinding;
|
||||
@@ -166,7 +166,7 @@ G4bool G4XmlRNtupleManager::SetNtupleDColumn(G4int ntupleId,
|
||||
" ntupleId " + to_string(ntupleId) + " " + columnName);
|
||||
|
||||
auto ntupleDescription = GetNtupleDescriptionInFunction(ntupleId, "SetNtupleDColumn");
|
||||
if ( ! ntupleDescription ) return false;
|
||||
if (ntupleDescription == nullptr) return false;
|
||||
|
||||
// not supported
|
||||
//tools::ntuple_binding* ntupleBinding = ntupleDescription->fNtupleBinding;
|
||||
@@ -194,7 +194,7 @@ G4bool G4XmlRNtupleManager::SetNtupleSColumn(G4int ntupleId,
|
||||
" ntupleId " + to_string(ntupleId) + " " + columnName);
|
||||
|
||||
auto ntupleDescription = GetNtupleDescriptionInFunction(ntupleId, "SetNtupleSColumn");
|
||||
if ( ! ntupleDescription ) return false;
|
||||
if (ntupleDescription == nullptr) return false;
|
||||
|
||||
// not supported
|
||||
//tools::ntuple_binding* ntupleBinding = ntupleDescription->fNtupleBinding;
|
||||
|
||||
Reference in New Issue
Block a user