import numpy
import random
import math
from matplotlib.pyplot import *
import testMetropolis
                

N=100
atomes = testMetropolis.Atomes(N)
atomes.temperature(1.0)
(en,E,delta)=atomes.boucle(10000)
figure(figsize=(10,6))
plot(en)
ylabel('E')
xlabel('iteration')
grid()
                

N=1000
atomes = testMetropolis.Atomes(N)
atomes.temperature(1.0)
(en,E,delta)=atomes.boucle(50000)
figure(figsize=(10,6))
plot(en)
ylabel('E')
xlabel('iteration')
grid()
                

(en,E,delta)=atomes.boucle(10000)
figure(figsize=(10,6))
plot(en)
axis([0,10000,0,1000])
ylabel('E')
xlabel('iteration')
grid()
                

temp = numpy.arange(1.0,10.0,1.0)
n = temp.size
Energie = numpy.zeros(n)
Delta = numpy.zeros(n)
for k in range(n):
    atomes.temperature(temp[k])
    atomes.boucle(10000)
    (en,E,delta) = atomes.boucle(10000)
    Energie[k] = E
    Delta[k] = delta
    
figure(figsize=(8,6))
errorbar(temp,Energie,yerr=Delta,fmt=None)
axis([0,10,0,6000])
ylabel("Energie")
xlabel("T")
axis()
                

figure(figsize=(8,6))
plot(atomes.etat,'.')
xlabel('atome')
ylabel('e')
axis()