ParticleWriterParquet.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>

namespace corsika {

  class ParticleWriterParquet : public BaseOutput {

  public:
    ParticleWriterParquet();

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

    void startOfShower(unsigned int const showerId) final override;

    void endOfShower(unsigned int const showerId) final override;

    void endOfLibrary() final override;

    void write(Code const pid, units::si::HEPEnergyType const energy,
               units::si::LengthType const x, units::si::LengthType const y,
               units::si::LengthType const z, units::si::TimeType const time,
               const double weight);

    YAML::Node getSummary() const final override;

    HEPEnergyType getEnergyGround() const { return totalEnergy_; }

  private:
    ParquetStreamer output_; 
    unsigned int showerId_;  

    double countHadrons_ = 0; 
    double countMuons_ = 0;   
    double countEM_ = 0;      
    double countOthers_ = 0;  

    HEPEnergyType totalEnergy_; 

  }; // class ParticleWriterParquet

} // namespace corsika

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