root/ossiedev/branches/jsnyder/ComponentProject/PythonSrc/MainFrameGlue.py @ 7254

import sys, os, copy
import wavedev.importResource
import wavedev.importNode
import wavedev.WaveformClass
import wavedev.PlatformClass
import wavedev.ComponentClass
import threading
import cPickle
from edu.vt.ossie.jyinterface.interfaces import MainFrame

class MainFrameTreeNode:
    def __init__(self, name, contents):
        self.name = name
        self.contents = contents


class RunInNewThread(threading.Thread):
    def __init__(self, inCommand):
        self.command = inCommand
        threading.Thread.__init__(self)


    def run(self):
        os.system(self.command)



class MainFrameGlue(MainFrame):
    def __init__(self):
        self.active_waveform = wavedev.WaveformClass.Waveform()
        self.active_platform = wavedev.PlatformClass.Platform()
        self.installPath = "/sdr/"


    def loadResources(self):
        self.Available_Components = []
        self.Available_Devices = []
        self.Available_Nodes = []
        resList = []

        # List possible resource directories (components and device)
        baseComponentPath = self.installPath + 'xml/'
        if os.path.isdir(baseComponentPath):
            for r in os.listdir(baseComponentPath):
                if r != 'dtd':  # ignore dtd directory
                    resList.append( (baseComponentPath,r) )
        else:
            errorMsg(self,"No component resources could be found in: " + self.installPath)
            return

        # find the .scd.xml files for each resource

        for r in resList:
            tmpResName = r[1]
            tmpResPath = r[0] + r[1]
            tmpComp = wavedev.importResource.getResource(tmpResPath,tmpResName,self)

            if tmpComp == None:
                continue
            if tmpComp.type == 'resource':
                self.Available_Components.append(tmpComp)

            elif tmpComp.type == 'executabledevice':
                self.Available_Devices.append(tmpComp)

            elif tmpComp.type == 'loadabledevice':
                self.Available_Devices.append(tmpComp)

            elif tmpComp.type == 'device':
                self.Available_Devices.append(tmpComp)


        nodeList = []
        if os.path.isdir(self.installPath + 'nodes'):
            nodeList = os.listdir(self.installPath + 'nodes')
            print nodeList
        else:
            errorMsg(self, "No nodes could be found in: " + self.installPath)

        # find the scd files for each node
        for node_name in nodeList:

            # check for existence of DomainManager XML files
            nodes_root_path = self.installPath + os.path.sep + 'nodes' + os.path.sep + node_name + os.path.sep
            if not os.path.exists(nodes_root_path + 'DeviceManager.dcd.xml'):
                errorMsg(self, "Could not find DeviceManager.dcd.xml in: " + nodes_root_path)
                continue
            elif not os.path.exists(nodes_root_path + 'DeviceManager.prf.xml'):
                errorMsg(self, "Could not find DeviceManager.prf.xml in: " + nodes_root_path)
                continue
            elif not os.path.exists(nodes_root_path + 'DeviceManager.scd.xml'):
                errorMsg(self, "Could not find DeviceManager.scd.xml in: " + nodes_root_path)
                continue
            elif not os.path.exists(nodes_root_path + 'DeviceManager.spd.xml'):
                errorMsg(self, "Could not find DeviceManager.spd.xml in: " + nodes_root_path)
                continue

            nodeName = node_name
            nodePath = self.installPath + 'nodes/' + nodeName + '/'

            print "calling getNode(", nodePath, ",", nodeName, ")"
            tmpNode = wavedev.importNode.getNode(nodePath,nodeName,self)

            if tmpNode == None:
                print "WARNING: possibly an error reading node " + nodePath + "/" + nodeName
                continue
            self.Available_Nodes.append(tmpNode)
        return [MainFrameTreeNode("Components", self.Available_Components),
                MainFrameTreeNode("Devices", self.Available_Devices),
                MainFrameTreeNode("Nodes", self.Available_Nodes)]


    def getActiveWaveform(self):
        return self.active_waveform


    def getActivePlatform(self):
        return self.active_platform


    def getOssieInstallPath(self):
        return self.installPath


    def setOssieInstallPath(self, newInstallPath):
        self.installPath = newInstallPath


    def generateTestWaveform(self):
        self.active_wave = wavedev.WaveformClass.Waveform()
        int1 = wavedev.ComponentClass.Interface('complexShort')
        op1 = wavedev.ComponentClass.Operation('pushPacket','void')
        param1 = wavedev.ComponentClass.Param('I','PortTypes::ShortSequence','in')
        param2 = wavedev.ComponentClass.Param('Q','PortTypes::ShortSequence','in')
        op1.params.extend([param1,param2])
        int1.operations.append(op1)

        t2 = wavedev.ComponentClass.Component("Transmitter",AC=True)
        p1 = wavedev.ComponentClass.Port('inPortTx1',copy.deepcopy(int1),'Provides')
        p2 = wavedev.ComponentClass.Port('outPortTx1',copy.deepcopy(int1),'Uses')
        t2.ports.append(p1); t2.ports.append(p2)
        self.active_wave.components.append(t2)

        t3 = wavedev.ComponentClass.Component("Channel")
        p1 = wavedev.ComponentClass.Port('inPortCh1',copy.deepcopy(int1),'Provides')
        p2 = wavedev.ComponentClass.Port('inPortCh2',copy.deepcopy(int1),'Provides')
        p3 = wavedev.ComponentClass.Port('inPortCh3',copy.deepcopy(int1),'Provides')
        p4 = wavedev.ComponentClass.Port('outPortCh1',copy.deepcopy(int1),'Uses')
        p5 = wavedev.ComponentClass.Port('outPortCh2',copy.deepcopy(int1),'Uses')
        p6 = wavedev.ComponentClass.Port('outPortCh3',copy.deepcopy(int1),'Uses')
        t3.ports.extend([p1,p2,p3,p4,p5,p6])
        self.active_wave.components.append(t3)

        t4 = wavedev.ComponentClass.Component("Receiver")
        p1 = wavedev.ComponentClass.Port('inPortRx1',copy.deepcopy(int1),'Provides')
        p2 = wavedev.ComponentClass.Port('outPortRx1',copy.deepcopy(int1),'Uses')
        t4.ports.append(p1); t4.ports.append(p2)
        self.active_wave.components.append(t4)

        temp_dev = wavedev.ComponentClass.Component("GPP")
        self.active_wave.devices.append(temp_dev)
        return [MainFrameTreeNode("Sample Waveform", self.active_wave)]

    def displayDoxygen(self, referenceMaterials):
        defaultHtml = 'index.html'
        defaultPdf = 'refman.pdf'
        docList = None

        docsPath = self.installPath + 'docs/' + referenceMaterials.name
        print docsPath
        RunInNewThread('evince ' + "/home/Ossie/Desktop/OSSIE_0.6.2_User_Guide.pdf").start()
        if os.path.isdir(docsPath):
            docList = os.listdir(docsPath)
            if os.path.isfile(docsPath + '/' + defaultHtml):
                webbrowser.open_new('file://' + docsPath + '/' + defaultHtml)
            elif os.path.isfile(docsPath + '/' + defaultPdf):
                #find more portable way to view pdf?
                try:
                    RunInNewThread('evince ' + docsPath + '/' + defaultPdf + ' &')
                finally:
                    errorMsg(self,'evince pdf viewer threw exception or not found')
                #webbrowser.open('file://' + docsPath + '/' + defaultPdf, 1)
            else:
                errorMsg(self,'Neither index.html nor refman.pdf found in ' + docsPath + ': directory listing: ' + str(docList))
        else:
            errorMsg(self,'No directory for ' + referenceMaterials.name + ' could be found in: ' + self.installPath + 'docs')
            return


    def errorMsg(self, message):
        sys.stderr.write(message)


    def saveProject(self, saveProjectPath):
        f = open(saveProjectPath,'w')
        cPickle.dump(('project',self.active_waveform,self.active_platform),f)