# Star Pusher (a Sokoban clone)
# By Al Sweigart al@inventwithpython.com
# http://inventwithpython.com/pygame
# Released under a "Simplified BSD" license
 
import random, sys, copy, os, pygame
from pygame.locals import *
 
FPS = 30 # frames per second to update the screen
WINWIDTH = 800 # width of the program's window, in pixels
WINHEIGHT = 600 # height in pixels
HALF_WINWIDTH = int(WINWIDTH / 2)
HALF_WINHEIGHT = int(WINHEIGHT / 2)
 
# The total width and height of each tile in pixels.
TILEWIDTH = 50
TILEHEIGHT = 85
TILEFLOORHEIGHT = 40
 
CAM_MOVE_SPEED = 5 # how many pixels per frame the camera moves
 
# The percentage of outdoor tiles that have additional
# decoration on them, such as a tree or rock.
OUTSIDE_DECORATION_PCT = 20
 
BRIGHTBLUE = ( 0, 170, 255)
WHITE = (255, 255, 255)
BGCOLOR = BRIGHTBLUE
TEXTCOLOR = WHITE
 
UP = 'up'
DOWN = 'down'
LEFT = 'left'
RIGHT = 'right'
 
 
def main():
 global FPSCLOCK, DISPLAYSURF, IMAGESDICT, TILEMAPPING, OUTSIDEDECOMAPPING, BASICFONT, PLAYERIMAGES, currentImage
 
 # Pygame initialization and basic set up of the global variables.
 pygame.init()
 FPSCLOCK = pygame.time.Clock()
 
 # Because the Surface object stored in DISPLAYSURF was returned
 # from the pygame.display.set_mode() function, this is the
 # Surface object that is drawn to the actual computer screen
 # when pygame.display.update() is called.
 DISPLAYSURF = pygame.display.set_mode((WINWIDTH, WINHEIGHT))
 
 pygame.display.set_caption('Star Pusher')
 BASICFONT = pygame.font.Font('freesansbold.ttf', 18)
 
 # A global dict value that will contain all the Pygame
 # Surface objects returned by pygame.image.load().
 IMAGESDICT = {'uncovered goal': pygame.image.load('RedSelector.png'),
   'covered goal': pygame.image.load('Selector.png'),
   'star': pygame.image.load('Star.png'),
   'corner': pygame.image.load('Wall_Block_Tall.png'),
   'wall': pygame.image.load('Wood_Block_Tall.png'),
   'inside floor': pygame.image.load('Plain_Block.png'),
   'outside floor': pygame.image.load('Grass_Block.png'),
   'title': pygame.image.load('star_title.png'),
   'solved': pygame.image.load('star_solved.png'),
   'princess': pygame.image.load('princess.png'),
   'boy': pygame.image.load('boy.png'),
   'catgirl': pygame.image.load('catgirl.png'),
   'horngirl': pygame.image.load('horngirl.png'),
   'pinkgirl': pygame.image.load('pinkgirl.png'),
   'rock': pygame.image.load('Rock.png'),
   'short tree': pygame.image.load('Tree_Short.png'),
   'tall tree': pygame.image.load('Tree_Tall.png'),
   'ugly tree': pygame.image.load('Tree_Ugly.png')}
 
 # These dict values are global, and map the character that appears
 # in the level file to the Surface object it represents.
 TILEMAPPING = {'x': IMAGESDICT['corner'],
   '#': IMAGESDICT['wall'],
   'o': IMAGESDICT['inside floor'],
   ' ': IMAGESDICT['outside floor']}
 OUTSIDEDECOMAPPING = {'1': IMAGESDICT['rock'],
    '2': IMAGESDICT['short tree'],
    '3': IMAGESDICT['tall tree'],
    '4': IMAGESDICT['ugly tree']}
 
 # PLAYERIMAGES is a list of all possible characters the player can be.
 # currentImage is the index of the player's current player image.
 currentImage = 0
 PLAYERIMAGES = [IMAGESDICT['princess'],
   IMAGESDICT['boy'],
   IMAGESDICT['catgirl'],
   IMAGESDICT['horngirl'],
   IMAGESDICT['pinkgirl']]
 
 startScreen() # show the title screen until the user presses a key
 
 # Read in the levels from the text file. See the readLevelsFile() for
 # details on the format of this file and how to make your own levels.
 levels = readLevelsFile('starPusherLevels.txt')
 currentLevelIndex = 0
 
 # The main game loop. This loop runs a single level, when the user
 # finishes that level, the next/previous level is loaded.
 while True: # main game loop
 # Run the level to actually start playing the game:
 result = runLevel(levels, currentLevelIndex)
 
 if result in ('solved', 'next'):
  # Go to the next level.
  currentLevelIndex += 1
  if currentLevelIndex >= len(levels):
  # If there are no more levels, go back to the first one.
  currentLevelIndex = 0
 elif result == 'back':
  # Go to the previous level.
  currentLevelIndex -= 1
  if currentLevelIndex < 0:
  # If there are no previous levels, go to the last one.
  currentLevelIndex = len(levels)-1
 elif result == 'reset':
  pass # Do nothing. Loop re-calls runLevel() to reset the level
 
 
def runLevel(levels, levelNum):
 global currentImage
 levelObj = levels[levelNum]
 mapObj = decorateMap(levelObj['mapObj'], levelObj['startState']['player'])
 gameStateObj = copy.deepcopy(levelObj['startState'])
 mapNeedsRedraw = True # set to True to call drawMap()
 levelSurf = BASICFONT.render('Level %s of %s' % (levelNum + 1, len(levels)), 1, TEXTCOLOR)
 levelRect = levelSurf.get_rect()
 levelRect.bottomleft = (20, WINHEIGHT - 35)
 mapWidth = len(mapObj) * TILEWIDTH
 mapHeight = (len(mapObj[0]) - 1) * TILEFLOORHEIGHT + TILEHEIGHT
 MAX_CAM_X_PAN = abs(HALF_WINHEIGHT - int(mapHeight / 2)) + TILEWIDTH
 MAX_CAM_Y_PAN = abs(HALF_WINWIDTH - int(mapWidth / 2)) + TILEHEIGHT
 
 levelIsComplete = False
 # Track how much the camera has moved:
 cameraOffsetX = 0
 cameraOffsetY = 0
 # Track if the keys to move the camera are being held down:
 cameraUp = False
 cameraDown = False
 cameraLeft = False
 cameraRight = False
 
 while True: # main game loop
 # Reset these variables:
 playerMoveTo = None
 keyPressed = False
 
 for event in pygame.event.get(): # event handling loop
  if event.type == QUIT:
  # Player clicked the "X" at the corner of the window.
  terminate()
 
  elif event.type == KEYDOWN:
  # Handle key presses
  keyPressed = True
  if event.key == K_LEFT:
   playerMoveTo = LEFT
  elif event.key == K_RIGHT:
   playerMoveTo = RIGHT
  elif event.key == K_UP:
   playerMoveTo = UP
  elif event.key == K_DOWN:
   playerMoveTo = DOWN
 
  # Set the camera move mode.
  elif event.key == K_a:
   cameraLeft = True
  elif event.key == K_d:
   cameraRight = True
  elif event.key == K_w:
   cameraUp = True
  elif event.key == K_s:
   cameraDown = True
 
  elif event.key == K_n:
   return 'next'
  elif event.key == K_b:
   return 'back'
 
  elif event.key == K_ESCAPE:
   terminate() # Esc key quits.
  elif event.key == K_BACKSPACE:
   return 'reset' # Reset the level.
  elif event.key == K_p:
   # Change the player image to the next one.
   currentImage += 1
   if currentImage >= len(PLAYERIMAGES):
   # After the last player image, use the first one.
   currentImage = 0
   mapNeedsRedraw = True
 
  elif event.type == KEYUP:
  # Unset the camera move mode.
  if event.key == K_a:
   cameraLeft = False
  elif event.key == K_d:
   cameraRight = False
  elif event.key == K_w:
   cameraUp = False
  elif event.key == K_s:
   cameraDown = False
 
 if playerMoveTo != None and not levelIsComplete:
  # If the player pushed a key to move, make the move
  # (if possible) and push any stars that are pushable.
  moved = makeMove(mapObj, gameStateObj, playerMoveTo)
 
  if moved:
  # increment the step counter.
  gameStateObj['stepCounter'] += 1
  mapNeedsRedraw = True
 
  if isLevelFinished(levelObj, gameStateObj):
  # level is solved, we should show the "Solved!" image.
  levelIsComplete = True
  keyPressed = False
 
 DISPLAYSURF.fill(BGCOLOR)
 
 if mapNeedsRedraw:
  mapSurf = drawMap(mapObj, gameStateObj, levelObj['goals'])
  mapNeedsRedraw = False
 
 if cameraUp and cameraOffsetY < MAX_CAM_X_PAN:
  cameraOffsetY += CAM_MOVE_SPEED
 elif cameraDown and cameraOffsetY > -MAX_CAM_X_PAN:
  cameraOffsetY -= CAM_MOVE_SPEED
 if cameraLeft and cameraOffsetX < MAX_CAM_Y_PAN:
  cameraOffsetX += CAM_MOVE_SPEED
 elif cameraRight and cameraOffsetX > -MAX_CAM_Y_PAN:
  cameraOffsetX -= CAM_MOVE_SPEED
 
 # Adjust mapSurf's Rect object based on the camera offset.
 mapSurfRect = mapSurf.get_rect()
 mapSurfRect.center = (HALF_WINWIDTH + cameraOffsetX, HALF_WINHEIGHT + cameraOffsetY)
 
 # Draw mapSurf to the DISPLAYSURF Surface object.
 DISPLAYSURF.blit(mapSurf, mapSurfRect)
 
 DISPLAYSURF.blit(levelSurf, levelRect)
 stepSurf = BASICFONT.render('Steps: %s' % (gameStateObj['stepCounter']), 1, TEXTCOLOR)
 stepRect = stepSurf.get_rect()
 stepRect.bottomleft = (20, WINHEIGHT - 10)
 DISPLAYSURF.blit(stepSurf, stepRect)
 
 if levelIsComplete:
  # is solved, show the "Solved!" image until the player
  # has pressed a key.
  solvedRect = IMAGESDICT['solved'].get_rect()
  solvedRect.center = (HALF_WINWIDTH, HALF_WINHEIGHT)
  DISPLAYSURF.blit(IMAGESDICT['solved'], solvedRect)
 
  if keyPressed:
  return 'solved'
 
 pygame.display.update() # draw DISPLAYSURF to the screen.
 FPSCLOCK.tick()
 
 
def isWall(mapObj, x, y):
 """Returns True if the (x, y) position on
 the map is a wall, otherwise return False."""
 if x < 0 or x >= len(mapObj) or y < 0 or y >= len(mapObj[x]):
 return False # x and y aren't actually on the map.
 elif mapObj[x][y] in ('#', 'x'):
 return True # wall is blocking
 return False
 
 
def decorateMap(mapObj, startxy):
 """Makes a copy of the given map object and modifies it.
 Here is what is done to it:
 * Walls that are corners are turned into corner pieces.
 * The outside/inside floor tile distinction is made.
 * Tree/rock decorations are randomly added to the outside tiles.
 Returns the decorated map object."""
 
 startx, starty = startxy # Syntactic sugar
 
 # Copy the map object so we don't modify the original passed
 mapObjCopy = copy.deepcopy(mapObj)
 
 # Remove the non-wall characters from the map data
 for x in range(len(mapObjCopy)):
 for y in range(len(mapObjCopy[0])):
  if mapObjCopy[x][y] in ('$', '.', '@', '+', '*'):
  mapObjCopy[x][y] = ' '
 
 # Flood fill to determine inside/outside floor tiles.
 floodFill(mapObjCopy, startx, starty, ' ', 'o')
 
 # Convert the adjoined walls into corner tiles.
 for x in range(len(mapObjCopy)):
 for y in range(len(mapObjCopy[0])):
 
  if mapObjCopy[x][y] == '#':
  if (isWall(mapObjCopy, x, y-1) and isWall(mapObjCopy, x+1, y)) or \
   (isWall(mapObjCopy, x+1, y) and isWall(mapObjCopy, x, y+1)) or \
   (isWall(mapObjCopy, x, y+1) and isWall(mapObjCopy, x-1, y)) or \
   (isWall(mapObjCopy, x-1, y) and isWall(mapObjCopy, x, y-1)):
   mapObjCopy[x][y] = 'x'
 
  elif mapObjCopy[x][y] == ' ' and random.randint(0, 99) < OUTSIDE_DECORATION_PCT:
  mapObjCopy[x][y] = random.choice(list(OUTSIDEDECOMAPPING.keys()))
 
 return mapObjCopy
 
 
def isBlocked(mapObj, gameStateObj, x, y):
 """Returns True if the (x, y) position on the map is
 blocked by a wall or star, otherwise return False."""
 
 if isWall(mapObj, x, y):
 return True
 
 elif x < 0 or x >= len(mapObj) or y < 0 or y >= len(mapObj[x]):
 return True # x and y aren't actually on the map.
 
 elif (x, y) in gameStateObj['stars']:
 return True # a star is blocking
 
 return False
 
 
def makeMove(mapObj, gameStateObj, playerMoveTo):
 """Given a map and game state object, see if it is possible for the
 player to make the given move. If it is, then change the player's
 position (and the position of any pushed star). If not, do nothing.
 Returns True if the player moved, otherwise False."""
 
 # Make sure the player can move in the direction they want.
 playerx, playery = gameStateObj['player']
 
 # This variable is "syntactic sugar". Typing "stars" is more
 # readable than typing "gameStateObj['stars']" in our code.
 stars = gameStateObj['stars']
 
 # The code for handling each of the directions is so similar aside
 # from adding or subtracting 1 to the x/y coordinates. We can
 # simplify it by using the xOffset and yOffset variables.
 if playerMoveTo == UP:
 xOffset = 0
 yOffset = -1
 elif playerMoveTo == RIGHT:
 xOffset = 1
 yOffset = 0
 elif playerMoveTo == DOWN:
 xOffset = 0
 yOffset = 1
 elif playerMoveTo == LEFT:
 xOffset = -1
 yOffset = 0
 
 # See if the player can move in that direction.
 if isWall(mapObj, playerx + xOffset, playery + yOffset):
 return False
 else:
 if (playerx + xOffset, playery + yOffset) in stars:
  # There is a star in the way, see if the player can push it.
  if not isBlocked(mapObj, gameStateObj, playerx + (xOffset*2), playery + (yOffset*2)):
  # Move the star.
  ind = stars.index((playerx + xOffset, playery + yOffset))
  stars[ind] = (stars[ind][0] + xOffset, stars[ind][1] + yOffset)
  else:
  return False
 # Move the player upwards.
 gameStateObj['player'] = (playerx + xOffset, playery + yOffset)
 return True
 
 
def startScreen():
 """Display the start screen (which has the title and instructions)
 until the player presses a key. Returns None."""
 
 # Position the title image.
 titleRect = IMAGESDICT['title'].get_rect()
 topCoord = 50 # topCoord tracks where to position the top of the text
 titleRect.top = topCoord
 titleRect.centerx = HALF_WINWIDTH
 topCoord += titleRect.height
 
 # Unfortunately, Pygame's font & text system only shows one line at
 # a time, so we can't use strings with \n newline characters in them.
 # So we will use a list with each line in it.
 instructionText = ['Push the stars over the marks.',
   'Arrow keys to move, WASD for camera control, P to change character.',
   'Backspace to reset level, Esc to quit.',
   'N for next level, B to go back a level.']
 
 # Start with drawing a blank color to the entire window:
 DISPLAYSURF.fill(BGCOLOR)
 
 # Draw the title image to the window:
 DISPLAYSURF.blit(IMAGESDICT['title'], titleRect)
 
 # Position and draw the text.
 for i in range(len(instructionText)):
 instSurf = BASICFONT.render(instructionText[i], 1, TEXTCOLOR)
 instRect = instSurf.get_rect()
 topCoord += 10 # 10 pixels will go in between each line of text.
 instRect.top = topCoord
 instRect.centerx = HALF_WINWIDTH
 topCoord += instRect.height # Adjust for the height of the line.
 DISPLAYSURF.blit(instSurf, instRect)
 
 while True: # Main loop for the start screen.
 for event in pygame.event.get():
  if event.type == QUIT:
  terminate()
  elif event.type == KEYDOWN:
  if event.key == K_ESCAPE:
   terminate()
  return # user has pressed a key, so return.
 
 # Display the DISPLAYSURF contents to the actual screen.
 pygame.display.update()
 FPSCLOCK.tick()
 
 
