Fun4AllDstOutputManager.h

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

// Tell emacs that this is a C++ source
//  -*- C++ -*-.
#ifndef FUN4ALL_FUN4ALLDSTOUTPUTMANAGER_H
#define FUN4ALL_FUN4ALLDSTOUTPUTMANAGER_H

#include "Fun4AllOutputManager.h"

#include <set>
#include <string>

class PHNodeIOManager;
class PHCompositeNode;

class Fun4AllDstOutputManager : public Fun4AllOutputManager
{
 public:
  Fun4AllDstOutputManager(const std::string &myname = "DSTOUT", const std::string &filename = "dstout.root");
  ~Fun4AllDstOutputManager() override;
  // Fun4AllDstOutputManager contains pointer to memory
  // copy ctor and = operator  need explicit implementation, do just delete it here
  Fun4AllDstOutputManager(const Fun4AllDstOutputManager &) = delete;
  Fun4AllDstOutputManager &operator=(Fun4AllDstOutputManager const &) = delete;
  int AddNode(const std::string &nodename) override;
  int AddRunNode(const std::string &nodename) override;
  int StripNode(const std::string &nodename) override;
  int StripRunNode(const std::string &nodename) override;
  void SaveRunNode(const int i) override { m_SaveRunNodeFlag = i; }
  void SaveDstNode(const int i) override { m_SaveDstNodeFlag = i; }
  int outfileopen(const std::string &fname) override;

  void Print(const std::string &what = "ALL") const override;

  int Write(PHCompositeNode *startNode) override;
  int WriteNode(PHCompositeNode *thisNode) override;
  std::string UsedOutFileName() const { return m_UsedOutFileName; }
  void CompressionSetting(const int i) { m_CompressionSetting = i; }

 private:
  int outfile_open_first_write();
  PHNodeIOManager *dstOut{nullptr};
  int m_SaveRunNodeFlag{1};
  int m_SaveDstNodeFlag{1};
  int m_CompressionSetting{505};
  int m_CurrentSegment{0};
  std::string m_FileNameStem;
  std::string m_UsedOutFileName;
  std::set<std::string> savenodes;
  std::set<std::string> saverunnodes;
  std::set<std::string> stripnodes;
  std::set<std::string> striprunnodes;
};

#endif