GGSSaveTGeoAction.cpp

/*
 * GGSSaveTGeoAction.cpp
 *
 *  Created on: Jun 3, 2020
 *      Author: Valerio Formato
 */

// GGS headers
#include "montecarlo/useractions/GGSSaveTGeoAction.h"
#include "geometry/GGSVGeometryConstruction.h"
#include "geometry/pluginmanagers/GGSGeoPluginManager.h"
#include "montecarlo/pluginmanagers/GGSMCPluginMacros.h"
#include "montecarlo/services/GGSRootFileService.h"
#include "utils/GGSSmartLog.h"

// VGM headers
#include "Geant4GM/volumes/Factory.h"
#include "RootGM/volumes/Factory.h"
#include "TGeoManager.h"

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......

RegisterUA(GGSSaveTGeoAction);

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......

GGSSaveTGeoAction::GGSSaveTGeoAction() : GGSUserAction(), _outFile(nullptr), _outBase("") {}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......

void GGSSaveTGeoAction::BeginOfRunAction(const G4Run *run) { // -----------------
  const std::string routineName("GGSSaveTGeoAction::BeginOfRunAction");

  // Open the run file
  // -----------------
  if (GGSRootFileService::GetInstance().GetOutputMode() == GGSRootFileService::OutputMode::MULTIFILE ||
      G4Threading::G4GetThreadId() <= 0) {
    _outFile = GGSRootFileService::GetInstance().GetFileForThisRun(_outBase, run);
    if (!_outFile || (_outFile && _outFile->IsZombie())) {
      GGSCOUT(ERROR) << "Cannot open ROOT file " << _outBase << "for run" << run->GetRunID() << GGSENDL;
      throw std::runtime_error("");
    }
  }
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
namespace {
G4Mutex mutexForSaveTGeo = G4MUTEX_INITIALIZER;
}
void GGSSaveTGeoAction::EndOfRunAction(const G4Run *run) {
  const std::string routineName("GGSSaveTGeoAction::EndOfRunAction");

  if (GGSRootFileService::GetInstance().GetOutputMode() == GGSRootFileService::OutputMode::MULTIFILE ||
      G4Threading::G4GetThreadId() <= 0) {

    // Acquire a lock since VGM is probably not thread-safe
    G4AutoLock lock(mutexForSaveTGeo);

    GGSVGeometryConstruction *geometry = GGSGeoPluginManager::GetInstance().GetGeoConstruction();
    if (!geometry) {
      GGSCOUT(ERROR) << "Can't get the geometry. Exit." << GGSENDL;
      throw std::runtime_error("Invalid geometry");
    }

    auto physVolume = geometry->GetVolume();

    auto g4Factory = std::make_unique<Geant4GM::Factory>();
    if (GGSSmartLog::verboseLevel >= GGSSmartLog::DEBUG)
      g4Factory->SetDebug(1);
    g4Factory->Import(physVolume);

    // Export VGM geometry to Root
    auto rtFactory = std::make_unique<RootGM::Factory>();
    if (GGSSmartLog::verboseLevel >= GGSSmartLog::DEBUG)
      rtFactory->SetDebug(1);
    g4Factory->Export(rtFactory.get());
    gGeoManager->CloseGeometry();

    _outFile->WriteObject(gGeoManager, "GGSGeometry");

    GGSRootFileService::GetInstance().CloseFileForThisRun(_outBase, run);
  }
}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......