From 59714029cfdfd13a98580dba31f4b90b7dfc01b8 Mon Sep 17 00:00:00 2001
From: Matthieu Schaller <matthieu.schaller@durham.ac.uk>
Date: Wed, 7 Sep 2016 18:16:34 +0100
Subject: [PATCH] Store the radiated energy in the xpart when performing
 cooling

---
 src/cooling/const_du/cooling.h     | 3 +++
 src/cooling/const_lambda/cooling.h | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/src/cooling/const_du/cooling.h b/src/cooling/const_du/cooling.h
index 8f005ee0bf..40bf2e919d 100644
--- a/src/cooling/const_du/cooling.h
+++ b/src/cooling/const_du/cooling.h
@@ -79,6 +79,9 @@ __attribute__((always_inline)) INLINE static void cooling_cool_part(
 
   /* Update the internal energy */
   hydro_set_internal_energy(p, u_new);
+
+  /* Store the radiated energy */
+  xp->cooling_data.radiated_energy += u_new - u_old;
 }
 
 /**
diff --git a/src/cooling/const_lambda/cooling.h b/src/cooling/const_lambda/cooling.h
index 458babf540..6d3f904d15 100644
--- a/src/cooling/const_lambda/cooling.h
+++ b/src/cooling/const_lambda/cooling.h
@@ -114,6 +114,9 @@ __attribute__((always_inline)) INLINE static void cooling_cool_part(
    * " */
   /*       "%g, du_dt*dt = %g, u_old + du_dt*dt = %g, u_new = %g\n", */
   /*       u_old, du_dt, dt, du_dt * dt, u_new, u_new_test); */
+
+  /* Store the radiated energy */
+  xp->cooling_data.radiated_energy += u_new - u_old;
 }
 
 /**
-- 
GitLab