from visual import *
from visual.graph import *

gd=gdisplay(xmin=0,xmax=8.0,ymin=0,ymax=8.0,
            title="fractal dimension",
            xtitle="n", ytitle="x",height=600,width=600,x=0,y=200,
            foreground=color.black, background=color.white)

BOXES=10.0
Q=2.0

number=arange(0,pow(2.0,BOXES))

xplot=gdots(color=color.blue)

nmax=100000

r=0.896

for i in arange(0,BOXES):
    x=0.3
    number_of_boxes=int(pow(2.0,i))
    box_length=1.0/number_of_boxes

    
    for j in arange(0,number_of_boxes):
        number[j]=0.0
        
    for n in arange(1,nmax):
        x=4.0*r*x*(1-x)
        if (n>10000):
            number[int(x/box_length)]=number[int(x/box_length)]+1.0

    count=0.0
    for j in arange(0,number_of_boxes):
        if(number[j]!=0.0):
            count=count+pow(number[j]/(nmax-10000),Q)
            
    xplot.plot(pos=(-1.0*log(box_length),-1.0/(Q-1.0)*log(count)))