Commit a46ff467 authored by augustin_s's avatar augustin_s 🐍
Browse files

quick camera check script

parent 9c819b06
#!/photonics/home/gac-maloja/.conda/envs/mdaq/bin/python
#!/usr/bin/env python
import logging
class DisabledLogger():
def __enter__(self):
logging.disable(logging.CRITICAL)
def __exit__(self, exit_type, exit_value, exit_traceback):
logging.disable(logging.NOTSET)
#from collections import defaultdict
from datetime import datetime, timedelta
from bsread import dispatcher
import data_api as dapi
import epics
from colorama import Fore, Back, Style
true = Fore.GREEN + " ✓" + Style.RESET_ALL
false = Fore.RED + " ✗" + Style.RESET_ALL
cam_names = [
# "DOES-NOT-EXIST",
"SATES21-CAMS154-M1",
"SATES24-CAMS161-M1"
]
cam_names += [f"SATES21-CAMS154-GIGE{i+1}" for i in range(8)]
width = max(len(str(i)) for i in cam_names)
fpics = [i + ":FPICTURE" for i in cam_names]
pvs = [epics.get_pv(c) for c in fpics]
pv_states = [pv.wait_for_connection(timeout=0.5) for pv in pvs]
bs_chans = dispatcher.get_current_channels()
bs_chans = set(x["name"] for x in bs_chans)
bs_states = [(i in bs_chans) for i in fpics]
#data = defaultdict(list)
#for f in fpics:
# res = dapi.search(f)
# for i in res:
# if not i["channels"]:
# continue
# be = i["backend"]
# data[f].append(be)
# print(f, be)
now = datetime.now()
start = now - timedelta(minutes=1)
end = start + timedelta(seconds=1)
ib_chans = ["sf-imagebuffer/" + i for i in fpics] #TODO does this speed this up?
aggregation = dapi.Aggregation(aggregation_type="value", aggregations=["sum"], nr_of_bins=1)
with DisabledLogger():
ib_data = dapi.get_data(channels=ib_chans, start=start, end=end, aggregation=aggregation)
def get_state(f):
try:
d = ib_data[f + ":sum"]
except KeyError:
try:
d = ib_data[f]
except KeyError:
return False
assert len(d) == 1
return not all(d.isna())
ib_states = [get_state(f) for f in fpics]
head = ("name", "PV", "BS", "IB")
bool_width = 2
bool_line = "-" * bool_width
line = ["-" * width] + [bool_line] * 3
res = [head, line]
for cam, pv_state, bs_state, ib_state in zip(cam_names, pv_states, bs_states, ib_states):
pv_state = true if pv_state else false
bs_state = true if bs_state else false
ib_state = true if ib_state else false
new = (cam, pv_state, bs_state, ib_state)
res.append(new)
for name, pv, bs, ib in res:
name = name.ljust(width)
# pv = pv.ljust(bool_width)
# bs = bs.ljust(bool_width)
# ib = ib.ljust(bool_width)
print(name, pv, bs, ib)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment