PhysiFun/Sprite.py at master · rohilvagarwal/PhysiFun · GitHub

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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
from ProjectConstants import *
import random


class Sprite:
	def __init__(self, image, centerX, centerY, velocity, angle, hitboxShape: str, hitboxScale):
		self.image = image.convert_alpha()
		self.centerX = centerX
		self.centerY = centerY
		self.velocity = velocity
		self.angle = angle
		self.xVelocity = math.cos(math.radians(self.angle)) * self.velocity
		self.yVelocity = -math.sin(math.radians(self.angle)) * self.velocity

		#define hitbox dimensions
		self.hitboxShape = hitboxShape
		if self.hitboxShape == 'rectangle':
			self.width = self.image.get_width() * hitboxScale
			self.height = self.image.get_height() * hitboxScale
		elif self.hitboxShape == 'circle':
			self.radius = max(self.image.get_width(), self.image.get_height()) / 2 * hitboxScale

	def get_centerX(self):
		return self.centerX

	def get_centerY(self):
		return self.centerY

	def set_centerX_and_centerY(self, centerX, centerY):
		self.centerX = centerX
		self.centerY = centerY

	def get_hitbox_shape(self):
		return self.hitboxShape

	def set_velocity_and_angle(self, velocity, angle):
		self.velocity = velocity
		self.angle = angle
		self.xVelocity = math.cos(math.radians(self.angle)) * self.velocity
		self.yVelocity = -math.sin(math.radians(self.angle)) * self.velocity

	def get_next_frame(self):
		self.centerX += self.xVelocity / FPS
		self.centerY += self.yVelocity / FPS
		self.yVelocity += 3

	def draw_static(self, screen):
		self.get_next_frame()
		blit_center(screen, self.image, (self.centerX, self.centerY))

	def get_hitbox(self):
		if self.hitboxShape == 'rectangle':
			# Calculate hitbox coordinates for a rectangle
			left = self.centerX - self.width / 2
			top = self.centerY - self.height / 2
			right = self.centerX + self.width / 2
			bottom = self.centerY + self.height / 2
			return left, top, right, bottom
		elif self.hitboxShape == 'circle':
			# Return hitbox coordinates for a circle
			return self.centerX, self.centerY, self.radius

	def collides_with(self, other_sprite: "Sprite"):
		self_hitbox = self.get_hitbox()
		other_hitbox = other_sprite.get_hitbox()

		if self.hitboxShape == 'rectangle' and other_sprite.get_hitbox_shape() == 'rectangle':
			# Check for collision between two rectangles
			if self_hitbox[2] >= other_hitbox[0] and self_hitbox[0] <= other_hitbox[2] and \
					self_hitbox[3] >= other_hitbox[1] and self_hitbox[1] <= other_hitbox[3]:
				return True
		elif self.hitboxShape == 'circle' and other_sprite.get_hitbox_shape() == 'circle':
			# Check for collision between two circles
			distance_squared = (self_hitbox[0] - other_hitbox[0]) ** 2 + (self_hitbox[1] - other_hitbox[1]) ** 2
			if distance_squared <= (self_hitbox[2] + other_hitbox[2]) ** 2:
				return True
		else:
			# Check for collision between a circle and a rectangle
			if self.hitboxShape == 'circle':
				circle_hitbox = self_hitbox
				rect_hitbox = other_hitbox
			else:
				circle_hitbox = other_hitbox
				rect_hitbox = self_hitbox

			closest_x = max(rect_hitbox[0], min(circle_hitbox[0], rect_hitbox[2]))
			closest_y = max(rect_hitbox[1], min(circle_hitbox[1], rect_hitbox[3]))

			distance_squared = (circle_hitbox[0] - closest_x) ** 2 + (circle_hitbox[1] - closest_y) ** 2
			if distance_squared <= circle_hitbox[2] ** 2:
				return True

		return False

	def randomly_move(self):
		# Generate random movement values
		move_x = random.choice([-20, 20])
		move_y = random.choice([-20, 20])

		# Update centerX and centerY with random movement
		self.centerX += move_x
		self.centerY += move_y

	def draw_hitbox(self, screen):
		hitbox = self.get_hitbox()
		if self.hitboxShape == 'rectangle':
			left, top, right, bottom = hitbox
			pygame.draw.rect(screen, (255, 0, 0), (left, top, right - left, bottom - top), 2)
		elif self.hitboxShape == 'circle':
			pygame.draw.circle(screen, RED, (int(hitbox[0]), int(hitbox[1])), int(hitbox[2]), 2)