pentris.py 文件源码

def get_input():
    for event in pygame.event.get():
        if event.type == pygame.QUIT: sys.exit()
        if event.type == pygame.KEYUP and event.key == pygame.K_DOWN \
            and state.state == "falling":
            state.state = "playing"
        if event.type == pygame.KEYDOWN:
            if state.state == "playing":
                if event.key == pygame.K_p:
                    state.state = "paused"
                if event.key == pygame.K_LEFT:
                    if valid(state.X - 1, state.Y, state.piece.form):
                        state.X -= 1
                if event.key == pygame.K_RIGHT:
                    if valid(state.X + 1, state.Y, state.piece.form):
                        state.X += 1
                if event.key == pygame.K_UP or event.key == pygame.K_SPACE:
                    new_form = Pentomino.rotate_right(state.piece.form,
                        state.piece.size)
                    if valid(state.X, state.Y, new_form):
                        state.piece.form = new_form
                if event.key == pygame.K_DOWN:
                    state.state = "falling"
            elif state.state == "paused":
                if event.key == pygame.K_p:
                    state.state = "playing"
            elif state.state == "lost":
                if event.key in [pygame.K_SPACE, pygame.K_RETURN]:
                    state.start_game()