pythonBoids/Flock.py at master · microtechno9000/pythonBoids · 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
from random import randrange

from Boids import Boid
from Mouse import Mouse
from ViewWindow import ViewWindow


class Flock:
    def __init__(self, width, height, max_boids, range_min, range_fov):
        """
        Create the flock
        """
        self.boids_list: list = []
        self.max_speed: int = 0
        self.avg_x: float = 0
        self.avg_y: float = 0
        self.buffer: int = 15
        self.colour: list = []

        self.view_window: ViewWindow = ViewWindow(width - (self.buffer * 2),
                                                  height - (self.buffer * 2),
                                                  self.buffer, )

        self.max_speed = randrange(3, 8, 1)
        self.colour = [randrange(0, 255, 1),
                       randrange(0, 255, 1),
                       randrange(0, 255, 1),
                       255]
        self.boid_height: int = randrange(4, 30, 1)
        self.boid_width: int = int(self.boid_height / 2)
        self.mouse = Mouse()

        # Loop and create the boids
        for _ in range(max_boids):
            # Initial Position
            pos_x = randrange(0, self.view_window.width, 10)
            pos_y = randrange(0, self.view_window.height, 10)
            # Initial Velocity
            vel_x = randrange(-5, 5, 1)
            vel_y = randrange(-5, 5, 1)

            new_boid = Boid(max_boids,
                            pos_x,
                            pos_y,
                            vel_x,
                            vel_y,
                            self.colour,
                            self.boid_width,
                            self.boid_height,
                            range_min,
                            range_fov,
                            self.max_speed,
                            self.view_window)
            # new_boid.debug_vals()
            self.boids_list.append(new_boid)

    def update(self, mouse: Mouse):
        self.mouse = mouse
        for key, boid in enumerate(self.boids_list):
            boid.move(self.boids_list, self.mouse)

    def get_boid_colour(self):
        return self.colour

    def get_boid_width_and_height(self):
        return (self.boid_width, self.boid_height)