From 66b49fd941fbe30ba1efb799cd48498fa3e18148 Mon Sep 17 00:00:00 2001
From: Sven Augustin <sven.augustin@psi.ch>
Date: Mon, 17 May 2021 11:46:22 +0200
Subject: [PATCH] added a bit of argparse

---
 rewrite_epicsdata_pids.py | 25 +++++++++++++++++++++----
 rewrite_scaninfo.py       | 22 ++++++++++++++++++++--
 2 files changed, 41 insertions(+), 6 deletions(-)

diff --git a/rewrite_epicsdata_pids.py b/rewrite_epicsdata_pids.py
index 434278d..5dc3f8f 100755
--- a/rewrite_epicsdata_pids.py
+++ b/rewrite_epicsdata_pids.py
@@ -1,5 +1,18 @@
 #!/usr/bin/env python
 
+import argparse
+
+parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+parser.add_argument("instrument")
+parser.add_argument("pgroup")
+parser.add_argument("-c", "--channel", default="SAR-CVME-TIFALL4:EvtSet", help="BS channel used as source for the timestamp/pulseID mapping")
+clargs = parser.parse_args()
+
+instrument = clargs.instrument
+pgroup = clargs.pgroup
+bs_source_channel = clargs.channel
+
+
 import os
 from glob import glob
 import numpy as np
@@ -25,8 +38,11 @@ def get_fn(fns, contains):
 
 
 print("globbing")
-fns = "/sf/alvra/data/p18938/raw/**/*.PVCHANNELS.h5"
-fns = sorted(glob(fns, recursive=True))
+fns_pattern = "/sf/{}/data/{}/raw/**/*.PVCHANNELS.h5".format(instrument, pgroup)
+fns = sorted(glob(fns_pattern, recursive=True))
+
+if not fns:
+    raise SystemExit("no file found for: {}".format(fns_pattern))
 
 for i, fn_pv in enumerate(fns):
     fn_bs = fn_pv.replace(".PVCHANNELS.h5", ".BSDATA.h5")
@@ -43,8 +59,9 @@ for i, fn_pv in enumerate(fns):
         print(new_fn_pv, "exists... skipping")
         continue
 
-    ch_evts = "SAR-CVME-TIFALL4:EvtSet"
-    #ch_evts = "SAR-CVME-TIFALL5:EvtSet"
+#    ch_evts = "SAR-CVME-TIFALL4:EvtSet"
+#    ch_evts = "SAR-CVME-TIFALL5:EvtSet"
+    ch_evts = bs_source_channel
 
     with h5py.File(fn_bs, "r") as f:
         try:
diff --git a/rewrite_scaninfo.py b/rewrite_scaninfo.py
index ed74f3e..45f0586 100755
--- a/rewrite_scaninfo.py
+++ b/rewrite_scaninfo.py
@@ -1,11 +1,29 @@
 #!/usr/bin/env python
 
+import argparse
+
+parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+parser.add_argument("instrument")
+parser.add_argument("pgroup")
+clargs = parser.parse_args()
+
+instrument = clargs.instrument
+pgroup = clargs.pgroup
+
+
 import os
 from glob import glob
 import json
 
-fns = "/sf/alvra/data/p18938/raw/scan_info/**.json"
-for fn_si in sorted(glob(fns, recursive=True)):
+
+print("globbing")
+fns_pattern = "/sf/{}/data/{}/raw/scan_info/**.json".format(instrument, pgroup)
+fns = sorted(glob(fns_pattern, recursive=True))
+
+if not fns:
+    raise SystemExit("no file found for: {}".format(fns_pattern))
+
+for fn_si in fns:
     new_fn_si = fn_si.replace("/raw/", "/res/epics/")
     print(fn_si, new_fn_si)