source: issm/trunk-jpl/src/m/plot/plot_icefront.py@ 21588

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 check
        icefront=np.where(np.abs(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)

        #hydro neumann
        hydro_neumann=np.where(md.hydrology.neumannflux!=0)
        #plot mesh
        ax.triplot(x,y,elements)

        #highlight elements on neumann
        if len(icefront[0])>0:
                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')
        if len(hydro_neumann[0])>0:
                colors=np.asarray([0.5 for element in elements[hydro_neumann]])
                ax.tripcolor(x,y,elements[hydro_neumann],facecolors=colors,alpha=0.5,label='non zero neumann flux for the hydrology')

        #apply options
        options.addfielddefault('title','Neumann boundary conditions')
        options.addfielddefault('colorbar','off')