auxiliar.py

import random
import pygame
import configurations as c


def adjust_list_creator(extent):
    first_range = [(x, y) for x in range(extent) for y in range(extent)]
    second_range = [(x, -y) for x in range(extent) for y in range(extent)]
    third_range = [(-x, y) for x in range(extent) for y in range(extent)]
    fourth_range = [(-x, -y) for x in range(extent) for y in range(extent)]
    return first_range + second_range + third_range + fourth_range


ADJUSTS_LIST = [adjust_list_creator(2),
                adjust_list_creator(3),
                adjust_list_creator(4),
                adjust_list_creator(5),
                adjust_list_creator(6)
                ]
# -------------------------------------------- WORLD DEFINITION --------------------------------------------------------
ADJUSTS = ADJUSTS_LIST[3]

# ------------------------------------------- WINDOW DEFINITION --------------------------------------------------------
WINDOW_WIDTH = c.ROWS * c.CELL_SIZE  # width of the simulation window based on rows number and cell size
WINDOW_HEIGHT = c.COLUMNS * c.CELL_SIZE  # length of the simulation window based on column number and cell size
CLOCK = pygame.time.Clock()  # clock to control the simulation loop regarding the frame rate


# --------------------------------------------- BEING CREATION ---------------------------------------------------------
def center_screen_y():
    return (WINDOW_HEIGHT - c.BEING_SIZE)//2


def center_screen_x():
    return (WINDOW_WIDTH - c.BEING_SIZE)//2


def random_change(speed):
    return (speed+random.randint(5, 10) * random.choice([-1, 1])) * random.choice([-1, 1])\
        if (random.random() > c.SPEED_CHANGE_PROBABILITY) else speed


def random_movement(x, y, x_speed, y_speed, events=None):
    x_speed = random_change(x_speed)
    y_speed = random_change(y_speed)
    x += x_speed
    y += y_speed
    return x, y, x_speed, y_speed


def controlled_movement(x, y, x_speed, y_speed, event=None):
    movements = {None: (0, 0), "r": (x_speed, 0), "l": (-x_speed, 0), "u": (0, -y_speed), "d": (0, x_speed)}[event]
    return x+movements[0], y+movements[1], x_speed, y_speed


def draw_filled_square(screen, x, y, size, color):
    pygame.draw.rect(screen, color, (x+size//2, y+size//2, size, size))


def draw_unfilled_square(screen, x, y, size, color):
    pygame.draw.rect(screen, color, (x+size//2, y+size//2, size, size), int(size*0.1))


def draw_filled_circle(screen, x, y, size, color):
    pygame.draw.circle(screen, color, (x+size/2, y+size//2), size//2)


def draw_unfilled_circle(screen, x, y, size, color):
    pygame.draw.circle(screen, color, (x+size/2, y+size//2), size//2, int(size*0.1))


def get_square_center(x, y):
    return x+c.BEING_SIZE//2, y+c.BEING_SIZE//2


def get_circle_center(x, y):
    return x, y


def get_being(code):
    # code 1 to 4 -> being with random movement | code 5 to 8 -> being with keyboard controlled movement
    # odd code -> being is a circle | even code -> being is square
    being_behaviours = {1: [draw_unfilled_circle, random_movement],
                        2: [draw_unfilled_square, random_movement],
                        3: [draw_filled_circle, random_movement],
                        4: [draw_filled_square, random_movement],
                        5: [draw_unfilled_circle, controlled_movement],
                        6: [draw_unfilled_square, controlled_movement],
                        7: [draw_filled_circle, controlled_movement],
                        8: [draw_filled_square, controlled_movement]}

    center_function = get_circle_center if code % 2 else get_square_center  # get corresponding function to get center

    being = [random.choice(c.COLORS),
             center_screen_x(),
             center_screen_y(),
             c.X_SPEED, c.Y_SPEED,
             c.BEING_SIZE,
             being_behaviours[code],
             center_function]

    return being