From b35375746c7f99f275f0cf505abc964829e0aadf Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lo=C3=AFc=20Le=20Guyader?= <loic.le.guyader@xfel.eu>
Date: Fri, 19 Jul 2019 03:40:31 +0200
Subject: [PATCH] checkTimApdWindow plots the mean of the MCPraw traces if they
 are loaded instead of showing only one train

---
 xgm.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/xgm.py b/xgm.py
index 9b62b14..2018e6a 100644
--- a/xgm.py
+++ b/xgm.py
@@ -739,14 +739,14 @@ def checkTimApdWindow(data, mcp=1, use_apd=True, intstart=None, intstop=None):
     apdChannels={1:3, 2:2, 3:1, 4:0}
     npulses_max = data['npulses_sase3'].max().values
     tid = data['npulses_sase3'].where(data['npulses_sase3'] == npulses_max,
-                                      drop=True)[0].trainId.values
+                                      drop=True).trainId.values
     if 'MCP{}raw'.format(mcp) not in data:
-        tid, data_from_train = data.attrs['run'].train_from_id(tid)
+        ntid, data_from_train = data.attrs['run'].train_from_id(tid[0])
         trace = data_from_train['SCS_UTC1_ADQ/ADC/1:network']['digitizers.channel_1_'
                                 +'{}.raw.samples'.format(mcpToChannel[mcp])]
         print('no raw data for MCP{}. Loading trace from MCP{}'.format(mcp, mcp))
     else:
-        trace = data['MCP{}raw'.format(mcp)].sel(trainId=tid).T
+        trace = data['MCP{}raw'.format(mcp)].sel({'trainId':tid}).mean(dim='trainId')
     if use_apd:
         pulseStart = data.attrs['run'].get_array(
             'SCS_UTC1_ADQ/ADC/1', 
-- 
GitLab