xephinh.py

import pygame
import random
import ImgModify
import os

"""
10x20 square grid
shapes: S Z I O J L T
represented in order by 0-6
"""

pygame.font.init()

# Global variables
s_width = 800
s_height = 700
play_width = 300  # meaning 300//10 = 30 width per block
play_height = 600  # meaning 600//20 = 30 height per block
block_size = 30

top_left_x = (s_width - play_width)//2
top_left_y = (s_height - play_height)

Tho = ImgModify.getImgColor("Tho.jpeg", 50, 50)
Winner1 = ImgModify.getImgColor("Winner1.jpeg", 50, 50)
Winner2 = ImgModify.getImgColor("Winner2.png", 50, 50)
Winner3 = ImgModify.getImgColor("Winner3.jpeg", 50, 50)
Winner4 = ImgModify.getImgColor("Winner4.jpeg", 50, 50)
Winner5 = ImgModify.getImgColor("Winner5.jpeg", 50, 50)
Winner6 = ImgModify.getImgColor("Winner6.png", 50, 50)
Winner7 = ImgModify.getImgColor("Winner7.jpeg", 50, 50)
Winner8 = ImgModify.getImgColor("Winner8.jpeg", 50, 50)
Winner9 = ImgModify.getImgColor("Winner9.jpeg", 50, 50)
Defeat = ImgModify.getImgColor("Defeat.jpg", 140, 140)
Defeat1 = ImgModify.getImgColor("Defeat1.jpeg", 140, 140)
Defeat2 = ImgModify.getImgColor("Defeat2.jpeg", 140, 140)
Defeat3 = ImgModify.getImgColor("Defeat2.jpeg", 140, 140)
# Shape formats

S = [['.....',
      '.....',
      '..00.',
      '.00..',
      '.....'],
     ['.....',
      '..0..',
      '..00.',
      '...0.',
      '.....']]

Z = [['.....',
      '.....',
      '.00..',
      '..00.',
      '.....'],
     ['.....',
      '..0..',
      '.00..',
      '.0...',
      '.....']]

I = [['..0..',
      '..0..',
      '..0..',
      '..0..',
      '.....'],
     ['.....',
      '0000.',
      '.....',
      '.....',
      '.....']]

O = [['.....',
      '.....',
      '.00..',
      '.00..',
      '.....']]

J = [['.....',
      '...0.',
      '...0.',
      '..00.',
      '.....'],
     ['.....',
      '.0...',
      '.000.',
      '.....',
      '.....'],
     ['.....',
      '..00.',
      '..0..',
      '..0..',
      '.....'],
     ['.....',
      '.000.',
      '...0.',
      '.....',
      '.....']]

L = [['.....',
      '..0..',
      '..0..',
      '..00.',
      '.....'],
     ['.....',
      '.....',
      '.000.',
      '.0...',
      '.....'],
     ['.....',
      '.00..',
      '..0..',
      '..0..',
      '.....'],
     ['.....',
      '...0.',
      '.000.',
      '.....',
      '.....']]


T = [['.....',
      '..0..',
      '.000.',
      '.....',
      '.....'],
     ['.....',
      '..0..',
      '..00.',
      '..0..',
      '.....'],
     ['.....',
      '.....',
      '.000.',
      '..0..',
      '.....'],
     ['.....',
      '..0..',
      '.00..',
      '..0..',
      '.....']]

C = [['.....',
      '.0.0.',
      '.000.',
      '.....',
      '.....'],
     ['.....',
      '..00.',
      '..0..',
      '..00.',
      '.....'],
     ['.....',
      '.000.',
      '.0.0.',
      '.....',
      '.....'],
     ['.....',
      '.00..',
      '..0..',
      '.00..',
      '.....']]

P = [['.....',
      '..0..',
      '.000.',
      '.0.0.',
      '.....'],
     ['.....',
      '..00.',
      '.00..',
      '..00.',
      '.....'],
     ['.....',
      '.0.0.',
      '.000.',
      '..0..',
      '.....'],
     ['.....',
      '.00..',
      '..00.',
      '.00..',
      '.....']]

H = [['.....',
      '.0.0.',
      '.000.',
      '.0.0.',
      '.....'],
     ['.....',
      '.000.',
      '..0..',
      '.000.',
      '.....'],
     ['.....',
      '.0.0.',
      '.000.',
      '.0.0.',
      '.....'],
     ['.....',
      '.000.',
      '..0..',
      '.000.',
      '.....']]

