-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
200 lines (155 loc) · 4.87 KB
/
main.py
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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
import pygame
pygame.init()
# Font that is used to render the text
font20 = pygame.font.Font('freesansbold.ttf', 20)
# RGB values of standard colors
BLACK = (0, 0, 0)
WHITE = (255, 255, 255)
GREEN = (0, 255, 0)
# Basic parameters of the screen
WIDTH, HEIGHT = 900, 600
screen = pygame.display.set_mode((WIDTH, HEIGHT))
pygame.display.set_caption("Pong")
clock = pygame.time.Clock()
FPS = 30
# Striker class
class Striker:
# Take the initial position, dimensions, speed and color of the object
def __init__(self, posx, posy, width, height, speed, color):
self.posx = posx
self.posy = posy
self.width = width
self.height = height
self.speed = speed
self.color = color
# Rect that is used to control the position and collision of the object
self.geekRect = pygame.Rect(posx, posy, width, height)
# Object that is blit on the screen
self.geek = pygame.draw.rect(screen, self.color, self.geekRect)
# Used to display the object on the screen
def display(self):
self.geek = pygame.draw.rect(screen, self.color, self.geekRect)
def update(self, yFac):
self.posy = self.posy + self.speed*yFac
# Restricting the striker to be below the top surface of the screen
if self.posy <= 0:
self.posy = 0
# Restricting the striker to be above the bottom surface of the screen
elif self.posy + self.height >= HEIGHT:
self.posy = HEIGHT-self.height
# Updating the rect with the new values
self.geekRect = (self.posx, self.posy, self.width, self.height)
def displayScore(self, text, score, x, y, color):
text = font20.render(text+str(score), True, color)
textRect = text.get_rect()
textRect.center = (x, y)
screen.blit(text, textRect)
def getRect(self):
return self.geekRect
# Ball class
class Ball:
def __init__(self, posx, posy, radius, speed, color):
self.posx = posx
self.posy = posy
self.radius = radius
self.speed = speed
self.color = color
self.xFac = 1
self.yFac = -1
self.ball = pygame.draw.circle(
screen, self.color, (self.posx, self.posy), self.radius)
self.firstTime = 1
def display(self):
self.ball = pygame.draw.circle(
screen, self.color, (self.posx, self.posy), self.radius)
def update(self):
self.posx += self.speed*self.xFac
self.posy += self.speed*self.yFac
# If the ball hits the top or bottom surfaces,
# then the sign of yFac is changed and
# it results in a reflection
if self.posy <= 0 or self.posy >= HEIGHT:
self.yFac *= -1
if self.posx <= 0 and self.firstTime:
self.firstTime = 0
return 1
elif self.posx >= WIDTH and self.firstTime:
self.firstTime = 0
return -1
else:
return 0
def reset(self):
self.posx = WIDTH//2
self.posy = HEIGHT//2
self.xFac *= -1
self.firstTime = 1
# Used to reflect the ball along the X-axis
def hit(self):
self.xFac *= -1
def getRect(self):
return self.ball
# Game Manager
def main():
running = True
# Defining the objects
geek1 = Striker(20, 0, 10, 100, 10, GREEN)
geek2 = Striker(WIDTH-30, 0, 10, 100, 10, GREEN)
ball = Ball(WIDTH//2, HEIGHT//2, 7, 7, WHITE)
listOfGeeks = [geek1, geek2]
# Initial parameters of the players
geek1Score, geek2Score = 0, 0
geek1YFac, geek2YFac = 0, 0
while running:
screen.fill(BLACK)
# Event handling
for event in pygame.event.get():
if event.type == pygame.QUIT:
running = False
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_UP:
geek2YFac = -1
if event.key == pygame.K_DOWN:
geek2YFac = 1
if event.key == pygame.K_w:
geek1YFac = -1
if event.key == pygame.K_s:
geek1YFac = 1
if event.type == pygame.KEYUP:
if event.key == pygame.K_UP or event.key == pygame.K_DOWN:
geek2YFac = 0
if event.key == pygame.K_w or event.key == pygame.K_s:
geek1YFac = 0
# Collision detection
for geek in listOfGeeks:
if pygame.Rect.colliderect(ball.getRect(), geek.getRect()):
ball.hit()
# Updating the objects
geek1.update(geek1YFac)
geek2.update(geek2YFac)
point = ball.update()
# -1 -> Geek_1 has scored
# +1 -> Geek_2 has scored
# 0 -> None of them scored
if point == -1:
geek1Score += 1
elif point == 1:
geek2Score += 1
# Someone has scored
# a point and the ball is out of bounds.
# So, we reset it's position
if point:
ball.reset()
# Displaying the objects on the screen
geek1.display()
geek2.display()
ball.display()
# Displaying the scores of the players
geek1.displayScore("Geek_1 : ",
geek1Score, 100, 20, WHITE)
geek2.displayScore("Geek_2 : ",
geek2Score, WIDTH-100, 20, WHITE)
pygame.display.update()
clock.tick(FPS)
if __name__ == "__main__":
main()
pygame.quit()