#-*- coding: utf-8 -*- 
 
################################################################################
## 使用wxPython的繪圖模塊wxPyPlot,需要數據可視化的時候,無需再借用其他的庫或模塊了
################################################################################
import numpy
import wx
import wx.lib.plot as wxPyPlot #導入繪圖模塊,并命名為wxPyPlot
 
#---------------------------------------------------------------------------
# 需要把數據封裝進入MyDataObject中
def MyDataObject():
 # 50 個點的sin函數,用藍色圓點表示
 data1 = 2.*numpy.pi*numpy.arange(100)/100.
 data1.shape = (50, 2)
 data1[:,1] = numpy.sin(data1[:,0])
 markers = wxPyPlot.PolyMarker(data1, legend='Green Markers', colour='blue', marker='circle',size=1)
 
 # 50個點的cos函數,用紅色表示
 data2 = 2.*numpy.pi*numpy.arange(100)/100.
 data2.shape = (50,2)
 data2[:,1] = numpy.cos(data2[:,0])
 lines = wxPyPlot.PolySpline(data2, legend= 'Red Line', colour='red')
 
 GraphTitle="Plot Data(Sin and Cos)"
 
 
 return wxPyPlot.PlotGraphics([markers, lines],GraphTitle, "X Axis", "Y Axis")
#-----------------------------------------------------------------------------
class TestFrame1(wx.Frame):
 def __init__(self, parent=None, id=wx.ID_ANY, title="Using wxPyPlot"):
  wx.Frame.__init__(self, parent, id, title,size=(600, 400))
  
  # 創建菜單欄
  self.mainmenu = wx.MenuBar()
 
  menu = wx.Menu()
  menu.Append(100, 'Draw1', 'Draw plots1')
  self.Bind(wx.EVT_MENU,self.OnPlotDraw1, id=100)
 
  self.mainmenu.Append(menu, '&Plot')
 
  self.SetMenuBar(self.mainmenu)
 
  # 創建狀態欄,顯示信息
  self.CreateStatusBar(2)
  
  self.pc = wxPyPlot.PlotCanvas(self) #此處導入繪圖面板
 
 def OnPlotDraw1(self, event): #繪圖函數
  self.pc.Draw(MyDataObject())
 
 
###########################################################################
## 測試wxPyPlot的代碼
###########################################################################
if __name__=='__main__':
  app = wx.App()
  tf=TestFrame1(None)
  tf.Show()
  app.MainLoop()