EventWriter.h

Go to the documentation of this file.

/*
 * Thermal-FIST package
 *
 * Copyright (c) 2019-2019 Volodymyr Vovchenko
 *
 * GNU General Public License (GPLv3 or later)
 */
#ifndef EVENTWRITER_H
#define EVENTWRITER_H
 
#include <fstream>
 
#include "SimpleEvent.h"
#include "HRGBase/ThermalParticleSystem.h"
 
 
namespace thermalfist {

  class EventWriter
  {
  public:
    EventWriter(const std::string & filename = "");
    virtual ~EventWriter();
 
    virtual bool OpenFile(const std::string& filename);
    
    virtual void CloseFile();
 
    virtual bool WriteEvent(const SimpleEvent& evt);
  protected:
    std::ofstream m_fout;
    int m_EventNumber;
  };

  class EventWriterAsciiExtended
    : public EventWriter
  {
  public:
    EventWriterAsciiExtended(const std::string& filename = "", const SimpleEvent::EventOutputConfig& config = SimpleEvent::EventOutputConfig()) : EventWriter(filename), m_config(config) { }
    void SetOutputConfig(const SimpleEvent::EventOutputConfig& config) { m_config = config; }
    bool WriteEvent(const SimpleEvent& evt);
  private:
    SimpleEvent::EventOutputConfig m_config;
  };

  class EventWriterForUrqmd
    : public EventWriter
  {
  public:
    EventWriterForUrqmd(const std::string& filename = "") : EventWriter(filename) { }
    bool WriteEvent(const SimpleEvent& evt);
  };

  class EventWriterForSmash
    : public EventWriter
  {
  public:
    EventWriterForSmash(const std::string& filename = "", const ThermalParticleSystem* TPS = NULL) : EventWriter(filename), m_TPS(TPS) { }
    bool WriteEvent(const SimpleEvent& evt);
  private:
    const ThermalParticleSystem* m_TPS;
  };
 
} // namespace thermalfist
 
 
 
#endif