Mosi: Multipendel

This commit is contained in:
madmaurice 2016-02-16 18:03:41 +01:00
parent 375e5f9662
commit 2a84801ba8

102
pendel.py Normal file
View file

@ -0,0 +1,102 @@
#!/usr/bin/env python
from game import *
from random import randint
class Vector(object):
def __init__(self, x, y):
self.x = float(x)
self.y = float(y)
def __add__(self, other):
return Vector( self.x + other.x, self.y + other.y )
def __neg__(self):
return Vector( -self.x, -self.y )
def __mul__(self, other):
return Vector( self.x * other, self.y * other )
def __rmul__(self, other):
return self.__mul__(other)
def __div__(self, other):
return Vector( self.x / other, self.y / other )
def __truediv__(self, other):
return self.__div__( other )
def __sub__(self, other):
return self.__add__(other.__neg__())
def len(self):
return math.sqrt( self.x * self.x + self.y * self.y )
def astuple(self):
return ( int(self.x) , int(self.y) )
def __repr__(self):
return "( %.2f, %.2f )" % (self.x, self.y)
class PendelSim(Game):
def __init__(self):
super(PendelSim,self).__init__(1000,600, 60)
self.init_data()
def init_data(self):
self.staff_lengths = staff_lengths = [ randint(100,200) for i in range(3) ]
fixed_joint = Vector( self._width/2, self._height/5 )
joint_a = fixed_joint - Vector(staff_lengths[0],0)
joint_b = joint_a - Vector(0,-staff_lengths[1])
joint_c = joint_b - Vector(-staff_lengths[2],0)
self.joints = [fixed_joint, joint_a, joint_b, joint_c ]
self.joint_speed = [ Vector(0,0), Vector(0,0), Vector(0,0), Vector(0,0) ]
def on_update(self, dtime):
new_joints = self.joints[:]
for i in range(1,4):
new_joints[i] += (self.joint_speed[i] + Vector(0,40)) * dtime
for i in range(0,4-1):
relvec = new_joints[i+1] - new_joints[i]
relvec = (relvec * self.staff_lengths[i]) / relvec.len()
new_joints[i+1] = new_joints[i] + relvec
for i in range(0,4):
self.joint_speed[i] = (new_joints[i] - self.joints[i]) / dtime
self.joints = new_joints
def on_render(self, surface):
BLACK = (0,0,0)
RED = (255,0,0)
GREEN = (0,255,0)
WHITE = (255,255,255)
BLUE = (0,0,255)
staff_colors = [ RED, GREEN, BLUE ]
surface.fill(BLACK)
for i in range(4-1):
pygame.draw.line(surface, staff_colors[i%3], self.joints[i].astuple(), self.joints[i+1].astuple(), 3)
for i in range(4):
pygame.draw.circle(surface, WHITE, self.joints[i].astuple(), 4)
pygame.display.flip()
def on_event(self, event):
if event.type == MOUSEBUTTONDOWN:
self.init_data()
else:
super(PendelSim, self).on_event(event)
def main():
PendelSim().on_execute()
if __name__ == '__main__':
main()