def readLevelsFile(filename):
 assert os.path.exists(filename), 'Cannot find the level file: %s' % (filename)
 mapFile = open(filename, 'r')
 # Each level must end with a blank line
 content = mapFile.readlines() + ['\r\n']
 mapFile.close()
 
 levels = [] # Will contain a list of level objects.
 levelNum = 0
 mapTextLines = [] # contains the lines for a single level's map.
 mapObj = [] # the map object made from the data in mapTextLines
 for lineNum in range(len(content)):
 # Process each line that was in the level file.
 line = content[lineNum].rstrip('\r\n')
 
 if ';' in line:
  # Ignore the ; lines, they're comments in the level file.
  line = line[:line.find(';')]
 
 if line != '':
  # This line is part of the map.
  mapTextLines.append(line)
 elif line == '' and len(mapTextLines) > 0:
  # A blank line indicates the end of a level's map in the file.
  # Convert the text in mapTextLines into a level object.
 
  # Find the longest row in the map.
  maxWidth = -1
  for i in range(len(mapTextLines)):
  if len(mapTextLines[i]) > maxWidth:
   maxWidth = len(mapTextLines[i])
  # Add spaces to the ends of the shorter rows. This
  # ensures the map will be rectangular.
  for i in range(len(mapTextLines)):
  mapTextLines[i] += ' ' * (maxWidth - len(mapTextLines[i]))
 
  # Convert mapTextLines to a map object.
  for x in range(len(mapTextLines[0])):
  mapObj.append([])
  for y in range(len(mapTextLines)):
  for x in range(maxWidth):
   mapObj[x].append(mapTextLines[y][x])
 
  # Loop through the spaces in the map and find the @, ., and $
  # characters for the starting game state.
  startx = None # The x and y for the player's starting position
  starty = None
  goals = [] # list of (x, y) tuples for each goal.
  stars = [] # list of (x, y) for each star's starting position.
  for x in range(maxWidth):
  for y in range(len(mapObj[x])):
   if mapObj[x][y] in ('@', '+'):
   # '@' is player, '+' is player & goal
   startx = x
   starty = y
   if mapObj[x][y] in ('.', '+', '*'):
   # '.' is goal, '*' is star & goal
   goals.append((x, y))
   if mapObj[x][y] in ('$', '*'):
   # '$' is star
   stars.append((x, y))
 
  # Basic level design sanity checks:
  assert startx != None and starty != None, 'Level %s (around line %s) in %s is missing a "@" or "+" to mark the start point.' % (levelNum+1, lineNum, filename)
  assert len(goals) > 0, 'Level %s (around line %s) in %s must have at least one goal.' % (levelNum+1, lineNum, filename)
  assert len(stars) >= len(goals), 'Level %s (around line %s) in %s is impossible to solve. It has %s goals but only %s stars.' % (levelNum+1, lineNum, filename, len(goals), len(stars))
 
  # Create level object and starting game state object.
  gameStateObj = {'player': (startx, starty),
    'stepCounter': 0,
    'stars': stars}
  levelObj = {'width': maxWidth,
   'height': len(mapObj),
   'mapObj': mapObj,
   'goals': goals,
   'startState': gameStateObj}
 
  levels.append(levelObj)
 
  # Reset the variables for reading the next map.
  mapTextLines = []
  mapObj = []
  gameStateObj = {}
  levelNum += 1
 return levels
 
 
