Coverage for /home/travis/build/RobotLocomotion/director/build/src/director-build/lib/python2.7/dist-packages/director/lcmgl : 0%

import director.vtkAll as vtk 

import director.objectmodel as om 

from director import lcmUtils 

# if bot_lcmgl cannot be important than this module will not be able to 

# support lcmgl, but it can still be imported in a disabled state 

try: 

    import bot_lcmgl 

    LCMGL_AVAILABLE = True 

except ImportError: 

    LCMGL_AVAILABLE = False 

class LCMGLObject(om.ObjectModelItem): 

    def __init__(self, name, actor): 

        om.ObjectModelItem.__init__(self, name, om.Icons.Robot) 

        self.actor = actor 

        self.actor.SetUseBounds(False) 

        self.addProperty('Visible', actor.GetVisibility()) 

        self.views = [] 

    def _onPropertyChanged(self, propertySet, propertyName): 

        om.ObjectModelItem._onPropertyChanged(self, propertySet, propertyName) 

        if propertyName == 'Visible': 

            self.actor.SetVisibility(self.getProperty(propertyName)) 

            self.renderAllViews() 

    def addToView(self, view): 

        if view in self.views: 

            return 

        self.views.append(view) 

        view.renderer().AddActor(self.actor) 

        view.render() 

    def renderAllViews(self): 

        for view in self.views: 

            view.render() 

    def onRemoveFromObjectModel(self): 

        self.removeFromAllViews() 

    def removeFromAllViews(self): 

        for view in list(self.views): 

            self.removeFromView(view) 

        assert len(self.views) == 0 

    def removeFromView(self, view): 

        assert view in self.views 

        self.views.remove(view) 

        view.renderer().RemoveActor(self.actor) 

        view.render() 

    def onMessage(self, msgBytes): 

        self.actor.UpdateGLData(msgBytes.data()) 

        self.renderAllViews() 

managerInstance = None 

class LCMGLManager(object): 

    def __init__(self, view): 

        assert LCMGL_AVAILABLE 

        self.view = view 

        self.subscriber = None 

        self.enable() 

    def isEnabled(self): 

        return self.subscriber is not None 

    def setEnabled(self, enabled): 

        if enabled and not self.subscriber: 

            self.subscriber = lcmUtils.addSubscriber('LCMGL.*', callback=self.onMessage) 

        elif not enabled and self.subscriber: 

            lcmUtils.removeSubscriber(self.subscriber) 

            self.subscriber = None 

    def enable(self): 

        self.setEnabled(True) 

    def disable(self): 

        self.setEnabled(False) 

    def onMessage(self, msgBytes, channel): 

        msg = bot_lcmgl.data_t.decode(msgBytes.data()) 

        drawObject = self.getDrawObject(msg.name) 

        if not drawObject: 

            drawObject = self.addDrawObject(msg.name, msgBytes) 

        drawObject.onMessage(msgBytes) 

    def getDrawObject(self, name): 

        parent = om.getOrCreateContainer('LCM GL') 

        return parent.findChild(name) 

    def addDrawObject(self, name, msgBytes): 

        actor = vtk.vtkLCMGLProp() 

        obj = LCMGLObject(name, actor) 

        om.addToObjectModel(obj, om.getOrCreateContainer('LCM GL')) 

        obj.addToView(self.view) 

        return obj 

def init(view): 

    if not hasattr(vtk, 'vtkLCMGLProp'): 

        return None 

    global managerInstance 

    managerInstance = LCMGLManager(view) 

    return managerInstance