RootFile_service.h

Go to the documentation of this file.

#pragma once
 
#include <iostream>
#include <vector>
#include <string>
#include <mutex>
 
#include <JANA/JApplicationFwd.h>
#include "services/log/Log_service.h"
#include <JANA/Services/JGlobalRootLock.h>
#include <JANA/Services/JServiceLocator.h>
 
#include <TFile.h>
 
class RootFile_service : public JService {
public:
  explicit RootFile_service(JApplication* app) : m_app(app) {}
  ~RootFile_service() override { CloseHistFile(); }
 
  void acquire_services(JServiceLocator* /* locayor */) override {
    auto log_service = m_app->GetService<Log_service>();
    m_log            = log_service->logger("RootFile");
  }
 
  TDirectory* GetHistFile(bool create_if_needed = true) {
 
    if (create_if_needed) {
      std::call_once(init_flag, &RootFile_service::CreateHistFile, this);
    }
    return m_histfile;
  }
 
  void CloseHistFile() {
    if (m_histfile) {
      std::string filename = m_histfile->GetName();
      m_histfile->Write();
      delete m_histfile;
      m_log->info("Closed user histogram file: {}", filename);
    }
    m_histfile = nullptr;
  }
 
private:
  void CreateHistFile() {
    // Get root file name
    std::string filename = "eicrecon.root";
    m_app->SetDefaultParameter("histsfile", filename,
                               "Name of root file to be created for plugin histograms/trees");
    if (!m_histfile) {
      try {
        m_histfile = new TFile(filename.c_str(), "RECREATE", "user histograms/trees");
        m_log->info("Created file: {} for user histograms", filename);
      } catch (std::exception& ex) {
        m_log->error("Problem opening root file for histograms: {}", ex.what());
        throw ex;
      }
    }
  }
 
  RootFile_service() = default;
 
  JApplication* m_app = nullptr;
  std::shared_ptr<spdlog::logger> m_log;
  TFile* m_histfile = nullptr;
  std::once_flag init_flag;
};