A PersistenceManager that writes variables to ROOT histograms (TH1D). More...

#include <ConcreteVariablesToHistogramPersistenceManager.h>

Inheritance diagram for ConcreteVariablesToHistogramPersistenceManager:

Public Member Functions
	ConcreteVariablesToHistogramPersistenceManager ()
	Default constructor.

void	initialize (const std::string &fileName, const std::string &directory, Variables &variables) override
	Initializes the manager by opening a ROOT file and creating histograms.

void	addEntry (const EvaluatedVariables &evaluatedVariables) override
	Fills histograms with the current set of evaluated variables.

void	store () override
	Writes histogram data to disk.

Private Member Functions
void	openFileWithGuards ()
	Safely opens the ROOT file specified by `m_fileName`.

void	registerHistograms ()
	Registers (books) the histograms based on `m_variables`.

Private Attributes
std::string	m_fileName
	Name of the ROOT file where histograms will be written.

std::string	m_directory
	Name of the directory (folder) within the ROOT file where histograms are stored.

Variables	m_variables
	A list of variables associated with the histograms.

std::shared_ptr< TFile >	m_file {nullptr}
	Pointer to the ROOT file object.

std::map< std::string, std::unique_ptr< StoreObjPtr< RootMergeable< TH1D > > > >	m_histograms
	A map of histogram names to their respective objects (wrapped in RootMergeable).

Detailed Description

A PersistenceManager that writes variables to ROOT histograms (TH1D).

This class creates and manages one-dimensional histograms for a set of variables. It handles opening a ROOT file, booking the histograms, and filling them with user-supplied data each event. The histograms can be merged in a multi-process environment through the RootMergeable mechanism.

Definition at line 38 of file ConcreteVariablesToHistogramPersistenceManager.h.

Constructor & Destructor Documentation

◆ ConcreteVariablesToHistogramPersistenceManager()

ConcreteVariablesToHistogramPersistenceManager ( )

Default constructor.

Definition at line 20 of file ConcreteVariablesToHistogramPersistenceManager.cc.

20{}

Member Function Documentation

◆ addEntry()

void addEntry ( const EvaluatedVariables & evaluatedVariables )

overridevirtual

Fills histograms with the current set of evaluated variables.

Parameters

evaluatedVariables A map of variable names to their current values.

Each call updates the corresponding histogram with the provided variable values.

Implements PersistenceManager.

Definition at line 34 of file ConcreteVariablesToHistogramPersistenceManager.cc.

  {
    for (const auto& variable : evaluatedVariables) {
      std::visit([&](auto&& val) {
        (*m_histograms[variable.first])->get().Fill(val);
      }, variable.second);
    }
  }

◆ initialize()

void initialize	(	const std::string &	fileName,
		const std::string &	directory,
		Variables &	variables )

overridevirtual

Initializes the manager by opening a ROOT file and creating histograms.

Parameters

fileName	The name of the ROOT file to create or update.
directory	The directory (folder) within the ROOT file where histograms will be placed.
variables	A list of variables that will be associated with the histograms.

This method configures the persistence manager with file/directory names and prepares the memory structures for histogram creation. It does not fill any data yet; that is handled by addEntry().

Implements PersistenceManager.

Definition at line 22 of file ConcreteVariablesToHistogramPersistenceManager.cc.

  {
    m_fileName = fileName;
    m_directory = directoryName;
    m_variables = variables;
 
    openFileWithGuards();
    registerHistograms();
  }

◆ openFileWithGuards()

void openFileWithGuards ( )

private

Safely opens the ROOT file specified by m_fileName.

Throws an exception if the file cannot be opened for writing or if any other file-related error occurs.

Definition at line 62 of file ConcreteVariablesToHistogramPersistenceManager.cc.

  {
    m_file = RootFileCreationManager::getInstance().getFile(m_fileName);
 
    if (!m_file) return;
 
    TDirectory::TContext directoryGuard(m_file.get());
    if (not m_directory.empty()) {
      m_directory = MakeROOTCompatible::makeROOTCompatible(m_directory);
      m_file->mkdir(m_directory.c_str());
      m_file->cd(m_directory.c_str());
    }
  }

