create_level.py

"""
This code creates the layout of level one.
"""

import pymunk
import arcade
import random
import sys
import logging
from random import randint, uniform

logging.basicConfig(format="%(levelname)s: %(message)s", level=logging.INFO)

from physics_utility import (
    PymunkSprite,
)

# from constants import *
import constants
from k8s_kill_pod import count_pods


def create_floor(space, sprite_list):
    """Create a bunch of blocks for the floor."""
    # Layer of grass
    for x in range(-constants.MAP_SIZE, constants.MAP_SIZE, constants.SPRITE_SIZE):
        sprite = arcade.Sprite("./images/tiles/grassMid.png", constants.SPRITE_SCALING)
        sprite.center_y = 0
        sprite.center_x = x
        sprite_list.append(sprite)
        # print(sprite.center_y)

    # Create one big rectangle as the floor physic body, instead of one per tile
    xpos = constants.SCREEN_WIDTH / 2
    ypos = 0
    body = pymunk.Body(body_type=pymunk.Body.STATIC)
    body.position = pymunk.Vec2d(xpos, ypos)
    shape = pymunk.Poly.create_box(
        body, (constants.MAP_SIZE * 2, constants.SPRITE_SIZE)
    )

    shape.friction = constants.DEFAULT_FRICTION
    space.add(body, shape)

    # First layer of dirt
    for x in range(-constants.MAP_SIZE, constants.MAP_SIZE, constants.SPRITE_SIZE):
        y = -constants.SPRITE_SIZE + 1
        # y = constants.SPRITE_SIZE - (constants.SPRITE_SIZE * 2)
        sprite = PymunkSprite(
            "./images/tiles/grassCenter.png",
            x,
            y,
            scale=constants.SPRITE_SCALING,
            body_type=pymunk.Body.STATIC,
        )
        sprite_list.append(sprite)

    # Extra layer of dirt
    for x in range(-constants.MAP_SIZE, constants.MAP_SIZE, constants.SPRITE_SIZE):
        y = -constants.SPRITE_SIZE - constants.SPRITE_SIZE + 1
        # y = constants.SPRITE_SIZE - (constants.SPRITE_SIZE * 3)
        sprite = PymunkSprite(
            "./images/tiles/grassCenter.png",
            x,
            y,
            scale=constants.SPRITE_SCALING,
            body_type=pymunk.Body.STATIC,
        )
        sprite_list.append(sprite)


def create_walls(space, sprite_list):
    """Create side walls"""
    # Left wall
    for y in range(96, 2000, constants.SPRITE_SIZE):  # was 96 - constants.SPRITE_SIZE
        # x = constants.SPRITE_SIZE / 4
        x = -1200 - constants.SPRITE_SIZE
        sprite = PymunkSprite(
            "./images/tiles/brickBrown.png",
            x,
            y,
            scale=0.5,
            body_type=pymunk.Body.STATIC,
        )
        sprite_list.append(sprite)
        space.add(sprite.body, sprite.shape)
    # Right wall
    for y in range(96, 2000, constants.SPRITE_SIZE):
        # x = constants.SPRITE_SIZE / 4
        x = 2200 - constants.SPRITE_SIZE
        sprite = PymunkSprite(
            "./images/tiles/brickBrown.png",
            x,
            y,
            scale=0.5,
            body_type=pymunk.Body.STATIC,
        )
        sprite_list.append(sprite)
        space.add(sprite.body, sprite.shape)


def create_platform(space, sprite_list, start_x, y, count):
    """Create a platform"""
    for x in range(
        start_x, start_x + count * constants.SPRITE_SIZE + 1, constants.SPRITE_SIZE
    ):
        sprite = PymunkSprite(
            "./images/tiles/grassMid.png", x, y, scale=0.5, body_type=pymunk.Body.STATIC
        )
        sprite_list.append(sprite)
        space.add(sprite.body, sprite.shape)


