import zmq
import time
import psana
import numpy as np

context = zmq.Context()
socket = context.socket(zmq.PUB)
socket.bind("tcp://*:12322")

#run = 8
#ds = psana.DataSource('exp=xppl3816:run=%d:smd' %run)
ds = psana.DataSource('shmem=XPP.0:stop=no')
epics = ds.env().epicsStore()
opal_0_detector = psana.Detector('opal_0')
opal_1_detector = psana.Detector('opal_1')
opal_1_detector = psana.Detector('opal_1')
#ipm3_src = psana.Source('BldInfo(XppSb3_Ipm)')

t0 = time.time()
for i, evt in enumerate(ds.events()):
    #if i % 20 != 0:
    #    continue
    opal_0 = opal_0_detector.raw(evt)
#    opal_2 = np.random.random((1024, 1024))#opal_2_detector.raw(evt)
    opal_1 = opal_1_detector.raw(evt)
    cntr = ds.env().configStore().get(psana.ControlData.ConfigV3, psana.Source()).pvControls()
    if len(cntr) > 0:
      cpvName = cntr.pvControlsu()[0].name()
      cpvValue = cntr.pvControls()[0].value()
    else:
      cpvName = None
      cpvValue = None
    
    if opal_0 is None or opal_1 is None:
        print 'missing opal'
        continue
    socket.send_pyobj( dict( opal0 = opal_0, opal1 = opal_1, cPv = dict(name=cpvName,value=cpvValue)) )
    print 'Shot',i, 'sent; time since starting:', time.time()-t0
    time.sleep(1.0)