From 6a695e7a88596d879959ae27de8cbf2955e3a8c3 Mon Sep 17 00:00:00 2001
From: Yngve Inntjore Levinsen <Yngve.Levinsen@esss.se>
Date: Mon, 23 Mar 2015 13:47:50 +0100
Subject: [PATCH] added TraceWin.plt.calc_std

---
 ess/TraceWin.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/ess/TraceWin.py b/ess/TraceWin.py
index a44a303..418be1d 100644
--- a/ess/TraceWin.py
+++ b/ess/TraceWin.py
@@ -244,6 +244,25 @@ class plt:
             for v in vals:
                 self.avg[v].append(numpy.average(data[v]))
 
+    def calc_std(self):
+        '''
+        Calculates the beam sizes
+
+        '''
+
+        import numpy
+
+        if not hasattr(self,'sigma'):
+               self.calc_sigma()
+
+        vals=self._columns[:-1]
+
+        self.std={}
+
+        for j in xrange(len(vals)):
+            v=vals[j]
+            self.std[v]=numpy.sqrt(self.sigma[:,j,j])
+
     def calc_minmax(self,pmin=5,pmax=95):
         '''
         Calculates min/max values of beam coordinates
-- 
GitLab