99 lines
2.8 KiB
Python
Executable File
99 lines
2.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
This little program synchronizes an EPICS substitutions file with values read
|
|
from a MasterMAC motor controller.
|
|
|
|
Mark Koennecke, February 2023
|
|
"""
|
|
import sys
|
|
import time
|
|
|
|
from macmaster import MasterMACS
|
|
|
|
def find_commas(rawline):
|
|
comma_list = []
|
|
for i in range(len(rawline)):
|
|
if rawline[i] == ',':
|
|
comma_list.append(i)
|
|
return comma_list
|
|
|
|
def pretty_line(newlist, comma_list):
|
|
"""
|
|
adds spaces to match the tabbing of the patter line
|
|
"""
|
|
newline = ''
|
|
for item, idx in zip(newlist, comma_list):
|
|
length = len(newline) + 1 + len(item)
|
|
if length < idx:
|
|
newline += ' ' * (idx - length)
|
|
newline += item
|
|
newline += ','
|
|
newline += newlist[-1]
|
|
return newline[0:-1]
|
|
|
|
def transact(command):
|
|
global mac
|
|
mac.send(command)
|
|
time.sleep(.1)
|
|
return mac.receive()
|
|
|
|
def fix_line(par_list, index_list):
|
|
# import pdb; pdb.set_trace()
|
|
addridx = index_list.index('ADDR')
|
|
motNo = int(par_list[addridx])
|
|
ack, reply = transact('%dR24' % motNo)
|
|
if ack == 'NO':
|
|
idx = reply.find('=')
|
|
lowlim = reply[idx+1:]
|
|
lowidx = index_list.index('DLLM')
|
|
par_list[lowidx] = lowlim.strip()
|
|
ack, reply = transact('%dR23' % motNo)
|
|
if ack == 'NO':
|
|
idx = reply.find('=')
|
|
highlim = reply[idx+1:]
|
|
highidx = index_list.index('DHLM')
|
|
par_list[highidx] = highlim.strip()
|
|
ack, reply = transact('%dR05' % motNo)
|
|
if ack == 'NO':
|
|
idx = reply.find('=')
|
|
speed = reply[idx+1:]
|
|
speedidx = index_list.index('VELO')
|
|
par_list[speedidx] = speed.strip()
|
|
return par_list
|
|
|
|
def scan_substitution_file(filename):
|
|
index_list = None
|
|
# import pdb; pdb.set_trace()
|
|
with open(filename, 'r') as fin:
|
|
rawline = fin.readline()
|
|
# import pdb; pdb.set_trace()
|
|
while rawline:
|
|
line = rawline.replace(' ','')
|
|
line = line.strip('{}')
|
|
l = line.split(',')
|
|
if line.find('DHLM') > 0:
|
|
index_list = l
|
|
comma_list = find_commas(rawline)
|
|
sys.stdout.write(rawline)
|
|
elif not index_list:
|
|
sys.stdout.write(rawline)
|
|
else:
|
|
if len(l) == len(index_list):
|
|
newlist = fix_line(l, index_list)
|
|
# newline = ','.join(newlist)
|
|
newline = pretty_line(newlist, comma_list)
|
|
sys.stdout.write('{' + newline + '\n')
|
|
else:
|
|
sys.stdout.write(rawline)
|
|
rawline = fin.readline()
|
|
|
|
#------------------ main
|
|
if len(sys.argv) < 4:
|
|
print('Usage:\n\tsyncMMACSub.py <host> <port> <substitutions-file>\n')
|
|
sys.exit(1)
|
|
|
|
mac = MasterMACS(sys.argv[1], sys.argv[2])
|
|
scan_substitution_file(sys.argv[3])
|
|
|
|
|