1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
from math import copysign
import pygame
from collision import detectTopCollision, handleTopCollision, \
detectBottomCollision, handleBottomCollision, detectLeftCollision, handleLeftCollision, handleRightCollision, \
detectRightCollision
from colours import *
class Particle:
speed_limit = 3
def __init__(self, initial_position, initial_velocity, initial_acceleration, radius, mass, collision_layer):
self.position = pygame.Vector2(initial_position)
self.velocity = pygame.Vector2(initial_velocity)
self.acceleration = pygame.Vector2(initial_acceleration)
self.radius = radius
self.mass = mass
self.left = self.position.x - self.radius
self.right = self.position.x + self.radius
self.top = self.position.y - self.radius
self.bottom = self.position.y + self.radius
self.collision_layer = collision_layer
self.colour = RED
self.collided_with_wall = {"top": False, "bottom": False, "left": False, "right": False}
def get_next_frame(self, position, velocity, delta):
vel = pygame.Vector2()
pos = pygame.Vector2()
vel.x = min(self.speed_limit, abs(velocity.x + self.acceleration.x * delta)) \
* copysign(1, velocity.x + self.acceleration.x * delta)
vel.y = min(self.speed_limit, abs(velocity.y + self.acceleration.y * delta)) \
* copysign(1, velocity.y + self.acceleration.y * delta)
pos.x = position.x + velocity.x * delta
pos.y = position.y + velocity.y * delta
return pos, vel
def update(self, delta):
self.position, self.velocity = self.get_next_frame(self.position, self.velocity, delta)
self.left = self.position.x - self.radius
self.right = self.position.x + self.radius
self.top = self.position.y - self.radius
self.bottom = self.position.y + self.radius
if not self.collided_with_wall["top"] and detectTopCollision(self, self.collision_layer[0]):
handleTopCollision(self, self.collision_layer[0])
self.collided_with_wall["top"] = True
if not detectTopCollision(self, self.collision_layer[0]):
self.collided_with_wall["top"] = False
if not self.collided_with_wall["bottom"] and detectBottomCollision(self, self.collision_layer[0]):
handleBottomCollision(self, self.collision_layer[0])
self.collided_with_wall["bottom"] = True
if not detectBottomCollision(self, self.collision_layer[0]):
self.collided_with_wall["bottom"] = False
if not self.collided_with_wall["left"] and detectLeftCollision(self, self.collision_layer[0]):
handleLeftCollision(self, self.collision_layer[0])
self.collided_with_wall["left"] = True
if not detectLeftCollision(self, self.collision_layer[0]):
self.collided_with_wall["left"] = False
if not self.collided_with_wall["right"] and detectRightCollision(self, self.collision_layer[0]):
handleRightCollision(self, self.collision_layer[0])
self.collided_with_wall["right"] = True
if not detectRightCollision(self, self.collision_layer[0]):
self.collided_with_wall["right"] = False
def draw(self, surf):
pygame.draw.circle(surf, self.colour, self.position, self.radius)
|
