prjemian
commented
4 years ago With the synApps docker IOC and prefix sky:, create a happi database (json text file) that describes 16 motors and one scaler.
#!/usr/bin/env python
"""
create_happi_db.py: create the JSON file for happi
========== ==============================
item description
========== ==============================
IOC prefix sky:
motors 16: sky:m1 .. sky:m16
scaler 1: sky:scaler1
========== ==============================
"""
# extracted lines
from apstools.utils import *
import json
import re
logging.basicConfig(level=logging.WARNING)
logger = logging.getLogger(__name__)
HAPPI_FILE = "sky_db.json"
IOC_PREFIX = "sky:"
MASTER_PREFIX = "zz_" # Why "zz_"? Sorts last, makes happi happy!
def prefix_to_mnemonic(prefix):
"""
make an IOC prefix safe to use with the happi name attribute
ioc: --> ioc_
AD42: --> ad42_
13ID:Sim1: --> zz_13id_sim1_
XF:11IDB-BI{Cam:09} --> xf_11idb_bi_cam_09_
"""
# required regexp from happi.item
compliant_pattern = r'^[a-z][a-z\_0-9]*'
replacement = '_'
safer = prefix.lower()
match = re.match(compliant_pattern, safer)
if match is None: # Fix for IOC prefixes that start with other characters
prefix = MASTER_PREFIX + prefix
safer = prefix.lower()
match = re.match(compliant_pattern, safer)
last = match.span()[-1]
while last < len(prefix):
safer = match.group(0) + replacement + safer[last+1:]
match = re.match(compliant_pattern, safer)
last = match.span()[-1]
return safer
def motors(prefix="ioc:", num=8):
"""
create ophyd motors provided by iocsky
"""
db = dict()
for i in range(num):
mne = prefix_to_mnemonic(prefix).rstrip("_") + f"_m{i+1}"
entry = dict(
documentation = f"{prefix} motor {i+1}",
prefix = f"{prefix}m{i+1}",
_id = mne,
name = mne,
active = True,
args = ['{{prefix}}'],
kwargs = dict(
name = '{{name}}',
labels = ["motors",]
),
type = 'OphydItem',
device_class = "ophyd.EpicsMotor",
)
db[mne] = entry
return db
def scaler(prefix="ioc:", num=1):
"""
create ophyd scaler ``num`` provided by iocsky
"""
db = dict()
mne = prefix_to_mnemonic(prefix).rstrip("_") + f"_scaler{num}"
entry = dict(
documentation = f"{prefix} scaler {num}",
prefix = f"{prefix}scaler{num}",
_id = mne,
name = mne,
active = True,
args = ['{{prefix}}'],
kwargs = dict(
name = '{{name}}',
labels = ["detectors",]
),
type = 'OphydItem',
device_class = "ophyd.scaler.ScalerCH",
)
db[mne] = entry
return db
def main():
ioc = {}
ioc.update(motors(IOC_PREFIX, 16))
ioc.update(scaler(IOC_PREFIX, 1))
print(dictionary_table(ioc))
with open(HAPPI_FILE, "w") as fp:
json.dump(ioc, fp, indent=2)
if __name__ == "__main__":
main()
tacaswell
Use [
happi.loader]() to define ophyd objects.