#M = [['.0.0.',
#      '.....',
#      '0...0',
#      '.000.',
#      '.....'],
#     ['..0..',
#      '.0..0',
#      '.0...',
#      '.0..0',
#      '..0..'],
#     ['.....',
#      '.000.',
#      '0...0',
#      '.....',
#      '.0.0.'],
#    ['...0.',
#      '0..0.',
#      '....0',
#      '0..0.',
#      '...0.']]
shapes = [S, Z, I, O, J, L, T, C, P, H]
shape_colors = [(0, 255, 0), (255, 0, 0), (0, 0, 255), (255, 255, 0), (0, 255, 255), (255, 0, 255), (255, 165, 0), (165, 165, 0), (165, 0, 165), (123, 234, 12)]
# index 0-6 represent shape

class Piece(object):
  rows = 20  # height
  columns = 10  # width

  def __init__(self, column, row, shape):
    self.x = column
    self.y = row
    self.shape = shape
    self.color = shape_colors[shapes.index(shape)]
    self.rotation = 0  # number from 0-3

def create_grid(locked_positions):
  grid = [[(0,0,0) for x in range(10)] for y in range(20)]

  for i in range(len(grid)):
    for j in range(len(grid[i])):
      if (j,i) in locked_positions:
        c = locked_positions[(j,i)]
        grid[i][j] = c
  return grid

def convert_shape_format(shape):
  positions = []
  ## Determine rotation direction of current shape
  format = shape.shape[shape.rotation % len(shape.shape)]
  for i, line in enumerate(format):
    row = list(line)
    for j, column in enumerate(row):
      if column == '0':
        positions.append((shape.x + j, shape.y + i))
  for i, pos in enumerate(positions):
    positions[i] = (pos[0] - 2, pos[1] - 4)
  return positions


def valid_space(shape, grid):
  accepted_positions = [[(j,i) for j in range(10) if grid[i][j] == (0,0,0)] for i in range(20)]
  accepted_positions = [j for sub in accepted_positions for j in sub]
  formatted = convert_shape_format(shape)
  for pos in formatted:
    if pos not in accepted_positions:
      if pos[1] > -1:
        return False
  return True



def check_lost(positions):
  for pos in positions:
    x,y = pos
    if y < 1:
      return True
  return False


def get_shape(level):
  global shapes, shape_colors
  if level == 'd':
      return Piece(5,0,random.choice(shapes[:-3]))
  else:
    return Piece(5,0,random.choice(shapes))


def draw_text_middle(text, size, color, surface):
  font = pygame.font.SysFont('comicsans', size, bold = True)
  label = font.render(text, True, color)
  surface.blit(label, (top_left_x + play_width/2 - (label.get_width()/2), top_left_y + play_height/3 - label.get_height()/2))


def draw_grid(surface, row, col):
  sx = top_left_x
  sy = top_left_y
  for i in range(row):
    pygame.draw.line(surface, (128,128,128), (sx, sy + i*30), (sx + play_width, sy + i* 30)) # horizontal line
    for j in range(col):
      pygame.draw.line(surface, (128, 128, 128), (sx + j*30, sy), (sx + j*30, sy * play_height)) # vertical line


def clear_rows(grid, locked):
  # need to see if row is clear the shift every other row above down one
  ind = 0
  inc = 0
  for i in range(len(grid)-1, -1, -1):
    row = grid[i]
    if (0,0,0) not in row:
      inc +=1
      # add positions to remove from locked
      ind = i
      for j in range(len(row)):
        try:
          del locked[(j,i)]
        except:
          continue
      break

  if inc > 0:
    for key in sorted(list(locked), key = lambda x: x[1])[::-1]:
      x, y = key
      if y < ind:
        newKey = (x, y+inc)
        locked[newKey] = locked.pop(key)
    return True
  return False

def draw_next_shape(shape, surface):
  font = pygame.font.SysFont('comicsans', 30)
  label = font.render('Hinh tiep theo', True, (255,255,255))
  sx = top_left_x + play_width + 20
  sy = top_left_y + play_height/2 - 200
  format = shape.shape[shape.rotation % len(shape.shape)]
  for i, line in enumerate(format):
    row = list(line)
    for j, column in enumerate(row):
      if column == '0':
        pygame.draw.rect(surface, shape.color, (sx + j*30, sy + i*30, 30,30), 0)
  surface.blit(label, (sx + 10, sy - 30))


