sketches/bouncing_ball_classes.py at master · hackingmath/sketches · 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
#Pygame Template

import pygame
import time
from random import randint, choice
from math import pi, sqrt, cos, sin, atan2

# define constants
BLACK = (0, 0, 0)
WHITE = (255, 255, 255)
RED = (255, 0, 0)
GREEN = (0, 255, 0)
BLUE = (0, 0, 255)
CYAN = (0, 255, 255)
VIOLET = (148, 0, 211)
YELLOW = (255,255,0)

colors = [BLACK,WHITE,RED,GREEN,BLUE,CYAN,VIOLET,YELLOW]

width,height = 600,600

class Ball(object):

    def __init__(self):
        self.xpos = randint(0,width)
        self.ypos = randint(0,height)
        self.xvel = randint(-5,5)
        self.yvel = randint(-5, 5)
        self.radius = randint(5,30)
        self.color = choice(colors)

    def update(self):
        self.xpos += self.xvel
        self.ypos += self.yvel

        if self.xpos >= 600 - self.radius:
            self.xpos = 600 - self.radius
            self.xvel *= (-1)
        if self.xpos <= self.radius:
            self.xpos = self.radius
            self.xvel *= (-1)
        if self.ypos >= 600 - self.radius:
            self.ypos = 600 - self.radius
            self.yvel *= (-1)
        if self.ypos <= self.radius:
            self.ypos = self.radius
            self.yvel *= (-1)
        pygame.draw.circle(screen, self.color, [self.xpos, self.ypos], 2 * self.radius)

# set up display
pygame.init()
screen = pygame.display.set_mode([width, height])

#in case you use fonts:
pygame.font.init()
myfont = pygame.font.SysFont('Consolas', 24)
scorefont = pygame.font.SysFont('Consolas', 72)

pygame.display.set_caption('Pygame Window') #add your own caption!
FPS = 60  # frames per second
clock = pygame.time.Clock()

counter = 0 #frame count

# loop until user clicks the close button
done = False

#Rendering the Game
ball_list = list()
for i in range(50):
    ball_list.append(Ball())

#display loop
while not done:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:  # if pygame window is closed by user
            done = True

    keys = pygame.key.get_pressed()
    if keys[pygame.K_LEFT]:
        pygame.key.set_repeat(1000)
        xvel -= 5
    if keys[pygame.K_RIGHT]:
        pygame.key.set_repeat(1000)
        xvel += 5

    if keys[pygame.K_p]:
        if FPS == 60:
            FPS = 300  #faster display
        else:
            FPS = 60
    # fill the screen with background color
    screen.fill(BLACK)

    #pygame.draw.polygon(screen, WHITE, [(100, 200), (xpos,ypos), (524, 307), (500, 200)], 0)
    #pygame.draw.rect(screen, WHITE, [xpos,ypos, 150,200])
    #pygame.draw.line(screen, GREEN, [100,100],[200,300],3)

    for b in ball_list:
        b.update()

    #print("xvel:",xvel)
    counter += 1

    pygame.display.update()

    clock.tick(FPS)
pygame.quit()