import matplotlib.pyplot as plt
import numpy as np
import threading
import sys
from random import random, randrange
from time import sleep

'''
繪制2x2的畫板
可設置窗口標題和4個子圖標題
可更新曲線數(shù)據(jù)
'''
quit_flag = False # 退出標志


class Plot2_2(object):
  """ 2x2的畫板 """

  def __init__(self, wtitle='Figure', p1title='1', p2title='2', p3title='3',
         p4title='4'):
    self.sub_title = [p1title, p2title, p3title, p4title] # 4個子圖的標題
    self.fig, self.ax = plt.subplots(2, 2) # 創(chuàng)建2X2子圖
    self.fig.subplots_adjust(wspace=0.3, hspace=0.3) # 設置子圖之間的間距
    self.fig.canvas.set_window_title(wtitle) # 設置窗口標題

    # 子圖字典，key為子圖的序號，value為子圖句柄
    self.axdict = {0: self.ax[0, 0], 1: self.ax[0, 1], 2: self.ax[1, 0], 3: self.ax[1, 1]}

  def showPlot(self):
    """ 顯示曲線 """
    plt.show()

  def setPlotStyle(self, index):
    """ 設置子圖的樣式，這里僅設置了標題 """
    self.axdict[index].set_title(self.sub_title[index], fontsize=12)

  def updatePlot(self, index, x, y):
    """
    更新指定序號的子圖
    :param index: 子圖序號
    :param x: 橫軸數(shù)據(jù)
    :param y: 縱軸數(shù)據(jù)
    :return:
    """
    # X軸數(shù)據(jù)必須和Y軸數(shù)據(jù)長度一致
    if len(x) != len(y):
      ex = ValueError("x and y must have same first dimension")
      raise ex

    self.axdict[index].cla() # 清空子圖數(shù)據(jù)
    self.axdict[index].plot(x, y) # 繪制最新的數(shù)據(jù)
    self.setPlotStyle(index) # 設置子圖樣式
    if min(x) < max(x):
      self.axdict[index].set_xlim(min(x), max(x)) # 根據(jù)X軸數(shù)據(jù)區(qū)間調(diào)整X軸范圍
    plt.draw()
    print("%s end" % sys._getframe().f_code.co_name)


def updatePlot(plot):
  """
  模擬收到實時數(shù)據(jù)，更新曲線的操作
  :param plot: 曲線實例
  :return:
  """
  print("Thread: %s" % threading.current_thread().getName())
  count = 0
  global quit_flag
  print("quit_flag[%s]" % str(quit_flag))
  while True:
    if quit_flag:
      print("quit_flag[%s]" % str(quit_flag))
      break
    count += 1
    print("count#%d" % count)
    x = np.arange(0, 100, 1)
    y = np.random.normal(loc=1, scale=1, size=100) # 產(chǎn)生隨機數(shù)，模擬變化的曲線
    index = randrange(4) # 隨機更新某一個子圖
    plot.updatePlot(index, x, y)
    sleep(random() * 3)


def main():
  p = Plot2_2() # 創(chuàng)建一個2X2畫板

  t = threading.Thread(target=updatePlot, args=(p,)) # 啟動一個線程更新曲線數(shù)據(jù)
  t.start()

  p.showPlot() # showPlot方法會阻塞當前線程，直到窗口關閉
  print("plot close")
  global quit_flag
  quit_flag = True # 通知更新曲線數(shù)據(jù)的線程退出

  t.join()
  print("Thread: %s end" % threading.current_thread().getName())


if __name__ == '__main__':
  main()