RootFileInfo.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2019  Belle II Collaboration                              *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Martin Ritter                                            *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
#include <memory>
#include <string>
#include <set>
 
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;
    };
  }
}