From 626e19a52267addf1b3711f9ccbef7d985985d5f Mon Sep 17 00:00:00 2001
From: Maximilian Reininghaus <maximilian.reininghaus@tu-dortmund.de>
Date: Tue, 25 Dec 2018 13:49:41 +0100
Subject: [PATCH] header for TrackWriter output

---
 Processes/TrackWriter/TrackWriter.cc | 9 ++++++++-
 Processes/TrackWriter/TrackWriter.h  | 6 +++---
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/Processes/TrackWriter/TrackWriter.cc b/Processes/TrackWriter/TrackWriter.cc
index ef15c3d2d..ffc67cde8 100644
--- a/Processes/TrackWriter/TrackWriter.cc
+++ b/Processes/TrackWriter/TrackWriter.cc
@@ -9,5 +9,12 @@
  */
 
 #include <corsika/process/track_writer/TrackWriter.h>
+#include <string>
 
-void corsika::process::TrackWriter::TrackWriter::Init() { fFile.open(fFilename); }
+void corsika::process::TrackWriter::TrackWriter::Init() {
+  using namespace std::string_literals;
+
+  fFile.open(fFilename);
+  fFile << "# PID, E / eV, start coordinates / m, displacement vector to end / m "s
+        << '\n';
+}
diff --git a/Processes/TrackWriter/TrackWriter.h b/Processes/TrackWriter/TrackWriter.h
index 50e17e6af..8434ec617 100644
--- a/Processes/TrackWriter/TrackWriter.h
+++ b/Processes/TrackWriter/TrackWriter.h
@@ -36,9 +36,9 @@ namespace corsika::process::TrackWriter {
       auto const delta = t.GetPosition(1).GetCoordinates() - start;
       auto const& name = corsika::particles::GetName(p.GetPID());
 
-      fFile << name << "    " << start[0] / 1_m << ' ' << start[1] / 1_m << ' '
-                << start[2] / 1_m << "   " << delta[0] / 1_m << ' ' << delta[1] / 1_m
-                << ' ' << delta[2] / 1_m << '\n';
+      fFile << name << "    " << p.GetEnergy() / 1_eV << ' ' << start[0] / 1_m << ' '
+            << start[1] / 1_m << ' ' << start[2] / 1_m << "   " << delta[0] / 1_m << ' '
+            << delta[1] / 1_m << ' ' << delta[2] / 1_m << '\n';
 
       return corsika::process::EProcessReturn::eOk;
     }
-- 
GitLab