From de22ebaf5cf3f3537b8cf987868b655b04b489d5 Mon Sep 17 00:00:00 2001 From: tlatorre Date: Fri, 24 May 2019 17:27:50 -0400 Subject: add a script to concatenate output from grid jobs --- utils/cat-grid-jobs | 42 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100755 utils/cat-grid-jobs (limited to 'utils/cat-grid-jobs') diff --git a/utils/cat-grid-jobs b/utils/cat-grid-jobs new file mode 100755 index 0000000..d43dfad --- /dev/null +++ b/utils/cat-grid-jobs @@ -0,0 +1,42 @@ +#!/usr/bin/env python +# Copyright (c) 2019, Anthony Latorre +# +# This program is free software: you can redistribute it and/or modify it +# under the terms of the GNU General Public License as published by the Free +# Software Foundation, either version 3 of the License, or (at your option) +# any later version. +# +# This program is distributed in the hope that it will be useful, but WITHOUT +# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or +# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for +# more details. +# +# You should have received a copy of the GNU General Public License along with +# this program. If not, see . + +from __future__ import print_function, division +import yaml + +if __name__ == '__main__': + import argparse + import matplotlib.pyplot as plt + import numpy as np + + parser = argparse.ArgumentParser("plot fit results") + parser.add_argument("filenames", nargs='+', help="input files") + args = parser.parse_args() + + cat = [] + + for filename in args.filenames: + with open(filename) as f: + data = yaml.load(f.read()) + + if data is None: + continue + + for event in data['data']: + if event['ev'] is not None: + cat.append(event) + + print(yaml.dump({'data':cat},default_flow_style=False)) -- cgit