EnergyLossWriterParquet.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 <corsika/output/BaseOutput.hpp>
#include <corsika/output/ParquetStreamer.hpp>
#include <corsika/framework/core/ParticleProperties.hpp>
#include <corsika/framework/core/PhysicalUnits.hpp>
#include <corsika/media/ShowerAxis.hpp>

#include <array>

namespace corsika {

  template <size_t NColumns>
  class EnergyLossWriterParquet : public BaseOutput {

  public:
    EnergyLossWriterParquet(std::array<const char*, NColumns> const& colNames);

    void startOfLibrary(boost::filesystem::path const& directory) override;

    void startOfShower(unsigned int const showerId) override;

    void endOfShower(unsigned int const showerId) override;

    void endOfLibrary() override;

    void write(unsigned int const showerId, GrammageType const grammage,
               std::array<HEPEnergyType, NColumns> const& data);

  private:
    std::array<const char*, NColumns> columns_; 
    ParquetStreamer output_;                    

  }; // namespace corsika

} // namespace corsika

#include <corsika/detail/modules/writers/EnergyLossWriterParquet.inl>