aiplatformer/Level.py at master · jtracks/aiplatformer · 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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
'''
Contains the level basic class, Levels will inherit from this class to make a level
'''

import pygame as pg
import random
from Collectibles import Collectible
from Checkpoints import Checkpoint
from itertools import chain
from Config import SCREEN, BACKGROUND, TERRAIN, COLLECTIBLE
from AssetLoader import BACKGROUNDS, TERRAINS, COLLECTIBLES

SCREEN_WIDTH, SCREEN_HIGHT = SCREEN['SIZE']

class Level():
    ''' Default level '''

    level_limit = -1000
    world_shift = 0
    tick = 0

    def __init__(self, player, lvl_type='Blue',  ter_type='Wood'):

        self.platforms = pg.sprite.Group()
        self.enemies = pg.sprite.Group()
        self.bots = pg.sprite.Group()
        self.player = player
        self.collectibles = pg.sprite.Group()
        self.checkpoints = pg.sprite.Group()
        self.level_type = lvl_type
        self.terrain_type = ter_type

        for width, height, x, y in self._platforms():
            self.platforms.add(Platform(width, height, x, y, ter_type))

        for x, y in self._collectibles():
            self.collectibles.add(Collectible(x,y))

        for x,y in self._checkpoints():
            self.checkpoints.add(Checkpoint(x,y))

    def _platforms(self):
        ''' Platform(width, height, x, y) '''

        ground = (5000, 96, self.level_limit, SCREEN_HIGHT - 64)
        left_block = (64, SCREEN_HIGHT-64, self.level_limit, 0)
        right_block = (64, SCREEN_HIGHT-500, 1600, 100)
        roof =(5000, 64, self.level_limit, - 64)
        wall = (96, 200, 1000, SCREEN_HIGHT - 550)
        big_block = (320, 160, 1200, SCREEN_HIGHT - 500)
        platform1 = (210, 70, 600, SCREEN_HIGHT - 200)
        platform2 = (210, 70, 600, SCREEN_HIGHT - 600)

        return [ground, left_block, right_block, roof, wall, big_block, platform1, platform2]

    def _collectibles(self):
        coins = []
        coins.append((800,400))
        coins.append((200,200))

        return coins

    def _checkpoints(self):
        checkpoints = []
        checkpoints.append((1700,184))

        return checkpoints


    def update(self):
        ''' Update everything '''

        self.platforms.update()
        self.enemies.update()
        self.collectibles.update()
        self.checkpoints.update()

    def draw(self, screen):
        ''' Draw content on screen '''

        if self.tick == 60:
            self.tick = 0

        offset = (BACKGROUND['SIZE'][0]*(self.tick/60-1), BACKGROUND['SIZE'][0]*(self.tick/60-1))

        screen.blit(BACKGROUNDS[self.level_type], offset)

        self.platforms.draw(screen)
        self.enemies.draw(screen)
        self.collectibles.draw(screen)
        self.checkpoints.draw(screen)

        self.tick += 1

    def shift_world(self, speed_x):
        ''' Scroll world '''

        self.world_shift += speed_x

        for item in chain(self.platforms, self.enemies, self.bots,self.collectibles,self.checkpoints):
            item.rect.x += speed_x

class Platform(pg.sprite.Sprite):
    ''' Platform the player can jump on '''

    def __init__(self, width, height, x, y, ter_type='Wood'):

        super(Platform,self).__init__()
        width -= (width % TERRAIN['SIZE'][0])
        height -= (height % TERRAIN['SIZE'][1])
        self.image = pg.Surface([width, height])

        self._add_terrain(width, height, ter_type)

        self.rect = self.image.get_rect()
        self.rect.x = x
        self.rect.y = y

    def _add_terrain(self, width, height, ter_type):

        terrain = TERRAINS[ter_type]
        ter_w, ter_h = TERRAIN['SIZE']

        # Add fill to every square
        for w in range(0, width+1 , ter_w):
            for h in range(0, height+1 , ter_h):
                self.image.blit(random.choice(terrain['FILL']),(w,h))

        # Add tops and bottoms
        for w in range(0, width+1 , ter_w):
            self.image.blit(terrain['TOP'], (w, 0))
            self.image.blit(terrain['BOTTOM'], (w, height-ter_h))

        # Add left and rights
        for h in range(0, height+1 , ter_h):
            self.image.blit(terrain['LEFT'], (0, h))
            self.image.blit(terrain['RIGHT'], (width-ter_w, h))

        # Add corners
        self.image.blit(terrain['TOP LEFT'], (0, 0))
        self.image.blit(terrain['TOP RIGHT'], (width-ter_w, 0))
        self.image.blit(terrain['BOTTOM LEFT'], (0, height-ter_h))
        self.image.blit(terrain['BOTTOM RIGHT'], (width-ter_w, height-ter_h))