RootFileInfo.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 <string>
#include <set>
#include <optional>
 
class TTree;
class TFile;
 
namespace Belle2 {
  class FileMetaData;
  namespace RootIOUtilities {
    class RootFileInfo {
    public:
      explicit RootFileInfo(const std::string& filename);
      ~RootFileInfo();
 
      TTree& getPersistentTree() { return *m_persistent; }
      TTree& getEventTree() { return *m_events; }
      const FileMetaData& getFileMetaData();
      const std::set<std::string>& getBranchNames(bool persistent = false);
      void checkMissingBranches(const std::set<std::string>& required, bool persistent = false);
    private:
      std::unique_ptr<TFile> m_file;
      std::unique_ptr<TTree> m_persistent;
      std::unique_ptr<TTree> m_events;
      std::unique_ptr<FileMetaData> m_metadata;
      std::optional<std::set<std::string>> m_persistentBranches;
      std::optional<std::set<std::string>> m_eventBranches;
    };
  }
}