58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
from typing import Type, Literal
|
||
|
||
from module import WIDTH, HEIGHT, JUMP_FORCE
|
||
from module.block import AirBlock, Block
|
||
|
||
|
||
# 玩家类
|
||
class Player:
|
||
def __init__(self, world: list[list[Block]]):
|
||
self.x = 100
|
||
self.y = 100
|
||
self.velocity = 0
|
||
self.on_ground = False
|
||
self.selected_block = AirBlock
|
||
self.world = world
|
||
|
||
def move(self, dx, dy):
|
||
world_position = {'x': int(self.x) // 20, 'y': int(self.y) // 20}
|
||
|
||
# TODO：Need to judgment x and y in order not to raise Index Out Of Range Error
|
||
top_block = self.world[world_position['x']][world_position['y'] - 1]
|
||
button_block = self.world[world_position['x']][world_position['y'] + 1]
|
||
left_block = self.world[world_position['x'] - 1][world_position['y']]
|
||
right_block = self.world[world_position['x'] + 1][world_position['y']]
|
||
|
||
if dx < 0 and left_block.collision:
|
||
new_x = left_block.get_screen_x() + 30
|
||
elif dx > 0 and right_block.collision:
|
||
new_x = right_block.get_screen_x() - 10
|
||
else:
|
||
new_x = self.x + dx
|
||
|
||
if dy < 0 and top_block.collision:
|
||
new_y = top_block.get_screen_y() + 30
|
||
elif dy > 0 and button_block.collision:
|
||
new_y = button_block.get_screen_y() - 10
|
||
else:
|
||
new_y = self.y + dy
|
||
|
||
if 0 <= new_x < WIDTH and 0 <= new_y < HEIGHT:
|
||
self.x = new_x
|
||
self.y = new_y
|
||
|
||
def jump(self):
|
||
if self.on_ground:
|
||
self.velocity = JUMP_FORCE
|
||
self.on_ground = False
|
||
|
||
def select_block(self, block: Type[Block]):
|
||
self.selected_block = block
|
||
|
||
def put_block(self, x: int, y: int, action: Literal['create', 'delete'] = 'create'):
|
||
if action == 'create':
|
||
self.world[x][y] = self.selected_block(x, y)
|
||
elif action == 'delete':
|
||
self.world[x][y] = AirBlock(x, y)
|
||
else:
|
||
pass |