Changeset 21256

Timestamp:

10/11/16 07:24:53 (8 years ago)

Author:

bdef

Message:

NEW:adding quiver to py plotmodel

Location:

issm/trunk-jpl/src/m/plot

Files:

• plot_quiver.py (added)
• plot_unit.py (modified) (2 diffs)

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

@@ -1 +1 @@
 from cmaptools import truncate_colormap
+from plot_quiver import plot_quiver
 try:
         import pylab as p
@@ -89 +90 @@
         # {{{ plotting quiver (TODO)
         elif datatype==3:
-                vx=data[:,0]
-                vy=data[:,1]
-                #TODO write plot_quiver.py to handle this here
-                color=np.sqrt(vx**2+vy**2)
-                scale=options.getfieldvalue('scale',1000)
-                width=options.getfieldvalue('width',0.005*(np.amax(x)-np.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)
+                if is2d:
+                        Q=plot_quiver(x,y,data,options,ax)
+                else:
+                        raise ValueError('plot_unit error: 3D node plot not supported yet')
                 return
+        
         # }}}
         # {{{ plotting P1 Patch (TODO)