50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
import time
|
|
import numpy as np
|
|
|
|
from bstrd import BSCache
|
|
from epics import PV
|
|
|
|
class SpectralAnalysis:
|
|
"""
|
|
Wrapper class to bundle all daq/io needed for adaptive orbit feedback.
|
|
"""
|
|
def __init__(self):
|
|
|
|
self.bs = BSCache(100000,10000) # 100 second timeout, size for 100 second data taken
|
|
self.bs.stop()
|
|
|
|
self.channel = ''
|
|
self.channels = ['SARFE10-PSSS059:SPECTRUM_Y',
|
|
'SATOP21-PMOS127-2D:SPECTRUM_Y',
|
|
'SATOP31-PMOS132-2D:SPECTRUM_Y']
|
|
self.isConnected = False
|
|
|
|
def connect(self,ich):
|
|
if ich < 0 or ich >= len(self.channels):
|
|
return False
|
|
self.channel = self.channels[ich]
|
|
print('Connecting to BS-Channel:',self.channel)
|
|
self.bs.channels.clear()
|
|
self.bs.get_var(self.channel) # this starts the stream into the cache
|
|
self.pv = PV(self.channel.replace('_Y','_X'))
|
|
|
|
def terminate(self):
|
|
print('Stopping BSStream Thread...')
|
|
self.bs.stop()
|
|
self.bs.pt.running.clear() # for some reason I have to
|
|
|
|
def flush(self):
|
|
self.bs.flush()
|
|
|
|
def read(self):
|
|
data=self.bs.__next__()
|
|
return data['pid'],data[self.channel]
|
|
|
|
def readPV(self):
|
|
return self.pv.value
|
|
|
|
def getSpectrometerName(self):
|
|
return self.channel
|
|
|
|
|