Changeset 21478

Timestamp:

01/11/17 01:22:30 (8 years ago)

Author:

bdef

Message:

CHG:prettyfying numbering plots

Location:

issm/trunk-jpl/src/m/plot

Files:

• plot_elementnumbering.py (modified) (2 diffs)
• plot_vertexnumbering.py (modified) (2 diffs)

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

@@ -26 +26 @@
                 ax.triplot(x,y,elements)
         else:
-                ax=inset_locator.inset_axes(axgrid[gridindex],width='100%',height='100%',loc=3,borderpad=0,axes_class=Axes3D)
+                print 'Not Implemented Yet'
 
         XLims=[np.min(x),np.max(x)]
@@ -43 +43 @@
         # and numbers
         for i,element in enumerate(elements):
-                ax.text(np.mean(x[element]),np.mean(y[element]),str(i+1),ha='center',va='center')
+                ax.text(np.mean(x[element]),np.mean(y[element]),str(i+1),ha='center',va='center',clip_on=True)
                 
         #apply options

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

@@ -28 +28 @@
                 ax.triplot(x,y,elements)
         else:
-                ax=inset_locator.inset_axes(axgrid[gridindex],width='100%',height='100%',loc=3,borderpad=0,axes_class=Axes3D)
+                print 'Not Implemented Yet'
 
-
-        XLims=[np.min(x),np.max(x)]
-        YLims=[np.min(y),np.max(y)]
+        XPad=0.1*(np.max(x)-np.min(x))
+        YPad=0.1*(np.max(y)-np.min(y))
         #plot mesh
         ax.triplot(x,y,elements)
-        highlightpos=options.getfieldvalue('highlight','none')
+        ax.set_xlim((np.min(x)-XPad,np.max(x)+XPad))
+        ax.set_ylim((np.min(y)-XPad,np.max(y)+XPad))
+
+        highlightpos=options.getfieldvalue('highlight',[])
         if highlightpos!='none':
                 #if just one element duplicate it to avoid coloring issues
@@ -42 +44 @@
                 #convert from to matlab numbering
                 highlightpos=[pos-1 for pos in highlightpos]
-                colors=np.asarray([0.5 for element in elements[highlightpos]])
-                ax.tripcolor(x,y,vertices[highlightpos],facecolors=colors,alpha=0.5)
+
         # and numbers
         for i,Xcoord in enumerate(x):
-                ax.text(x[i],y[i],str(i+1))     
-        
+                if i in highlightpos:
+                        props = dict(boxstyle='circle', pad=0.1,color='r')
+                else:
+                        props = dict(boxstyle='circle', pad=0.1,color='w')
+                ax.text(x[i],y[i],str(i+1),ha='center',va='center',backgroundcolor='w',clip_on=True,bbox=props) 
+                
         #apply options
         options.addfielddefault('title','Vertex numbers (matlab indexation)')