#!/usr/bin/env python
"""
Tetris Tk - A tetris clone written in Python using the Tkinter GUI library.

Controls:
    Left/a      Move left
    Right/d     Move right
    Down/s      Move down
    Up/w        Rotate anti-clockwise (to the left)
"""
from Tkinter import *
from time import sleep
import random
import sys
from tetrisGUI import GUI
from tetris_shape import *

MAXX = 10
MAXY = 18
NO_OF_LEVELS = 10

LEFT = "left"
RIGHT = "right"
DOWN = "down"
direction_d = { "left": (-1, 0), "right": (1, 0), "down": (0, 1) }

COLORS = ["orange", "red", "green", "blue", "purple", "yellow", "magenta"]
#COLORS = ["gray"]

class Board():
    """
    The board represents the tetris playing area. A grid of x by y blocks.
    """
    def __init__(self, max_x=10, max_y=20): 
        # blocks are stored in dict of (x,y)->"color"
        self.landed = {}
        self.max_x = max_x
        self.max_y = max_y
  
    def receive_lines(self, num_lines):
        #shift lines up
        for y in range(self.max_y-num_lines):
            for x in xrange(self.max_x):
                block_color = self.landed.pop((x,y+num_lines),None)
                if block_color:
                    self.landed[(x,y)] = block_color
        #put in new lines
        for j in range(num_lines):
            for i in random.sample(xrange(self.max_x), random.choice([6,7])):
                self.landed[(i,self.max_y-1-j)] = random.choice(COLORS)
                
    def check_for_complete_row( self, blocks ):
        """
        Look for a complete row of blocks, from the bottom up until the top row
        or until an empty row is reached.
        """
        rows_deleted = 0
        
        # Add the blocks to those in the grid that have already 'landed'
        for block in blocks:
            self.landed[ block.coord() ] = block.color
        
        empty_row = 0
        # find the first empty row
        for y in xrange(self.max_y -1, -1, -1):
            row_is_empty = True
            for x in xrange(self.max_x):
                if self.landed.get((x,y), None):
                    row_is_empty = False
                    break;
            if row_is_empty:
                empty_row = y
                break

        # Now scan up and until a complete row is found. 
        y = self.max_y - 1
        while y > empty_row:
 
            complete_row = True
            for x in xrange(self.max_x):
                if self.landed.get((x,y), None) is None:
                    complete_row = False
                    break;

            if complete_row:
                rows_deleted += 1
                
                #delete the completed row
                for x in xrange(self.max_x):
                    self.landed.pop((x,y))
                    
                # move all the rows above it down
                for ay in xrange(y-1, empty_row, -1):
                    for x in xrange(self.max_x):
                        block_color = self.landed.pop((x,ay), None)
                        if block_color:
                            dx,dy = direction_d[DOWN]
                            self.landed[(x+dx, ay+dy)] = block_color

                # move the empty row down index down too
                empty_row +=1
                # y stays same as row above has moved down.       
            else:
                y -= 1
            
        # return the score, calculated by the number of rows deleted.        
        return rows_deleted
                
    def output( self ):
        for y in xrange(self.max_y):
            line = []
            for x in xrange(self.max_x):
                if self.landed.get((x,y), None): line.append("X")
                else: line.append(".")
            print "".join(line)
    
    def check_block( self, (x, y) ):
        """
        Check if the x, y coordinate can have a block placed there.
        That is; if there is a 'landed' block there or it is outside the
        board boundary, then return False, otherwise return true.
        """
        if x < 0 or x >= self.max_x or y < 0 or y >= self.max_y:
            return False
        elif self.landed.has_key( (x, y) ):
            return False
        else:
            return True

#represents a player. each player has a board and can get new shapes...
#
class Player():
    def __init__(self,parent, gs, myBoard, otherBoard):
        print "initialize player"
        self.parent = parent
        self.board = myBoard
        self.other_board = otherBoard
        self.score = 0
        self.shapes = [square_shape,
                      t_shape,
                      l_shape,
                      reverse_l_shape,
                      z_shape,
                      s_shape,
                      i_shape ]
        self.gs = gs
        self.shape = self.get_next_shape()
        
    def handle_move(self, direction):
        #if you can't move then you've hit something
        success = self.shape.move( direction )
        if not success:
                
            # if you're heading down then the shape has 'landed'
            if direction == DOWN:
                points = self.board.check_for_complete_row(
                    self.shape.blocks)
                del self.shape
                self.shape = self.get_next_shape()
                
                self.score += points
                if points > 1:
                    self.other_board.receive_lines(points-1) 
   
                # If the shape returned is None, then this indicates that
                # that the check before creating it failed and the
                # game is over!
                if self.shape is None:
                    self.end_game() #loss!
                
                # do we go up a level?
                if (self.gs.level < NO_OF_LEVELS and 
                    self.score >= self.gs.thresholds[self.gs.level]):
                    self.gs.level+=1
                    self.gs.delay-=100
                
                # Signal that the shape has 'landed'
                return False
        return True

    def left(self):
        if self.shape:
            self.handle_move( LEFT )
        
    def right(self):
        if self.shape:
            self.handle_move( RIGHT )
            
    def down(self):
        if self.shape:
            self.handle_move( DOWN )
            
    def up(self):
        if self.shape:
            self.shape.rotate(clockwise=False)
 
    def move_my_shape( self ):
        if self.shape:
            self.handle_move( DOWN )
        
    def get_next_shape( self ):
        #Randomly select which tetrominoe will be used next.
        the_shape = self.shapes[ random.randint(0,len(self.shapes)-1) ]
        return the_shape.check_and_create(self.board)

#contains variables that are shared between the players:
#levels, delay time, etc?
class GameState():
    def __init__(self, gui):
        self.level = 1
        self.delay = 1000
        self.thresholds = range(10,100,10)
       
        
#runs the overall game. initializes both player and any displays
class TwoPlayerGame(object):
    """
    Main game loop and receives GUI callback events for keypresses etc...
    """
    def __init__(self, parent):
        print "initialize game"
        self.parent = parent
        self.gui = GUI(parent,20,MAXX,MAXY)
        self.gui.pack(side=BOTTOM)
        self.gameState = GameState(self.gui)
        
        board0 = Board(MAXX,MAXY)
        board1 = Board(MAXX,MAXY)
        player0 = Player(parent, self.gameState, board0, board1)
        player1 = Player(parent, self.gameState, board1, board0)
        self.players = [player0, player1]

        self.parent.bind("<Key>", self.handle_key_press)
        self.parent.after( self.gameState.delay, self.gravity)

    def gravity(self):
        for p in self.players:
            p.move_my_shape()
        self.update_gui()
        self.parent.after( self.gameState.delay, self.gravity)

    def handle_key_press(self,event):
        k = event.keysym
        if k=="Left":
            self.players[1].left()
        elif k=="Right":
            self.players[1].right()
        elif k=="Down":
            self.players[1].down()
        elif k=="Up":
            self.players[1].up()
        elif k=="a":
            self.players[0].left()
        elif k=="d":
            self.players[0].right()
        elif k=="s":
            self.players[0].down()
        elif k=="w":
            self.players[0].up()
        self.update_gui()
            
    def update_gui(self):
        self.gui.draw_board(self.players)
        
    def end_game(self):
        #fix this
        #show_end_seq()
        Toplevel().destroy()
        self.parent.destroy()
        sys.exit(0)

        
        
if __name__ == "__main__":
    root = Tk()
    root.title("Tetris Tk")
    theGame = TwoPlayerGame( root )
    root.mainloop()