def draw_score(score, surface):
  font = pygame.font.SysFont('comicsans', 40)
  font2 = pygame.font.SysFont('comicsans', 50)
  label1 = font.render('Diem so', True, (255,255,0))
  label2 = font2.render(str(score), True, (255,255,255))
  surface.blit(label1, (top_left_x - 180,  top_left_y + play_height/2 + 30))
  surface.blit(label2, (top_left_x - 150,  top_left_y + play_height/2 + 80))


def draw_player(Img, surface):
    font = pygame.font.SysFont('comicsans', 60)
    label = font.render('Tho', True, (255, 165, 255))
    surface.blit(label, (top_left_x - 170,  top_left_y + play_height/2 - 300))
    sx = top_left_x - 185
    sy = top_left_y + play_height/2 - 200
    for i in range(len(Img)):
        for j in range(len(Img[i])):
            pygame.draw.rect(surface, Img[i][j], (sx + 3*j,  sy + 3*i, 3, 3), 0)


def draw_winnericon(Img, surface):
    sx = top_left_x +play_width + 60
    sy = top_left_y + play_height / 2 + 50
    for i in range(len(Img)):
        for j in range(len(Img[i])):
            pygame.draw.rect(surface, Img[i][j], (sx + 3 * j, sy + 3 * i, 3, 3), 0)


def draw_window(surface):
  surface.fill((0,0,0))
  #Tetris Title
  font = pygame.font.SysFont('comicsans', 60)
  label = font.render('TRO CHOI XEP HINH', True, (255,255,255))
  surface.blit(label, (top_left_x + play_width/2 - (label.get_width()/2), 10))
  for i in range(len(grid)):
    for j in range(len(grid[i])):
      pygame.draw.rect(surface, grid[i][j], (top_left_x + j*30, top_left_y + i*30, 30,30), 0)
  # draw grid and border
  draw_grid(surface, 20, 10)
  pygame.draw.rect(surface, (255,0,0), (top_left_x, top_left_y, play_width, play_height), 5)
  # pygame.display.update()


def draw_lose_window(Img, surface):
    surface.fill((0,0,0))
    font = pygame.font.SysFont('comicsans', 80)
    label = font.render('An ui ne!', True, (0, 255, 255))
    for i in range(len(Img)):
        for j in range(len(Img[i])):
            pygame.draw.rect(surface, Img[i][j], (50+j*5, i*5, 5, 5), 0)
    surface.blit(label, (top_left_x + play_width/2 - (label.get_width()/2), 10))


def draw_champion_window(surface, record):
    Champion = ImgModify.getImgColor('Champion.jpeg', 140, 140)
    surface.fill((0, 0, 0))
    font = pygame.font.SysFont('comicsans', 45)
    label = font.render('Chuc mung ky luc moi cua Tho:' + str(record), True, (255, 0, 0))
    for i in range(len(Champion)):
        for j in range(len(Champion[i])):
            pygame.draw.rect(surface, Champion[i][j], (50+j*5, i*5, 5, 5), 0)
    surface.blit(label, (top_left_x + play_width/2 - (label.get_width()/2), 10))


