menu.py

import pygame, os, sys
from pygame.locals import *
from statemanager import *
from bitmapfont import *


class MainMenuState(GameState):
    def __init__(self, game):
        super().__init__(game)
        self.playGameState = None
        self.font = BitmapFont('fasttracker2-style_12x12.png', 12, 12)
        self.index = 0
        self.inputTick = 0
        self.menuItems = ['Play', 'Quit']
        self.background = pygame.image.load("menu_wallpaper.jpg").convert()
        pygame.mixer.music.load("bg_music.mp3")
        pygame.mixer.music.play()

    def setPlayState(self, state):
        self.playGameState = state

    def update(self, gameTime):
        print("Menu Updating")
        keys = pygame.key.get_pressed()
        if (keys[K_UP] or keys[K_DOWN]) and self.inputTick == 0:
            '''
            The user presses the up and down button to select a menu item.
            To prevent the menu selection from spinning out of control,
            the updates are clamped to four per second (250 milliseconds).
            '''
            self.inputTick = 250
            if keys[K_UP]:
                self.index -= 1
                if self.index < 0:
                    self.index = len(self.menuItems) - 1
            elif keys[K_DOWN]:
                self.index += 1
                if self.index == len(self.menuItems):
                    self.index = 0

        elif self.inputTick > 0:
            self.inputTick -= gameTime
        if self.inputTick < 0:
            self.inputTick = 0

        if keys[K_SPACE]:
            if self.index == 1:
                self.game.changeState(None)  # exit the game
            elif self.index == 0:
                self.game.changeState(self.playGameState)

    def draw(self, surface):
        surface.blit(self.background, (0,0))
        self.font.centre(surface, "team#263:whack&roll - POKERBRICKS", 48)

        count = 0
        y = surface.get_rect().height - len(self.menuItems) * 100
        for item in self.menuItems:
            itemText = "  "
            if count == self.index:
                itemText = "> "

            itemText += item
            self.font.draw(surface, itemText, 25, y)
            y += 24
            count += 1