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(), _outBase("") {}

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

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

  // Open the run file
  // -----------------
  _outFile = GGSRootFileService::GetInstance().GetFileForThisRun(_outBase, run);
  if (!_outFile || (_outFile && _outFile->IsZombie())) {
    COUT(ERROR) << "Cannot open ROOT file " << _outBase << "for run" << run->GetRunID() << ENDL;
    throw std::runtime_error("");
  }
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......

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

  GGSVGeometryConstruction *geometry = GGSGeoPluginManager::GetInstance().GetGeoConstruction();
  if (!geometry) {
    COUT(ERROR) << "Can't get the geometry. Exit." << ENDL;
    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->cd();
  gGeoManager->Write("GGSGeometry");

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

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