source: issm/trunk/src/m/plot/plot_icefront.py@ 21341

try:
        import pylab as p
except ImportError:
        print "could not import pylab, matplotlib has not been installed, no plotting capabilities enabled"
import numpy as  np
from processmesh import processmesh
from applyoptions import applyoptions

def plot_icefront(md,options,fig,ax):
        #PLOT_ICEFRONT - plot segment on neumann BC
        #
        #   Usage:
        #      plot_icefront(md,options,width,i)
        #
        #   See also: PLOTMODEL
#process mesh and data
        x,y,z,elements,is2d,isplanet=processmesh(md,[],options)
        icefront=np.where(np.logical_and(np.sum(md.mask.ice_levelset[elements],1)<3,np.sum(md.mask.ice_levelset[elements],1)>-3)) 
        onlyice=np.where(np.sum(md.mask.ice_levelset[elements],1)==-3)
        noice=np.where(np.sum(md.mask.ice_levelset[elements],1)==3)

        #plot mesh
        ax.triplot(x,y,elements)

        #highlight elements on neumann
        colors=np.asarray([0.5 for element in elements[icefront]])
        
        ax.tripcolor(x,y,elements[icefront],facecolors=colors,alpha=0.5,label='elements on ice front')#,facecolor='b'

        #apply options
        options.addfielddefault('title','Neumann boundary conditions')
        options.addfielddefault('colorbar','off')
        applyoptions(md,[],options,fig,ax)