root/experimental/display/trunk/display/display.py @ 3851

#!/usr/bin/env python

'''Write incomming packet(s) to file(s)'''

import wx         #needed for display stuff
from omniORB import CORBA      #use this for the CORBA orb stuff (pushing packets)
import sys       #for system commands (e.g., argv and argc stuff)
import CosNaming   #narrowing naming context stuff
import CF, CF__POA    #core framework stuff
import standardInterfaces__POA
from wx.lib.anchors import LayoutAnchors  #used by splitter window
# import wx.lib.buttons as buttons #for special wx buttons

class display_short(standardInterfaces__POA.complexShort):
    '''Writes I and Q data to 2 files'''

    def __init__(self, orb, gui):        
        self.orb = orb 
        self.gui = gui    #usually parent
        self.delimiter = ','

    def pushPacket(self, I_data, Q_data):
        '''Store the data to be written to file when the Write Packet button is bushed'''
        self.gui.I_data = I_data
        self.gui.Q_data = Q_data
  


def create(parent,namespace, interface, ns_name, port_name):
    #return MainFrame(parent, -1, namespace, interface, ns_name, port_name)
    return MainFrame(parent, -1, "Don't know what this should be", namespace, interface, ns_name, port_name)

#generate wx ids for my wx controls
[wxID_MAINFRAME, wxID_SPLITTERWINDOW1] = [wx.NewId() for _init_ctrls in range(2)]


class MainFrame(wx.Frame):
    def __init__(self, parent, id, title, namespace, interface, component_name, port_name):

        self._init_ctrls(parent)

        self.I_data = []
        self.Q_data = []
         
        self.parent = parent
        self.namespace = namespace
        self.interface = interface
        self.my_local_controls = None
        self.component_name = component_name
        self.port_name = port_name
        self.setup_graphics()
        
        # Now Create the menu bar and items
        self.mainmenu = wx.MenuBar()

        menu = wx.Menu()
        menu.Append(205, 'E&xit', 'Enough of this already!')
        self.Bind(wx.EVT_MENU, self.OnFileExit, id=205)
        self.mainmenu.Append(menu, '&File')
        
        menu = wx.Menu()
        menu.Append(300, '&About', 'About this thing...')
        self.Bind(wx.EVT_MENU, self.OnHelpAbout, id=300)
        self.mainmenu.Append(menu, '&Help')

        self.SetMenuBar(self.mainmenu)

        # Bind the close event so we can disconnect the ports
        self.Bind(wx.EVT_CLOSE, self.OnCloseWindow)

        self.Show(True)
        
        self.iFileNameEditor.write('i_out.dat')
        self.qFileNameEditor.write('q_out.dat')


    def _init_ctrls(self, prnt):
        wx.Frame.__init__(self, id=wxID_MAINFRAME, name='', parent=prnt,
              pos=wx.Point(530, 680), size=wx.Size(520, 270),
              style=wx.DEFAULT_FRAME_STYLE, title='Write to File Tool')

        self.splitterWindow1 = wx.SplitterWindow(id=wxID_SPLITTERWINDOW1,
              name='splitterWindow1', parent=self, point=wx.Point(1, 1),
              size=wx.Size(570, 270), style=wx.SP_3D)
        self.splitterWindow1.SetConstraints(LayoutAnchors(self.splitterWindow1,
              True, True, True, True))



    
    def OnFileExit(self, event):
        '''This is what will happen when you select File -> Exit in the menu bar'''
        self.Close()      #close the frame
  
    def OnHelpAbout(self, event):
        '''Stuff that gets displayed when you select Help -> About in the menu bar'''
        from wx.lib.dialogs import ScrolledMessageDialog
        about = ScrolledMessageDialog(self, "Write to file tool.\nA product of Wireless@VT.", "About...")
        about.ShowModal()

    def setup_graphics(self):
        self.CORBA_being_used = False

        if True:                
         self.CORBA_being_used = True
         self.orb = CORBA.ORB_init(sys.argv, CORBA.ORB_ID)
         obj = self.orb.resolve_initial_references("NameService")
         rootContext = obj._narrow(CosNaming.NamingContext)
         if rootContext is None:
             print "Failed to narrow the root naming context"
             sys.exit(1)
         name = [CosNaming.NameComponent(self.component_name[0],""),
             CosNaming.NameComponent(self.component_name[1],""),
             CosNaming.NameComponent(self.component_name[2],"")]
     
         try:
             ResourceRef = rootContext.resolve(name)
     
         except:
             print "Required resource not found"
             sys.exit(1)
     
         ResourceHandle = ResourceRef._narrow(CF.Resource)
         PortReference = ResourceHandle.getPort(self.port_name)
         if PortReference is None:
             print "Failed to get Port reference"
         self.PortHandle = PortReference._narrow(CF.Port)
         
        
         #create the class instance of the display class 
         self.my_file_writer = display_short(self.orb, self)

         obj_poa = self.orb.resolve_initial_references("RootPOA")
         poaManager = obj_poa._get_the_POAManager()
         poaManager.activate()
         obj_poa.activate_object(self.my_file_writer)
         self.PortHandle.connectPort(self.my_file_writer._this(), "thisismyconnectionid_disp")
         #orb.run()

    def OnCloseWindow(self,event):
        if hasattr(self.parent, 'removeToolFrame'):
            self.parent.removeToolFrame(self)
        self = None
        event.Skip()

    def __del__(self):
        if self.CORBA_being_used:
            self.PortHandle.disconnectPort("thisismyconnectionid_disp")
            while (_time.time() - self.my_local_plot.end_time) < 1.5:
                #print (time.time() - self.my_local_plot.end_time)
                pass
                #_time.sleep(1)