def floodFill(mapObj, x, y, oldCharacter, newCharacter):
 """Changes any values matching oldCharacter on the map object to
 newCharacter at the (x, y) position, and does the same for the
 positions to the left, right, down, and up of (x, y), recursively."""
 
 # In this game, the flood fill algorithm creates the inside/outside
 # floor distinction. This is a "recursive" function.
 # For more info on the Flood Fill algorithm, see:
 # http://en.wikipedia.org/wiki/Flood_fill
 if mapObj[x][y] == oldCharacter:
 mapObj[x][y] = newCharacter
 
 if x < len(mapObj) - 1 and mapObj[x+1][y] == oldCharacter:
 floodFill(mapObj, x+1, y, oldCharacter, newCharacter) # call right
 if x > 0 and mapObj[x-1][y] == oldCharacter:
 floodFill(mapObj, x-1, y, oldCharacter, newCharacter) # call left
 if y < len(mapObj[x]) - 1 and mapObj[x][y+1] == oldCharacter:
 floodFill(mapObj, x, y+1, oldCharacter, newCharacter) # call down
 if y > 0 and mapObj[x][y-1] == oldCharacter:
 floodFill(mapObj, x, y-1, oldCharacter, newCharacter) # call up
 
 
def drawMap(mapObj, gameStateObj, goals):
 """Draws the map to a Surface object, including the player and
 stars. This function does not call pygame.display.update(), nor
 does it draw the "Level" and "Steps" text in the corner."""
 
 # mapSurf will be the single Surface object that the tiles are drawn
 # on, so that it is easy to position the entire map on the DISPLAYSURF
 # Surface object. First, the width and height must be calculated.
 mapSurfWidth = len(mapObj) * TILEWIDTH
 mapSurfHeight = (len(mapObj[0]) - 1) * TILEFLOORHEIGHT + TILEHEIGHT
 mapSurf = pygame.Surface((mapSurfWidth, mapSurfHeight))
 mapSurf.fill(BGCOLOR) # start with a blank color on the surface.
 
 # Draw the tile sprites onto this surface.
 for x in range(len(mapObj)):
 for y in range(len(mapObj[x])):
  spaceRect = pygame.Rect((x * TILEWIDTH, y * TILEFLOORHEIGHT, TILEWIDTH, TILEHEIGHT))
  if mapObj[x][y] in TILEMAPPING:
  baseTile = TILEMAPPING[mapObj[x][y]]
  elif mapObj[x][y] in OUTSIDEDECOMAPPING:
  baseTile = TILEMAPPING[' ']
 
  # First draw the base ground/wall tile.
  mapSurf.blit(baseTile, spaceRect)
 
  if mapObj[x][y] in OUTSIDEDECOMAPPING:
  # Draw any tree/rock decorations that are on this tile.
  mapSurf.blit(OUTSIDEDECOMAPPING[mapObj[x][y]], spaceRect)
  elif (x, y) in gameStateObj['stars']:
  if (x, y) in goals:
   # A goal AND star are on this space, draw goal first.
   mapSurf.blit(IMAGESDICT['covered goal'], spaceRect)
  # Then draw the star sprite.
  mapSurf.blit(IMAGESDICT['star'], spaceRect)
  elif (x, y) in goals:
  # Draw a goal without a star on it.
  mapSurf.blit(IMAGESDICT['uncovered goal'], spaceRect)
 
  # Last draw the player on the board.
  if (x, y) == gameStateObj['player']:
  # Note: The value "currentImage" refers
  # to a key in "PLAYERIMAGES" which has the
  # specific player image we want to show.
  mapSurf.blit(PLAYERIMAGES[currentImage], spaceRect)
 
 return mapSurf
 
 
def isLevelFinished(levelObj, gameStateObj):
 """Returns True if all the goals have stars in them."""
 for goal in levelObj['goals']:
 if goal not in gameStateObj['stars']:
  # Found a space with a goal but no star on it.
  return False
 return True
 
 
def terminate():
 pygame.quit()
 sys.exit()
 
 
if __name__ == '__main__':
 main()