TFileDirectory.h

Go to the documentation of this file.

#ifndef art_Framework_Services_Optional_TFileDirectory_h
#define art_Framework_Services_Optional_TFileDirectory_h

/* class TFileDirectory
 *
 * Original author: Luca Lista, INFN
 *
 */

#include "art/Framework/Services/Optional/detail/RootDirectorySentry.h"

#include "TDirectory.h"

class TFile;
class TGraph;

#include <functional>
#include <map>
#include <string>
#include <utility>

namespace art {

  class TFileDirectory {
  public:
    virtual ~TFileDirectory() = default;

    template <typename T, typename... ARGS>
    T* make(ARGS... args) const;

    template <typename T, typename... ARGS>
    T* makeAndRegister(char const* name, char const* title, ARGS... args) const;

    template <typename T, typename... ARGS>
    T* makeAndRegister(std::string const& name,
                       std::string const& title,
                       ARGS... args) const;

    TFileDirectory mkdir(std::string const& dir, std::string const& descr = "");

  protected:
    using Callback_t = std::function<void()>;

    TFileDirectory(std::string const& dir,
                   std::string const& descr,
                   TFile* file,
                   std::string const& path);

    void invokeCallbacks();
    void registerCallback(Callback_t);

    std::string fullPath() const;

    TFile* file_;
    std::string dir_;
    std::string descr_;
    bool requireCallback_{false};

  private:
    std::string path_;
    void cd() const;
    std::map<std::string, std::vector<Callback_t>> callbacks_{};
  };

  template <typename T, typename... ARGS>
  T*
  TFileDirectory::make(ARGS... args) const
  {
    detail::RootDirectorySentry sentry;
    cd();
    return new T(args...);
  }

  template <typename T, typename... ARGS>
  T*
  TFileDirectory::makeAndRegister(char const* name,
                                  char const* title,
                                  ARGS... args) const
  {
    detail::RootDirectorySentry sentry;
    cd();
    T* p = new T(args...);
    p->SetName(name);
    p->SetTitle(title);
    gDirectory->Append(p);
    return p;
  }

  template <typename T, typename... ARGS>
  T*
  TFileDirectory::makeAndRegister(std::string const& name,
                                  std::string const& title,
                                  ARGS... args) const
  {
    return makeAndRegister(name.c_str(), title.c_str(), args...);
  }
}

#endif /* art_Framework_Services_Optional_TFileDirectory_h */