From e67aeaccf80d5037df516b5db4775b9213b335d6 Mon Sep 17 00:00:00 2001
From: Matthieu Schaller <schaller@strw.leidenuniv.nl>
Date: Thu, 27 Feb 2020 12:32:35 +0800
Subject: [PATCH] Add the number of chemistry elements to the snapshot headers
---
src/chemistry/EAGLE/chemistry_io.h | 2 ++
src/chemistry/GEAR/chemistry_io.h | 2 ++
src/chemistry/none/chemistry_io.h | 1 +
3 files changed, 5 insertions(+)
diff --git a/src/chemistry/EAGLE/chemistry_io.h b/src/chemistry/EAGLE/chemistry_io.h
index 393952e0b2..02420e1932 100644
--- a/src/chemistry/EAGLE/chemistry_io.h
+++ b/src/chemistry/EAGLE/chemistry_io.h
@@ -283,6 +283,8 @@ INLINE static int chemistry_write_bparticles(const struct bpart* bparts,
INLINE static void chemistry_write_flavour(hid_t h_grp) {
io_write_attribute_s(h_grp, "Chemistry Model", "EAGLE");
+ io_write_attribute_d(h_grp, "Chemistry element count",
+ chemistry_element_count);
for (int elem = 0; elem < chemistry_element_count; ++elem) {
char buffer[20];
sprintf(buffer, "Element %d", elem);
diff --git a/src/chemistry/GEAR/chemistry_io.h b/src/chemistry/GEAR/chemistry_io.h
index 0943f369b1..531765d313 100644
--- a/src/chemistry/GEAR/chemistry_io.h
+++ b/src/chemistry/GEAR/chemistry_io.h
@@ -116,6 +116,8 @@ INLINE static int chemistry_write_bparticles(const struct bpart* bparts,
INLINE static void chemistry_write_flavour(hid_t h_grp) {
io_write_attribute_s(h_grp, "Chemistry Model", "GEAR");
+ io_write_attribute_d(h_grp, "Chemistry element count",
+ GEAR_CHEMISTRY_ELEMENT_COUNT);
}
#endif
diff --git a/src/chemistry/none/chemistry_io.h b/src/chemistry/none/chemistry_io.h
index 698d92c7ca..80657726fe 100644
--- a/src/chemistry/none/chemistry_io.h
+++ b/src/chemistry/none/chemistry_io.h
@@ -98,6 +98,7 @@ INLINE static int chemistry_write_bparticles(const struct bpart* bparts,
INLINE static void chemistry_write_flavour(hid_t h_grp) {
io_write_attribute_s(h_grp, "Chemistry Model", "None");
+ io_write_attribute_d(h_grp, "Chemistry element count", 0);
}
#endif
--
GitLab