def create_hill(space, sprite_list, start_x, y, count):
    """Create a hill"""
    # Left edge
    sprite = PymunkSprite(
        "./images/tiles/grassHill_right.png",
        start_x - constants.SPRITE_SIZE,
        y,
        scale=constants.SPRITE_SCALING,
        body_type=pymunk.Body.STATIC,
    )
    t = pymunk.Transform(tx=-60, ty=-96)
    sprite.shape = pymunk.Poly(
        sprite.body, [(80, 120), (80, 60), (20, 60)], transform=t
    )
    sprite.shape.friction = constants.DEFAULT_FRICTION / 4
    sprite_list.append(sprite)
    space.add(sprite.body, sprite.shape)
    # Add dirt under
    dirt_y = 0
    sprite = PymunkSprite(
        "./images/tiles/grassCorner_right.png",
        start_x - constants.SPRITE_SIZE,
        dirt_y,
        scale=constants.SPRITE_SCALING,
        body_type=pymunk.Body.STATIC,
    )
    sprite_list.append(sprite)

    # Middle
    for x in range(
        start_x, start_x + count * constants.SPRITE_SIZE + 1, constants.SPRITE_SIZE
    ):
        sprite = PymunkSprite(
            "./images/tiles/grassMid.png",
            x,
            y,
            scale=constants.SPRITE_SCALING,
            body_type=pymunk.Body.STATIC,
        )
        sprite_list.append(sprite)
        space.add(sprite.body, sprite.shape)
        # Add dirt under
        dirt_y = 0
        sprite = PymunkSprite(
            "./images/tiles/grassCenter.png",
            x,
            dirt_y,
            scale=constants.SPRITE_SCALING,
            body_type=pymunk.Body.STATIC,
        )
        sprite_list.append(sprite)

    # Right edge
    sprite = PymunkSprite(
        "./images/tiles/grassHill_left.png",
        start_x + constants.SPRITE_SIZE * count + constants.SPRITE_SIZE,
        y,
        scale=constants.SPRITE_SCALING,
        body_type=pymunk.Body.STATIC,
    )
    sprite.shape.friction = constants.DEFAULT_FRICTION / 4
    t = pymunk.Transform(tx=-60, ty=-96)
    sprite.shape = pymunk.Poly(
        sprite.body, [(20, 120), (20, 60), (80, 60)], transform=t
    )
    sprite_list.append(sprite)
    space.add(sprite.body, sprite.shape)
    # Add dirt under
    dirt_y = 0
    sprite = PymunkSprite(
        "./images/tiles/grassCorner_left.png",
        start_x + constants.SPRITE_SIZE * count + constants.SPRITE_SIZE,
        dirt_y,
        scale=constants.SPRITE_SCALING,
        body_type=pymunk.Body.STATIC,
    )
    sprite_list.append(sprite)


def decorate_cactus(sprite_list, start_x, y, count):
    """Create a cactus"""
    for x in range(0, count):
        # print(x)
        X_POSITION = randint(-3000, 3000)  # Initial random position
        POSITION_OK = False
        while POSITION_OK == False:
            if any(
                X_POSITION in range(s.center_x - 40, s.center_x + 40)
                for s in sprite_list
            ):
                # print(X_POSITION, s)
                X_POSITION = randint(-3000, 3000)
            else:
                # print("Outside of if any, cactus position good!")
                POSITION_OK = True
                sprite = arcade.Sprite("./images/tiles/cactus.png", 0.5)
                sprite.center_y = y
                sprite.center_x = X_POSITION  # Randomly place cacti
                sprite_list.append(sprite)
                # print(len(sprite_list))


def decorate_cactus_large(sprite_list, start_x, y, count):
    """Create a cactus"""
    for x in range(0, count):
        # print(x)
        X_POSITION = randint(-3000, 3000)  # Initial random position
        POSITION_OK = False
        while POSITION_OK == False:
            if any(
                X_POSITION in range(s.center_x - 100, s.center_x + 100)
                for s in sprite_list
            ):
                # print(X_POSITION, s)
                X_POSITION = randint(-3000, 3000)
            else:
                # print("Outside of if any, cactus position good!")
                POSITION_OK = True
                sprite = arcade.Sprite("./images/tiles/cactus.png", 1)
                sprite.center_y = y
                sprite.center_x = X_POSITION  # Randomly place cacti
                sprite_list.append(sprite)


