From d268a030098b9f79c042d2ac176be0c448f0d5d0 Mon Sep 17 00:00:00 2001
From: Jameson Graef Rollins <jrollins@finestructure.net>
Date: Fri, 1 Jun 2018 18:13:04 -0700
Subject: [PATCH] CLI: support saving budget to HDF5 file
If an HDF5 file is passed to the --save option (indicated by '.hdf5' extension)
then save the data to an HDF5 file without plotting.
---
gwinc/__main__.py | 22 +++++++++++++++++++---
1 file changed, 19 insertions(+), 3 deletions(-)
diff --git a/gwinc/__main__.py b/gwinc/__main__.py
index 53146b15..d2e2538b 100644
--- a/gwinc/__main__.py
+++ b/gwinc/__main__.py
@@ -15,6 +15,7 @@ from .precomp import precompIFO
from .gwinc import gwinc as pygwinc
from . import gwinc_matlab
from . import plot_noise
+from . import util
##################################################
@@ -62,10 +63,10 @@ parser.add_argument('--displacement', '-D', action='store_true', default=False,
group = parser.add_mutually_exclusive_group()
group.add_argument('--dump', '-d', dest='dump', action='store_true',
help="print IFO parameters to stdout and exit")
-group.add_argument('--save', '-s',
- help="save figure to file")
group.add_argument('--interactive', '-i', action='store_true',
help="open interactive shell when plotting")
+group.add_argument('--save', '-s',
+ help="save noise (hdf5) or figure (pdf/png/etc) to file")
group.add_argument('--no-plot', '-np', action='store_false', dest='plot',
help="supress plotting")
parser.add_argument('IFO', default=IFO,
@@ -136,7 +137,20 @@ def main():
)
title += '\n{}'.format(fom_title)
- if args.interactive:
+ ##########
+ # output
+
+ # save noise traces to HDF5 file
+ if args.save and os.path.splitext(args.save)[1] == '.hdf5':
+ util.save_hdf5(
+ title,
+ ifo,
+ noises,
+ args.save,
+ )
+
+ # interactive shell plotting
+ elif args.interactive:
ipshell = InteractiveShellEmbed(
user_ns={'freq': freq,
'noises': noises,
@@ -155,6 +169,8 @@ You may interact with plot using "plt." methods, e.g.:
ipshell.run_code("plot_noise(ifo, noises)")
ipshell.run_code("plt.title('{}')".format(title))
ipshell()
+
+ # standard plotting
elif args.plot:
fig = plt.figure()
ax = fig.add_subplot(1, 1, 1)
--
GitLab