diff --git a/app/__init__.py b/app/__init__.py new file mode 100644 index 0000000..b2338f3 --- /dev/null +++ b/app/__init__.py @@ -0,0 +1 @@ +from .adaptiveorbit import AdaptiveOrbit diff --git a/app/adaptiveorbit.py b/app/adaptiveorbit.py new file mode 100644 index 0000000..48ee0d7 --- /dev/null +++ b/app/adaptiveorbit.py @@ -0,0 +1,53 @@ +import time +import numpy as np + +from sfbd.ext.reichebscombined import ReicheBSCombined +from slic.core.adjustable import PVAdjustable + +class AdaptiveOrbit: + """ + Wrapper class to bundle all daq/io needed for adaptive orbit feedback. + """ + def __init__(self,beamline='Aramis'): + self.beamline=None + self.sensor = None + self.actuator = None + self.getBeamline(beamline) + + def read(self): + if not self.sensor or not self.beamline: + return None + data=self.sensor.get_current_value() + if not data: + return None + + retval={'pid':data['pid'], + 'Signal':data[self.channel0[0]], + 'X':np.array([data[i] for i in self.channelX]), + 'Y':np.array([data[i] for i in self.channelY])} + return retval + + + def read_adj(self): + if not self.actuator or not self.beamline: + return None + return [pv.get_current_value() for pv in self.actuator] + + + def getBeamline(self,beamline='Aramis'): + if beamline.upper() == 'ARAMIS': + self.beamline='Aramis' + self.channelX=['SARUN%2.2d-DBPM070:X1' % id for id in range(1,17)] + self.channelY=['SARUN%2.2d-DBPM070:Y1' % id for id in range(1,17)] + self.channel0 = ['SARFE10-PBIG050-EVR0:CALCI'] + FB_bpms = [x.replace(':X1',':X-REF-FB') for x in self.channelX] + [x.replace(':Y1',':Y-REF-FB') for x in self.channelY] + elif beamline.upper() == 'ATHOS': + self.beamline=None + else: + self.beamline=None + return + channels=self.channel0+self.channelX+self.channelY + self.sensor = ReicheBSCombined('AdapiveOrbit',channels) + self.actuator = [PVAdjustable(pv) for pv in FB_bpms] + + diff --git a/ext/__init__.py b/ext/__init__.py new file mode 100644 index 0000000..ae07e6b --- /dev/null +++ b/ext/__init__.py @@ -0,0 +1 @@ +from .reichebscombined import ReicheBSCombined diff --git a/ext/reichebscombined.py b/ext/reichebscombined.py new file mode 100644 index 0000000..1541705 --- /dev/null +++ b/ext/reichebscombined.py @@ -0,0 +1,12 @@ +from slic.core.sensor.bsmonitor import BSMonitor + +class ReicheBSCombined(BSMonitor): + + # Du brauchst kein extra init. BSMonitor tut schon das richtige... + + def _unpack(self, data): + # data ist ein dict mit allen Deinen Kanälen + pid = data["pid"] # der effektive Channel-Name der Pulse ID + # hier dein Code + # am Ende sollte eine Zahl rauskommen: + return data diff --git a/util/__init__.py b/util/__init__.py new file mode 100644 index 0000000..e69de29