root/tools/alf_plugins/AWG/branches/AWG-0.6.2/AWG/AWG.py @ 5791

#!/usr/bin/env python

'''get signals and output them to a port'''

import sources    # within the AWG directory.  This will generate the signals
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
from ossie.cf import CF, CF__POA    # core framework stuff
from ossie.standardinterfaces import standardInterfaces__POA
from wx.lib.anchors import LayoutAnchors  # used by splitter window

class AWG:
    '''this class makes the calls to get the actual signal from the 
       sources class'''

    def __init__(self, parent):        
        self.parent = parent

    def get_signal(self, signal_type, file_name):
        #signal I want is hard-coded for now
        self.signal_type = signal_type
        self.signal_len = 8192   #length of packet
        self.frequency = 5   #cycles per packet
        self.rand_type = 'random'
        self.file_name = file_name
        self.delimiter = ','

        my_signal = sources.sources(self)    #instance of signals class
        self.the_signal = my_signal.gen_signal()

        return self.the_signal              
    

def create(parent,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, wxID_PUSHPACKETBTN, wxID_ISOURCECHOICE, wxID_IFILENAMEEDITOR, wxID_QSOURCECHOICE, wxID_QFILENAMEEDITOR, wxID_ISOURCESTATICTEXT, wxID_QSOURCESTATICTEXT, wxID_IFILESTATICTEXT,wxID_QFILESTATICTEXT] = [wx.NewId() for _init_ctrls in range(11)]


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

        tmp_sources_instance = sources.sources(self)
        self.available_sources = tmp_sources_instance.get_sources_list()

        self._init_ctrls(parent)
         
        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.dat')
        self.qFileNameEditor.write('q.dat')

        #set the default source
        #this way if the user presses the "PushPacket" button before selecting
        #a source, the source that is being displayed will be used
        self.iSourceChoice.SetSelection(0)
        self.qSourceChoice.SetSelection(0)

    def _init_ctrls(self, prnt):
        wx.Frame.__init__(self, id=wxID_MAINFRAME, name='', parent=prnt,
              pos=wx.Point(1, 570), size=wx.Size(520, 370),
              style=wx.DEFAULT_FRAME_STYLE, title='Arbitrary Waveform Generator')

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

        self.PushPacketBtn = wx.Button(id=wxID_PUSHPACKETBTN, label='Push Packet',
              name='PushPacketBtn', parent=self.splitterWindow1, pos=wx.Point(155, 250),
              size=wx.Size(145, 50))
        self.PushPacketBtn.SetFont(wx.Font(16, wx.SWISS, wx.NORMAL, wx.BOLD, False))
        self.PushPacketBtn.SetBackgroundColour("green")
        self.PushPacketBtn.Bind(wx.EVT_BUTTON, self.OnPushPacketBtnButton,
              id=wxID_PUSHPACKETBTN)
 
        self.iSourceChoice = wx.Choice(choices=self.available_sources.keys(), 
              id=wxID_ISOURCECHOICE,
              name=u'iSourceChoice', parent=self.splitterWindow1, pos=wx.Point(215, 50),
              size=wx.Size(136, 28), style=0)
        self.iSourceChoice.Bind(wx.EVT_CHOICE, self.OnSourceChoiceChoice,
              id=wxID_ISOURCECHOICE)

        self.iFileNameEditor = wx.TextCtrl(id=wxID_IFILENAMEEDITOR,
              name=u'iFileNameEditor', parent=self.splitterWindow1, pos=wx.Point(215, 150),
              size=wx.Size(250, 30), style=0, value=u'')
                          
        self.qSourceChoice = wx.Choice(choices=self.available_sources.keys(), 
              id=wxID_QSOURCECHOICE,
              name=u'qSourceChoice', parent=self.splitterWindow1, pos=wx.Point(215, 100),
              size=wx.Size(136, 28), style=0)
        self.qSourceChoice.Bind(wx.EVT_CHOICE, self.OnSourceChoiceChoice,
              id=wxID_QSOURCECHOICE)

        self.qFileNameEditor = wx.TextCtrl(id=wxID_QFILENAMEEDITOR,
              name=u'qFileNameEditor', parent=self.splitterWindow1, pos=wx.Point(215, 200),
              size=wx.Size(250, 30), style=0, value=u'')

        self.iSourceStaticText = wx.StaticText(id=wxID_ISOURCESTATICTEXT,
              label=u'I channel source:', name='qSourceStaticText', parent=self.splitterWindow1,
              pos=wx.Point(55, 50), size= wx.Size(100, 20), style=0)
        self.iSourceStaticText.SetFont(wx.Font(10,wx.SWISS,wx.NORMAL,wx.BOLD,True,u'Sans'))

        self.qSourceStaticText = wx.StaticText(id=wxID_QSOURCESTATICTEXT,
              label=u'Q channel Source:', name='qSourceStaticText', parent=self.splitterWindow1,
              pos=wx.Point(55, 100), size= wx.Size(100, 20), style=0)
        self.qSourceStaticText.SetFont(wx.Font(10,wx.SWISS,wx.NORMAL,wx.BOLD,True,u'Sans'))
  

        self.iFileStaticText = wx.StaticText(id=wxID_IFILESTATICTEXT,
              label=u'I channel file:', name='qFileStaticText', parent=self.splitterWindow1,
              pos=wx.Point(55, 150), size= wx.Size(100, 20), style=0)
        self.iFileStaticText.SetFont(wx.Font(10,wx.SWISS,wx.NORMAL,wx.BOLD,True,u'Sans'))

        self.qFileStaticText = wx.StaticText(id=wxID_QFILESTATICTEXT,
              label=u'Q channel file:', name='qFileStaticText', parent=self.splitterWindow1,
              pos=wx.Point(55, 200), size= wx.Size(100, 20), style=0)
        self.qFileStaticText.SetFont(wx.Font(10,wx.SWISS,wx.NORMAL,wx.BOLD,True,u'Sans'))
  
 

    def OnSourceChoiceChoice(self,event):
        '''If a source option other than "file" is selected, "grey out" the file input field'''

        i_source = self.iSourceChoice.GetStringSelection()
        if i_source == 'file':
            self.iFileNameEditor.Enable(True)
        else:
            self.iFileNameEditor.Enable(False)

        q_source = self.qSourceChoice.GetStringSelection()
        if q_source == 'file':
            self.qFileNameEditor.Enable(True)
        else:
            self.qFileNameEditor.Enable(False)
        

    def OnPushPacketBtnButton(self,event):
        '''accesses the AWG class to get a signal, then sends the signal'''
        self.i_file_name = self.iFileNameEditor.GetLineText(0)
        self.q_file_name = self.qFileNameEditor.GetLineText(0)
        AWG_instance = AWG(self)
        i_source_selected = self.iSourceChoice.GetStringSelection()
        q_source_selected = self.qSourceChoice.GetStringSelection()
        self.I = AWG_instance.get_signal(i_source_selected, self.i_file_name)
        self.Q = AWG_instance.get_signal(q_source_selected, self.q_file_name)


        # temporary hack:
        # skip the data type conversion for data that is already integers
        if i_source_selected == 'file' or\
           i_source_selected == 'zeros' or\
           i_source_selected == 'ones': 
            self.I = [int(x) for x in self.I]    #convert from string to integer
        else:        #convert from float to integer
            self.I = self.convert_data_type(self.I, self.interface)

        if q_source_selected == 'file' or\
           q_source_selected == 'zeros' or\
           q_source_selected == 'ones': 
            self.I = [int(x) for x in self.I]
        else:
            self.Q = self.convert_data_type(self.Q, self.interface)


        if len(self.I) == 0 and len(self.Q) == 0:  #if there is no data for whatever reason
            print "Packet is empty.  Not sending anything."
        else:  #there is data, try to send it
            self.PortHandle.pushPacket(self.I,self.Q)


    def convert_data_type(self, data, type):
        if type == "complexShort":   #short and int are the same in python
            #TODO: test the robustness of the max() module
            #WARNING: if the maximum value of the data is greater than 1, this will probably crash 
            #data_max = max(data)
            data_max = 1
            SHRT_MAX = 32767    #have to use -1 because round might round up to 32768
            data = [int(round(float(x)*(SHRT_MAX/data_max))) for x in data]  #conver to integer format
        elif type == "complexFloat":
            data = [float(x) for x in data]
        else:
            print "types other than complexFloat and complexShort are not yet supported in sources.convert"
        
        return data


    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, "Arbitrary Waveform Generator.\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)
     
         #connect to an existing port
         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(standardInterfaces__POA.complexShort)
         
    def OnCloseWindow(self,event):
        if hasattr(self.parent, 'removeToolFrame'):
            self.parent.removeToolFrame(self)
        self = None
        event.Skip()