forked from NSLS2/lsdc
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathfmx_annealer.py
122 lines (93 loc) · 3.48 KB
/
fmx_annealer.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
''' Originally FMX's annealer code from the file 2021-02-02 FMX annealer 4 Jun.html'''
import logging
logger = logging.getLogger(__name__)
import epics
import time
import socket
def blStrGet():
"""
Return beamline string
blStr: 'AMX' or 'FMX'
Beamline is determined by querying hostname
"""
hostStr = socket.gethostname()
if hostStr == 'xf17id2-ca1':
blStr = 'FMX'
elif hostStr == 'xf17id1-ca1':
blStr = 'AMX'
else:
logger.error('Error - this code must be executed on one of the -ca1 machines')
blStr = -1
return blStr
def govMsgGet(configStr = 'Robot'):
"""
Returns Governor message
configStr: Governor configuration, 'Robot' or 'Human', default: 'Robot'
Examples:
govMsgGet()
govMsgGet(configStr = 'Human')
"""
blStr = blStrGet()
if blStr == -1: return -1
sysStr = 'XF:17IDC-ES:' + blStr
devStr = '{Gov:' + configStr + '}'
stsStr = 'Sts:Msg-Sts'
pvStr = sysStr + devStr + stsStr
govMsg = epics.caget(pvStr)
return govMsg
def govStatusGet(stateStr, configStr = 'Robot'):
"""
Returns the current active status for a Governor state
configStr: Governor configuration, 'Robot' or 'Human', default: 'Robot'
stateStr: Governor short version state. Example: 'SA' for sample alignment
one of ['M', 'SE', 'SA', 'DA', 'XF', 'BL', 'BS', 'AB', 'CB', 'DI']
Examples
govStatusGet('SA')
govStatusGet('SA', configStr = 'Human')
"""
blStr = blStrGet()
if blStr == -1: return -1
if stateStr not in ['M', 'SE', 'SA', 'DA', 'XF', 'BL', 'BS', 'AB', 'CB', 'DI']:
logger.error('stateStr must be one of: M, SE, SA, DA, XF, BL, BS, AB, CB, DI]')
return -1
sysStr = 'XF:17IDC-ES:' + blStr
devStr = '{Gov:' + configStr + '-St:' + stateStr + '}'
stsStr = 'Sts:Active-Sts'
pvStr = sysStr + devStr + stsStr
govStatus = epics.caget(pvStr)
return govStatus
def govStateSet(stateStr, configStr = 'Robot'):
"""
Sets Governor state
configStr: Governor configuration, 'Robot' or 'Human', default: 'Robot'
stateStr: Governor short version state. Example: 'SA' for sample alignment
one of ['M', 'SE', 'SA', 'DA', 'XF', 'BL', 'BS', 'AB', 'CB', 'DI']
Examples:
govStateSet('SA')
govStateSet('AB', configStr = 'Human')
"""
blStr = blStrGet()
if blStr == -1: return -1
if stateStr not in ['M', 'SE', 'SA', 'DA', 'XF', 'BL', 'BS', 'AB', 'CB', 'DI']:
logger.error('stateStr must be one of: M, SE, SA, DA, XF, BL, BS, AB, CB, DI]')
return -1
sysStr = 'XF:17IDC-ES:' + blStr
devStr = '{Gov:' + configStr + '}'
cmdStr = 'Cmd:Go-Cmd'
pvStr = sysStr + devStr + cmdStr
epics.caput(pvStr, stateStr)
while not govStatusGet(stateStr, configStr = configStr):
print(govMsgGet(configStr = configStr))
time.sleep(2)
print(govMsgGet(configStr = configStr))
return
from ophyd import PVPositioner, PVPositionerPC, Device, Component as Cpt, EpicsMotor, EpicsSignal, EpicsSignalRO
# XF:17IDC-ES:FMX{Wago:1}AnnealerIn-Sts
# XF:17IDC-ES:FMX{Wago:1}AnnealerAir-Sel
## DONE Redo ophyd object with proper PV
class Annealer(Device):
air = Cpt(EpicsSignal, 'Air-Sel')
status = Cpt(EpicsSignalRO, 'In-Sts') # status: 0 (Not In), 1 (In)
annealer = Annealer('XF:17IDC-ES:FMX{Wago:1}Annealer', name='annealer',
read_attrs=[],
labels=['fmx'])