Rotation attribute added. Now is possible to define which blocks make

sense to rotate.

Author: benycze

block.py

@@ -30,7 +30,7 @@ class Block(object):
     Class for handling of tetris block
     """    
 
-    def __init__(self,shape,x,y,screen,color):
+    def __init__(self,shape,x,y,screen,color,rotate_en):
         """
         Initialize the tetris block class
 
@@ -41,6 +41,7 @@ def __init__(self,shape,x,y,screen,color):
             - y - Y coordinate of first tetris shape block
             - screen  - screen to draw on
             - color - the color of each shape block in RGB notation
+            - rotate_en - enable or disable the rotation
         """
         # The initial shape (convert all to Rect objects)
         self.shape = []
@@ -49,6 +50,8 @@ def __init__(self,shape,x,y,screen,color):
             by = sh[1]*constants.BHEIGHT + y
             block = pygame.Rect(bx,by,constants.BWIDTH,constants.BHEIGHT)
             self.shape.append(block)     
+        # Setup the rotation attribute
+        self.rotate_en = rotate_en
         # Setup the rest of variables
         self.x = x
         self.y = y
@@ -134,8 +137,10 @@ def rotate(self):
         Setup the rotation value to 90 degrees.
         """
         # Setup the rotation and update coordinates of all shape blocks.
-        self.diff_rotation = 90
-        self._update()
+        # The block is rotated iff the rotation is enabled
+        if self.rotate_en:
+            self.diff_rotation = 90
+            self._update()
 
     def _update(self):
         """

tetris.py

@@ -52,15 +52,16 @@ def __init__(self,bx,by):
         self.start_x = math.ceil(self.resx/2.0)
         self.start_y = constants.BOARD_UP_MARGIN + constants.BOARD_HEIGHT + constants.BOARD_MARGIN
         # Blocka data (shapes and colors). The shape is encoded in the list of [X,Y] points. Each point
-        # represents the relative position.
+        # represents the relative position. The true/false value is used for the configuration of rotation where
+        # False means no rotate and True allows the rotation.
         self.block_data = (
-            ([[0,0],[1,0],[2,0],[3,0]],constants.RED),     # I block 
-            ([[0,0],[1,0],[0,1],[-1,1]],constants.GREEN),  # S block 
-            ([[0,0],[1,0],[2,0],[2,1]],constants.BLUE),    # J block
-            ([[0,0],[0,1],[1,0],[1,1]],constants.ORANGE),  # O block
-            ([[-1,0],[0,0],[0,1],[1,1]],constants.GOLD),   # Z block
-            ([[0,0],[1,0],[2,0],[1,1]],constants.PURPLE),  # T block
-            ([[0,0],[1,0],[2,0],[0,1]],constants.CYAN),    # J block
+            ([[0,0],[1,0],[2,0],[3,0]],constants.RED,True),     # I block 
+            ([[0,0],[1,0],[0,1],[-1,1]],constants.GREEN,True),  # S block 
+            ([[0,0],[1,0],[2,0],[2,1]],constants.BLUE,True),    # J block
+            ([[0,0],[0,1],[1,0],[1,1]],constants.ORANGE,False), # O block
+            ([[-1,0],[0,0],[0,1],[1,1]],constants.GOLD,True),   # Z block
+            ([[0,0],[1,0],[2,0],[1,1]],constants.PURPLE,True),  # T block
+            ([[0,0],[1,0],[2,0],[0,1]],constants.CYAN,True),    # J block
         )
         # Compute the number of blocks. When the number of blocks is even, we can use it directly but 
         # we have to decrese the number of blocks in line by one when the number is odd (because of the used margin).
@@ -303,7 +304,7 @@ def get_block(self):
             # Get the block and add it into the block list(static for now)
             tmp = random.randint(0,len(self.block_data)-1)
             data = self.block_data[tmp]
-            self.active_block = block.Block(data[0],self.start_x,self.start_y,self.screen,data[1])
+            self.active_block = block.Block(data[0],self.start_x,self.start_y,self.screen,data[1],data[2])
             self.blk_list.append(self.active_block)
             self.new_block = False