diff --git a/xgm.py b/xgm.py index 11294d641519de9f8d436d1e26f1c30ab4a372c3..d4daf34ed36713283335de8093df0fdd756470a4 100644 --- a/xgm.py +++ b/xgm.py @@ -700,7 +700,6 @@ def checkTimApdWindow(data, mcp=1, use_apd=True, intstart=None, intstop=None): step = sa3.where(data['npulses_sase3']>1, drop=True)[0,:2].values step = int(step[1] - step[0]) nsamples = 440 * step - print(nsamples) else: nsamples = 0 @@ -754,16 +753,19 @@ def matchXgmTimPulseId(data, use_apd=True, intstart=None, intstop=None, mergeList = [res, res2] for mcp in range(1,5): if 'MCP{}apd'.format(mcp) in data or 'MCP{}raw'.format(mcp) in data: - MCPapd = getTIMapd(data, mcp=mcp, use_apd=use_apd, intstart=intstart, intstop=intstop, - bkgstart=bkgstart, bkgstop=bkgstop, t_offset=t_offset, - npulses=npulses).rename('MCP{}apd'.format(mcp)) + MCPapd = getTIMapd(data, mcp=mcp, use_apd=use_apd, intstart=intstart, + intstop=intstop,bkgstart=bkgstart, bkgstop=bkgstop, + t_offset=t_offset, + npulses=npulses).rename('MCP{}apd'.format(mcp)) if use_apd: MCPapd = MCPapd.rename({'apdId':'pId'}) else: MCPapd = MCPapd.rename({'MCP{}fromRaw'.format(mcp):'pId'}) - dropList.append('MCP{}raw'.format(mcp)) mergeList.append(MCPapd) - dropList.append('MCP{}apd'.format(mcp)) + if 'MCP{}raw'.format(mcp) in data: + dropList.append('MCP{}raw'.format(mcp)) + if 'MCP{}apd'.format(mcp) in data: + dropList.append('MCP{}apd'.format(mcp)) mergeList.append(data.drop(dropList)) subset = xr.merge(mergeList, join='inner') return subset