import psana
from xtcav2.LasingOnCharacterization import *
import numpy as np

from matplotlib import pyplot as plt

data_source = psana.DataSource("exp=amox23616:run=74")
XTCAVRetrieval = LasingOnCharacterization()

for evt in data_source.events():
    try:
        XTCAVRetrieval.processEvent(evt)
        roi = XTCAVRetrieval.processedXTCAVImage()
        print(XTCAVRetrieval.physicalUnits())
        plt.imshow(roi[0])
        plt.show()
    except:
        print "issue finding pulse"