OutputStreamData.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/

 
 
 
#ifndef EVENT_LOOP__OUTPUT_STREAM_DATA_H
#define EVENT_LOOP__OUTPUT_STREAM_DATA_H
 
#include <EventLoop/Global.h>
 
#include <SampleHandler/DiskWriter.h>
#include <TFile.h>
#include <memory>
#include <unordered_map>
 
class TH1;
class StatusCode;
class TTree;
 
namespace EL
{
  namespace Detail
  {
 
    class OutputStreamData final
    {
      //
      // public interface
      //

    public:
      void testInvariant () const;
 

    public:
      explicit OutputStreamData (const std::string& val_fileName, const std::string& mode);
 

    public:
      explicit OutputStreamData (std::unique_ptr<TFile> val_file);
 

    public:
      explicit OutputStreamData (std::unique_ptr<SH::DiskWriter> val_writer);
 

    public:
      const std::string& mainStreamName () const noexcept;
      void setMainStreamName (const std::string& val_mainStreamName);
 

    public:
      TFile *file () const noexcept;
 

    public:
      void saveOutput ();
 

    public:
      void close ();
 

    public:
      std::string finalFileName () const;
 

    public:
      void addOutput (std::unique_ptr<TObject> outputObject);
 

    public:
      void addClone (const TObject& prototypeObject);
 

    public:
      TDirectory *makeDirectoryFor (std::string& name);
 

    public:
      TObject *getOutputHist (const std::string& name) const noexcept;
 

    public:
      TTree *getOutputTree (const std::string& name) const noexcept;
 
 
 
      //
      // private interface
      //

    private:
      std::string m_mainStreamName;

    private:
      std::unique_ptr<SH::DiskWriter> m_writer;

    private:
      std::vector<std::unique_ptr<TObject> > m_output;

    private:
      std::unordered_map<std::string,TObject*> m_outputHistMap;

    private:
      std::unordered_map<std::string,TTree*> m_outputTreeMap;
    };
  }
}
 
#endif