I want to make the ball go into the basket to produce a score but I have no clue how to.I used pygame.draw.circle as the ball to project. The basket is an image loaded from disk and is embedded as a rectangle and moves from left to right and back again. I was also thinking about making the two objects collide somehow.
import math
import random
import pygame
#from functions import *
from pygame.locals import *
pygame.init()
SCREEN = WIDTH, HEIGHT = 800, 534
screen = pygame.display.set_mode((720, 700))
info = pygame.display.Info()
width = info.current_w
height = info.current_h
if width >= height:
win = pygame.display.set_mode(SCREEN, pygame.NOFRAME)
else:
win = pygame.display.set_mode(SCREEN, pygame.NOFRAME | pygame.SCALED | pygame.FULLSCREEN)
clock = pygame.time.Clock()
FPS = 60
BLACK = (18, 18, 18)
WHITE = (217, 217, 217)
RED = (252, 91, 122)
GREEN = (29, 161, 16)
BLUE = (78, 193, 246)
ORANGE = (252,76,2)
YELLOW = (254,221,0)
PURPLE = (155,38,182)
AQUA = (0,249,182)
COLORS = [RED, GREEN, BLUE, ORANGE, YELLOW, PURPLE]
font = pygame.font.SysFont('verdana', 12)
basket = pygame.image.load("basket.png")
rect = basket.get_rect()
origin = (60, 400)
radius = 250
u = 80
g = 9.8
platform_vel = 1
x = 100
y = 150
rect = Rect(x, y, 200, 50)
class Projectile(pygame.sprite.Sprite):
def __init__(self, u, theta):
super(Projectile, self).__init__()
self.u = u
self.theta = toRadian(abs(theta))
self.x, self.y = origin
self.color = random.choice(COLORS)
self.ch = 0
self.dx = 2
self.f = self.getTrajectory()
self.range = self.x + abs(self.getRange())
self.path = []
def timeOfFlight(self):
return round((2 * self.u * math.sin(self.theta)) / g, 2)
def getRange(self):
range_ = ((self.u ** 2) * 2 * math.sin(self.theta) * math.cos(self.theta)) / g
return round(range_, 2)
def getMaxHeight(self):
h = ((self.u ** 2) * (math.sin(self.theta)) ** 2) / (2 * g)
return round(h, 2)
def getTrajectory(self):
return round(g / (2 * (self.u ** 2) * (math.cos(self.theta) ** 2)), 4)
def getProjectilePos(self, x):
return x * math.tan(self.theta) - self.f * x ** 2
def update(self):
if self.x >= self.range:
self.dx = 0
self.x += self.dx
self.ch = self.getProjectilePos(self.x - origin[0])
self.path.append((self.x, self.y-abs(self.ch)))
self.path = self.path[-50:]
sprite1 = pygame.sprite.Sprite()
sprite1.image = pygame.Surface((80, 80), pygame.SRCALPHA)
pygame.draw.circle(sprite1.image, self.color, self.path[-1], 5)
sprite1.rect = pygame.Rect(*window.get_rect().center, 0, 0).inflate(40, 40)
sprite2 = pygame.sprite.Sprite()
sprite2.image = pygame.Surface((80, 89), pygame.SRCALPHA)
pygame.draw.circle(sprite2.image, (0, 0, 0), (40, 40), 40)
sprite2.rect = pygame.Rect(*window.get_rect().center, 0, 0).inflate(80, 80)
all_group = pygame.sprite.Group([sprite2, sprite1])
test_group = pygame.sprite.Group(sprite2)
pygame.draw.circle(win, WHITE, self.path[-1], 5, 1)
for pos in self.path[:-1:5]:
pygame.draw.circle(win, WHITE, pos, 1)
projectile_group = pygame.sprite.Group()
clicked = False
currentp = None
theta = -30
end = getPosOnCircumeference(theta, origin)
arct = toRadian(theta)
arcrect = pygame.Rect(origin[0]-30, origin[1]-30, 60, 60)
running = True
while running:
win.fill(GREEN)
if rect.left >= 400 or rect.left < 100:
platform_vel *= -1
rect.left += platform_vel
for event in pygame.event.get():
if event.type == pygame.QUIT:
running = False
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_ESCAPE or event.key == pygame.K_q:
running = False
if event.key == pygame.K_r:
projectile_group.empty()
currentp = None
if event.type == pygame.MOUSEBUTTONDOWN:
clicked = True
if event.type == pygame.MOUSEBUTTONUP:
clicked = False
pos = event.pos
theta = getAngle(pos, origin)
if -90 < theta <= 0:
projectile = Projectile(u, theta)
projectile_group.add(projectile)
currentp = projectile
if event.type == pygame.MOUSEMOTION:
if clicked:
pos = event.pos
theta = getAngle(pos, origin)
if -90 < theta <= 0:
end = getPosOnCircumeference(theta, origin)
arct = toRadian(theta)
sprite1.rect.center = pygame.mouse.get_pos()
collide = pygame.sprite.spritecollide(sprite1, test_group, False, pygame.sprite.collide_circle)
window.fill(0)
all_group.draw(window)
for s in collide:
pygame.draw.circle(window, (255, 255, 255), s.rect.center, s.rect.width // 2, 6)
pygame.display.flip()
screen.blit(basket, rect)
pygame.draw.line(win, WHITE, origin, (origin[0] + 250, origin[1]), 2)
pygame.draw.line(win, WHITE, origin, (origin[0], origin[1] - 250), 2)
pygame.draw.line(win, AQUA, origin, end, 2)
pygame.draw.circle(win, WHITE, origin, 3)
pygame.draw.arc(win, AQUA, arcrect, 0, -arct, 2)
projectile_group.update()
# Info *******************************************************************
title = font.render("Projectile Motion", True, WHITE)
fpstext = font.render(f"FPS : {int(clock.get_fps())}", True, WHITE)
thetatext = font.render(f"Angle : {int(abs(theta))}", True, WHITE)
degreetext = font.render(f"{int(abs(theta))}°", True, YELLOW)
win.blit(title, (80, 30))
win.blit(fpstext, (20, 400))
win.blit(thetatext, (20, 420))
win.blit(degreetext, (origin[0]+38, origin[1]-20))
if currentp:
veltext = font.render(f"Velocity : {currentp.u}m/s", True, WHITE)
timetext = font.render(f"Time : {currentp.timeOfFlight()}s", True, WHITE)
rangetext = font.render(f"Range : {currentp.getRange()}m", True, WHITE)
heighttext = font.render(f"Max Height : {currentp.getMaxHeight()}m", True, WHITE)
win.blit(veltext, (WIDTH-150, 400))
win.blit(timetext, (WIDTH-150, 420))
win.blit(rangetext, (WIDTH-150, 440))
win.blit(heighttext, (WIDTH-150, 460))
pygame.draw.rect(win, (0,0,0), (0, 0, WIDTH, HEIGHT), 5)
clock.tick(FPS)
pygame.display.update()
pygame.quit()
