ObservationPlane.hpp

/*
 * (c) Copyright 2020 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/framework/core/PhysicalUnits.hpp>
#include <corsika/framework/geometry/Plane.hpp>
#include <corsika/framework/process/ContinuousProcess.hpp>
#include <corsika/modules/writers/ObservationPlaneWriterParquet.hpp>

namespace corsika {

  template <typename TTracking, typename TOutputWriter = ObservationPlaneWriterParquet>
  class ObservationPlane
      : public ContinuousProcess<ObservationPlane<TTracking, TOutputWriter>>,
        public TOutputWriter {

  public:
    ObservationPlane(Plane const&, DirectionVector const&, bool = true);

    template <typename TParticle, typename TTrajectory>
    ProcessReturn doContinuous(TParticle& vParticle, TTrajectory& vTrajectory,
                               bool const stepLimit);

    template <typename TParticle, typename TTrajectory>
    LengthType getMaxStepLength(TParticle const&, TTrajectory const& vTrajectory);

    void showResults() const;
    void reset();
    HEPEnergyType getEnergyGround() const { return energy_ground_; }
    YAML::Node getConfig() const;

  private:
    Plane const plane_;
    bool const deleteOnHit_;
    HEPEnergyType energy_ground_;
    unsigned int count_ground_;
    DirectionVector const xAxis_;
    DirectionVector const yAxis_;
  };
} // namespace corsika

#include <corsika/detail/modules/ObservationPlane.inl>