source: issm/trunk/src/m/geometry/FlagElements.py@ 16137

import numpy
import os
#from basinzoom import *
from ContourToMesh import *
from MatlabFuncs import *
from PythonFuncs import *

def FlagElements(md,region):
        """
        FLAGELEMENTS - flag the elements in an region

           The region can be given with an exp file, a list of elements or vertices

           Usage: 
              flag=FlagElements(md,region);

           Example:
              flag=FlagElements(md,'all');
              flag=FlagElements(md,'');
              flag=FlagElements(md,'Domain.exp');
              flag=FlagElements(md,'~Domain.exp');
        """

        if   isinstance(region,(str,unicode)):
                if   not region:
                        flag=numpy.zeros(md.mesh.numberofelements,bool)
                        invert=0
                elif strcmpi(region,'all'):
                        flag=numpy.ones(md.mesh.numberofelements,bool)
                        invert=0
                else:
                        #make sure that we actually don't want the elements outside the domain outline!
                        if strcmpi(region[0],'~'):
                                region=region[1:]
                                invert=1
                        else:
                                invert=0

                        #does the region domain outline exist or do we have to look for xlim,ylim in basinzoom?
                        if not os.path.exists(region):
                                if len(region)>3 and not strcmp(region[-4:],'.exp'):
                                        raise IOError("Error: File 'region' not found!" % region)
                                raise RuntimeError("FlagElements.py calling basinzoom.py is not complete.")
                                xlim,ylim=basinzoom('basin',region)
                                flag_nodes=logical_and_n(md.mesh.x<xlim[1],md.mesh.x>xlim[0],md.mesh.y<ylim[1],md.mesh.y>ylim[0])
                                flag=numpy.prod(flag_nodes[md.mesh.elements],axis=1).astype(bool)
                        else:
                                #ok, flag elements
                                [flag,dum]=ContourToMesh(md.mesh.elements[:,0:3].copy(),md.mesh.x,md.mesh.y,region,'element',1)
                                flag=flag.astype(bool)

                if invert:
                        flag=numpy.logical_not(flag)

        elif isinstance(region,numpy.ndarray) or isinstance(region,bool):
                if numpy.size(region,0)==md.mesh.numberofelements:
                        flag=region
                elif numpy.size(region,0)==md.mesh.numberofvertices:
                        flag=(numpy.sum(region[md.mesh.elements-1]>0,axis=1)==numpy.size(md.mesh.elements,1))
                else:
                        raise TypeError("Flaglist for region must be of same size as number of elements in model.")

        else:
                raise TypeError("Invalid region option")

        return flag