From 0e65f71705ebac800b6578066f8889c6d8e1a4bb Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Wed, 4 Feb 2015 18:46:55 +0800
Subject: [PATCH] artiq_run: HDF5 output support

---
 artiq/frontend/artiq_run.py | 22 +++++++++++++++++-----
 1 file changed, 17 insertions(+), 5 deletions(-)

diff --git a/artiq/frontend/artiq_run.py b/artiq/frontend/artiq_run.py
index cea992fe2..73a733ecf 100755
--- a/artiq/frontend/artiq_run.py
+++ b/artiq/frontend/artiq_run.py
@@ -6,6 +6,8 @@ from inspect import isclass
 from operator import itemgetter
 from itertools import chain
 
+import h5py
+
 from artiq.language.db import *
 from artiq.protocols import pyon
 from artiq.protocols.file_db import FlatFileDB
@@ -44,6 +46,9 @@ def get_argparser():
                         help="run ELF binary")
     parser.add_argument("-u", "--unit", default=None,
                         help="unit to run")
+    parser.add_argument("-o", "--hdf5", default=None,
+                        help="write results to specified HDF5 file"
+                             " (default: print them)")
     parser.add_argument("file",
                         help="file containing the unit to run")
     parser.add_argument("arguments", nargs="*",
@@ -107,11 +112,18 @@ def main():
             unit_inst = unit(dbh, **arguments)
             unit_inst.run()
 
-            if rdb.data.read or rdb.realtime_data.read:
-                print("Results:")
-                for k, v in chain(rdb.realtime_data.read.items(),
-                                  rdb.data.read.items()):
-                    print("{}: {}".format(k, v))
+            if args.hdf5 is not None:
+                f = h5py.File(args.hdf5, "w")
+                try:
+                    rdb.write_hdf5(f)
+                finally:
+                    f.close()
+            else:
+                if rdb.data.read or rdb.realtime_data.read:
+                    print("Results:")
+                    for k, v in chain(rdb.realtime_data.read.items(),
+                                      rdb.data.read.items()):
+                        print("{}: {}".format(k, v))
     finally:
         dbh.close()