Insanely good

main
Rolands 2025-12-15 11:48:43 +02:00
parent 1b403c0391
commit e7c444292f
1 changed files with 226 additions and 127 deletions

349
main.py
View File

@ -2,101 +2,167 @@ import pygame
import random
import math
# -----------------------------
# Settings
# -----------------------------
SCREEN_WIDTH = 1280
SCREEN_HEIGHT = 720
FPS = 80
NUM_ENEMIES = 14
FPS = 60
# =========================
# DIFFICULTY SETTINGS
# =========================
difficulty = 2 # 1 = Easy, 2 = Medium, 3 = Hard
difficulty_flash_timer = 0
difficulty_cooldown_timer = 0 # 10-second cooldown
# =========================
# POLARITY SWITCH COOLDOWN
# =========================
polarity_cooldown = 0
POLARITY_COOLDOWN_TIME = int(FPS * 0.7)
polarity_flash_timer = 0
# -----------------------------
# Initialize Pygame
# -----------------------------
pygame.init()
pygame.mixer.init() # Initialize mixer for music
pygame.mixer.init()
screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
pygame.display.set_caption("Polarity Field")
clock = pygame.time.Clock()
font = pygame.font.SysFont(None, 36)
font = pygame.font.SysFont(None, 42)
# -----------------------------
# Load background music
# -----------------------------
pygame.mixer.music.load("sprites/music.ogg")
pygame.mixer.music.set_volume(0.5)
pygame.mixer.music.play(-1) # Loop indefinitely
pygame.mixer.music.play(-1)
# -----------------------------
# Load sprites AFTER display is ready
# -----------------------------
player_surf = pygame.image.load("sprites/orb_purple.png").convert_alpha()
player_surf = pygame.transform.scale(player_surf, (60, 60))
player_surf = pygame.transform.scale(player_surf, (50, 50))
player_rect = player_surf.get_rect(center=(SCREEN_WIDTH // 2, SCREEN_HEIGHT // 2))
player_polarity = 1
player_lives = 3
# =========================
# INITIAL ENEMIES BASED ON DIFFICULTY (easier)
# =========================
if difficulty == 1:
NUM_ENEMIES = 12
difficulty_text = "Easy"
difficulty_color = (0, 200, 0)
elif difficulty == 2:
NUM_ENEMIES = 22
difficulty_text = "Medium"
difficulty_color = (255, 200, 0)
elif difficulty == 3:
NUM_ENEMIES = 35
difficulty_text = "Hard"
difficulty_color = (255, 60, 60)
# =========================
# SPAWN ENEMIES LIST
# =========================
enemies = []
for _ in range(NUM_ENEMIES):
polarity = random.choice([1, -1])
if polarity == 1:
surf = pygame.image.load("sprites/orb_red.png").convert_alpha()
else:
surf = pygame.image.load("sprites/orb_blue.png").convert_alpha()
surf = pygame.transform.scale(surf, (50, 50))
# Spawn 500 pixels away from player
while True:
surf = pygame.image.load("sprites/orb_red.png" if polarity == 1 else "sprites/orb_blue.png").convert_alpha()
surf = pygame.transform.scale(surf, (40, 40))
side = random.choice(["top", "bottom", "left", "right"])
if side == "top":
x = random.randint(0, SCREEN_WIDTH)
y = -100
elif side == "bottom":
x = random.randint(0, SCREEN_WIDTH)
y = SCREEN_HEIGHT + 100
elif side == "left":
x = -100
y = random.randint(0, SCREEN_HEIGHT)
else:
x = SCREEN_WIDTH + 100
y = random.randint(0, SCREEN_HEIGHT)
dx = x - player_rect.centerx
dy = y - player_rect.centery
dist = math.hypot(dx, dy)
if dist >= 500:
break
rect = surf.get_rect(center=(x, y))
speed_x = 0 # stationary until game starts
speed_y = 0
enemies.append({
"surf": surf,
"rect": rect,
"polarity": polarity,
"speed_x": speed_x,
"speed_y": speed_y
"speed_x": random.uniform(-3.5, 3.5),
"speed_y": random.uniform(-3.5, 3.5)
})
# -----------------------------
# Score & cooldown
# -----------------------------
score = 0
hit_cooldown = 0
game_started = False # Game begins only after first movement
game_started = False
difficulty_timer = 0
# -----------------------------
# Main loop
# -----------------------------
running = True
while running:
clock.tick(FPS)
screen.fill((20, 20, 20))
# -----------------------------
# Events
# -----------------------------
for event in pygame.event.get():
if event.type == pygame.QUIT:
running = False
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_ESCAPE:
running = False
if event.key == pygame.K_SPACE and game_started:
player_polarity *= -1
# -----------------------------
# Player movement
# -----------------------------
# POLARITY SWITCH
if event.key == pygame.K_SPACE and game_started and polarity_cooldown == 0:
player_polarity *= -1
polarity_cooldown = POLARITY_COOLDOWN_TIME
polarity_flash_timer = int(FPS * 0.2)
# DIFFICULTY CHANGE
if game_started and difficulty_cooldown_timer == 0:
if event.key in [pygame.K_1, pygame.K_2, pygame.K_3]:
if event.key == pygame.K_1:
difficulty = 1
elif event.key == pygame.K_2:
difficulty = 2
elif event.key == pygame.K_3:
difficulty = 3
# Easier versions
if difficulty == 1:
NUM_ENEMIES = 12
difficulty_text = "Easy"
difficulty_color = (0, 200, 0)
elif difficulty == 2:
NUM_ENEMIES = 22
difficulty_text = "Medium"
difficulty_color = (255, 200, 0)
elif difficulty == 3:
NUM_ENEMIES = 35
difficulty_text = "Hard"
difficulty_color = (255, 60, 60)
# RESET ENEMIES
enemies = []
for _ in range(NUM_ENEMIES):
polarity = random.choice([1, -1])
surf = pygame.image.load("sprites/orb_red.png" if polarity == 1 else "sprites/orb_blue.png").convert_alpha()
surf = pygame.transform.scale(surf, (40, 40))
side = random.choice(["top", "bottom", "left", "right"])
if side == "top":
x = random.randint(0, SCREEN_WIDTH)
y = -100
elif side == "bottom":
x = random.randint(0, SCREEN_WIDTH)
y = SCREEN_HEIGHT + 100
elif side == "left":
x = -100
y = random.randint(0, SCREEN_HEIGHT)
else:
x = SCREEN_WIDTH + 100
y = random.randint(0, SCREEN_HEIGHT)
rect = surf.get_rect(center=(x, y))
enemies.append({
"surf": surf,
"rect": rect,
"polarity": polarity,
"speed_x": random.uniform(-3.5, 3.5),
"speed_y": random.uniform(-3.5, 3.5)
})
difficulty_flash_timer = FPS // 2
difficulty_cooldown_timer = FPS * 10
keys = pygame.key.get_pressed()
dx = dy = 0
if keys[pygame.K_w] or keys[pygame.K_UP]:
@ -108,119 +174,152 @@ while running:
if keys[pygame.K_d] or keys[pygame.K_RIGHT]:
dx += 6
# Start game on first movement
if not game_started and (dx != 0 or dy != 0):
if not game_started and (dx or dy):
game_started = True
# Activate enemy movement now
for enemy in enemies:
enemy["speed_x"] = random.uniform(-3, 3)
enemy["speed_y"] = random.uniform(-3, 3)
player_rect.move_ip(dx, dy)
player_rect.clamp_ip(pygame.Rect(0, 0, SCREEN_WIDTH, SCREEN_HEIGHT))
moved = dx != 0 or dy != 0
moved = dx or dy
# -----------------------------
# Enemy updates & scoring (only if game started)
# -----------------------------
if game_started:
for enemy in enemies:
# Polarity effect
if enemy["polarity"] == player_polarity: # repel
if enemy["rect"].centerx < player_rect.centerx:
enemy["rect"].x -= 4
# Polarity-based movement
if enemy["polarity"] == player_polarity:
enemy["rect"].x += -4 if enemy["rect"].centerx < player_rect.centerx else 3
enemy["rect"].y += -4 if enemy["rect"].centery < player_rect.centery else 3
else:
enemy["rect"].x += 4
if enemy["rect"].centery < player_rect.centery:
enemy["rect"].y -= 4
else:
enemy["rect"].y += 4
else: # attract
if enemy["rect"].centerx < player_rect.centerx:
enemy["rect"].x += 4
else:
enemy["rect"].x -= 4
if enemy["rect"].centery < player_rect.centery:
enemy["rect"].y += 4
else:
enemy["rect"].y -= 4
enemy["rect"].x += 4 if enemy["rect"].centerx < player_rect.centerx else -3
enemy["rect"].y += 4 if enemy["rect"].centery < player_rect.centery else -3
# Random drift
enemy["rect"].x += enemy["speed_x"] * 0.3
enemy["rect"].y += enemy["speed_y"] * 0.3
# Bounce on screen edges
if enemy["rect"].left < 0 or enemy["rect"].right > SCREEN_WIDTH:
enemy["speed_x"] *= -1
if enemy["rect"].top < 0 or enemy["rect"].bottom > SCREEN_HEIGHT:
enemy["speed_y"] *= -1
# RESPAWN IF OFF SCREEN (+/-100 pixels)
if (enemy["rect"].right < -100 or enemy["rect"].left > SCREEN_WIDTH + 100 or
enemy["rect"].bottom < -100 or enemy["rect"].top > SCREEN_HEIGHT + 100):
side = random.choice(["top", "bottom", "left", "right"])
if side == "top":
x = random.randint(0, SCREEN_WIDTH)
y = -100
elif side == "bottom":
x = random.randint(0, SCREEN_WIDTH)
y = SCREEN_HEIGHT + 100
elif side == "left":
x = -100
y = random.randint(0, SCREEN_HEIGHT)
else:
x = SCREEN_WIDTH + 100
y = random.randint(0, SCREEN_HEIGHT)
enemy["rect"].center = (x, y)
enemy["speed_x"] = random.uniform(-3.5, 3.5)
enemy["speed_y"] = random.uniform(-3.5, 3.5)
# Prevent stacking
for other in enemies:
if other != enemy:
dx_sep = enemy["rect"].centerx - other["rect"].centerx
dy_sep = enemy["rect"].centery - other["rect"].centery
dist_sep = math.hypot(dx_sep, dy_sep)
min_dist = 30
if dist_sep < min_dist and dist_sep != 0:
push_x = (dx_sep / dist_sep) * (min_dist - dist_sep) / 2
push_y = (dy_sep / dist_sep) * (min_dist - dist_sep) / 2
enemy["rect"].move_ip(push_x, push_y)
other["rect"].move_ip(-push_x, -push_y)
# Collision
# Collision with player
if hit_cooldown == 0 and player_rect.colliderect(enemy["rect"]):
player_lives -= 1
hit_cooldown = FPS
for e in enemies:
dx_push = e["rect"].centerx - player_rect.centerx
dy_push = e["rect"].centery - player_rect.centery
e["rect"].move_ip(dx_push//2, dy_push//2)
# Draw enemy
# RESET ALL ENEMIES WHEN HIT
enemies = []
for _ in range(NUM_ENEMIES):
polarity = random.choice([1, -1])
surf = pygame.image.load("sprites/orb_red.png" if polarity == 1 else "sprites/orb_blue.png").convert_alpha()
surf = pygame.transform.scale(surf, (40, 40))
side = random.choice(["top", "bottom", "left", "right"])
if side == "top":
x = random.randint(0, SCREEN_WIDTH)
y = -100
elif side == "bottom":
x = random.randint(0, SCREEN_WIDTH)
y = SCREEN_HEIGHT + 100
elif side == "left":
x = -100
y = random.randint(0, SCREEN_HEIGHT)
else:
x = SCREEN_WIDTH + 100
y = random.randint(0, SCREEN_HEIGHT)
rect = surf.get_rect(center=(x, y))
enemies.append({
"surf": surf,
"rect": rect,
"polarity": polarity,
"speed_x": random.uniform(-3.5, 3.5),
"speed_y": random.uniform(-3.5, 3.5)
})
break
screen.blit(enemy["surf"], enemy["rect"])
# Update score
# Score multiplier based on difficulty
score_multiplier = 1 + (difficulty - 1) * 0.5
if moved:
score += 1
score += score_multiplier
# Update cooldown
if hit_cooldown > 0:
hit_cooldown -= 1
# -----------------------------
# Draw player
# -----------------------------
# TIMERS
if difficulty_flash_timer > 0:
difficulty_flash_timer -= 1
if polarity_cooldown > 0:
polarity_cooldown -= 1
if polarity_flash_timer > 0:
polarity_flash_timer -= 1
if difficulty_cooldown_timer > 0:
difficulty_cooldown_timer -= 1
# DRAW PLAYER WITH FLASH EFFECT
if polarity_flash_timer > 0:
flash_color = (255, 255, 255)
flash_surf = player_surf.copy()
flash_surf.fill(flash_color, special_flags=pygame.BLEND_ADD)
screen.blit(flash_surf, player_rect)
else:
screen.blit(player_surf, player_rect)
# -----------------------------
# Draw HUD
# -----------------------------
hud = font.render(f"Lives: {player_lives} Score: {score}", True, (255, 255, 255))
# HUD
hud = font.render(f"Lives: {player_lives} Score: {int(score)}", True, (255, 255, 255))
screen.blit(hud, (20, 20))
# -----------------------------
# Check for game over
# -----------------------------
# Difficulty flash effect
if difficulty_flash_timer > 0 and difficulty_flash_timer % 10 < 5:
diff_color = (255, 255, 255)
else:
diff_color = difficulty_color
difficulty_hud = font.render(f"Difficulty: {difficulty_text}", True, diff_color)
screen.blit(difficulty_hud, (20, 60))
# Show difficulty cooldown timer
if difficulty_cooldown_timer > 0:
seconds_left = difficulty_cooldown_timer // FPS
cooldown_hud = font.render(f"Change in: {seconds_left}s", True, (200, 200, 255))
screen.blit(cooldown_hud, (20, 100))
# Show player polarity at top right (- for Red, + for Blue)
polarity_text = "-" if player_polarity == 1 else "+"
polarity_hud = font.render(f"Polarity: {polarity_text}", True, (255, 255, 255))
screen.blit(polarity_hud, (SCREEN_WIDTH - polarity_hud.get_width() - 20, 20))
if player_lives <= 0:
running = False
pygame.display.flip()
# -----------------------------
# Game Over screen
# -----------------------------
# GAME OVER SCREEN
screen.fill((20, 20, 20))
game_over_text = font.render("GAME OVER", True, (255, 0, 0))
score_text = font.render(f"Final Score: {score}", True, (255, 255, 255))
score_text = font.render(f"Final Score: {int(score)}", True, (255, 255, 255))
# Center text
screen.blit(game_over_text, (SCREEN_WIDTH//2 - game_over_text.get_width()//2, SCREEN_HEIGHT//2 - 40))
screen.blit(score_text, (SCREEN_WIDTH//2 - score_text.get_width()//2, SCREEN_HEIGHT//2 + 10))
screen.blit(game_over_text,
(SCREEN_WIDTH // 2 - game_over_text.get_width() // 2,
SCREEN_HEIGHT // 2 - 40))
screen.blit(score_text,
(SCREEN_WIDTH // 2 - score_text.get_width() // 2,
SCREEN_HEIGHT // 2 + 10))
pygame.display.flip()
# Wait until the player closes the window
game_over = True
while game_over:
for event in pygame.event.get():