OutputWriter.h

Go to the documentation of this file.

#ifndef RDFAnalysis_OutputWriter_H
#define RDFAnalysis_OutputWriter_H

// package includes
#include "RDFAnalysis/INodeWriter.h"
#include "RDFAnalysis/Scheduler.h"

// ROOT includes
#include <TDirectory.h>

// STL includes
#include <memory>

namespace RDFAnalysis {
  template <typename Detail>
    class OutputWriter {
      public:
        using detail_t = Detail;
        OutputWriter(const std::shared_ptr<TDirectory>& directory);

        OutputWriter(
            const std::string& fileName,
            bool overwrite = false);

        void write(Node<Detail>& node) { writeFullTree(node, m_directory.get() ); }

        void write(Scheduler<Detail>& scheduler)
        { write(scheduler.regions() ); }

        void write(std::map<std::string, typename  Scheduler<Detail>::Region>& regions);

        void addWriter(const std::shared_ptr<INodeWriter<Detail>>& writer)
        { m_writers.push_back(writer); }

        template <typename T> std::enable_if_t<std::is_base_of<INodeWriter<Detail>, T>{},
          void> addWriter(T&& writer)
          { addWriter(std::make_shared<T>(std::move(writer) ) ); }

        template <template<class> class T, typename... Args>
          std::enable_if_t<std::is_base_of<INodeWriter<Detail>, T<Detail>>{}, void> addWriter(
              Args&&... args)
          { addWriter(std::make_shared<T<Detail>>(std::forward<Args>(args)...) ); }

        std::vector<std::shared_ptr<INodeWriter<Detail>>>& writers()
        { return m_writers; }

        const std::vector<std::shared_ptr<INodeWriter<Detail>>>& writers() const
        { return m_writers; }

      private:
        std::shared_ptr<TDirectory> m_directory;

        std::vector<std::shared_ptr<INodeWriter<Detail>>> m_writers;

        void writeFullTree(
            Node<Detail>& node,
            TDirectory* directory,
            std::size_t depth = 0);

        void writeNode(
            Node<Detail>& node,
            TDirectory* directory,
            std::size_t depth);
    }; //> end class OutputWriter
} //> end namespace RDFAnalysis
#include "RDFAnalysis/OutputWriter.icc"
#endif //> !RDFAnalysis_OutputWriter_H