#!/usr/bin/python

import numpy as np
from ROOT import *

a = TMinuit()

def twoPartParabFCN(nPar, grad, fval, par, iflag):
	print "aaa1"
	fval=1

def twoPartParab(pos, par):
	x = pos[0]
	return par[4]*(par[0]+par[3]+par[1]*x+par[2]*x*x)

l1=0
l2=10

a = TF1("f", twoPartParab, l1, l2, 5)

g = TGraph(10, np.array(xrange(l1, l2), dtype=float), np.array(xrange(l1, l2), dtype=float))

#f = TVirtualFitter.Fitter(g, 5)

TVirtualFitter.SetDefaultFitter("Minuit")
print gMinuit
gMinuit.SetFCN(twoPartParabFCN)

g.Fit(a, "U")

c = TCanvas("c", "Original Image", 930, 700)
g.Draw("A*")
c.Modified()
c.Update()


if __name__ == '__main__':
	rep = ''
	while not rep in [ 'q', 'Q' ]:
		rep = raw_input( 'enter "q" to quit, "m" to markup stars, "r" to redraw main image: ' )
		if 1 < len(rep):
			rep = rep[0]