◆ registerHistograms()

void registerHistograms ( )

private

Registers (books) the histograms based on m_variables.

For each variable that is defined as "binned" (i.e., has bin information), a TH1D histogram is created in m_histograms.

Definition at line 76 of file ConcreteVariablesToHistogramPersistenceManager.cc.

  {
    for (const auto& variable : m_variables) {
      std::visit([&](const auto & typedVariable) {
        using T = std::decay_t<decltype(typedVariable)>;
        if constexpr(std::is_same_v<T, BinnedVariable>) {
          std::string varStr = typedVariable.getName();
          int varNbins = typedVariable.getNbins();
          float low = typedVariable.getLowBin();
          float high = typedVariable.getHighBin();
 
          std::string compatibleName = MakeROOTCompatible::makeROOTCompatible(varStr);
 
          auto ptr = std::make_unique<StoreObjPtr<RootMergeable<TH1D>>>("", DataStore::c_Persistent);
          ptr->registerInDataStore(m_fileName + m_directory + varStr, DataStore::c_DontWriteOut);
          ptr->construct(compatibleName.c_str(), compatibleName.c_str(), varNbins, low, high);
 
          m_histograms[compatibleName] = std::move(ptr);
        } else {
          B2WARNING("Incompatible variable type. Skipping histogram registration.");
        }
      }, variable);
    }
  }

◆ store()

void store ( )

overridevirtual

Writes histogram data to disk.

After filling the histograms through addEntry(), this function ensures the data is written (or merged) into the ROOT file.

Implements PersistenceManager.

Definition at line 43 of file ConcreteVariablesToHistogramPersistenceManager.cc.

  {
    if (!ProcHandler::parallelProcessingUsed() or ProcHandler::isOutputProcess()) {
      TDirectory::TContext directoryGuard(m_file.get());
      if (not m_directory.empty()) {
        m_file->cd(m_directory.c_str());
      }
      B2INFO("Writing Histograms to " << gDirectory->GetPath());
      for (auto it = m_histograms.begin(); it != m_histograms.end(); ++it) {
        (*it->second)->write(gDirectory);
      }
      const bool writeError = m_file->TestBit(TFile::kWriteError);
      m_file.reset();
      if (writeError) {
        B2FATAL("A write error occurred while saving '" << m_fileName  << "', please check if enough disk space is available.");
      }
    }
  }

Member Data Documentation

◆ m_directory

std::string m_directory

private

Name of the directory (folder) within the ROOT file where histograms are stored.

Definition at line 84 of file ConcreteVariablesToHistogramPersistenceManager.h.

◆ m_file

std::shared_ptr<TFile> m_file {nullptr}

private

Pointer to the ROOT file object.

Definition at line 94 of file ConcreteVariablesToHistogramPersistenceManager.h.

94{nullptr};

◆ m_fileName

std::string m_fileName

private

Name of the ROOT file where histograms will be written.

Definition at line 79 of file ConcreteVariablesToHistogramPersistenceManager.h.

◆ m_histograms

std::map<std::string, std::unique_ptr<StoreObjPtr<RootMergeable<TH1D> > > > m_histograms

private

A map of histogram names to their respective objects (wrapped in RootMergeable).

Definition at line 99 of file ConcreteVariablesToHistogramPersistenceManager.h.

◆ m_variables

Variables m_variables

private

A list of variables associated with the histograms.

Definition at line 89 of file ConcreteVariablesToHistogramPersistenceManager.h.

The documentation for this class was generated from the following files:

reconstruction/persistenceManager/include/ConcreteVariablesToHistogramPersistenceManager.h
reconstruction/persistenceManager/src/ConcreteVariablesToHistogramPersistenceManager.cc

Public Member Functions

Private Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ ConcreteVariablesToHistogramPersistenceManager()

Member Function Documentation

◆ addEntry()

◆ initialize()

◆ openFileWithGuards()

◆ registerHistograms()

◆ store()

Member Data Documentation

◆ m_directory

◆ m_file

◆ m_fileName

◆ m_histograms

◆ m_variables