#!/usr/bin/env python
#
#   l o d e 4 . p y
#
import time, ttyLinux
from util import *

class Player :
	def setDirection (self, ch) : pass

	def move (self) :
		global inPlay, you
		spot = getSpot(self.row,self.col)
		lspot = getSpot(self.row,self.col-1)
		rspot = getSpot(self.row,self.col+1)
  
		horz,vert = self.dir
		writeScreen(self.row,self.col,spot)
		if   spot == '_' : self.col += horz  # by left/right arself.row
		elif spot == '|' : self.row -= vert  # by up/down arrow
		elif spot == ' ' : self.row += 1     # always fall in air
		# Ok to walk horizontally past a ladder or obstacle
		if   lspot == '_' and horz == -1 : self.col -= 1
		elif rspot == '_' and horz ==  1 : self.col += 1
		writeScreen(self.row,self.col,self.face)
		if self == you and self.row > 23 : inPlay = 0

class You(Player) :
	def __init__ (self, Row=1, Col=10, Face='^') :
		self.row = Row; self.col = Col; self.face=Face
		self.dir = (0,0); self.score=0

	def setDirection (self, ch) :
		here = getSpot(self.row,self.col)
		if here == '*' :
			self.score += 10
			setSpot(self.row,self.col,"_")
			writeScreen(self.row,self.col,"_")   # update the screen

		if ch == '\033[A' : self.dir=( 0, 1)  # up
		if ch == '\033[B' : self.dir=( 0,-1)  # down
		if ch == '\033[C' : self.dir=( 1, 0)  # right
		if ch == '\033[D' : self.dir=(-1, 0)  # left
		if ch == 'a' : burn(self.row, self.col-1)
		if ch == 's' : burn(self.row, self.col+1)

class Robot(Player) :
	def __init__ (self, Row=1, Col=12, Face='&') :
		self.row = Row; self.col = Col; self.face=Face
		self.dir = (0,0)

	def setDirection (self, ch) :
		global inPlay, you
		# did we tag him?
		if you.row == self.row and you.col == self.col : inPlay=0
		# same level. run toward you
		if self.row == you.row :
			if self.col > you.col : self.dir=(-1,0) # left
			if self.col < you.col : self.dir=( 1,0) # right
		else :
			me = getSpot(self.row,self.col)  # where am I
			if me == "|" : # on a ladder
				if self.row > you.row : self.dir=(0, 1) # up
				if self.row < you.row : self.dir=(0,-1) # down
	
def burn (row,col) :
	"Burn a hole in the catwalk at row,col"
	setSpot(row,col," ")       # set the board
	writeScreen(row,col," ")   # update the screen

def main () :
	setBoard(0)
	try :
		ttyLinux.setSpecial()
		playGame()
	finally : ttyLinux.setNormal()

def playGame() :
	global inPlay, you
	you = You()
	writeBoard ()
	players = [you]
	clock = 0
	inPlay = 1
	while inPlay :
		clock += 1
		if clock == 90 : players.append(Robot())
		time.sleep(.1)
		keys = ttyLinux.readLookAhead()
		for player in players :
			player.setDirection(keys)
			player.move()
		writeScreen (20,0,'Score: %d' % you.score)
	writeScreen (20,0,'')

if __name__ == "__main__" : main()