python wxpython一例

hanyh

前些天wxpython做得简单代码，目标是：
比如说c:\photos\....下面有很多头像图片，这些图片在上面或下面有一个圆点，呈现图片一段时间，判断圆点在上面还是下面，按1，2进行反应，记录反应时间。
注：是帮朋友进行孤独症儿童注意力分配实验写的，后面因为在处理*.gif图片的时候响应速度不够好，改为用delphi7写了。

程序涉及：
wxpython
事件响应
图片缩放和GIF动画操作

# -*- coding: cp936 -*-import timeimport wx,osimport wx.animatefrom operator import or_from func import *from  threading import *class M(object):    idx = 0    total = 0    right = 0    wrong = 0    starticker = 0 #开始时间    steps = []    answer = 0    pause = 1000    pre = 'bmp' #0 is bmp,1 is if,2 is text         class FullScreenPanel(wx.Panel):    def __init__(self,parent):        wx.Panel.__init__(self,parent,name="FullScreenPanel",style= wx.DEFAULT_FRAME_STYLE | wx.WANTS_CHARS )        self.SetBackgroundColour('black')                    self.TrialBtn = wx.Button(self,label="练习")        self.TrialBtn.Bind(wx.EVT_BUTTON,self.BeginTrial)        self.TestBtn = wx.Button(self,label="测试")        self.TestBtn.Bind(wx.EVT_BUTTON,self.BeginTest)        sizer = wx.BoxSizer(orient=wx.VERTICAL)          sizer.AddStretchSpacer()        sizer.Add(self.TrialBtn,0, wx.ALIGN_CENTER )        sizer.Add(self.TestBtn,0, wx.ALIGN_CENTER )        sizer.AddStretchSpacer()         self.SetSizer(sizer)                self.Layout()                def hideImg(self):        try:            self.curImg.Show(False)        except:            pass    def hideGif(self):        try:            self.curGif.Stop()            self.curGif.Close()            self.curGif.Show(False)        except:            print "hide gif error"            pass    def hideText(self):        try:            self.curText.Show(False)        except:            pass                    def KeyDownResponse(self,evt):                code = evt.GetKeyCode()        if code in (49,50):            self.stopTimer()            self.Bind(wx.EVT_KEY_DOWN,self.KeyDownNull)            if code == 49:                code = 1            else:                code = 2            if code == M.answer:                slapedTime = time.time() - M.starticker                 self.show(response['right'])                M.right = M.right + 1                M.total = M.total + 1                rate = M.right/float(M.total)            else:                slapedTime = time.time() - M.starticker                 self.show(response['wrong'])                M.wrong = M.wrong + 1                M.total = M.total + 1                rate = M.right/float(M.total)            text = "正确率:%.3f 耗时:%.3f" % (rate,slapedTime)            if self.practice:                self.addText(text)            self.fp.write(text+"\n")             self.timer = wx.CallLater(M.pause,self.next)                  else:            pass            def addText(self,text):        mm=wx.DisplaySize()        x0 = 100        y0 = 600        self.curText = wx.StaticText(self,-1,text,(x0,y0),(160,0),wx.ALIGN_CENTER)        font=wx.Font(18,wx.DECORATIVE,wx.NORMAL,wx.NORMAL)                self.curText.SetFont(font)        self.curText.SetForegroundColour("white")        self.curText.SetBackgroundColour("black")        self.curText.Show(True)                    def KeyDownNull(self,evt):        pass                        def showText(self,text):        mm=wx.DisplaySize()        x0 = (mm[0] - 60) / 2        y0 = (mm[1] - 60) / 2        self.curText = wx.StaticText(self,-1,text,(x0,y0),(160,0),wx.ALIGN_CENTER)        font=wx.Font(28,wx.DECORATIVE,wx.NORMAL,wx.NORMAL)                self.curText.SetFont(font)        self.curText.SetForegroundColour("white")        self.curText.SetBackgroundColour("black")        self.curText.Show(True)                    def showBmp(self,fpath):        try:            self.curImg.Show(False)        except:            pass        mm=wx.DisplaySize()        img = wx.Image(fpath, wx.BITMAP_TYPE_BMP)        x0 = (mm[0] - img.GetWidth()) / 2        y0 = (mm[1] - img.GetHeight()) / 2        wxBitmap = img.ConvertToBitmap()       # self.curImg.SetBitmap(wxBitmap)        self.curImg = wx.StaticBitmap(self, -1, img.ConvertToBitmap(), (x0,y0), (img.GetWidth(), img.GetHeight()))        self.curImg.Show(True)        print "end.."+fpath    def showGif(self,fpath):        mm=wx.DisplaySize()                img = wx.Image(fpath, wx.BITMAP_TYPE_GIF)        x0 = (mm[0] - img.GetWidth()) / 2        y0 = (mm[1] - img.GetHeight()) / 2        self.curGif = wx.animate.GIFAnimationCtrl(self, -1, fpath, (x0,y0), (img.GetWidth(), img.GetHeight()))                    self.curGif.GetPlayer().UseBackgroundColour(True)        self.curGif.Play()            def show(self,fpath):         type = getType(fpath)        self.hideText()        self.hideGif()        if not type == M.pre:            if type == "gif":                self.hideText()                self.hideImg()                self.hideGif()            elif type == "bmp":                self.hideGif()                self.hideText()            else:                self.hideImg()                self.hideGif()            M.pre = type                        print 'show '+fpath        if  type == "bmp":            self.hideText()            self.showBmp(fpath)        elif type == "gif":            self.hideText()            self.showGif(fpath)        elif type == "":            self.showText(fpath)        def stopTimer(self):        self.timer.Stop()                    def timeout(self):        self.stopTimer()        self.Bind(wx.EVT_KEY_DOWN,self.KeyDownNull)        self.show(response['timeout'])             M.total += 1        rate = M.right / float(M.total)        slapedTime = time.time() - M.starticker                 text = "正确率:%.3f %.3f" % (rate,slapedTime)        if self.practice:            self.addText(text)                       self.fp.write(text+"\n")         self.timer = wx.CallLater(M.pause,self.next)                    def next(self):        idx = M.idx        if idx > len(M.steps)-1:            self.stopTimer()            print "over"            return ""        print 'current idx %s' %idx        opType =  M.steps[idx][0]        fpath = M.steps[idx][1]        type = getType(fpath)        sepTime = M.steps[idx][2]        M.idx = idx+1        self.show(fpath)        if opType == 0: #退出            self.Bind(wx.EVT_KEY_DOWN,self.KeyDownOver)        if opType == 1: #等待space            self.Bind(wx.EVT_KEY_DOWN,self.KeyDownTrigger)        elif opType in (2,3): #空操作            self.Bind(wx.EVT_KEY_DOWN,self.KeyDownNull)            self.timer = wx.CallLater(sepTime, self.next)            elif opType == 4:  #等待反映,1500ms 后            self.Bind(wx.EVT_KEY_DOWN,self.KeyDownResponse) #多做一次也无所谓，让逻辑            M.starticker = time.time()            self.timer = wx.CallLater(sepTime, self.timeout)            elif opType == 9:            M.answer = M.steps[idx][3]            self.Bind(wx.EVT_KEY_DOWN,self.KeyDownResponse)            self.timer = wx.CallLater(sepTime, self.next)                def KeyDownOver(self,evt):        self.stopTimer()        self.hideGif()        self.hideText()        self.hideImg()        self.TrialBtn.Show()        self.TestBtn.Show()         try:            self.fp.close()        except:            pass        M.steps = []        M.idx = 0        M.total = 0        M.right = 0        M.wrong = 0        M.starticker = 0 #开始时间        M.steps = []        M.answer = 0            def KeyDownTrigger(self,evt):        if evt.GetKeyCode() == 32:            self.next()            def getSavePath(self):        sp = time.strftime("%Y-%m-%d-%H-%M-%S",time.localtime())+".txt"        return os.path.join(os.getcwd(),sp)                def BeginTest(self,evt):        fpath = self.getSavePath()        self.fp = open(fpath, 'w')                 self.practice = False        self.TrialBtn.Hide()        self.TestBtn.Hide()         M.steps = getSteps(testConf)         print M.steps        self.next()        self.FullScreen(evt)                 def BeginTrial(self,evt):        fpath = self.getSavePath()        self.fp = open(fpath, 'w')         self.practice = True        self.TrialBtn.Hide()        self.TestBtn.Hide()         M.steps = getSteps(trialConf)         self.next()        self.FullScreen(evt)            def GetFlags(self):        res = []        val = "FULLSCREEN_ALL"        res.append(getattr(wx,val))        return reduce(or_,res,0)    def FullScreen(self,evt):        top = self.GetTopLevelParent()        top.OnFullScreen(evt)class MainFrame(wx.Frame):    def __init__(self, parent):        wx.Frame.__init__(self, parent, wx.ID_ANY, 'Test FullScreen', size=(600, 400))        self.Bind(wx.EVT_CLOSE, self.OnCloseWindow)        self.client = FullScreenPanel(self)        mbar = wx.MenuBar()        the_menu = wx.Menu()        fullscreenMI = the_menu.Append(wx.ID_ANY,"Full Screen\tF12","Toggles Full Screen Mode")        mbar.Append(the_menu,"File")        self.SetMenuBar(mbar)        self.Bind(wx.EVT_MENU,self.OnFullScreen,id=fullscreenMI.GetId())    def OnCloseWindow(self, event):        self.Destroy()    def OnFullScreen(self,event):        flags = self.client.GetFlags()        self.ShowFullScreen(not self.IsFullScreen(),flags)class MainApp(wx.App):    def OnInit(self):        self.main = MainFrame(None)        self.main.Show()        self.SetTopWindow(self.main)        return Truedef main():    application = MainApp(redirect=False)    application.MainLoop()if __name__=='__main__':    main()

