PHTFileServer.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PHTFileServer.h

// Tell emacs that this is a C++ source
//  -*- C++ -*-.



#ifndef FUN4ALL_PHTFILESERVER_H
#define FUN4ALL_PHTFILESERVER_H

#include <TFile.h>

#include <map>
#include <sstream>
#include <string>

class PHTFileServer
{
 public:
  static PHTFileServer& get(void)
  {
    static PHTFileServer singleton;
    return singleton;
  }

  virtual ~PHTFileServer();

  void open(const std::string& filename, const std::string& type = "RECREATE");

  bool flush(const std::string& filename);

  bool cd(const std::string& filename);

  bool write(const std::string& filename);

  void close(void);

 private:
  PHTFileServer(void)
  {
  }

  class SafeTFile : public TFile
  {
   public:
    SafeTFile(const std::string& filename, const std::string& type = "RECREATE")
      : TFile(filename.c_str(), type.c_str())
      , _filename(filename)
      , _counter(1)
    {
    }

    ~SafeTFile(void) override;

    int& counter()
    {
      return _counter;
    }

    const int& counter() const
    {
      return _counter;
    }

    typedef std::map<std::string, SafeTFile*> TFileMap;

    static TFileMap& file_map(void)
    {
      return _map;
    }

   private:
    std::string _filename;

    int _counter;

    static TFileMap _map;
  };
};

#endif