Changeset 20500 for issm/trunk/src/m/plot/plot_unit.py

Timestamp:

04/12/16 21:32:01 (9 years ago)

Author:

Mathieu Morlighem

Message:

merged trunk-jpl and trunk for revision 20497

Location:

issm/trunk

Files:

• . (modified) (2 props)
• src (modified) (1 prop)
• src/m/plot/plot_unit.py (modified) (1 diff)

issm/trunk

@@ +1 @@
+build-js
+build-esmf
+build-gcm
 build-fw
 build-ad

@@ +1 @@
+build-js
+build-esmf
+build-gcm
 build-fw
 build-ad

issm/trunk/src/m/plot/plot_unit.py

@@ -1 +1 @@
 from cmaptools import truncate_colormap
 try:
-        import pylab as p
-        import matplotlib as mpl
-        import matplotlib.pyplot as plt
+    import pylab as p
+    import matplotlib as mpl
+    import matplotlib.pyplot as plt
+    import numpy as npy
 except ImportError:
-        print "could not import pylab, matplotlib has not been installed, no plotting capabilities enabled"
+    print "could not import pylab, matplotlib has not been installed, no plotting capabilities enabled"
 
 def plot_unit(x,y,z,elements,data,is2d,isplanet,datatype,options,ax):
-        """
-        PLOT_UNIT - unit plot, display data
-        
-                Usage:
-                        plot_unit(x,y,z,elements,data,is2d,isplanet,datatype,options)
-        
-                See also: PLOTMODEL, PLOT_MANAGER
-        """
-        
-        #edgecolor
-        edgecolor=options.getfieldvalue('edgecolor','None')
-        
-        #number of colorlevels for plots
-        colorlevels=options.getfieldvalue('colorlevels',128)
-        
-        alpha=options.getfieldvalue('alpha',1)
-        
-        #colormap
-        # default sequential colormap
-        defaultmap=truncate_colormap(mpl.cm.gnuplot2,0.1,0.9,128)
-        cmap=options.getfieldvalue('colormap',defaultmap)
-        if options.exist('cmap_set_over'):
-           over=options.getfieldvalue('cmap_set_over','0.5')
-           cmap.set_over(over)
-        if options.exist('cmap_set_under'):
-           under=options.getfieldvalue('cmap_set_under','0.5')
-           cmap.set_under(under)
-        
-        #normalize colormap if clim/caxis specified
-        if options.exist('clim'):
-           lims=options.getfieldvalue('clim',[min(data),max(data)])
-        elif options.exist('caxis'):
-           lims=options.getfieldvalue('caxis',[min(data),max(data)])
-        else:
-                if min(data)==max(data):
-                        lims=[min(data)-0.5,max(data)+0.5]
-                else:
-                        lims=[min(data),max(data)]
-        norm = mpl.colors.Normalize(vmin=lims[0], vmax=lims[1])
-        if datatype==1:
-           #element plot
-                if is2d:
-                        tri=ax.tripcolor(x,y,elements,data,colorlevels,cmap=cmap,norm=norm,alpha=alpha,edgecolors=edgecolor)
-                else:
-                        raise ValueError('plot_unit error: 3D element plot not supported yet')
-                return 
-        
-        elif datatype==2:
-                #node plot
-                if is2d:
-                        tri=ax.tricontourf(x,y,elements,data,colorlevels,cmap=cmap,norm=norm,alpha=alpha,extend='both')
-                        if edgecolor != 'None':
-                                ax.triplot(x,y,elements,color=edgecolor)
-                else:
-                        raise ValueError('plot_unit error: 3D node plot not supported yet')
-                return
-        
-        elif datatype==3:
-                print 'plot_unit message: quiver plot not implemented yet'
-                return
-        
-        elif datatype==4:
-                #P1 patch plot
-                print 'plot_unit message: P1 patch plot not implemented yet'
-                return
-        
-        elif datatype==5:
-                print 'plot_unit message: P0 patch plot not implemented yet'
-                return
-        
-        else:
-                raise ValueError('datatype=%d not supported' % datatype)
-
+    """
+    PLOT_UNIT - unit plot, display data
+    
+        Usage:
+                plot_unit(x,y,z,elements,data,is2d,isplanet,datatype,options)
+    
+        See also: PLOTMODEL, PLOT_MANAGER
+    """
+    
+    #edgecolor
+    edgecolor=options.getfieldvalue('edgecolor','None')
+    
+    #number of colorlevels for plots
+    colorlevels=options.getfieldvalue('colorlevels',128)
+    
+    alpha=options.getfieldvalue('alpha',1)
+    
+    #colormap
+    # default sequential colormap
+    defaultmap=truncate_colormap(mpl.cm.gnuplot2,0.1,0.9,128)
+    cmap=options.getfieldvalue('colormap',defaultmap)
+    if options.exist('cmap_set_over'):
+        over=options.getfieldvalue('cmap_set_over','0.5')
+        cmap.set_over(over)
+    if options.exist('cmap_set_under'):
+        under=options.getfieldvalue('cmap_set_under','0.5')
+        cmap.set_under(under)
+    
+    #normalize colormap if clim/caxis specified
+    if options.exist('clim'):
+        lims=options.getfieldvalue('clim',[npy.amin(data),npy.amax(data)])
+    elif options.exist('caxis'):
+        lims=options.getfieldvalue('caxis',[npy.amin(data),npy.amax(data)])
+    else:
+        if npy.amin(data)==npy.amax(data):
+            lims=[npy.amin(data)-0.5,npy.amax(data)+0.5]
+        else:
+            lims=[npy.amin(data),npy.amax(data)]
+    norm = mpl.colors.Normalize(vmin=lims[0], vmax=lims[1])
+    if datatype==1:
+       #element plot
+        if is2d:
+            tri=ax.tripcolor(x,y,elements,data,colorlevels,cmap=cmap,norm=norm,alpha=alpha,edgecolors=edgecolor)
+        else:
+            raise ValueError('plot_unit error: 3D element plot not supported yet')
+        return 
+    
+    elif datatype==2:
+        #node plot
+        if is2d:
+            tri=ax.tricontourf(x,y,elements,data,colorlevels,cmap=cmap,norm=norm,alpha=alpha,extend='both')
+            if edgecolor != 'None':
+                ax.triplot(x,y,elements,color=edgecolor)
+        else:
+            raise ValueError('plot_unit error: 3D node plot not supported yet')
+        return
+    
+    elif datatype==3:
+        vx=data[:,0]
+        vy=data[:,1]
+        #TODO write plot_quiver.py to handle this here
+        color=npy.sqrt(vx**2+vy**2)
+        scale=options.getfieldvalue('scale',1000)
+        width=options.getfieldvalue('width',0.005*(npy.amax(x)-npy.amin(y)))
+        headwidth=options.getfieldvalue('headwidth',3)
+        headlength=options.getfieldvalue('headlength',5)
+        Q=ax.quiver(x,y,vx,vy,color,cmap=cmap,norm=norm,scale=scale,
+                width=width,headwidth=headwidth,headlength=headlength)
+        return
+    
+    elif datatype==4:
+        #P1 patch plot
+        print 'plot_unit message: P1 patch plot not implemented yet'
+        return
+    
+    elif datatype==5:
+        print 'plot_unit message: P0 patch plot not implemented yet'
+        return
+    
+    else:
+        raise ValueError('datatype=%d not supported' % datatype)
+    

@@ +1 @@
+build-js
+build-esmf
+build-gcm
 build-fw
 build-ad