planets/controls.py at develop · gm2211/planets · 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
import pygame

from collisions import find_first_collision
from objects import GameState, PendingPlanet


def handle_interrupts(state: GameState) -> GameState:
    new_state = state

    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            new_state = new_state.copy(quit=True)
        if event.type == pygame.MOUSEBUTTONDOWN:
            mouse_x, mouse_y = pygame.mouse.get_pos()
            pending_planet = PendingPlanet(mouse_x, mouse_y, radius=new_state.radius)
            new_state = new_state.copy(pending_planet=pending_planet)
        if event.type == pygame.MOUSEBUTTONUP:
            if new_state.pending_planet is not None:
                planet = new_state.pending_planet.to_planet()
                if find_first_collision(new_state, planet) is None:
                    new_state = new_state.with_append_planet(planet)
                new_state = new_state.copy(pending_planet=None)
        if event.type == pygame.KEYDOWN:
            if pygame.key.name(event.key) == 'k':
                new_state = new_state.copy(radius_change=1)
            elif pygame.key.name(event.key) == 'j':
                new_state = new_state.copy(radius_change=-1)
            elif pygame.key.name(event.key) == 'f':
                new_state = new_state.copy(time_warp_change=1)
            elif pygame.key.name(event.key) == 's':
                new_state = new_state.copy(time_warp_change=-1)
            elif pygame.key.name(event.key) == 'e':
                new_state = new_state.copy(new_planet_density_change=1)
            elif pygame.key.name(event.key) == 'w':
                new_state = new_state.copy(new_planet_density_change=-1)
        if event.type == pygame.KEYUP:
            if pygame.key.name(event.key) == 'r':
                new_state = GameState()
            elif pygame.key.name(event.key) == 'd':
                new_state = new_state.copy(debug=not new_state.debug)
            elif pygame.key.name(event.key) == 'q':
                new_state = new_state.copy(quit=True)
            elif pygame.key.name(event.key) == 'p':
                new_state = new_state.copy(paused=not new_state.paused)
            elif pygame.key.name(event.key) in ('k', 'j'):
                new_state = new_state.copy(radius_change=0)
            elif pygame.key.name(event.key) in ('f', 's'):
                new_state = new_state.copy(time_warp_change=0)
            elif pygame.key.name(event.key) in ('w', 'e'):
                new_state = new_state.copy(new_planet_density_change=0)
            elif pygame.key.name(event.key) == 'g':
                new_state = new_state.copy(new_planet_fixed_position=not new_state.new_planet_fixed_position)

    # If we inside the if statement, it means either 'k' or 'j' was being held down.
    # Also, we don't want a negative radius
    new_radius = new_state.radius + new_state.radius_change
    if new_state.radius_change != 0 and new_radius >= 0:
        new_state = new_state.copy(radius=new_radius)

    # If we inside the if statement, it means either 'f' or 's' was being held down.
    # Also, we don't want a negative time warp factor
    new_time_warp = new_state.time_warp + new_state.time_warp_change
    if new_state.time_warp_change != 0 and new_time_warp >= 1:
        new_state = new_state.copy(time_warp=new_time_warp)

    # If we inside the if statement, it means either 'd' or 'e' was being held down.
    # Also, we don't want a negative density
    new_new_planet_density = (new_state.new_planet_density + new_state.new_planet_density_change)
    if new_state.new_planet_density_change != 0 and new_new_planet_density >= 1:
        new_new_planet_density_change = (
                new_state.new_planet_density_change +
                + (100 if new_state.new_planet_density_change > 0 else -100)
        )
        print(new_new_planet_density_change)
        new_state = new_state.copy(
            new_planet_density=new_new_planet_density,
            new_planet_density_change=new_new_planet_density_change
        )

    if new_state.pending_planet is not None:
        mouse_x, mouse_y = pygame.mouse.get_pos()
        pending_planet = new_state.pending_planet
        new_momentum = (
            -(pending_planet.x - mouse_x) / new_state.momentum_input_scale,
            -(pending_planet.y - mouse_y) / new_state.momentum_input_scale
        )
        new_pending_planet = pending_planet.copy(
            momentum=new_momentum,
            radius=new_state.radius,
            density=new_state.new_planet_density,
            fixed_position=new_state.new_planet_fixed_position
        )
        new_state = new_state.copy(pending_planet=new_pending_planet)

    return new_state