Fun4AllHepMCOutputManager.h

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

#ifndef PHHEPMC_FUN4ALLHEPMCOUTPUTMANAGER_H
#define PHHEPMC_FUN4ALLHEPMCOUTPUTMANAGER_H

#include <fun4all/Fun4AllOutputManager.h>

#include <fstream>
#include <string>

namespace HepMC
{
  class IO_GenEvent;
}

class PHCompositeNode;

class Fun4AllHepMCOutputManager : public Fun4AllOutputManager
{
 public:
  Fun4AllHepMCOutputManager(const std::string &myname = "HEPMCOUT",
                            const std::string &filename = "hepmcout.txt");

  ~Fun4AllHepMCOutputManager() override;

  int outfileopen(const std::string & /*fname*/) override { return 0; }

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

  int Write(PHCompositeNode *startNode) override;

  int AddComment(const std::string &text);

  int get_embedding_id() const { return _embedding_id; }
  //
  void set_embedding_id(int id) { _embedding_id = id; }

 protected:
  std::string outfilename;
  HepMC::IO_GenEvent *ascii_out;
  std::string comment;
  int comment_written;

  // some pointers for use in compression handling
  std::ofstream *filestream;  // holds compressed filestream
  std::ostream *zipstream;    // feed into HepMC

  int _embedding_id;
};

#endif /* PHHEPMC_FUN4ALLHEPMCOUTPUTMANAGER_H */