mirror of https://github.com/python/cpython
56 lines
1.3 KiB
Python
56 lines
1.3 KiB
Python
|
# Brownian motion -- an example of a NON multi-threaded Tkinter program ;)
|
||
|
|
# By Michele Simoniato, inspired by brownian.py
|
||
|
|
|
||
|
|
from Tkinter import *
|
||
|
|
import random
|
||
|
|
import sys
|
||
|
|
|
||
|
|
WIDTH = 400
|
||
|
|
HEIGHT = 300
|
||
|
|
SIGMA = 10
|
||
|
|
BUZZ = 2
|
||
|
|
RADIUS = 2
|
||
|
|
LAMBDA = 10
|
||
|
|
FILL = 'red'
|
||
|
|
|
||
|
|
stop = 0 # Set when main loop exits
|
||
|
|
root = None # main window
|
||
|
|
|
||
|
|
def particle(canvas): # particle = iterator over the moves
|
||
|
|
r = RADIUS
|
||
|
|
x = random.gauss(WIDTH/2.0, SIGMA)
|
||
|
|
y = random.gauss(HEIGHT/2.0, SIGMA)
|
||
|
|
p = canvas.create_oval(x-r, y-r, x+r, y+r, fill=FILL)
|
||
|
|
while not stop:
|
||
|
|
dx = random.gauss(0, BUZZ)
|
||
|
|
dy = random.gauss(0, BUZZ)
|
||
|
|
try:
|
||
|
|
canvas.move(p, dx, dy)
|
||
|
|
except TclError:
|
||
|
|
break
|
||
|
|
else:
|
||
|
|
yield None
|
||
|
|
|
||
|
|
def move(particle): # move the particle at random time
|
||
|
|
particle.next()
|
||
|
|
dt = random.expovariate(LAMBDA)
|
||
|
|
root.after(int(dt*1000), move, particle)
|
||
|
|
|
||
|
|
def main():
|
||
|
|
global root, stop
|
||
|
|
root = Tk()
|
||
|
|
canvas = Canvas(root, width=WIDTH, height=HEIGHT)
|
||
|
|
canvas.pack(fill='both', expand=1)
|
||
|
|
np = 30
|
||
|
|
if sys.argv[1:]:
|
||
|
|
np = int(sys.argv[1])
|
||
|
|
for i in range(np): # start the dance
|
||
|
|
move(particle(canvas))
|
||
|
|
try:
|
||
|
|
root.mainloop()
|
||
|
|
finally:
|
||
|
|
stop = 1
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|