import random
import tkinter
import tkinter.messagebox


class RandomBall():
 '''
 單個(gè)球定義、運(yùn)動(dòng)的類
 '''
 def __init__(self, root_canvas, width, height):
  '''
  參數(shù)說明：
  canvas：從ScreenSaver類中傳入的畫布
  width，height：從SS類中傳入的寬高，即屏幕寬高
  '''

  # 將傳入變量賦為RB類的屬性
  self.canvas = root_canvas
  self.width = width
  self.height = height

  # 隨機(jī)生成球的中心坐標(biāo)
  self.xcenter = random.randint(100, width-100)
  self.ycenter = random.randint(100, height-100)
  # 隨機(jī)生成球的運(yùn)動(dòng)速度
  self.xvelocity = random.randint(8,16)
  self.yvelocity = random.randint(8,16)
  # 計(jì)算球的半徑
  self.radius = random.randint(60, 100)
  # 利用十六進(jìn)制隨機(jī)數(shù)與lambda表達(dá)式生成球的顏色
  # RGB表示法：三個(gè)數(shù)字，每個(gè)數(shù)字的值是0-255之間，表示紅綠藍(lán)三個(gè)顏色的大小
  # 在某些系統(tǒng)中，直接用英文單詞表示也可以，比如red，green
  color = lambda : random.randint(0,255)
  self.color = '#%02x%02x%02x' % (color(),color(),color())

 # 創(chuàng)建球的函數(shù)
 def create_ball(self):
  '''
  用構(gòu)造函數(shù)定義的變量值，在canvas上畫一個(gè)球
  '''
  # tkinter沒有畫圓形函數(shù)
  # 只有一個(gè)畫橢圓函數(shù)，畫橢圓需要定義兩個(gè)坐標(biāo)，
  # 在一個(gè)長方形內(nèi)畫橢圓，我們只需要定義長方形左上角和右下角就好
  # 求兩個(gè)坐標(biāo)的方法是，已知圓心的坐標(biāo)，則圓心坐標(biāo)減去半徑能求出
  # 左上角坐標(biāo)，加上半徑能求出右下角坐標(biāo)(向右x為正，向下y為正)
  xleftup = self.xcenter - self.radius
  yleftup = self.ycenter - self.radius
  xrightdown = self.xcenter + self.radius
  yrightdown = self.ycenter + self.radius
  # 創(chuàng)建球
  self.item = self.canvas.create_oval(xleftup,yleftup,
           xrightdown,yrightdown,
           fill=self.color,
           outline=self.color)

 # 球運(yùn)動(dòng)的函數(shù)
 def move_ball(self):
  # 計(jì)算球移動(dòng)后的中心點(diǎn)坐標(biāo)
  self.xcenter += self.xvelocity
  self.ycenter += self.yvelocity
  # 當(dāng)球與邊框發(fā)生碰撞時(shí)，需要進(jìn)行回彈操作，即對應(yīng)方向的速度取負(fù)
  if self.xcenter + self.radius >= self.width:
   self.xvelocity = - self.xvelocity
  if self.xcenter - self.radius <= 0:
   self.xvelocity = - self.xvelocity
  if self.ycenter + self.radius >= self.height:
   self.yvelocity = - self.yvelocity
  if self.ycenter - self.radius <= 0:
   self.yvelocity = - self.yvelocity
  # 在canvas上移動(dòng)球,前提是create_ball已經(jīng)調(diào)用
  self.canvas.move(self.item, self.xvelocity, self.yvelocity)



class ScreenSaver():
 '''
 屏保定義類
 程序啟動(dòng)
 '''
 def __init__(self):
  # 創(chuàng)建球存儲(chǔ)列表
  self.balls = []
  # 隨機(jī)生成球的數(shù)量
  self.num = random.randint(10,20)

  # 利用tkinter生成root窗口
  self.root = tkinter.Tk()
  # 獲取屏幕寬、高尺寸
  root_w, root_h = self.root.winfo_screenwidth(), self.root.winfo_screenheight()
  # 取消邊框
  self.root.overrideredirect(1)
  # 綁定退出函數(shù)與相應(yīng)動(dòng)作
  self.root.bind('<Motion>', self.myquit)
  self.root.bind('<Key>', self.myquit)
  self.root.bind('<Any-Button>', self.myquit)

  # 創(chuàng)建畫布，配置尺寸與顏色屬性
  self.canvas = tkinter.Canvas(self.root, width=root_w, height=root_h)
  self.canvas.pack()

  # 利用循環(huán)與RandomBall類在畫布上畫球，并append到列表中
  for i in range(self.num):
   ball = RandomBall(self.canvas, width=root_w, height=root_h)
   ball.create_ball()
   self.balls.append(ball)

  # 調(diào)用球運(yùn)動(dòng)函數(shù)
  self.run_screen_saver()
  # 啟用tkinter時(shí)間消息循環(huán)mainloop
  self.root.mainloop()

 # 球運(yùn)動(dòng)函數(shù)
 def run_screen_saver(self):
  # 循環(huán)實(shí)例化的ball調(diào)用move_ball函數(shù)
  for ball in self.balls:
   ball.move_ball()
  # 使用after實(shí)現(xiàn)遞歸，通過不斷調(diào)用各球的move_ball函數(shù)，實(shí)現(xiàn)位置刷新
  self.root.after(50, self.run_screen_saver)

 # 停止運(yùn)行
 # 此處e只是利用了事件處理機(jī)制,際上并不關(guān)心事件的類型
 def myquit(self, e):
  # 擴(kuò)展：
  # 此屏保程序擴(kuò)展成，一旦捕獲事件，則判斷屏保不退出
  # 顯示一個(gè)Button，Button上顯示事件類型，點(diǎn)擊Button后屏保才退出
  if tkinter.messagebox.askokcancel("彩球碰撞", '確定退出？'):
   self.root.destroy()
  else:
   pass


if __name__ == '__main__':
 # 啟動(dòng)屏保
 ScreenSaver()