def main(win, difflevel):
  global grid 
  locked_positions = {}  # (x,y): (255,0,0)
  grid = create_grid(locked_positions)
  change_piece = False
  run = True
  current_piece = get_shape(difflevel)
  next_piece = get_shape(difflevel)
  clock = pygame.time.Clock()
  fall_time = 0
  level_time = 0
  fall_speed = 0.34
  score = 0
  winnercounter = 0
  winnerfile = [Winner1, Winner2, Winner3, Winner4, Winner5, Winner6, Winner7, Winner8, Winner9]
  defeatfile = [Defeat, Defeat1, Defeat2, Defeat3]
  while run:

     
    grid = create_grid(locked_positions)
    fall_time += clock.get_rawtime()
    level_time += clock.get_rawtime()
    clock.tick()

    if level_time/1000 > 4:
      level_time = 0
      if fall_speed > 0.15:
        fall_speed -= 0.005


    #PIECE FALLING
    if fall_time/1000 >= fall_speed:
      fall_time = 0
      current_piece.y += 1
      if not (valid_space(current_piece, grid)) and current_piece.y > 0:
        current_piece.y -= 1
        change_piece = True


    for event in pygame.event.get():
      if event.type == pygame.QUIT:
        run = False
        pygame.display.quit()
        quit()

      if event.type == pygame.KEYDOWN:
        if event.key == pygame.K_q:
          run = False
          pygame.display.quit()
          quit()
        
        if event.key == pygame.K_LEFT:
          current_piece.x -= 1
          if not valid_space(current_piece, grid):
            current_piece.x += 1
        
        elif event.key == pygame.K_RIGHT:
          current_piece.x +=1
          if not valid_space(current_piece, grid):
            current_piece.x -= 1
        
        elif event.key == pygame.K_UP:
          # rotate shape
          current_piece.rotation = (current_piece.rotation + 1)%len(current_piece.shape)
          if not valid_space(current_piece, grid):
            current_piece.rotation = (current_piece.rotation -1)%len(current_piece.shape)
        
        if event.key == pygame.K_DOWN:
          #move shape down
          current_piece.y +=1
          if not valid_space(current_piece, grid):
            current_piece.y -= 1
          
        if event.key == pygame.K_SPACE:
          while valid_space(current_piece, grid):
            current_piece.y += 1
          current_piece.y -= 1
    
    shape_pos = convert_shape_format(current_piece)

    # add piece to the  grid for drawing
    for i in range(len(shape_pos)):
      x, y = shape_pos[i]
      if y > -1:
        grid[y][x] = current_piece.color
    
    # IF PIECE HIT GROUND
    if change_piece:
      for pos in shape_pos:
        p = (pos[0], pos[1])
        locked_positions[p] = current_piece.color
      current_piece = next_piece
      next_piece = get_shape(difflevel)
      change_piece = False

      for i in range(4):
        if clear_rows(grid, locked_positions):
          grid = create_grid(locked_positions)
          score += 10
          winnercounter += 1
        else:
          break
    
    draw_window(win)
    draw_next_shape(next_piece, win)
    draw_score(score, win)
    draw_player(Tho, win)
    draw_winnericon(winnerfile[winnercounter%len(winnerfile)],win)

    pygame.display.update()

    if check_lost(locked_positions):
      run = False


  draw_text_middle("Thua rui Tho lam lai nhe!", 40, (0,255,255), win)
  pygame.display.update()
  pygame.time.delay(2000)

  draw_lose_window(defeatfile[random.randint(0,3)], win)
  pygame.display.update()
  pygame.time.delay(10000)

  curdir = os.path.dirname(__file__)
  with open(curdir + '/record.txt', "r+") as f:
      record1 = int(f.readline())
      record2 = int(f.readline())
      f.truncate(0)
      f.close()
  if difflevel == 'd':
      if score > record1:
          record1 = int(score)

          draw_champion_window(win, record1)
          pygame.display.update()
          pygame.time.delay(10000)
  else:
      if score > record2:
          record2 = int(score)
          draw_champion_window(win, record2)
          pygame.display.update()
          pygame.time.delay(10000)

  with open(curdir + '/record.txt', "w+") as f:
      f.write(str(record1) + "\n")
      f.write(str(record2) + "\n")
      f.close()




IntroList = ['Tho hay an mot nut bat ky',
            'Chao mung ban Tho da den voi game',
            'XEP HINH CHO THO',
            'De minh huong dan Tho nhe',
            'Hay an nhung nut mui ten trai, phai, duoi',
            'De di chuyen hinh',
            'An mui ten len tren de xoay hinh.',
            'An nut space de ha hinh xuong nhanh hon',
            'Bay gio Tho hay chon do kho cho game',
            'Che do de (D)  Che do kho (K)',
            'Ban Tho da san sang chua',
            'Chung ta cung bat dau thoi'
            ]

def main_menu(win):

  run = True
  while run:
    # Intro
    counter = 0
    while counter < len(IntroList)-2:
        win.fill((0,0,0))
        draw_text_middle(IntroList[counter], 35, (255,255,255), win)
        pygame.display.update()
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                run = False
                break
            if event.type == pygame.KEYDOWN:
                counter +=1
        if run == False:
            break

    if run == False:
        break

    choselevel = False

    while not choselevel:
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                run = False
                break
            if event.type == pygame.KEYDOWN:
                if event.key == pygame.K_d:
                    difflevel = 'd'
                    choselevel = True
                elif event.key == pygame.K_k:
                    difflevel = 'k'
                    choselevel = True
        if run == False:
            break

    if run == False:
        break

    while counter < len(IntroList):
        win.fill((0,0,0))
        draw_text_middle(IntroList[counter], 35, (255,255,255), win)
        pygame.display.update()
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                run = False
                break
            elif event.type == pygame.KEYDOWN:
                counter += 1
        if run == False:
            break

    if run == False:
        break
    main(win, difflevel)
  pygame.quit()



def main_process():
    win = pygame.display.set_mode((s_width, s_height))
    pygame.display.set_caption('XEP HINH CUA THO')
    main_menu(win)