Page History
...
Code Block | ||
---|---|---|
| ||
from psana import DataSource import numpy as np def test_callback(data_dict): # called back for every event print(data_dict) ds = DataSource(exp='xpptut13', run=1, dir='.tmp', filter=lambda x : True, batch_size=2) smd = ds.smalldata(filename='my.h5', batch_size=5, callbacks=[test_callback]) run = next(ds.runs()) arrsum = None for i,evt in enumerate(run.events()): myones = np.ones(2, dtype=np.int) smd.event(evt, myfloat=2.0, arrint=myones) if arrsum is None: arrsum = myones else: arrsum += myones if smd.summary: # add up array across "big data" cores and save to h5 smd.sum(arrsum) # assumes that all cores have received events smd.save_summary({'summary_array' : arrsum}, summary_int=1) smd.done() |
...
Overview
Content Tools