import pylab
import numpy

class IsingMeanField:
    def __init__(self,temp=0.):
        from numpy import tanh, cosh
        self.temp = temp

    def setTemp(self,temp):
        self.temp = temp

    def __call__(self,x,mode=0):
        ## fix this
        if mode==0:  
            # return the function
            return 1.
        elif mode==1:
            # return its first derivative
            return 1.

    def makeHisto(self,T0=0.0,T1=6.0,dT=0.1,accuracy=0.001):
        from RootFinder import RootFinder
        Ts     = []
        ss     = []
        solver = RootFinder(self,'NR')

        while (T0<T1):
            T0 += dT
            Ts.append(T0)
            self.setTemp(T0)
            ss.append(solver.solveFor0(accuracy))

        return Ts,ss

if __name__ == "__main__":
    isinganalytical = IsingMeanField()
    Ts,ss = isinganalytical.makeHisto()

    plot1 = pylab.plot(Ts,ss,'bo',linewidth=1.5)
    pylab.title("Magnetization from mean field approximation")
    pylab.ylabel(r"$M/M_{\rm max}$")
    pylab.xlabel(r"$T [jJ/k_B]$")
    pylab.grid(True)
    pylab.show()