JetScapeWriterAsciiGZ.h

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

// -----------------------------------------
// JetScape (modular/task) based framework
// Intial Design: Joern Putschke (2017)
//                (Wayne State University)
// -----------------------------------------
// License and Doxygen-like Documentation to be added ...


// jetscape writer ascii + gzip class

#ifndef JETSCAPEWRITERASCIIGZ_H
#define JETSCAPEWRITERASCIIGZ_H

#include "gzstream.h"

#include <fstream>
#include <string>

#include "JetScapeWriter.h"

namespace Jetscape {

// Maybe a templeate for Ascii and AsciiGZ !? Or in general !???

class JetScapeWriterAsciiGZ : public JetScapeWriter
{

 public:

  JetScapeWriterAsciiGZ() {};
  JetScapeWriterAsciiGZ(string m_file_name_out);
  virtual ~JetScapeWriterAsciiGZ();

  void Init();
  void Exec();
  
  bool GetStatus() {return output_file.good();}
  void Close() {output_file.close();}

  void Write(weak_ptr<PartonShower> ps);
  void Write(weak_ptr<Parton> p);
  void Write(weak_ptr<Vertex> v);
  void Write(weak_ptr<Hadron> h);
  void WriteHeaderToFile();

  void Write(string s) {output_file<<s<<endl;}
  void WriteComment(string s) {output_file<<"# "<<s<<endl;}
  void WriteWhiteSpace(string s) {output_file<<s<<" ";}
  void WriteEvent();
 
 private:

  ogzstream output_file; 
  //int m_precision; //!< Output precision
  
};

} // end namespace Jetscape

#endif