source: tk1.py @ 0:3ce177d96be2

importˆàTkinter
importˆàtime
fromˆàmathˆàimportˆà*
importˆàrandom

classˆàVector(object):
ˆà ˆà defˆà__init__(self,x,y):
ˆà ˆà ˆà ˆà self.x=x
ˆà ˆà ˆà ˆà self.y=y
ˆà ˆà defˆà__abs__(self):
ˆà ˆà ˆà ˆà returnˆàsqrt(self.x**2+self.y**2)
ˆà ˆà defˆà__add__(self,other):
ˆà ˆà ˆà ˆà returnˆàVector(self.x+other.x,self.y+other.y)
ˆà ˆà defˆà__mul__(self,digit):
ˆà ˆà ˆà ˆà returnˆàVector(digit*self.x,digit*self.y)
ˆà ˆà defˆàangle(self):
ˆà ˆà ˆà ˆà ifˆàself.x==0:
ˆà ˆà ˆà ˆà ˆà ˆà ifˆàself.y>0:
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà returnˆà-pi/2
ˆà ˆà ˆà ˆà ˆà ˆà else:
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà returnˆàpi/2
ˆà ˆà ˆà ˆà ifˆàself.y==0:
ˆà ˆà ˆà ˆà ˆà ˆà ifˆàself.x>0:
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà returnˆà0
ˆà ˆà ˆà ˆà ˆà ˆà else:
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà returnˆàpi
ˆà ˆà ˆà ˆà m=abs(self)
ˆà ˆà ˆà ˆà ifˆàacos(self.x/m)>pi/2ˆàorˆà(acos(self.x/m)>pi/2ˆàandˆàasin(self.y/m)<0):
ˆà ˆà ˆà ˆà ˆà ˆà returnˆàatan(self.y/self.x)-pi
ˆà ˆà ˆà ˆà else:
ˆà ˆà ˆà ˆà ˆà ˆà returnˆàatan(self.y/self.x)
ˆà ˆà defˆàangleToCoord(self,angle):
ˆà ˆà ˆà ˆà magn=abs(self)
ˆà ˆà ˆà ˆà self.x=magn*cos(angle)
ˆà ˆà ˆà ˆà self.y=magn*sin(angle)
ˆà ˆà ˆà ˆà returnˆàself
ˆà ˆà defˆàmagnitToCoord(self,m):
ˆà ˆà ˆà ˆà ang=self.angle()
ˆà ˆà ˆà ˆà self.x=m*cos(ang)
ˆà ˆà ˆà ˆà self.y=m*sin(ang)
ˆà ˆà ˆà ˆà returnˆàself

classˆàBall(object):
ˆà ˆà defˆà__init__(self,position,velocity):
ˆà ˆà ˆà ˆà self.position=position
ˆà ˆà ˆà ˆà self.velocity=velocity
ˆà ˆà defˆàbounds(self):
ˆà ˆà ˆà ˆà ifˆàself.position.x>=595:
ˆà ˆà ˆà ˆà ˆà ˆà self.position.x=595
ˆà ˆà ˆà ˆà ˆà ˆà self.velocity=self.velocity.angleToCoord(pi-self.velocity.angle())
ˆà ˆà ˆà ˆà elifˆàself.position.x<=2:
ˆà ˆà ˆà ˆà ˆà ˆà self.position.x=2
ˆà ˆà ˆà ˆà ˆà ˆà self.velocity=self.velocity.angleToCoord(pi-self.velocity.angle())
ˆà ˆà ˆà ˆà ifˆàself.position.y>=595:
ˆà ˆà ˆà ˆà ˆà ˆà self.position.y=595
ˆà ˆà ˆà ˆà ˆà ˆà self.velocity=self.velocity.angleToCoord(-self.velocity.angle())
ˆà ˆà ˆà ˆà elifˆàself.position.y<=2:
ˆà ˆà ˆà ˆà ˆà ˆà self.position.y=2
ˆà ˆà ˆà ˆà ˆà ˆà self.velocity=self.velocity.angleToCoord(-self.velocity.angle())
ˆà ˆà defˆàcollision(self,other):
ˆà ˆà ˆà ˆà ifˆàabs(self.position.y-other.position.y)<70ˆàandˆàabs(self.position.x-other.position.x)<70:
ˆà ˆà ˆà ˆà ˆà ˆà self.velocity.angleToCoord(2*pi*random.random())
ˆà ˆà ˆà ˆà ˆà ˆà other.velocity.angleToCoord(2*pi*random.random())
root =ˆàTkinter.Tk()ˆà
c =ˆàTkinter.Canvas(ˆàroot,ˆàwidth =ˆà670,ˆàheight =ˆà670ˆà)
c.pack()

b1=Ball(Vector(440,100),Vector(5,7))
b2=Ball(Vector(40,90),Vector(-3,-1))
b3=Ball(Vector(100,250),Vector(0,5))
b4=Ball(Vector(150,250),Vector(-1,2))
ball1 =ˆàc.create_oval(0,0,0,0)
ball2 =ˆàc.create_oval(0,0,0,0)
ball3 =ˆàc.create_oval(0,0,0,0)
ball4 =ˆàc.create_oval(0,0,0,0)
whileˆàTrue:
ˆà ˆà time.sleep(ˆà0.02ˆà)
ˆà ˆà c.delete(ball1)
ˆà ˆà c.delete(ball2)
ˆà ˆà c.delete(ball3)
ˆà ˆà c.delete(ball4)
ˆà ˆà ball1=c.create_oval(b1.position.x,ˆàb1.position.y,ˆàb1.position.x+75,ˆàb1.position.y+75,fill="blue"ˆà)
ˆà ˆà ball2=c.create_oval(b2.position.x,ˆàb2.position.y,ˆàb2.position.x+75,ˆàb2.position.y+75,fill="red")
ˆà ˆà ball3=c.create_oval(b3.position.x,ˆàb3.position.y,ˆàb3.position.x+75,ˆàb3.position.y+75,fill="yellow")
ˆà ˆà ball4=c.create_oval(b4.position.x,ˆàb4.position.y,ˆàb4.position.x+75,ˆàb4.position.y+75,fill="green")
ˆà ˆà b1.bounds()
ˆà ˆà b2.bounds()
ˆà ˆà b3.bounds()
ˆà ˆà b4.bounds()
ˆà ˆà b1.collision(b2)
ˆà ˆà b2.collision(b3)
ˆà ˆà b1.collision(b3)
ˆà ˆà b1.collision(b4)
ˆà ˆà b3.collision(b4)
ˆà ˆà b2.collision(b4)
ˆà ˆà b1.position =ˆàb1.position+b1.velocity
ˆà ˆà b2.position =ˆàb2.position+b2.velocity
ˆà ˆà b3.position =ˆàb3.position+b3.velocity
ˆà ˆà b4.position =ˆàb4.position+b4.velocity
ˆà ˆà c.update()