diff --git a/script/test/WireScanTest2.py b/script/test/WireScanTest2.py index 941b633..67f6e0b 100644 --- a/script/test/WireScanTest2.py +++ b/script/test/WireScanTest2.py @@ -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)) + +