add dumping of index list

JulienDoerner · Sep 5, 2024 · ee91511 · ee91511
1 parent bb34f63
commit ee91511
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/include/crpropa/module/TextOutput.h b/include/crpropa/module/TextOutput.h
@@ -71,6 +71,8 @@ class TextOutput: public Output {
 	 */
 	static void load(const std::string &filename, ParticleCollector *collector);
 	std::string getDescription() const;
+
+	void dumpIndexList(std::vector<int> indicies);
 };
 /** @}*/
 

diff --git a/src/module/TextOutput.cpp b/src/module/TextOutput.cpp
@@ -7,6 +7,7 @@
 
 #include "kiss/string.h"
 
+#include <sstream>
 #include <cstdio>
 #include <stdexcept>
 #include <iostream>
@@ -378,4 +379,17 @@ void TextOutput::gzip() {
 #endif
 }
 
+void TextOutput::dumpIndexList(std::vector<int> indices) {
+#pragma omp critical(FileOutput)
+	{
+		std::stringstream ss;
+		ss << "#" << "\t";
+		for (int i = 0; i < indices.size(); i++)
+			ss << indices[i] << "\t";
+
+		const std::string cstr = ss.str();
+		out-> write(cstr.c_str(), cstr.length());
+	}
+}
+
 } // namespace crpropa