ConcreteVariablesToNtuplePersistenceManager.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <memory>
#include <map>
#include <string>
#include <TTree.h>
#include <TFile.h>
 
#include <reconstruction/persistenceManager/PersistenceManager.h>
#include <framework/pcore/RootMergeable.h>
#include <framework/datastore/StoreObjPtr.h>
 
namespace Belle2::VariablePersistenceManager {

  class ConcreteVariablesToNtuplePersistenceManager : public PersistenceManager {
  public:
    ConcreteVariablesToNtuplePersistenceManager();

    void initialize(const std::string& fileName,
                    const std::string& treeName,
                    Variables& variables) override;

    void addEntry(const EvaluatedVariables& evaluatedVariables) override;

    void store() override;
 
  private:
    std::string m_fileName;

    std::string m_treeName;

    Variables m_variables;

    int m_basketSize{1600};

    std::shared_ptr<TFile> m_file{nullptr};

    StoreObjPtr<RootMergeable<TTree>> m_tree;

    std::map<std::string, double> m_branchesDouble; 
    std::map<std::string, int>    m_branchesInt;    
    std::map<std::string, bool>   m_branchesBool;   

    void openFileWithGuards();

    void registerBranches();

    void updateBranch(const std::string& variableName,
                      const VariableType& value);
  };
 
} // namespace Belle2::VariablePersistenceManager