GraphFrame.py

Go to the documentation of this file.

#!/usr/bin/env python
#----------------------------------------------------------------------------
#----------------------------------------------------------------------------

"""
This program will load and run one of the individual demos in this
directory within its own frame window.  Just specify the module name
on the command line.
"""

import wx
import wx.xrc  as  xrc
import wx.lib.mixins.inspection

import sys, os

import images

from libCatKernel import *

from wrapper import *
from tools import opj

#----------------------------------------------------------------------------
class GraphFrame(wx.Frame):    
    overviewText = "CAT Graphical Window"
    def __init__(self,app,parent,title): 
        wx.Frame.__init__(self, parent, wx.NewId(), title)
        self.figId=0
        self.parent=parent
        self.app=app
        self.path=os.path.join(os.environ.get("CATPATH"), "CatPython", "python")
        self.makeToolBar()
        res=xrc.XmlResource(os.path.join(self.path,"xrc/GraphPanel.xrc"))
        self.panel=res.LoadPanel(self, "GraphPanel")
        self.notebook=xrc.XRCCTRL(self, 'notebook')
        self.panels=list()
        self.objs=list()
        self.figs=list()

        self.Bind(wx.EVT_CLOSE, self.onIconize)
        self.Bind(wx.EVT_NOTEBOOK_PAGE_CHANGED, self.onChange)

    def getControl(self, panel, xmlid):
        '''Retrieves the given control (within a dialog) by its xmlid'''
        control = panel.FindWindowById(xrc.XRCID(xmlid))
        assert control != None, 'Programming error: a control with xml id ' + xmlid + ' was not found.'
        return control
        
    def makeToolBar(self):
        TB_PRINT=wx.NewId()
        TB_SAVE=wx.NewId()
        TB_CLOSEALL=wx.NewId()
        TB_CLOSE=wx.NewId()
        #
        self.toolBar = self.CreateToolBar(wx.TB_DOCKABLE)
        self.toolBar.AddLabelTool(TB_CLOSEALL, '', wx.Bitmap(os.path.join(self.path,"xrc/icons/closeall.png")))
        self.toolBar.AddLabelTool(TB_CLOSE  , '', wx.Bitmap(os.path.join(self.path,"xrc/icons/close.png")))
        self.toolBar.AddSeparator()        
        self.toolBar.AddLabelTool(TB_PRINT, '', wx.Bitmap(os.path.join(self.path,"xrc/icons/fileprint.png")))
        self.toolBar.AddLabelTool(TB_SAVE, '', wx.Bitmap(os.path.join(self.path,"xrc/icons/filesave.png")))        
        self.toolBar.Realize()

        self.Bind(wx.EVT_TOOL, self.onCloseAll , id=TB_CLOSEALL)
        self.Bind(wx.EVT_TOOL, self.onClose , id=TB_CLOSE)
        self.Bind(wx.EVT_TOOL, self.onPrint , id=TB_PRINT)
        self.Bind(wx.EVT_TOOL, self.onSave  , id=TB_SAVE)
                        
    def onIdle(self, event):
        '''Responds to idle time in the system'''
        # when the timer says it's time, we do the actual downloading in the main thread (wx doesn't work well in secondary threads)
        print "hello"        

    def onChange(self, event):
        self.update()

    def onIconize(self, event):
        self.parent.gphFrame.Show(False)
        self.parent.gphState=False

    def onReLoad(self, event):
        self.update(self.notebook.GetSelection())

    def onPrint(self, event):
        print "printing..."

    def onSave(self, event):
        print "save..."

    def onClose(self, event):
        pos=self.notebook.GetSelection()
        self.notebook.DeletePage(pos)
#        self.panels(pos).killFigure()
#        self.panels(pos).Destroy()
        self.panels.pop(pos)
        self.objs.pop(pos)
        self.figs.pop(pos)

    def onCloseAll(self, event):
        self.notebook.DeleteAllPages()
        self.panels=[]
        self.objs  =[]
        self.figs  =[]
        
    def onPlot(self, obj):
        wrap=wrapper(self.app,obj,"proc")
        if (wrap.ok):
            module=wrap.GetActive()
            panel=module.Plot(self.app,obj,self.notebook,self.figId)
            if panel!=None :
                self.figId+=1
                self.panels.append(panel)
                self.objs.append(obj)
                self.figs.append(panel)
                self.notebook.AddPage(panel,obj.name(),True)
                
    def update(self):
        pos=self.notebook.GetSelection()
        self.figs[pos].current()

#----------------------------------------------------------------------------