394 lines
13 KiB
Python
394 lines
13 KiB
Python
|
'''
|
||
|
|
|
||
|
|
Game: Angry Birds
|
||
|
|
File: physics_engine.py
|
||
|
|
|
||
|
|
Contents: Class Vector
|
||
|
|
Class PIG
|
||
|
|
Class BIRD
|
||
|
|
Class BLOCK
|
||
|
|
Class SLINGSHOT
|
||
|
|
func collision_handler, block_collision_handler
|
||
|
|
|
||
|
|
Requirements: Pygame, sys, math, random
|
||
|
|
|
||
|
|
By: Jatin Kumar Mandav
|
||
|
|
|
||
|
|
Blog: https://www.jatinmandav.wordpress.com
|
||
|
|
Twitter: @jatinmandav
|
||
|
|
YouTube: https://www.youtube.com/mandav
|
||
|
|
|
||
|
|
'''
|
||
|
|
import pygame
|
||
|
|
import sys
|
||
|
|
from math import *
|
||
|
|
import random
|
||
|
|
|
||
|
|
pygame.init()
|
||
|
|
width = None
|
||
|
|
height = None
|
||
|
|
display = None
|
||
|
|
ground = 50
|
||
|
|
clock = pygame.time.Clock()
|
||
|
|
|
||
|
|
def init(screen):
|
||
|
|
global width, height, display
|
||
|
|
display = screen
|
||
|
|
(width, height) = display.get_rect().size
|
||
|
|
height -= ground
|
||
|
|
|
||
|
|
class Vector:
|
||
|
|
def __init__(self, magnitude=0, angle=radians(0)):
|
||
|
|
self.magnitude = magnitude
|
||
|
|
self.angle = angle
|
||
|
|
|
||
|
|
def add_vectors(vector1, vector2):
|
||
|
|
x = sin(vector1.angle)*vector1.magnitude + sin(vector2.angle)*vector2.magnitude
|
||
|
|
y = cos(vector1.angle)*vector1.magnitude + cos(vector2.angle)*vector2.magnitude
|
||
|
|
|
||
|
|
new_angle = 0.5*pi - atan2(y, x)
|
||
|
|
new_magnitude = hypot(x, y)
|
||
|
|
|
||
|
|
new_vector = Vector(new_magnitude, new_angle)
|
||
|
|
return new_vector
|
||
|
|
|
||
|
|
gravity = Vector(0.2, pi)
|
||
|
|
inverse_friction = 0.99
|
||
|
|
elasticity = 0.8
|
||
|
|
block_elasticity = 0.7
|
||
|
|
|
||
|
|
class Pig:
|
||
|
|
def __init__(self, x, y, r, v=None, type="PIG", loaded = False, color=(255, 255, 255)):
|
||
|
|
self.x = x
|
||
|
|
self.y = y
|
||
|
|
self.r = r
|
||
|
|
if v == None:
|
||
|
|
self.velocity = Vector()
|
||
|
|
else:
|
||
|
|
self.velocity = v
|
||
|
|
|
||
|
|
self.pig1_image = pygame.image.load("Images/pig1.png")
|
||
|
|
self.pig2_image = pygame.image.load("Images/pig3.png")
|
||
|
|
|
||
|
|
self.pig_dead = pygame.image.load("Images/pig_damaged.png")
|
||
|
|
|
||
|
|
self.bird_image = pygame.image.load("Images/bird.png")
|
||
|
|
|
||
|
|
if type == "PIG":
|
||
|
|
self.image = random.choice([self.pig1_image, self.pig2_image])
|
||
|
|
else:
|
||
|
|
self.image = self.bird_image
|
||
|
|
|
||
|
|
self.type = type
|
||
|
|
self.color = color
|
||
|
|
self.loaded = loaded
|
||
|
|
self.path = []
|
||
|
|
self.count = 0
|
||
|
|
self.animate_count = 0
|
||
|
|
self.isDead = False
|
||
|
|
|
||
|
|
def draw(self):
|
||
|
|
self.animate_count += 1
|
||
|
|
|
||
|
|
if self.type == "BIRD" and not self.loaded:
|
||
|
|
for point in self.path:
|
||
|
|
pygame.draw.ellipse(display, self.color, (point[0], point[1], 3, 3), 1)
|
||
|
|
|
||
|
|
if (self.type == "PIG") and (not self.animate_count%20) and (not self.isDead):
|
||
|
|
self.image = random.choice([self.pig1_image, self.pig2_image])
|
||
|
|
|
||
|
|
display.blit(self.image, (self.x - self.r, self.y - self.r))
|
||
|
|
|
||
|
|
|
||
|
|
def dead(self):
|
||
|
|
self.isDead = True
|
||
|
|
self.image = self.pig_dead
|
||
|
|
|
||
|
|
def move(self):
|
||
|
|
self.velocity = add_vectors(self.velocity, gravity)
|
||
|
|
|
||
|
|
self.x += self.velocity.magnitude*sin(self.velocity.angle)
|
||
|
|
self.y -= self.velocity.magnitude*cos(self.velocity.angle)
|
||
|
|
|
||
|
|
self.velocity.magnitude *= inverse_friction
|
||
|
|
|
||
|
|
if self.x > width - self.r:
|
||
|
|
self.x = 2*(width - self.r) - self.x
|
||
|
|
self.velocity.angle *= -1
|
||
|
|
self.velocity.magnitude *= elasticity
|
||
|
|
elif self.x < self.r:
|
||
|
|
self.x = 2*self.r - self.x
|
||
|
|
self.velocity.angle *= -1
|
||
|
|
self.velocity.magnitude *= elasticity
|
||
|
|
|
||
|
|
if self.y > height - self.r:
|
||
|
|
self.y = 2*(height - self.r) - self.y
|
||
|
|
self.velocity.angle = pi - self.velocity.angle
|
||
|
|
self.velocity.magnitude *= elasticity
|
||
|
|
elif self.y < self.r:
|
||
|
|
self.y = 2*self.r - self.y
|
||
|
|
self.velocity.angle = pi - self.velocity.angle
|
||
|
|
self.velocity.magnitude *= elasticity
|
||
|
|
|
||
|
|
self.count += 1
|
||
|
|
if self.count%1 == 0:
|
||
|
|
self.path.append((self.x, self.y))
|
||
|
|
|
||
|
|
class Bird(Pig):
|
||
|
|
def load(self, slingshot):
|
||
|
|
self.x = slingshot.x
|
||
|
|
self.y = slingshot.y
|
||
|
|
self.loaded = True
|
||
|
|
|
||
|
|
def mouse_selected(self):
|
||
|
|
pos = pygame.mouse.get_pos()
|
||
|
|
dx = pos[0] - self.x
|
||
|
|
dy = pos[1] - self.y
|
||
|
|
dist = hypot(dy, dx)
|
||
|
|
if dist < self.r:
|
||
|
|
return True
|
||
|
|
|
||
|
|
return False
|
||
|
|
|
||
|
|
def reposition(self, slingshot, mouse_click):
|
||
|
|
pos = pygame.mouse.get_pos()
|
||
|
|
if self.mouse_selected():
|
||
|
|
self.x = pos[0]
|
||
|
|
self.y = pos[1]
|
||
|
|
|
||
|
|
dx = slingshot.x - self.x
|
||
|
|
dy = slingshot.y - self.y
|
||
|
|
self.velocity.magnitude = int(hypot(dx, dy)/2)
|
||
|
|
if self.velocity.magnitude > 80:
|
||
|
|
self.velocity.magnitude = 80
|
||
|
|
self.velocity.angle = pi/2 + atan2(dy, dx)
|
||
|
|
|
||
|
|
def unload(self):
|
||
|
|
self.loaded = False
|
||
|
|
|
||
|
|
def project_path(self):
|
||
|
|
if self.loaded:
|
||
|
|
path = []
|
||
|
|
ball = Pig(self.x, self.y, self.r, self.velocity, self.type)
|
||
|
|
for i in range(30):
|
||
|
|
ball.move()
|
||
|
|
if i%5 == 0:
|
||
|
|
path.append((ball.x, ball.y))
|
||
|
|
|
||
|
|
for point in path:
|
||
|
|
pygame.draw.ellipse(display, self.color, (point[0], point[1], 2, 2))
|
||
|
|
|
||
|
|
|
||
|
|
class Block:
|
||
|
|
def __init__(self, x, y, r, v=None, color=( 120, 40, 31 ), colorBoundary = ( 28, 40, 51 )):
|
||
|
|
self.r = 50
|
||
|
|
self.w = 100
|
||
|
|
self.h = 100
|
||
|
|
|
||
|
|
self.x = x
|
||
|
|
self.y = y
|
||
|
|
|
||
|
|
self.block_image = pygame.image.load("Images/block1.png")
|
||
|
|
self.block_destroyed_image = pygame.image.load("Images/block_destroyed1.png")
|
||
|
|
|
||
|
|
self.image = self.block_image
|
||
|
|
|
||
|
|
if v == None:
|
||
|
|
self.velocity = Vector()
|
||
|
|
else:
|
||
|
|
self.velocity = v
|
||
|
|
|
||
|
|
self.color = color
|
||
|
|
self.colorDestroyed = ( 100, 30, 22 )
|
||
|
|
self.colorBoundary = colorBoundary
|
||
|
|
self.rotateAngle = radians(0)
|
||
|
|
self.anchor = (self.r/2, self.r/2)
|
||
|
|
|
||
|
|
self.isDestroyed = False
|
||
|
|
|
||
|
|
def rotate(self, coord, angle, anchor=(0, 0)):
|
||
|
|
corr = 0
|
||
|
|
return ((coord[0] - anchor[0])*cos(angle + radians(corr)) - (coord[1] - anchor[1])*sin(angle + radians(corr)),
|
||
|
|
(coord[0] - anchor[0])*sin(angle + radians(corr)) + (coord[1] - anchor[1])*cos(angle + radians(corr)))
|
||
|
|
|
||
|
|
def translate(self, coord):
|
||
|
|
return [coord[0] + self.x, coord[1] + self.y]
|
||
|
|
|
||
|
|
def draw(self):
|
||
|
|
pygame.transform.rotate(self.image, self.rotateAngle)
|
||
|
|
display.blit(self.image, (self.x - self.w/2, self.y))
|
||
|
|
|
||
|
|
def destroy(self):
|
||
|
|
self.isDestroyed = True
|
||
|
|
self.image = self.block_destroyed_image
|
||
|
|
|
||
|
|
def move(self):
|
||
|
|
self.velocity = add_vectors(self.velocity, gravity)
|
||
|
|
|
||
|
|
self.x += self.velocity.magnitude*sin(self.velocity.angle)
|
||
|
|
self.y -= self.velocity.magnitude*cos(self.velocity.angle)
|
||
|
|
|
||
|
|
self.velocity.magnitude *= inverse_friction
|
||
|
|
|
||
|
|
if self.x > width - self.w:
|
||
|
|
self.x = 2*(width - self.w) - self.x
|
||
|
|
self.velocity.angle *= -1
|
||
|
|
self.rotateAngle = - self.velocity.angle
|
||
|
|
self.velocity.magnitude *= block_elasticity
|
||
|
|
elif self.x < self.w:
|
||
|
|
self.x = 2*self.w - self.x
|
||
|
|
self.velocity.angle *= -1
|
||
|
|
self.rotateAngle = - self.velocity.angle
|
||
|
|
self.velocity.magnitude *= block_elasticity
|
||
|
|
|
||
|
|
if self.y > height - self.h:
|
||
|
|
self.y = 2*(height - self.h) - self.y
|
||
|
|
self.velocity.angle = pi - self.velocity.angle
|
||
|
|
self.rotateAngle = pi - self.velocity.angle
|
||
|
|
self.velocity.magnitude *= block_elasticity
|
||
|
|
elif self.y < self.h:
|
||
|
|
self.y = 2*self.h - self.y
|
||
|
|
self.velocity.angle = pi - self.velocity.angle
|
||
|
|
self.rotateAngle = pi - self.velocity.angle
|
||
|
|
self.velocity.magnitude *= block_elasticity
|
||
|
|
|
||
|
|
|
||
|
|
class Slingshot:
|
||
|
|
def __init__(self, x, y, w, h, color=( 66, 73, 73 )):
|
||
|
|
self.x = x
|
||
|
|
self.y = y
|
||
|
|
self.w = w
|
||
|
|
self.h = h
|
||
|
|
self.color = color
|
||
|
|
|
||
|
|
def rotate(self, coord, angle, anchor=(0, 0)):
|
||
|
|
corr = 0
|
||
|
|
return ((coord[0] - anchor[0])*cos(angle + radians(corr)) - (coord[1] - anchor[1])*sin(angle + radians(corr)),
|
||
|
|
(coord[0] - anchor[0])*sin(angle + radians(corr)) + (coord[1] - anchor[1])*cos(angle + radians(corr)))
|
||
|
|
|
||
|
|
def translate(self, coord):
|
||
|
|
return [coord[0] + self.x, coord[1] + self.y]
|
||
|
|
|
||
|
|
def draw(self, loaded=None):
|
||
|
|
pygame.draw.rect(display, self.color, (self.x, self.y + self.h*1/3, self.w, self.h*2/3))
|
||
|
|
|
||
|
|
if (not loaded == None) and loaded.loaded:
|
||
|
|
pygame.draw.line(display, ( 100, 30, 22 ), (self.x - self.w/4 + self.w/4, self.y + self.h/6), (loaded.x, loaded.y + loaded.r/2), 10)
|
||
|
|
pygame.draw.line(display, ( 100, 30, 22 ), (self.x + self.w, self.y + self.h/6), (loaded.x + loaded.r, loaded.y + loaded.r/2), 10)
|
||
|
|
|
||
|
|
pygame.draw.rect(display, self.color, (self.x - self.w/4, self.y, self.w/2, self.h/3), 5)
|
||
|
|
pygame.draw.rect(display, self.color, (self.x + self.w - self.w/4, self.y, self.w/2, self.h/3), 5)
|
||
|
|
|
||
|
|
def collision_handler(b_1, b_2, type):
|
||
|
|
collision = False
|
||
|
|
if type == "BALL":
|
||
|
|
dx = b_1.x - b_2.x
|
||
|
|
dy = b_1.y - b_2.y
|
||
|
|
|
||
|
|
dist = hypot(dx, dy)
|
||
|
|
if dist < b_1.r + b_2.r:
|
||
|
|
tangent = atan2(dy, dx)
|
||
|
|
angle = 0.5*pi + tangent
|
||
|
|
|
||
|
|
angle1 = 2*tangent - b_1.velocity.angle
|
||
|
|
angle2 = 2*tangent - b_2.velocity.angle
|
||
|
|
|
||
|
|
magnitude1 = b_2.velocity.magnitude
|
||
|
|
magnitude2 = b_1.velocity.magnitude
|
||
|
|
|
||
|
|
b_1.velocity = Vector(magnitude1, angle1)
|
||
|
|
b_2.velocity = Vector(magnitude2, angle2)
|
||
|
|
|
||
|
|
b_1.velocity.magnitude *= elasticity
|
||
|
|
b_2.velocity.magnitude *= elasticity
|
||
|
|
|
||
|
|
overlap = 0.5*(b_1.r + b_2.r - dist + 1)
|
||
|
|
b_1.x += sin(angle)*overlap
|
||
|
|
b_1.y -= cos(angle)*overlap
|
||
|
|
b_2.x -= sin(angle)*overlap
|
||
|
|
b_2.y += cos(angle)*overlap
|
||
|
|
collision = True
|
||
|
|
#print(collision)
|
||
|
|
|
||
|
|
#print(collision)
|
||
|
|
return b_1, b_2, collision
|
||
|
|
elif type == "BALL_N_BLOCK":
|
||
|
|
dx = b_1.x - b_2.x
|
||
|
|
dy = b_1.y - b_2.y
|
||
|
|
|
||
|
|
dist = hypot(dx, dy)
|
||
|
|
if dist < b_1.r + b_2.w:
|
||
|
|
tangent = atan2(dy, dx)
|
||
|
|
angle = 0.5*pi + tangent
|
||
|
|
|
||
|
|
angle1 = 2*tangent - b_1.velocity.angle
|
||
|
|
angle2 = 2*tangent - b_2.velocity.angle
|
||
|
|
|
||
|
|
magnitude1 = b_2.velocity.magnitude
|
||
|
|
magnitude2 = b_1.velocity.magnitude
|
||
|
|
|
||
|
|
b_1.velocity = Vector(magnitude1, angle1)
|
||
|
|
b_2.velocity = Vector(magnitude2, angle2)
|
||
|
|
|
||
|
|
b_1.velocity.magnitude *= elasticity
|
||
|
|
b_2.velocity.magnitude *= block_elasticity
|
||
|
|
|
||
|
|
overlap = 0.5*(b_1.r + b_2.w - dist + 1)
|
||
|
|
b_1.x += sin(angle)*overlap
|
||
|
|
b_1.y -= cos(angle)*overlap
|
||
|
|
b_2.x -= sin(angle)*overlap
|
||
|
|
b_2.y += cos(angle)*overlap
|
||
|
|
collision = True
|
||
|
|
|
||
|
|
return b_1, b_2, collision
|
||
|
|
|
||
|
|
def block_collision_handler(block, block2):
|
||
|
|
collision = False
|
||
|
|
if (block.y + block.h > block2.y) and (block.y < block2.y + block2.h):
|
||
|
|
if (block.x < block2.x + block2.w) and (block.x + block.w > block2.x + block2.w):
|
||
|
|
block.x = 2*(block2.x + block2.w) - block.x
|
||
|
|
block.velocity.angle = - block.velocity.angle
|
||
|
|
block.rotateAngle = - block.velocity.angle
|
||
|
|
block.velocity.magnitude *= block_elasticity
|
||
|
|
|
||
|
|
block2.velocity.angle = - block2.velocity.angle
|
||
|
|
block2.rotateAngle = - block2.velocity.angle
|
||
|
|
block2.velocity.magnitude *= block_elasticity
|
||
|
|
collision = True
|
||
|
|
|
||
|
|
elif block.x + block.w > block2.x and (block.x < block2.x):
|
||
|
|
block.x = 2*(block2.x - block.w) - block.x
|
||
|
|
block.velocity.angle = - block.velocity.angle
|
||
|
|
block.rotateAngle = - block.velocity.angle
|
||
|
|
block.velocity.magnitude *= block_elasticity
|
||
|
|
|
||
|
|
block2.velocity.angle = - block2.velocity.angle
|
||
|
|
block2.rotateAngle = - block2.velocity.angle
|
||
|
|
block2.velocity.magnitude *= block_elasticity
|
||
|
|
collision = True
|
||
|
|
|
||
|
|
if (block.x + block.w > block2.x) and (block.x < block2.x + block2.w):
|
||
|
|
if block.y + block.h > block2.y and block.y < block2.y:
|
||
|
|
block.y = 2*(block2.y - block.h) - block.y
|
||
|
|
block.velocity.angle = pi - block.velocity.angle
|
||
|
|
block.rotateAngle = pi - block.velocity.angle
|
||
|
|
block.velocity.magnitude *= block_elasticity
|
||
|
|
|
||
|
|
block2.velocity.angle = pi - block2.velocity.angle
|
||
|
|
block2.rotateAngle = pi - block2.velocity.angle
|
||
|
|
block2.velocity.magnitude *= block_elasticity
|
||
|
|
collision = True
|
||
|
|
|
||
|
|
elif (block.y < block2.y + block2.h) and (block.y + block.h > block2.y + block2.h):
|
||
|
|
block.y = 2*(block2.y + block2.h) - block.y
|
||
|
|
block.velocity.angle = pi - block.velocity.angle
|
||
|
|
block.rotateAngle = pi - block.velocity.angle
|
||
|
|
block.velocity.magnitude *= block_elasticity
|
||
|
|
|
||
|
|
block2.velocity.angle = pi - block2.velocity.angle
|
||
|
|
block2.rotateAngle = pi - block2.velocity.angle
|
||
|
|
block2.velocity.magnitude *= block_elasticity
|
||
|
|
collision = True
|
||
|
|
|
||
|
|
return block, block2, collision
|