LongitudinalWriter.hpp

Go to the documentation of this file.

/*
 * (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/framework/core/ParticleProperties.hpp>
#include <corsika/framework/core/PhysicalUnits.hpp>
#include <corsika/media/ShowerAxis.hpp>
#include <corsika/modules/writers/WriterOff.hpp>
#include <corsika/modules/writers/LongitudinalProfileWriterParquet.hpp>

#include <vector>
#include <array>

namespace corsika {

  namespace number_profile {

    enum class ProfileIndex {
      Charged,
      Hadron,
      Photon,
      Electron,
      Positron,
      MuPlus,
      MuMinus,
      Entries
    };

    size_t constexpr NColumns = static_cast<size_t>(ProfileIndex::Entries);

    static std::array<char const*, NColumns> constexpr ProfileIndexNames{
        {"charged", "hadron", "photon", "electron", "positron", "muplus", "muminus"}};

    typedef std::array<double, NColumns> ProfileData;
  } // namespace number_profile

  // clang-format-off
  // clang-format-on

  template <typename TOutput = LongitudinalProfileWriterParquet<number_profile::NColumns>>
  class LongitudinalWriter : public TOutput {

  public:
    LongitudinalWriter(ShowerAxis const& axis,
                       GrammageType dX = 10_g / square(1_cm)); // profile binning

    LongitudinalWriter(ShowerAxis const& axis, size_t nbins,
                       GrammageType dX = 10_g / square(1_cm));

    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(Point const& p0, Point const& p1, Code const pid, double const weight);

    void write(GrammageType const Xstart, GrammageType const Xend, Code const pid,
               double const weight);

    YAML::Node getSummary() const;

    YAML::Node getConfig() const;

    number_profile::ProfileData const& getProfile(
        number_profile::ProfileIndex index) const {
      return profile_.at(static_cast<int>(index));
    }

  private:
    ShowerAxis const& showerAxis_; 
    GrammageType const dX_;        
    size_t const nBins_;           
    std::vector<number_profile::ProfileData> profile_; // longitudinal profile
  };

} // namespace corsika

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