OutputManager.hpp

/*
 * (c) Copyright 2021 CORSIKA Project, corsika-project@lists.kit.edu
 *
 * This software is distributed under the terms of the GNU General Public
 * Licence version 3 (GPL Version 3). See file LICENSE for a full version of
 * the license.
 */
#pragma once

#include <chrono>
#include <string>
#include <boost/filesystem.hpp>
#include <corsika/framework/core/Logging.hpp>
#include <corsika/output/BaseOutput.hpp>
#include <corsika/output/YAMLStreamer.hpp>

namespace corsika {

  class OutputManager : public YAMLStreamer {

    enum class OutputState {
      NoInit,
      LibraryReady,
      ShowerInProgress,
      LibraryFinished,
    };

  public:
    OutputManager(std::string const& name, const long& vseed,
                  std::string const& input_args, boost::filesystem::path const& dir);

    ~OutputManager();

    template <typename TOutput>
    void add(std::string const& name, TOutput& output);

    YAML::Node getSummary() const;

    YAML::Node getConfig() const;

  private:
    void writeConfig() const;

    void writeSummary() const;

  public:
    void startOfLibrary();

    void startOfShower();

    void endOfShower();

    void endOfLibrary();

    int getEventId() const;

  private:
    boost::filesystem::path root_;           
    OutputState state_{OutputState::NoInit}; 
    std::string const name_;                 
    std::string const cmnd_line_args_; 
    int count_{0};                     
    long seed_{0};
    std::chrono::time_point<std::chrono::system_clock> const start_time{
        std::chrono::system_clock::now()}; 
    inline static auto logger_{get_logger("output")}; 

    std::map<std::string, std::reference_wrapper<BaseOutput>> outputs_;

  }; // class OutputManager

} // namespace corsika

#include <corsika/detail/output/OutputManager.inl>