TreeStreamRedirector.h

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#ifndef ALICEO2_TREESTREAMREDIRECTOR_H
#define ALICEO2_TREESTREAMREDIRECTOR_H
 
#include <Rtypes.h>
#include <TDirectory.h>
#include "CommonUtils/TreeStream.h"
 
namespace o2
{
namespace utils
{
class TreeStreamRedirector
{
 public:
  TreeStreamRedirector(const char* fname = "", const char* option = "recreate");
  virtual ~TreeStreamRedirector();
  void Close();
  TFile* GetFile() { return mDirectory->GetFile(); }
  TDirectory* GetDirectory() { return mDirectory; }
  virtual TreeStream& operator<<(Int_t id);
  virtual TreeStream& operator<<(const char* name);
  void SetDirectory(TDirectory* sfile);
  void SetFile(TFile* sfile);
  static void FixLeafNameBug(TTree* tree);
 
 private:
  TreeStreamRedirector(const TreeStreamRedirector& tsr);
  TreeStreamRedirector& operator=(const TreeStreamRedirector& tsr);
 
  std::unique_ptr<TDirectory> mOwnDirectory;             // own directory of the redirector
  TDirectory* mDirectory = nullptr;                      // output directory
  std::vector<std::unique_ptr<TreeStream>> mDataLayouts; // array of data layouts
 
  ClassDefNV(TreeStreamRedirector, 0);
};
} // namespace utils
} // namespace o2
 
#endif