def plot(infile, outfile=None): yt = [None]*4 for i in [1]: yt[i-1],t,_,_,_,_ = wfm2read_fast.wfm2read(infile) print "Read file: ", infile y = np.array(yt) return (t, y[0]) # , 'r')
def read_wfm(infolder, num_traces=100): print 'Reading WFMs from', infolder # Can add a check to see that we are not reading a file that is not # availabled for file_count in xrange(1,num_traces+1): fname = 'W%d.wfm' % file_count print 'Reading %s [%d/%d]' % (fname, file_count, num_traces) trace_data = wfm2read_fast.wfm2read(infolder+'/'+fname)[0] trace_trimmed = trace_data[:] try: traces except NameError: traces = np.zeros((num_traces, len(trace_trimmed))) traces[file_count-1, :] = trace_trimmed # np.save(outfile, traces) # print 'Finished processing all WFMs. Output in',outfile return traces