def decorate_cactus_tiny(sprite_list, start_x, y, count):
    """Create a distant cactus"""
    for x in range(0, count):
        # print(x)
        X_POSITION = randint(-3000, 3000)  # Initial random position
        POSITION_OK = False
        while POSITION_OK == False:
            if any(
                X_POSITION in range(s.center_x - 200, s.center_x + 200)
                for s in sprite_list
            ):
                # print(X_POSITION, s)
                X_POSITION = randint(-3000, 3000)
            else:
                # print("Outside of if any, cactus position good!")
                POSITION_OK = True
                sprite = arcade.Sprite("./images/tiles/cactus2.png", 0.2)
                sprite.center_y = y
                sprite.center_x = X_POSITION  # Randomly place cacti
                sprite_list.append(sprite)


def decorate_grass(sprite_list, start_x, y, count):
    """Create some grass"""
    for x in range(0, count):
        X_POSITION = randint(-3000, 3000)  # Initial random position
        POSITION_OK = False
        while POSITION_OK == False:
            if any(
                X_POSITION in range(s.center_x - 20, s.center_x + 20)
                for s in sprite_list
            ):
                # print(X_POSITION, s)
                X_POSITION = randint(-3000, 3000)
            else:
                # print("Outside of if any, grass position good!")
                POSITION_OK = True
                sprite = arcade.Sprite("./images/tiles/grass_sprout.png", 0.5)
                sprite.center_y = y
                sprite.center_x = X_POSITION  # Randomly place grass
                sprite_list.append(sprite)


def decorate_rock(sprite_list, start_x, y, count):
    """Create some rocks"""
    for x in range(0, count):
        X_POSITION = randint(-3000, 3000)  # Initial random position
        POSITION_OK = False
        while POSITION_OK == False:
            if any(
                X_POSITION in range(s.center_x - 10, s.center_x + 10)
                for s in sprite_list
            ):
                # print(X_POSITION, s)
                X_POSITION = randint(-1000, 2000)
            else:
                # print("Outside of if any, rock position good!")
                POSITION_OK = True
                sprite = arcade.Sprite("./images/tiles/rock.png", 0.5)
                sprite.center_y = y
                sprite.center_x = X_POSITION  # Randomly place rock
                sprite_list.append(sprite)


def decorate_rock_small(sprite_list, start_x, y, count):
    """Create some rocks"""
    for x in range(0, count):
        X_POSITION = randint(-3000, 3000)  # Initial random position
        POSITION_OK = False
        while POSITION_OK == False:
            if any(
                X_POSITION in range(s.center_x - 10, s.center_x + 10)
                for s in sprite_list
            ):
                # print(X_POSITION, s)
                X_POSITION = randint(-3000, 3000)
            else:
                # print("Outside of if any, rock position good!")
                POSITION_OK = True
                sprite = arcade.Sprite("./images/tiles/rock.png", 0.2)
                sprite.center_y = y
                sprite.center_x = X_POSITION  # Randomly place rock
                sprite_list.append(sprite)


def decorate_clouds(sprite_list, count):
    """Create some clouds"""
    for x in range(0, count):
        # sprite = arcade.Sprite("./images/misc/cloud" + str(randint(1, 5)) + ".png", random.uniform(0.3,1))
        sprite = arcade.Sprite(
            "./images/misc/mbcloud" + str(randint(1, 2)) + ".png",
            random.uniform(0.2, 1),
        )
        sprite.center_y = randint(700, 3000)
        sprite.center_x = randint(-2500, 2500)
        # print(sprite.center_x)
        sprite_list.append(sprite)


