eco/xoptics/slits.py

from ..devices_general.motors import MotorRecord

class SlitBlades:
    def __init__(self,Id,name=None,elog=None):
        self.Id = Id
        self.name = name
        self._x1 = MotorRecord(Id+':MOTOR_X1')
        self._x2 = MotorRecord(Id+':MOTOR_X2')
        self._y1 = MotorRecord(Id+':MOTOR_Y1')
        self._y2 = MotorRecord(Id+':MOTOR_Y2')

    def get_hg(self):
        return self._x2.get_current_value()-self._x1.get_current_value()
    def get_vg(self):
        return self._y2.get_current_value()-self._y1.get_current_value()
    def get_ho(self):
        return (self._x1.get_current_value()+self._x2.get_current_value())/2
    def get_vo(self):
        return (self._y1.get_current_value()+self._y2.get_current_value())/2
    def set_hg(self,value):
        ho = self.get_ho()
        c1 = self._x1.changeTo(ho-value/2)
        c2 = self._x2.changeTo(ho+value/2)
        return c1,c2
    def set_vg(self,value):
        vo = self.get_vo()
        c1 = self._y1.changeTo(vo-value/2)
        c2 = self._y2.changeTo(vo+value/2)
        return c1,c2

    def set_ho(self,value):
        hg = self.get_hg()
        c1 = self._x1.changeTo(value-hg/2)
        c2 = self._x2.changeTo(value+hg/2)
        return c1,c2
    def set_vo(self,value):
        vg = self.get_vg()
        c1 = self._y1.changeTo(value-vg/2)
        c2 = self._y2.changeTo(value+vg/2)
        return c1,c2

    def __call__(self,width,height):
        self.set_hg(width)
        self.set_vg(height)

    def __repr__(self):
        return '(%s,%s) mm'%(self.get_hg(),self.get_vg())


class SlitFourBlades:
    def __init__(self,Id,name=None,elog=None):
        self.Id = Id
        self.name = name
        self._ax1 = MotorRecord(Id+':MOTOR_AX1')
        self._ax2 = MotorRecord(Id+':MOTOR_AX2')
        self._ay1 = MotorRecord(Id+':MOTOR_AY1')
        self._ay2 = MotorRecord(Id+':MOTOR_AY2')
        self._bx1 = MotorRecord(Id+':MOTOR_BX1')
        self._bx2 = MotorRecord(Id+':MOTOR_BX2')
        self._by1 = MotorRecord(Id+':MOTOR_BY1')
        self._by2 = MotorRecord(Id+':MOTOR_BY2')

    def get_hg(self):
        return self._ax2.get_current_value()-self._ax1.get_current_value()
    def get_vg(self):
        return self._ay2.get_current_value()-self._ay1.get_current_value()
    def get_ho(self):
        return (self._ax1.get_current_value()+self._ax2.get_current_value())/2
    def get_vo(self):
        return (self._ay1.get_current_value()+self._ay2.get_current_value())/2
    def set_hg(self,value):
        ho = self.get_ho()
        c1 = self._ax1.changeTo(ho-value/2)
        c2 = self._ax2.changeTo(ho+value/2)
        c3 = self._bx1.changeTo(ho-value/2)
        c4 = self._bx2.changeTo(ho+value/2)
        return c1,c2,c3,c4
    def set_vg(self,value):
        vo = self.get_vo()
        c1 = self._ay1.changeTo(vo-value/2)
        c2 = self._ay2.changeTo(vo+value/2)
        c3 = self._by1.changeTo(vo-value/2)
        c4 = self._by2.changeTo(vo+value/2)
        return c1,c2,c3,c4

    def set_ho(self,value):
        hg = self.get_hg()
        c1 = self._ax1.changeTo(value-hg/2)
        c2 = self._ax2.changeTo(value+hg/2)
        c3 = self._bx1.changeTo(value-hg/2)
        c4 = self._bx2.changeTo(value+hg/2)
        return c1,c2,c3,c4
    def set_vo(self,value):
        vg = self.get_vg()
        c1 = self._ay1.changeTo(value-vg/2)
        c2 = self._ay2.changeTo(value+vg/2)
        c3 = self._by1.changeTo(value-vg/2)
        c4 = self._by2.changeTo(value+vg/2)
        return c1,c2,c3,c4

    def __call__(self,width,height):
        self.set_hg(width)
        self.set_vg(height)

    def __str__(self):
        return '(%g,%g) mm'%(self.get_hg(),self.get_vg())

    def __repr__(self):
        return self.__str__()



class SlitPosWidth:
    def __init__(self,Id,name=None,elog=None):
        self.Id = Id
        self.name = name
        self._xoffs = MotorRecord(Id+':MOTOR_X')
        self._yoffs = MotorRecord(Id+':MOTOR_Y')
        self._xgap = MotorRecord(Id+':MOTOR_W')
        self._ygap = MotorRecord(Id+':MOTOR_H')

    def get_hg(self):
        return self._xgap.get_current_value()
    def get_vg(self):
        return self._ygap.get_current_value()
    def get_ho(self):
        return self._xoffs.get_current_value()
    def get_vo(self):
        return self._yoffs.get_current_value()
    def set_hg(self,value):
        c = self._xgap.changeTo(value)
        return c
    def set_vg(self,value):
        c = self._ygap.changeTo(value)
        return c

    def set_ho(self,value):
        c = self._xoffs.changeTo(value)
        return c
    def set_vo(self,value):
        c = self._yoffs.changeTo(value)
        return c

    def __call__(self,width,height):
        self.set_hg(width)
        self.set_vg(height)

    def __repr__(self):
        return '(%s,%s) mm'%(self.get_hg(),self.get_vg())