Subversion Repositories pdp1

import numpy as np

import matplotlib.pyplot as plt

import matplotlib.colors as colors

import matplotlib.animation as animation

from random import randint

width=1024

height=1024

def generator():

    while True:

        x = randint(0,width-1)

        y = randint(0,height-1)

        yield (x,y)

def init():

    return [im]

def run(point):

    # TODO: This doesn't update the image?

    fb = im.get_array()

    fb *= 0.99   # decay

    if point:     # plot new point

        #print point

        x,y=point

        fb[y,x]=1.0

    im.set_array(fb)

    return [im]

fig,ax = plt.subplots()

# TODO: this is white to green, want black to green?

cmap = plt.get_cmap("Greens")

#ax.axis('off')

fb = np.ndarray((width,height), np.float32)

fb.fill(0)

#for x in range(width):

#    fb[x,:] *= x/float(width)

#print fb

im = ax.imshow(fb, cmap, colors.Normalize(0,1))

ani = animation.FuncAnimation(fig, run, generator, interval=1)

plt.show()