59 lines
1.9 KiB
Python
Executable File
59 lines
1.9 KiB
Python
Executable File
"""
|
|
Function fitting and peak search with mathutils facade
|
|
"""
|
|
from mathutils import fit_polynomial,fit_gaussian, fit_harmonic, calculate_peaks
|
|
from mathutils import PolynomialFunction, Gaussian, HarmonicOscillator
|
|
import math
|
|
|
|
|
|
start = 0
|
|
end = 10
|
|
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)
|
|
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, start, end)
|
|
|
|
plots = plot([readable, fit_polinomial, fit_gaussian, fit_harmonic] ,
|
|
["data", "polinomial", "gaussian", "harmonic"], xdata = [positions,x,x,x], title="Data")
|
|
|
|
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)
|
|
|