From e0018a6da4ecf7deb3a883501d4e69b915d2fbb0 Mon Sep 17 00:00:00 2001
From: Anne de Jong <anne@amdj.nl>
Date: Mon, 30 Jan 2017 21:48:51 +0100
Subject: [PATCH] Added some hdf5 attributes to the root /.

---
 src/duct/duct.cpp    |  2 +-
 src/material/gas.h   |  2 +-
 src/sys/tasystem.cpp | 52 +++++++++++++++++++++++++++++++++++++++-----
 3 files changed, 48 insertions(+), 8 deletions(-)

diff --git a/src/duct/duct.cpp b/src/duct/duct.cpp
index f66e328..fc39f21 100644
--- a/src/duct/duct.cpp
+++ b/src/duct/duct.cpp
@@ -372,7 +372,7 @@ void Duct::exportHDF5(const hid_t group_id) const {
     u.x = dmat(sys.Ns(),ngp());
     for(us gp=0;gp<ngp();gp++){
         u.x.col(gp) = ut(gp);
-    }
+     }
     u.exportHDF5(group_id);
 
     TXData T;
diff --git a/src/material/gas.h b/src/material/gas.h
index 8c68c0e..42daea3 100644
--- a/src/material/gas.h
+++ b/src/material/gas.h
@@ -49,7 +49,7 @@ public:
     static Gas* newGas(const GasType gastype,d T0, d p0);
 
     operator GasType() { return _gastype;}
-
+    const string& gasName() const { return pb::GasType_Name(_gastype); }
     d T0() const {return _T0;}
     d p0() const {return _p0;}
 
diff --git a/src/sys/tasystem.cpp b/src/sys/tasystem.cpp
index 7aa5aa1..0a5cc10 100644
--- a/src/sys/tasystem.cpp
+++ b/src/sys/tasystem.cpp
@@ -5,7 +5,6 @@
 // Description:
 //
 //////////////////////////////////////////////////////////////////////
-#include <hdf5.h>
 
 #include "segment.h"
 #include "tasystem.h"
@@ -18,6 +17,9 @@
 #include "duct.h"
 #include "ductbc.h"
 
+#include <hdf5.h>
+// Lite version of the library
+#include <hdf5_hl.h>
 
 TaSystem::TaSystem(const pb::System& sys):
     GlobalConf(sys.nf(),sys.freq())
@@ -423,27 +425,65 @@ void TaSystem::exportHDF5(const string& filename) const {
     herr_t status;
 
     file_id = H5Fcreate (filename.c_str(),
-                         H5F_ACC_TRUNC,
+                         H5F_ACC_TRUNC, // Truncate any existing file
                          H5P_DEFAULT,
                          H5P_DEFAULT);    
 
     
+    d freq = getfreq();
+    H5LTset_attribute_double(file_id,
+                             "/",
+                             "freq",
+                             &freq,1);
+
+
+    int Ns = this->Ns();
+    H5LTset_attribute_int(file_id,
+                             "/",
+                             "Ns",
+                             &Ns,1);
+
+    vd timeInstances = this->timeInstances();
+    H5LTset_attribute_double(file_id,
+                             "/",
+                             "t",
+                             timeInstances.memptr(),
+                             timeInstances.size());
+
+    d p0 = _gas->p0();
+    H5LTset_attribute_double(file_id,
+                             "/",
+                             "p0",
+                             &p0,1);
+
+    d T0 = _gas->T0();
+    H5LTset_attribute_double(file_id,
+                             "/",
+                             "T0",
+                             &T0,1);
+
+
+    
+    const string& gasName = _gas->gasName();
+    H5LTset_attribute_string(file_id,
+                             "/",
+                             "gas",
+                             gasName.c_str());
 
     for(const auto& seg_ : _segs) {
         // Create a group for each segment
         hid_t grp_id = H5Gcreate(file_id,
-                                  (string("/") + std::to_string(seg_.first)).c_str(),
+                                  (string("/") +
+                                   std::to_string(seg_.first)).c_str(),
                                   H5P_DEFAULT,
                                   H5P_DEFAULT,
                                   H5P_DEFAULT);
 
         seg_.second->exportHDF5(grp_id);
 
+
         H5Gclose(grp_id);
 
-        // ^^ uncommenting above results in segfault. Why?
-        // seg_.second->exportHDF5(file_id);
-        
     }
 
     status = H5Fclose(file_id);