From c69f733a9a766873321e2b2d5b416b29eb64755f Mon Sep 17 00:00:00 2001
From: "chad.hanna" <crh184@psu.edu>
Date: Sun, 20 Oct 2019 06:48:56 -0700
Subject: [PATCH] gstlal_inspiral_activation_counts_aggregator: write result to
h5 file
---
...tlal_inspiral_activation_counts_aggregator | 20 +++++++++++--------
1 file changed, 12 insertions(+), 8 deletions(-)
diff --git a/gstlal-ugly/bin/gstlal_inspiral_activation_counts_aggregator b/gstlal-ugly/bin/gstlal_inspiral_activation_counts_aggregator
index 05f380ccbc..f6aa6cad29 100755
--- a/gstlal-ugly/bin/gstlal_inspiral_activation_counts_aggregator
+++ b/gstlal-ugly/bin/gstlal_inspiral_activation_counts_aggregator
@@ -1,4 +1,5 @@
#!/usr/bin/python
+import numpy
import h5py
import sys, os
@@ -20,11 +21,12 @@ class AcCounts(object):
for line in f.readlines()[1:]:
name, _, _, _, _, count = line.split(",")
try:
- self.counts[binnum][name] = float(count)
- except:
self.counts[binnum][name] += float(count)
+ except:
+ self.counts[binnum][name] = float(count)
except ValueError, IOError:
print "%s could not be processed" % fname
+ raise
def __str__(self):
total_counts = sorted([(sum(self.counts[b].values()), b, self.counts[b].values()) for b in self.counts])
@@ -44,12 +46,14 @@ class AcCounts(object):
h5 = h5py.File("activation_counts.h5", "w")
-counts = AcCounts()
+ac_counts = AcCounts()
for fname in sys.argv[1:]:
binnum = os.path.split(fname)[1].split("-")[0]
- counts.insert(fname, binnum)
-counts.normalize()
-#print counts.counts['0212']
-print counts
-
+ ac_counts.insert(fname, binnum)
+ac_counts.normalize()
+for b, counts in sorted(ac_counts.counts.items()):
+ grp = h5.create_group(b)
+ for cat, count in counts.items():
+ grp.create_dataset(cat, data = numpy.array(count))
+h5.close()
--
GitLab