def create_level_1(
    space, static_sprite_list, dynamic_sprite_list, bg_sprite_list, fg_sprite_list
):
    """Create level one."""
    create_floor(space, static_sprite_list)

    # Add hills
    create_hill(
        space, static_sprite_list, -constants.SPRITE_SIZE * 40, constants.SPRITE_SIZE, 3
    )
    create_hill(
        space, static_sprite_list, constants.SPRITE_SIZE * 34, constants.SPRITE_SIZE, 4
    )

    # create_walls(space, static_sprite_list)
    create_platform(space, static_sprite_list, 200, constants.SPRITE_SIZE * 3, 3)
    create_platform(space, static_sprite_list, 500, constants.SPRITE_SIZE * 6, 3)
    create_platform(space, static_sprite_list, 200, constants.SPRITE_SIZE * 9, 3)
    create_platform(space, static_sprite_list, -300, constants.SPRITE_SIZE * 3, 3)
    create_platform(space, static_sprite_list, -600, constants.SPRITE_SIZE * 6, 2)

    # Far left platforms
    create_platform(space, static_sprite_list, -900, constants.SPRITE_SIZE * 9, 3)
    create_platform(space, static_sprite_list, -1200, constants.SPRITE_SIZE * 3, 1)
    create_platform(space, static_sprite_list, -1600, constants.SPRITE_SIZE * 6, 2)
    create_platform(space, static_sprite_list, -1800, constants.SPRITE_SIZE * 13, 4)
    create_platform(space, static_sprite_list, -2100, constants.SPRITE_SIZE * 9, 3)

    create_platform(space, static_sprite_list, -840, constants.SPRITE_SIZE * 13, 1)
    create_platform(space, static_sprite_list, 0, constants.SPRITE_SIZE * 13, 3)

    # Add some more to the right
    create_platform(space, static_sprite_list, 1040, constants.SPRITE_SIZE * 4, 3)
    create_platform(space, static_sprite_list, 1440, constants.SPRITE_SIZE * 6, 2)
    create_platform(space, static_sprite_list, 1840, constants.SPRITE_SIZE * 8, 1)
    create_platform(space, static_sprite_list, 800, constants.SPRITE_SIZE * 8, 1)
    create_platform(space, static_sprite_list, 2300, constants.SPRITE_SIZE * 10, 1)

    # Add decorations
    decorate_cactus(bg_sprite_list, 0, constants.SPRITE_SIZE, 12)  # Cacti along ground
    decorate_cactus_large(bg_sprite_list, 0, constants.SPRITE_SIZE + 30, 4)
    decorate_cactus_tiny(bg_sprite_list, 0, constants.SPRITE_SIZE - 20, 4)
    decorate_grass(bg_sprite_list, 0, constants.SPRITE_SIZE, 20)
    decorate_rock(fg_sprite_list, 0, constants.SPRITE_SIZE, 3)
    decorate_rock_small(bg_sprite_list, 0, constants.SPRITE_SIZE - 20, 10)
    decorate_clouds(bg_sprite_list, 20)

    # Create the stacks of boxes based on number of running pods or create random ones if offline mode
    # print(constants.OFFLINE_MODE)
    if constants.OFFLINE_MODE == True:
        CRATE_COUNT = constants.OFFLINE_CRATE_COUNT
    else:
        logging.info("Attempting to connect to Kubernetes API host..")
        try:
            CRATE_COUNT = count_pods()
        except:
            logging.error("Unable to connect to Kubernetes API host")
            logging.error("Check your Kubernetes environment/kubeconfig file")
            logging.error("Or feel free to start with --offline yes set")
            sys.exit(1)

    logging.info("Creating %s crates", int(CRATE_COUNT * constants.CONTAINER_FACTOR))

    # Create crates in random locations, based on number of pods * CONTAINER_FACTOR
    i = 0
    while i < int(CRATE_COUNT * constants.CONTAINER_FACTOR):
        x = random.randrange(-2000, 2200)
        y = random.randrange(200, 7000)  # Drop crates in from random heights
        # print(x)
        # print(y)
        sprite = PymunkSprite(
            "./images/tiles/boxCrate_double.png",
            x,
            y,
            scale=constants.SPRITE_SCALING,
            friction=0.6,
        )
        dynamic_sprite_list.append(sprite)
        space.add(sprite.body, sprite.shape)
        # fall_sound = arcade.load_sound("./sounds/wooddrop.wav")
        # arcade.play_sound(fall_sound)
        i += 1

    # logging.info("Number of crates created: %s", len(dynamic_sprite_list))