# -*- coding: cp936 -*-import osimport randomimport wxdef opj(path):    st = apply(os.path.join, tuple(path.split('/')))    if path.startswith('/'):        st = '/' + st    return stimgTypeDict = {    'bmp':wx.BITMAP_TYPE_BMP,    'gif':wx.BITMAP_TYPE_GIF,    'png':wx.BITMAP_TYPE_PNG,    'jpg':wx.BITMAP_TYPE_JPEG}#获取后缀def getType(o):    if o.rfind(".") > -1:        idx=o.rindex(".")+1        return o[idx:].lower()     else:        return ""        p1 = opj('photos/right.bmp')p2 = opj('photos/wrong.bmp')p3 = opj('photos/no response.bmp')response = {    'right':p1,    'wrong':p2,    'timeout':p3}    sep = (2,'+',1000)sep0 = (2,'',1000)trialConf = (    (1,'photos/welcome.bmp'),          (1,'photos/instruction.bmp'),     (2,'+',1), #1000ms 文本    (3,'photos/practice/face1', 400,200,1500), #探测的文件目录时间（面空时间，探测点时间，等待回答时间）    (0,'photos/goodbye.bmp',5000))testConf = (    (1,'photos/welcome.bmp'),          (1,'photos/instruction.bmp'),     (2,'+',1000), #1000ms 文本    (3,'photos/donghua', 400,50,1500), #探测的文件目录时间（面空时间，探测点时间，等待回答时间）    (0,'photos/goodbye.bmp',5000))def getSteps(conf):    steps=[]    for x in conf:        type = x[0]        img = x[1]        if type == 0: #end            item = (0,opj(x[1]),-1)            steps.append(item)                    if type == 1: #单个图片            item = (1,opj(x[1]),-1) #-1 means infinate            steps.append(item)        elif type == 2: #            item = (2,x[1],x[2]) #内容，时间            steps.append(item)        elif type == 3:            dir1 = x[1]               t1 = x[2]            t2 = x[3]            t3 = x[4]                     face=[]            probe=[]            for root, dirs, files in os.walk(dir1, topdown=False):                for name in files:                    if name.endswith(".gif") or name.endswith(".bmp"):                        face.append(opj(os.path.join(root, name)))            face.sort()            print face            count = len(face)            i=0            while(i<count):                curImg = face[i]                i += 1                nextImg = face[i]                if nextImg.find("_2") > 0:   #存在2，就是下面有点                    answer = 2                else:                    answer = 1                i += 1                curItem = (curImg,nextImg,answer)                probe.append(curItem)           # print face            random.shuffle(probe)                        i=0            count = len(probe)          #  print probe            while(i<count):                curItem = probe[i]                item = (3,curItem[0],t1)                steps.append(item)                item = (9,curItem[1],t2,curItem[2])                 #马上要激活keyboard反应,并置相应的时间处理函数                                 #(类型4需要响应,图片路径，呈现时间，正确答案，超时时间，                steps.append(item)                item = (4,curItem[0],t3)                steps.append(item)                steps.append(sep0)                steps.append(sep)                i += 1                return steps