From a5b7db763d27aaf00c757804e7dbe29f3785e792 Mon Sep 17 00:00:00 2001
From: Pauluhn Anuschka <anuschka.pauluhn@psi.ch>
Date: Tue, 12 May 2015 14:54:30 +0200
Subject: [PATCH] Script execution

---
 script/PolynomialFit.py | 57 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 57 insertions(+)
 create mode 100644 script/PolynomialFit.py

diff --git a/script/PolynomialFit.py b/script/PolynomialFit.py
new file mode 100644
index 0000000..1c6998d
--- /dev/null
+++ b/script/PolynomialFit.py
@@ -0,0 +1,57 @@
+"""
+Function fitting and peak search with mathutils facade
+""" 
+from mathutils i mport fit_polynomial,fit_gaussian, fit_harmonic, calculate_peaks
+from mathutils import PolynomialFunction, Gaussian, HarmonicOscillator
+import math
+
+
+start = 0
+end = 5
+step_size = 0.1
+
+result= lscan(sout,sinp,start,end,[step_size,],0.01)
+
+readable = result.getReadable(0)
+positions = result.getPositions(0)
+
+def get_function_data(function, start, end, resolution):
+    ret = []
+    for x in frange(start, end, resolution, True):
+        fit_polinomial.append(function.value(x))
+
+
+pars_polynomial = (a0, a1, a2, a3, a4, a5, a6) = fit_polynomial(readable, positions, 6)
+fitted_polynomial_function = PolynomialFunction(pars_polynomial)
+print pars_polynomial 
+
+(normalization, mean, sigma) = fit_gaussian(readable, positions, True)
+fitted_gaussian_function = Gaussian(normalization, mean, sigma)
+print (normalization, mean, sigma)
+
+(amplitude, angular_frequency, phase) = fit_harmonic(readable, positions)
+fitted_harmonic_function = HarmonicOscillator(amplitude, angular_frequency, phase)
+print (amplitude, angular_frequency, phase) 
+
+
+resolution = step_size/100
+fit_polinomial = []
+fit_gaussian = []
+fit_harmonic = []
+for x in frange(start,end,resolution, True):
+    fit_polinomial.append(fitted_polynomial_function.value(x))
+    fit_gaussian.append(fitted_gaussian_function.value(x))
+    fit_harmonic.append(fitted_harmonic_function.value(x))
+x = frange(start, end+resolution, resolution)
+
+
+peaks = calculate_peaks(fitted_polynomial_function)
+
+plots = plot([readable, fit_polinomial, fit_gaussian, fit_harmonic] , 
+    ["data", "polinomial", "gaussian", "harmonic"], xdata = [positions,x,x,x] )
+
+for p in peaks:    
+    print "Max: " + str(p)
+    plots[0].addMarker(p, None, "Max=" + str(round(p,2)), None)
+import java.awt.Color
+plots[0].addMarker(mean, None, "Mean=" + str(round(mean,2)), java.awt.Color.LIGHT_GRAY)