dispersiveXanes/sigma_vs_shot.py

44 lines
976 B
Python
Raw Permalink Normal View History

import matplotlib.pyplot as plt
import numpy as np
import xppl37_spectra
import xanes_analyzeRun
from datastorage import DataStorage as ds
r=xppl37_spectra.xanes_analyzeRun.AnalyzeRun(81)
r.load()
if len(r.results.keys()) > 1:
p1 = np.vstack( ( r.results[c].p1 for c in range(0,10,2) ) )
p2 = np.vstack( ( r.results[c].p2 for c in range(0,10,2) ) )
else:
p1 = r.results[0].p1
p2 = r.results[0].p2
N = int(len(p1)/2)
ref = slice(0,1000)
sam = slice(1000,1200)
#ref = slice(200)
#sam = slice(200)
ref = ds(p1=p1[ref],p2=p2[ref])
sam = ds(p1=p1[sam],p2=p2[sam])
# "ratioOfAverage medianOfRatios"
data=xppl37_spectra.calcAbs(ref,sam,refKind="medianOfRatios")
abs = data[-1]
idx = slice(200,800)
abs = abs[:,idx]
fom = []
fom1 = []
x = np.arange(abs.shape[1])
N = range(1,len(abs))
#N = 2**np.arange(15)
for i in N:
av = np.nanmean(abs[:i],axis=0)
p = np.polyfit(x,av,5)
diff = av-np.polyval(p,x)
fom.append( np.nanstd(av) )
fom1.append( np.nanstd(diff) )