zombie-gone:4ba33e2 is merged to master

Author: hrani

python/rdesigneur/moogul.py

@@ -39,6 +39,7 @@ class MooView:
     rgb = []
     viewList = []
     consolidatedTitle = ""
+    colorbarViewIdx = 0    # Which View does the colorbar serve?
 
     def __init__( self, swx = 10, swy = 10, hideAxis = True, title = "view", colormap = 'jet'
     ):
@@ -56,7 +57,9 @@ def __init__( self, swx = 10, swy = 10, hideAxis = True, title = "view", colorma
         self.colorbar = None
         self.valMin = 0.0
         self.valMmax = 1.0
+        self.simTime = 0.0
         self.plotFlag_ = True
+        self.cbox = []
 
     @staticmethod
     def replayLoop():
@@ -68,7 +71,12 @@ def replayLoop():
                 for view in MooView.viewList:
                     view.replaySnapshot( idx )
                 vp.sleep( MooView.viewList[0].sleep )
-            vp.sleep( 0.5 ) # Pause 0.5 sec between replays
+            t = time.time() + 0.5
+            while time.time() < t:
+                for view in MooView.viewList:
+                    view.rotateFunc()
+                vp.sleep( MooView.viewList[0].sleep )
+            #vp.sleep( 0.5 ) # Pause 0.5 sec between replays
 
     def notifySimulationEnd( self ):
         if self.viewIdx == 0:
@@ -98,7 +106,11 @@ def toggleReplay( self ):
     def setSleepTime( self ):
         idx = int( round( self.sleepSlider.value ) )
         self.sleep = sleepTimes[idx]
-        self.sleepLabel.text = "    Frame dt = {:1.3f} sec".format( self.sleep )
+        self.sleepLabel.text = "Frame dt = {:1.4f} s".format( self.sleep )
+
+    def setRotation( self, slider ):
+        self.doRotation = ( abs( self.rotation ) > 0.005 )
+        self.rotation = slider.value
 
     def updateAxis( self ):
         if not self.colorbar:
@@ -119,26 +131,87 @@ def updateAxis( self ):
         self.zAx.axis = vp.vector( z.dot( right ), z.dot( up ), 0.0 )
         self.axisLength.text = "{:.2f} <i>u</i>m".format( dx * 1e6*self.scene.range * self.colorbar.width / self.scene.width )
 
+    def raiseMax( self, isDouble = False ):
+        moov = MooView.viewList[MooView.colorbarViewIdx]
+        valRange = moov.valMax - moov.valMin
+        if isDouble:
+            moov.valMax = moov.valMin + 2 * valRange
+        else:
+            moov.valMax = moov.valMin + valRange * 1.1111111111111111111111
+        return moov.valMax
+
+    def lowerMax( self, isDouble = False ):
+        moov = MooView.viewList[MooView.colorbarViewIdx]
+        valRange = moov.valMax - moov.valMin
+        if isDouble:
+            moov.valMax = moov.valMin + 0.5 * valRange
+        else:
+            moov.valMax = moov.valMin + 0.9 * valRange
+        return moov.valMax
+
+    def raiseMin( self, isDouble = False ):
+        moov = MooView.viewList[MooView.colorbarViewIdx]
+        valRange = moov.valMax - moov.valMin
+        if isDouble:
+            moov.valMin = moov.valMmax - 0.5 * valRange
+        else:
+            moov.valMin = moov.valMax - valRange * 0.9
+        return moov.valMin
+
+    def lowerMin( self, isDouble = False ):
+        moov = MooView.viewList[MooView.colorbarViewIdx]
+        valRange = moov.valMax - moov.valMin
+        if isDouble:
+            moov.valMin = moov.valMax - 2 * valRange
+        else:
+            moov.valMin = moov.valMax - valRange * 1.1111111111111111111111
+        return moov.valMin
+
+    def scaleColorbar( self, event ):
+        loc = event.pos
+        if loc.y > 6:
+            idx = MooView.colorbarViewIdx + 1
+            if idx >= len( MooView.viewList ):
+                idx = 0
+            self.selectCbar( idx )
+            return
+        if loc.y > 4 and loc.y < 4.7:
+            ret = self.lowerMax( self.scene.mouse.shift )
+            self.barMax.text = "{:.3e}".format(ret)
+        elif loc.y > 4.85 and loc.y < 5.5:
+            ret = self.raiseMax( self.scene.mouse.shift )
+            self.barMax.text = "{:.3e}".format(ret)
+        elif loc.y > -3.25 and loc.y < -2.6:
+            ret = self.raiseMin( self.scene.mouse.shift )
+            self.barMin.text = "{:.3e}".format(ret)
+        elif loc.y > -3.95 and loc.y < -3.45:
+            ret = self.lowerMin( self.scene.mouse.shift )
+            self.barMin.text = "{:.3e}".format(ret)
+        for moov in MooView.viewList:
+            moov.drawables_[0].updateLimits( moov.valMin, moov.valMax )
+
     def innerColorbar( self, title, bg ):
         barWidth = SCALE_SCENE * 1.5
         if ( bgLookup(bg).mag < 1 ):
             barTextColor = vp.color.white
         else:
             barTextColor = vp.color.black
-        self.colorbar = vp.canvas( title = title, width = barWidth, height = self.swy * SCALE_SCENE, background = bgLookup(bg), align = 'left', range = 1, autoscale = False )
-        #self.colorbar = vp.canvas( title = title, width = barWidth, height = self.swy * SCALE_SCENE, background = vp.color.cyan, align = 'left', range = 1, autoscale = False )
+        self.colorbar = vp.canvas( title = "Datasets =  ", width = barWidth, height = self.swy * SCALE_SCENE, background = bgLookup(bg), align = 'left', range = 1, autoscale = False )
         self.colorbar.userzoom = False
         self.colorbar.userspin = False
         self.colorbar.userpan = False
+        #self.colorbar.bind( 'keydown', self.scaleColorbar )
+        self.colorbar.bind( 'click', self.scaleColorbar )
         height = 0.10
         width = 5
         axOrigin = vp.vector( 0, -5.5, 0 )
         for idx, rgb in enumerate( self.rgb ):
             cbox = vp.box( canvas = self.colorbar, pos = vp.vector( 0, height * (idx - 26), 0), width = width, height = height, color = rgb )
-        barName = self.title.replace( ' ', '\n' )
+        barName = self.title.replace( '.', '\n' )
+        barName = barName.replace( '/', '\n', 1 )
         self.barName = vp.label( canvas = self.colorbar, align = 'left', pixel_pos = True, pos = vp.vector( 2, (self.swy - 0.32) * SCALE_SCENE, 0), text = barName, height = 15, color = barTextColor, box = False, opacity = 0 )
-        self.barMin = vp.label( canvas = self.colorbar, align = 'center', pixel_pos = True, pos = vp.vector( barWidth/2, self.swy * SCALE_SCENE * 0.22, 0), text = "{:.3f}".format(self.valMin), height = 12, color = barTextColor, box = False, opacity = 0 )
-        self.barMax = vp.label( canvas = self.colorbar, align = 'center', pixel_pos = True, pos = vp.vector( barWidth/2, (self.swy - 1.2) * SCALE_SCENE, 0), text = "{:.3f}".format(self.valMax), height = 12, color = barTextColor, box = False, opacity = 0 )
+        self.barMin = vp.label( canvas = self.colorbar, align = 'center', pixel_pos = True, pos = vp.vector( barWidth/2, self.swy * SCALE_SCENE * 0.22, 0), text = "{:.3e}".format(self.valMin), height = 12, color = barTextColor, box = False, opacity = 0 )
+        self.barMax = vp.label( canvas = self.colorbar, align = 'center', pixel_pos = True, pos = vp.vector( barWidth/2, (self.swy - 1.2) * SCALE_SCENE, 0), text = "{:.3e}".format(self.valMax), height = 12, color = barTextColor, box = False, opacity = 0 )
         self.xAx = vp.cylinder( canvas = self.colorbar, pos = axOrigin, axis = vp.vector( 0.8, 0, 0 ), radius = 0.04, color = vp.color.red )
         self.yAx = vp.cylinder( canvas = self.colorbar, pos = axOrigin, axis = vp.vector( 0, 0.8, 0 ), radius = 0.04, color = vp.color.green )
         self.zAx = vp.cylinder( canvas = self.colorbar, pos = axOrigin, axis = vp.vector( 0, 0, 0 ), radius = 0.04, color = vp.color.blue )
@@ -150,12 +223,34 @@ def makeColorbar( self, doOrnaments = True, colorscale = 'jet', bg = 'default' )
             title = MooView.consolidatedTitle + "\n"
         self.innerColorbar( title, bg )
         if doOrnaments:
-            self.timeLabel = vp.wtext( text = "Time =  0.000 sec", pos = self.colorbar.title_anchor )
-            self.sleepLabel = vp.wtext( text = "    Frame dt = 0.005 sec", pos = self.colorbar.title_anchor )
+            for idx, mv in enumerate( MooView.viewList ):
+                chk = vp.checkbox( bind = mv.toggleView, checked = True, text = mv.title + "    ",  pos = self.colorbar.title_anchor )
+                chk.idx = idx
+                self.cbox.append( chk )
+            self.colorbar.append_to_title("\n")
+            self.timeLabel = vp.wtext( text = "Time =  0.000 s\n", pos = self.colorbar.title_anchor )
+            self.sleepLabel = vp.wtext( text = "Frame dt = 0.0050 s", pos = self.colorbar.title_anchor )
             self.sleepSlider = vp.slider( pos = self.colorbar.title_anchor, length = 200, bind = self.setSleepTime, min = 0, max = len( sleepTimes ) -1, value = min( len( sleepTimes ), 2  ) )
             self.replayButton = vp.button( text = "Start Replay", pos = self.colorbar.title_anchor, bind=self.toggleReplay, disabled = True )
+            self.rotateLabel = vp.wtext( text = "      Rotation", pos = self.colorbar.title_anchor )
+            self.rotateSlider = vp.slider( pos = self.colorbar.title_anchor, length = 200,  bind=self.setRotation, min = -0.1, max = 0.1, value = 0.0 )
             self.colorbar.append_to_title("\n")
 
+    def selectCbar( self, idx ):
+        MooView.colorbarViewIdx = idx
+        moov = MooView.viewList[idx]
+        view0 = MooView.viewList[0]
+        view0.barName.text = view0.cbox[idx].text.replace( '.', '\n' )
+        view0.barName.text = view0.barName.text.replace( '/', '\n', 1 )
+        view0.barMin.text = "{:.3e}".format(moov.valMin)
+        view0.barMax.text = "{:.3e}".format(moov.valMax)
+
+    def toggleView( self, cbox ):
+        for d in self.drawables_:
+            d.setVisible( cbox.checked )
+        if cbox.checked:    # The colorbar is assigned to selected view:
+            self.selectCbar( cbox.idx )
+
     def pickObj( self ):
         obj = self.scene.mouse.pick
         if obj == None:
@@ -236,27 +331,29 @@ def firstDraw( self, mergeDisplays, rotation=0.0, elev=0.0, azim=0.0, center = [
                 self.doAutoscale()
             self.updateAxis()
         if self.viewIdx == (MooView.viewIdx-1):
-            self.graph = vp.graph( title = "Graph", xtitle = "Time (s)", ytitle = " Units here", width = 700, fast=False, align = "left" )
-            self.graphPlot1 = vp.gcurve( color = vp.color.blue, interval=-1)
-            #self.graphPlot1.data =  [[0,0], [1,1],[2,0],[3,4],[4,0], [5,1]]
-            #self.graphPlot1.plot( [[0,0], [1,1],[2,0],[3,4],[4,0]] )
+            MooView.viewList[0].graph = vp.graph( title = "Graph", xtitle = "Time (s)", ytitle = " Units here", width = 700, fast=False, align = "left" )
+            MooView.viewList[0].graphPlot1 = vp.gcurve( color = vp.color.blue, interval=-1)
 
+    def rotateFunc(self ):
+        if self.doRotation and abs( self.rotation ) < 2.0 * 3.14 / 3.0:
+            self.scene.forward = vp.rotate( self.scene.forward, angle = self.rotation, axis = self.scene.up )
+            self.updateAxis()
 
     def updateValues( self, simTime ):
+        self.simTime = simTime
         for i in self.drawables_:
             i.updateValues( simTime )
-        if self.doRotation and abs( self.rotation ) < 2.0 * 3.14 / 3.0:
-            self.scene.forward = vp.rotate( self.scene.forward, angle = self.rotation, axis = self.scene.up )
-            self.updateAxis()
+        self.rotateFunc()
         if self.viewIdx == 0:
-            self.timeLabel.text = "Time = {:7.3f} sec".format( simTime )
+            self.timeLabel.text = "Time = {:7.3f} s\n".format( simTime )
             vp.sleep( self.sleep )
 
     def replaySnapshot( self, idx ):
         for i in self.drawables_:
             simTime = i.replaySnapshot( idx )
+        self.rotateFunc()
         if self.viewIdx == 0:
-            self.timeLabel.text = "Time = {:7.3f} sec".format( simTime )
+            self.timeLabel.text = "Time = {:7.3f} s\n".format( simTime )
             self.updateAxis()
 
     def doAutoscale( self ):
@@ -320,13 +417,15 @@ def moveView(self, event):
             self.scene.camera.rotate( angle = -dtheta, axis = camAxis, origin = self.scene.camera.pos )
             return
         if event.key == "d": # Diameter scaling down
-            for dbl in self.drawables_:
-                dbl.diaScale *= 1.0 - self.sensitivity * 4
+            moov = MooView.viewList[MooView.colorbarViewIdx]
+            for dbl in moov.drawables_:
+                dbl.diaScale *= 1.0 - moov.sensitivity * 4
                 dbl.updateDiameter()
             return
         if event.key == "D":
-            for dbl in self.drawables_:
-                dbl.diaScale *= 1.0 + self.sensitivity * 4
+            moov = MooView.viewList[MooView.colorbarViewIdx]
+            for dbl in moov.drawables_:
+                dbl.diaScale *= 1.0 + moov.sensitivity * 4
                 dbl.updateDiameter()
             return
         if event.key == "s": # Scale down sleep time, make it faster.
@@ -430,6 +529,7 @@ def __init__( self,
         self.valMax = valMax
         self.segments = []
         self.snapshot = []
+        self.visible = True
         #cmap = plt.get_cmap( self.colormap, lut = NUM_CMAP )
         #self.rgb = [ list2vec(cmap(i)[0:3]) for i in range( NUM_CMAP ) ]
 
@@ -450,10 +550,22 @@ def updateValues( self, simTime ):
         indices = np.maximum( np.minimum( scaleVal, NUM_CMAP-0.5), 0.0).astype(int)
 
         # Have to figure how this will work with multiple update rates.
-        self.snapshot.append( [simTime, indices] )
+        self.snapshot.append( [simTime, self.val] )
+
+        self.displayValues( indices )
 
+    def updateLimits( self, vmin, vmax ):
+        if self.autoscale:
+            valMin = min( self.val )
+            valMax = max( self.val )
+        else:
+            valMin = self.valMin = vmin
+            valMax = self.valMax = vmax
+        scaleVal = NUM_CMAP * (self.val - valMin) / (valMax - valMin)
+        indices = np.maximum( np.minimum( scaleVal, NUM_CMAP-0.5), 0.0).astype(int)
         self.displayValues( indices )
 
+
     def displayValues( self, indices ):
         for idx, seg in zip( indices, self.segments ): 
             seg.color = self.rgb[ idx]
@@ -462,7 +574,9 @@ def displayValues( self, indices ):
     def replaySnapshot( self, idx ):
         if idx >= len( self.snapshot ):
             return 0.0
-        self.displayValues( self.snapshot[idx][1] )
+        scaleVal = NUM_CMAP * (self.snapshot[idx][1] - self.valMin) / (self.valMax - self.valMin)
+        indices = np.maximum( np.minimum( scaleVal, NUM_CMAP-0.5), 0.0).astype(int)
+        self.displayValues( indices )
         return self.snapshot[idx][0]    # return frame time
 
     def updateDiameter( self ):
@@ -495,20 +609,13 @@ def plotHistory( self, path, field, graph, plot ):
         graph.title = path + "." + field
         dat = [[x,y] for x, y in zip( t, v ) ]
         plot.data = dat
-        #print (dat)
-        #print( "IN plotHistory, ", len( dat), len( v ) )
-        #plot.data = [[x,y] for x, y in zip( t, v ) ]
-        #plot.data = [[x,sin(x)] for x in range( 0.0, 10.0, 0.1 ) ]
-        '''
-        fig = plt.figure( 1 )
-        plt.ion()
-        plt.title( path + "." + field )
-        plt.xlabel( "Time (s)" )
-        plt.ylabel( field + " um, units?" )
-        plt.plot( t, v )
-        plt.show( block = False )
-        fig.canvas.draw()
-        '''
+
+    def setVisible( self, state ):
+        if self.visible == state:
+            return
+        self.visible = state
+        for s in self.segments:
+            s.visible = state
 
 
 #####################################################################