From 6b9a2f86f2a4fd13194b79118f00f7c0234d45a3 Mon Sep 17 00:00:00 2001
From: Alex Yatskov <alex@foosoft.net>
Date: Mon, 19 May 2014 12:43:32 +0900
Subject: [PATCH] Support for preview tetrad

---
 tetris.py | 34 +++++++++++++++++++++++++++-------
 1 file changed, 27 insertions(+), 7 deletions(-)

diff --git a/tetris.py b/tetris.py
index 0b046ed..6fd4614 100755
--- a/tetris.py
+++ b/tetris.py
@@ -70,6 +70,7 @@ class Tetrad:
 
 class Board:
     border_color = 0xeeeeec
+    preview_color = 0xd3d7cf
     block_colors = [
         (0x555753, 0x2e3436), # Aluminium
         (0xedd400, 0xfce94f), # Butter
@@ -93,7 +94,7 @@ class Board:
         self.blocks = [[0]*grid_dims[0] for i in range(grid_dims[1])]
 
 
-    def render(self, surface, tetrad):
+    def render(self, surface):
         self.render_frame(surface)
         self.render_blocks(surface)
 
@@ -105,19 +106,27 @@ class Board:
     def render_blocks(self, surface):
         for y in xrange(self.grid_dims[1]):
             for x in xrange(self.grid_dims[0]):
-                self.render_block(surface, self.blocks[y][x], False, (x, y))
+                self.render_block(surface, self.blocks[y][x], (x, y))
 
 
-    def render_tetrad(self, surface, tetrad, transparent):
+    def render_tetrad(self, surface, tetrad, preview=False):
         color = tetrad.color()
         for point in tetrad.layout():
-            self.render_block(surface, color, transparent, point)
+            self.render_block(surface, color, point, preview)
 
 
-    def render_block(self, surface, color, transparent, position):
+    def render_block(self, surface, color, position, preview=False):
         block_rect = self.block_screen_rect(position)
-        pygame.draw.rect(surface, self.block_colors[color][1], block_rect)
-        pygame.draw.rect(surface, self.block_colors[color][0], block_rect, 1)
+        if preview:
+            color = 0
+
+        color_outer, color_inner = self.block_colors[color]
+        pygame.draw.rect(surface, color_inner, block_rect)
+        pygame.draw.rect(surface, color_outer, block_rect, 1)
+
+        if preview:
+            position = block_rect.centerx, block_rect.centery
+            pygame.draw.circle(surface, self.preview_color, position, 2)
 
 
     def block_screen_rect(self, position):
@@ -176,6 +185,7 @@ class Game:
 
         self.tetrad = Tetrad.random()
         self.tetrad_next = Tetrad.random()
+        self.tetrad_preview = None
 
         self.counter = 0
         self.active = True
@@ -187,6 +197,8 @@ class Game:
 
     def render(self, surface):
         self.board.render(surface)
+        if self.tetrad_preview is not None:
+            self.board.render_tetrad(surface, self.tetrad_preview, True)
         self.board.render_tetrad(surface, self.tetrad)
 
         self.board_prev.render(surface)
@@ -197,6 +209,12 @@ class Game:
         if not self.active:
             return
 
+        self.tetrad_preview = None
+        tetrad_preview = self.tetrad.moved_down()
+        while self.board.can_place_tetrad(tetrad_preview):
+            self.tetrad_preview = tetrad_preview
+            tetrad_preview = self.tetrad_preview.moved_down()
+
         self.counter += elapsed
         if self.counter > self.interval:
             self.lower_tetrad()
@@ -314,6 +332,8 @@ class Engine:
                 self.game.rotate()
             elif event.key == pygame.K_SPACE:
                 self.game.drop()
+            elif event.key == pygame.K_n:
+                self.game.new_game()
             elif event.key == pygame.K_ESCAPE:
                 return False