-
Notifications
You must be signed in to change notification settings - Fork 1
/
pong-game.py
114 lines (96 loc) · 4 KB
/
pong-game.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
# Implementation of classic arcade game Pong
import simplegui
import random
# initialize globals - pos and vel encode vertical info for paddles
WIDTH = 600
HEIGHT = 400
BALL_RADIUS = 20
PAD_WIDTH = 8
PAD_HEIGHT = 80
HALF_PAD_WIDTH = PAD_WIDTH / 2
HALF_PAD_HEIGHT = PAD_HEIGHT / 2
LEFT = False
RIGHT = True
# initialize ball_pos and ball_vel for new bal in middle of table
# if direction is RIGHT, the ball's velocity is upper right, else upper left
def spawn_ball(direction):
global ball_pos, ball_vel # these are vectors stored as lists
ball_pos = [WIDTH / 2, HEIGHT / 2]
if direction:
ball_vel = [random.randrange(120, 240) / 60, -random.randrange(60, 180) / 60]
else:
ball_vel = [-random.randrange(120, 240) / 60, -random.randrange(60, 180) / 60]
# define event handlers
def new_game():
global paddle1_pos, paddle2_pos, paddle1_vel, paddle2_vel # these are numbers
global score1, score2 # these are ints
paddle1_pos = HEIGHT / 2 - HALF_PAD_HEIGHT
paddle2_pos = HEIGHT / 2 - HALF_PAD_HEIGHT
paddle1_vel = 0
paddle2_vel = 0
score1 = 0
score2 = 0
spawn_ball(random.choice([LEFT, RIGHT]))
def draw(canvas):
global score1, score2, paddle1_pos, paddle2_pos, ball_pos, ball_vel
# draw mid line and gutters
canvas.draw_line([WIDTH / 2, 0],[WIDTH / 2, HEIGHT], 1, "White")
canvas.draw_line([PAD_WIDTH, 0],[PAD_WIDTH, HEIGHT], 1, "White")
canvas.draw_line([WIDTH - PAD_WIDTH, 0],[WIDTH - PAD_WIDTH, HEIGHT], 1, "White")
# update ball
ball_pos[0] += ball_vel[0]
ball_pos[1] += ball_vel[1]
if ball_pos[1] + BALL_RADIUS >= HEIGHT or ball_pos[1] <= BALL_RADIUS:
ball_vel[1] = -ball_vel[1]
# draw ball
canvas.draw_circle(ball_pos, BALL_RADIUS, 2, "White")
# update paddle's vertical position, keep paddle on the screen
if paddle1_pos + paddle1_vel > 0 and paddle1_pos + paddle1_vel + PAD_HEIGHT < HEIGHT:
paddle1_pos += paddle1_vel
if paddle2_pos + paddle2_vel > 0 and paddle2_pos + paddle2_vel + PAD_HEIGHT < HEIGHT:
paddle2_pos += paddle2_vel
# draw paddles
canvas.draw_line([HALF_PAD_WIDTH, paddle1_pos], [HALF_PAD_WIDTH, paddle1_pos + PAD_HEIGHT], PAD_WIDTH, "Red")
canvas.draw_line([WIDTH - HALF_PAD_WIDTH, paddle2_pos], [WIDTH - HALF_PAD_WIDTH, paddle2_pos + PAD_HEIGHT], PAD_WIDTH, "Green")
# determine whether paddle and ball collide
if ball_pos[0] <= BALL_RADIUS + PAD_WIDTH:
if ball_pos[1] > paddle1_pos - BALL_RADIUS and ball_pos[1] < paddle1_pos + PAD_HEIGHT + BALL_RADIUS:
ball_vel[0] *= -1.1
ball_vel[1] *= 1.1
else:
spawn_ball(RIGHT)
score2 += 1
elif ball_pos[0] + BALL_RADIUS + PAD_WIDTH >= WIDTH:
if ball_pos[1] > paddle2_pos - BALL_RADIUS and ball_pos[1] < paddle2_pos + PAD_HEIGHT + BALL_RADIUS:
ball_vel[0] *= - 1.1
ball_vel[1] *= 1.1
else:
spawn_ball(LEFT)
score1 += 1
# draw scores
canvas.draw_text(str(score1) + " : "+ str(score2), [255, 100], 50, "White")
def keydown(key):
global paddle1_vel, paddle2_vel
if key == simplegui.KEY_MAP["w"]:
paddle1_vel = -5
elif key == simplegui.KEY_MAP["s"]:
paddle1_vel = 5
elif key == simplegui.KEY_MAP["up"]:
paddle2_vel = -5
elif key == simplegui.KEY_MAP["down"]:
paddle2_vel = 5
def keyup(key):
global paddle1_vel, paddle2_vel
if key == simplegui.KEY_MAP["w"] or key == simplegui.KEY_MAP["s"]:
paddle1_vel = 0
elif key == simplegui.KEY_MAP["up"] or key == simplegui.KEY_MAP["down"]:
paddle2_vel = 0
# create frame
frame = simplegui.create_frame("Pong", WIDTH, HEIGHT)
frame.set_draw_handler(draw)
frame.set_keydown_handler(keydown)
frame.set_keyup_handler(keyup)
button = frame.add_button("Restart", new_game, 100)
# start frame
new_game()
frame.start()