pshell_config/sf-op
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Script execution

Browse Source
This commit is contained in:
gobbo_a
2017-05-05 12:30:59 +02:00
parent 61777fc33c
commit ba10e10370
1 changed files with 19 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
script/test/WireScanTest2.py
View File

@@ -9,16 +9,24 @@ for i in range(len(blms)):
bg = get_attributes(path_back)["Mean"]
for cycle in range (cycles):
pos = load_data(root + wire+"_" + ("%04d" % (cycle+1)) + "/w_pos")
path = root + wire+"_" + ("%04d" % (cycle+1)) + "/blm" + str(j+1)
sig = load_data(path)
sig = blm_remove_spikes(sig)
if bg is not None:
sig = [v-bg for v in sig]
path = root + wire+"_" + ("%04d" % (cycle+1)) + "/blm" + str(i+1)
data = load_data(path)
sp = blm_remove_spikes(data)
sig = sp if bg is None else [v-bg for v in sp]
[com, rms] = profile_rms_stats(pos, sig,noise_std=0, n_sigma=3.5)
set_attribute(path, "RMS", float("nan") if (rms is None) else rms)
set_attribute(path, "RMS centroid", float("nan") if (com is None) else com)
#print i, j, [com, rms]
#set_attribute(path, "RMS", float("nan") if (rms is None) else rms)
#set_attribute(path, "RMS centroid", float("nan") if (com is None) else com)
print i, j, [com, rms]
[off, amp, com, sigma] = profile_gauss_stats(pos, sig, off=None, amp=None, com=None, sigma=None)
set_attribute(path, "Gauss COM", float("nan") if (com is None) else com)
set_attribute(path, "Gauss Sigma", float("nan") if (sigma is None) else sigma)
#print i, j, [off, amp, com, sigma]
#set_attribute(path, "Gauss COM", float("nan") if (com is None) else com)
#set_attribute(path, "Gauss Sigma", float("nan") if (sigma is None) else sigma)
print i, j, [off, amp, com, sigma]
from mathutils import Gaussian
g = Gaussian(amp, com, sigma)
gauss = [g.value(i)+off for i in pos]
plot([data, sp, sig, gauss], ["data", "sp", "signal", "gauss", ], xdata = pos, title="Fit blm" + str(i+1) + " - " + str(cycle+1))