IN_POSITION_BAND = 10.0


#cam_server.start("SAROP21-PSRD103_sp", True)
#cam_server.start("SAROP21-PPRM102_sp", True)
#cam_server.start("SAROP21-PSRD103_sp", True)
#cam_server.start("SAROP11-PSCR106_sp", True)
#cam_server.start("SAROP11-PPRM122_sp", True)
cam_server.start("SATOP21-PSCR162_sp", True)
cam_server.stream.waitCacheChange(-1)
i   = cam_server.stream.getChild("intensity")
iavg = create_averager(i, 20, interval = -1)

#ycent   = cam_server.stream.getChild("y_center_of_mass")
#ycentavg = create_averager(ycent, 20, interval = -1)
#xcent   = cam_server.stream.getChild("x_center_of_mass")
#xcentavg = create_averager(xcent, 20, interval = -1)


#iode = Channel("SLAAR21-LSCP-DRS1FNS:CH2:BOXCAR.VALP")
#iavg = create_averager(diode, 30, interval = -1)

def beam_ok():
    I0 = caget ("SATFE10-PEPG046:PHOTON-ENERGY-PER-PULSE-AVG")
    if I0 > 100: 
        return "Yes" 
    else :
        return "No"


def before_sampling(rec):
    while beam_ok() == "No":
        time.sleep(0.1)
        print(".")
 
def after_sampling(rec):
    if beam_ok() == "No":
        rec.invalidate()

           
i0 = Channel("SATFE10-PEPG046:PHOTON-ENERGY-PER-PULSE-AVG")

class DCM(RegisterBase):
    def doWrite(self, value):
        caput ("ATHOS:SetEnergy", value)
        time.sleep(0.5)
        cawait('ATHOS:MOVING', 0, timeout = 10.0, type = 'i')    
       cawait('SFB_BEAM_ENERGY_ECOL:SUM-ERROR-OK', 1, timeout = 10.0, type = 'i')    
        if abs(value - self.read()) >  IN_POSITION_BAND :
            raise Exception("Cannot set mono")   
    def doRead(self):
         return caget("SAROP11-ODCM105:ENERGY")

alvra = DCM()
alvra.initialize()
i0 = Channel("SARFE10-PBPG050:PHOTON-ENERGY-PER-PULSE-US")
acc = Channel("SARCL02-MBND100:P-READ")
k    = Channel("SARUN-UIND-MOT:KALL.VAL")
lscan( alvra, [iavg,i0,k,acc,ycentavg,xcentavg], 4930.0, 5030.0,1.0,0.2)

#scan( (pitch2,alvra), [diode,i0], (5.0,2100.0),(8.0,2300.0),(0.1,1.0), latency= 0.2, before_read=before_sampling, after_read=after_