import sys
import cfg
import pygame
from modules import *
 
 
'''定義按鈕'''
def Button(screen, position, text, button_size=(200, 50)):
 left, top = position
 bwidth, bheight = button_size
 pygame.draw.line(screen, (150, 150, 150), (left, top), (left+bwidth, top), 5)
 pygame.draw.line(screen, (150, 150, 150), (left, top-2), (left, top+bheight), 5)
 pygame.draw.line(screen, (50, 50, 50), (left, top+bheight), (left+bwidth, top+bheight), 5)
 pygame.draw.line(screen, (50, 50, 50), (left+bwidth, top+bheight), (left+bwidth, top), 5)
 pygame.draw.rect(screen, (100, 100, 100), (left, top, bwidth, bheight))
 font = pygame.font.Font(cfg.FONTPATH, 30)
 text_render = font.render(text, 1, (255, 235, 205))
 return screen.blit(text_render, (left+50, top+10))
 
 
'''
Function:
 開始界面
Input:
 --screen: 游戲界面
Return:
 --game_mode: 1(單人模式)/2(雙人模式)
'''
def startInterface(screen):
 clock = pygame.time.Clock()
 while True:
  screen.fill((41, 36, 33))
  button_1 = Button(screen, (150, 175), '1 Player')
  button_2 = Button(screen, (150, 275), '2 Player')
  for event in pygame.event.get():
if event.type == pygame.QUIT:
 pygame.quit()
 sys.exit()
if event.type == pygame.MOUSEBUTTONDOWN:
 if button_1.collidepoint(pygame.mouse.get_pos()):
  return 1
 elif button_2.collidepoint(pygame.mouse.get_pos()):
  return 2
  clock.tick(10)
  pygame.display.update()
 
 
'''結(jié)束界面'''
def endInterface(screen, score_left, score_right):
 clock = pygame.time.Clock()
 font1 = pygame.font.Font(cfg.FONTPATH, 30)
 font2 = pygame.font.Font(cfg.FONTPATH, 20)
 msg = 'Player on left won!' if score_left > score_right else 'Player on right won!'
 texts = [font1.render(msg, True, cfg.WHITE),
font2.render('Press ESCAPE to quit.', True, cfg.WHITE),
font2.render('Press ENTER to continue or play again.', True, cfg.WHITE)]
 positions = [[120, 200], [155, 270], [80, 300]]
 while True:
  screen.fill((41, 36, 33))
  for event in pygame.event.get():
if event.type == pygame.QUIT:
 pygame.quit()
 sys.exit()
if event.type == pygame.KEYDOWN:
 if event.key == pygame.K_RETURN:
  return
 elif event.key == pygame.K_ESCAPE:
  sys.exit()
  pygame.quit()
  for text, pos in zip(texts, positions):
screen.blit(text, pos)
  clock.tick(10)
  pygame.display.update()
 
 
'''運(yùn)行游戲Demo'''
def runDemo(screen):
 # 加載游戲素材
 hit_sound = pygame.mixer.Sound(cfg.HITSOUNDPATH)
 goal_sound = pygame.mixer.Sound(cfg.GOALSOUNDPATH)
 pygame.mixer.music.load(cfg.BGMPATH)
 pygame.mixer.music.play(-1, 0.0)
 font = pygame.font.Font(cfg.FONTPATH, 50)
 # 開始界面
 game_mode = startInterface(screen)
 # 游戲主循環(huán)
 # --左邊球拍(ws控制, 僅雙人模式時(shí)可控制)
 score_left = 0
 racket_left = Racket(cfg.RACKETPICPATH, 'LEFT', cfg)
 # --右邊球拍(↑↓控制)
 score_right = 0
 racket_right = Racket(cfg.RACKETPICPATH, 'RIGHT', cfg)
 # --球
 ball = Ball(cfg.BALLPICPATH, cfg)
 clock = pygame.time.Clock()
 while True:
  for event in pygame.event.get():
if event.type == pygame.QUIT:
 pygame.quit()
 sys.exit(-1)
  screen.fill((41, 36, 33))
  # 玩家操作
  pressed_keys = pygame.key.get_pressed()
  if pressed_keys[pygame.K_UP]:
racket_right.move('UP')
  elif pressed_keys[pygame.K_DOWN]:
racket_right.move('DOWN')
  if game_mode == 2:
if pressed_keys[pygame.K_w]:
 racket_left.move('UP')
elif pressed_keys[pygame.K_s]:
 racket_left.move('DOWN')
  else:
racket_left.automove(ball)
  # 球運(yùn)動(dòng)
  scores = ball.move(ball, racket_left, racket_right, hit_sound, goal_sound)
  score_left += scores[0]
  score_right += scores[1]
  # 顯示
  # --分隔線
  pygame.draw.rect(screen, cfg.WHITE, (247, 0, 6, 500))
  # --球
  ball.draw(screen)
  # --拍
  racket_left.draw(screen)
  racket_right.draw(screen)
  # --得分
  screen.blit(font.render(str(score_left), False, cfg.WHITE), (150, 10))
  screen.blit(font.render(str(score_right), False, cfg.WHITE), (300, 10))
  if score_left == 11 or score_right == 11:
return score_left, score_right
  clock.tick(100)
  pygame.display.update()
 
 
'''主函數(shù)'''
def main():
 # 初始化
 pygame.init()
 pygame.mixer.init()
 screen = pygame.display.set_mode((cfg.WIDTH, cfg.HEIGHT))
 pygame.display.set_caption('pingpong —— 九歌')
 # 開始游戲
 while True:
  score_left, score_right = runDemo(screen)
  endInterface(screen, score_left, score_right)
 
 
'''run'''
if __name__ == '__main__':
 main()