Minor updates to ObservationPlane.

corsika/modules/ObservationPlane.hpp

@@ -15,9 +15,6 @@
 #include <corsika/setup/SetupTrajectory.hpp>
 #include <corsika/modules/writers/ObservationPlaneWriterParquet.hpp>
 
-#include <fstream>
-#include <string>
-
 namespace corsika {
 
   /**
@@ -42,6 +39,7 @@ namespace corsika {
     void reset();
     HEPEnergyType getEnergyGround() const { return energy_ground_; }
     YAML::Node getConfig() const;
+
   private:
     Plane const plane_;
     bool const deleteOnHit_;

corsika/modules/writers/ObservationPlaneWriterParquet.hpp

@@ -21,7 +21,7 @@ namespace corsika {
 
   public:
     /**
-     * Write an observation plane to a directory.
+     * Construct an ObservationPlane.
      *
      * @param name    The name of this output.
      */

corsika/output/ParquetStreamer.hpp

+
 /*
  * (c) Copyright 2021 CORSIKA Project, corsika-project@lists.kit.edu
  *
@@ -26,11 +27,11 @@ namespace corsika {
   class ParquetStreamer {
 
   protected:
-    std::shared_ptr<parquet::StreamWriter> writer_; ///< The stream writer to 'outfile'
-    parquet::WriterProperties::Builder builder_;    ///< The writer properties builder.
-    parquet::schema::NodeVector fields_;            ///< The fields in this file.
+    parquet::WriterProperties::Builder builder_; ///< The writer properties builder.
+    parquet::schema::NodeVector fields_;         ///< The fields in this file.
     std::shared_ptr<parquet::schema::GroupNode> schema_;   ///< The schema for this file.
     std::shared_ptr<arrow::io::FileOutputStream> outfile_; ///< The output file.
+    std::shared_ptr<parquet::StreamWriter> writer_; ///< The stream writer to 'outfile'
 
   public:
     /**
@@ -59,6 +60,11 @@ namespace corsika {
      */
     void closeStreamer();
 
+    /**
+     * Return a reference to the underlying writer.
+     */
+    std::shared_ptr<parquet::StreamWriter> getWriter();
+
   }; // class ParquetHelper
 } // namespace corsika
 

python/corsika/io/outputs/observation_plane.py

@@ -9,7 +9,7 @@
 """
 import logging
 import os.path as op
-from typing import Any, Dict
+from typing import Any
 
 import pyarrow.parquet as pq