# Run detached as:
# pshell -p=DemoScan.java -d -sbar -tbar -l -dplt
from mathutils import fit_harmonic, HarmonicOscillator


if get_exec_pars().innerArgs is None:
    print 1
    START = 0.0
    END = 0.400
    STEPS = 10
    PLOT = None
if PLOT is None:
    PLOT = plot(None,title="HFit")[0]
PLOT.clear()
    
sd = lscan(inp, (sin,out,arr), START, END, STEPS, 0.1)


ydata = sd[sin]
xdata = sd[inp]

max_y= max(ydata)    
index_max = ydata.index(max_y)    
max_x= xdata[index_max]
    
start,end = min(xdata), max(xdata)
(amplitude, angular_frequency, phase) = fit_harmonic(ydata, xdata)
fitted_harmonic_function = HarmonicOscillator(amplitude, angular_frequency, phase)

print "amplitude = ", amplitude
print "angular frequency = ", angular_frequency
print "phase = ", phase

f = angular_frequency/ (2* math.pi)  
print "frequency = ", f  

resolution = 0.01
fit_y = []
for x in frange(start,end,resolution, True):
    fit_y.append(fitted_harmonic_function.value(x))
fit_x = frange(start, end+resolution, resolution)


PLOT.addSeries(LinePlotSeries("data"))
PLOT.addSeries(LinePlotSeries("fit"))
PLOT.getSeries(0).setData(xdata, ydata)    
PLOT.getSeries(1).setData(fit_x, fit_y)    

#m = (phase + math.pi)/ angular_frequency
m = -phase / angular_frequency
if (m<start):
    m+=(1.0/f)
    
if start <= m <=end:
    print "fit = ", m
    PLOT.addMarker(m, None, "Fit="+str(round(m ,2)), Color.MAGENTA.darker())
    valid_fit = True
else:
    print "max = ",max_x
    PLOT.addMarker(max_x, None, "Max="+str(round(max_x ,2)), Color.MAGENTA.darker())        
    valid_fit = False

set_return([amplitude, angular_frequency, phase, valid_fit, m])