def addFoamScalarField(self, fieldpath, name, time, projectedField=False):
     """
     """
     tSurf = TriSurfaceScalar.readFromFoamFile(
         fieldpath, self.triSurfaceMesh, time=time, projectedField=projectedField
     )
     self.fields[name] = tSurf
 def addFoamScalarField(self, fieldpath, name, time, projectedField=False):
     '''
     '''
     tSurf = TriSurfaceScalar.readFromFoamFile(
         fieldpath,
         self.triSurfaceMesh,
         time=time,
         projectedField=projectedField)
     self.fields[name] = tSurf
    def readScalarFromFoamFile(self,
                               varsFile,
                               pointsFile,
                               facesFile,
                               viewAnchor=(0,0,0),
                               xViewBasis=(1,0,0),
                               yViewBasis=(0,1,0),
                               dx=None,
                               dy=None,
                               interpolationMethod='cubic',
                               kind='min_E'):
        '''
        '''
        varName=os.path.basename(varsFile)        
        tsm = TriSurfaceMesh.readFromFoamFile(pointsFile=pointsFile,
                                              facesFile=facesFile,
                                              viewAnchor=viewAnchor,
                                              xViewBasis=xViewBasis,
                                              yViewBasis=yViewBasis)
                                              
        tss = TriSurfaceScalar.readFromFoamFile(varsFile=varsFile,
                                                triSurfaceMesh=tsm,
                                                time=0,
                                                projectedField=False)                  

        points = np.vstack((tss.x,tss.y)).T

        #if not hasattr(self,'data'):
            #print 'dict does not exists'
        if not self.data.has_key('dx') or self.data.has_key('dy'):
            print 'keys dx and dy does not exist'
            if dx==None:
                dxlist=[a for a in np.abs(np.diff(points[:,0])) if a>0]
                dx=np.min(dxlist)
            if dy==None:
                dylist=[a for a in np.abs(np.diff(points[:,1])) if a>0]
                dy=np.min(dylist)

            MaxX=np.max(points[:,0])
            MinX=np.min(points[:,0])
            MaxY=np.max(points[:,1])
            MinY=np.min(points[:,1])
            extent=[MinX,MaxX,MinY,MaxY]
            #print MinX,MaxX,MinY,MaxY

            cellsX=int((MaxX-MinX)/dx)
            cellsY=int((MaxY-MinY)/dy)
            #print cellsX,cellsY
            grid_y, grid_x = np.mgrid[MinY:MaxY:np.complex(0,cellsY),MinX:MaxX:np.complex(0,cellsX)]
            triang = tss.triangulation
#            scalar_i=doInterp(triang,tss.s,grid_x, grid_y)
            scalar_i=self.interpolateField(tss.s,grid_x, grid_y, triang, method=interpolationMethod, kind=kind)
            vx_i=np.empty(scalar_i.shape)
            vy_i=np.empty(scalar_i.shape)
            vz_i=np.empty(scalar_i.shape)
            vx_i[:]=np.NAN
            vy_i[:]=np.NAN
            vz_i[:]=np.NAN

            self.vx=np.flipud(vx_i)
            self.vy=np.flipud(vy_i)
            self.vz=np.flipud(vz_i)
            self.extent=extent
            self.minX=MinX
            self.maxX=MaxX
            self.minY=MinY
            self.maxY=MaxY
            self.dx=dx
            self.dy=dy
            self.createDataDict()

            self.data[varName]=np.flipud(scalar_i)
        else:
            print 'dict exists'
            MaxX=self.extent[1]
            MinX=self.extent[0]
            MaxY=self.extent[3]
            MinY=self.extent[2]

            cellsX=int((MaxX-MinX)/self.dx)
            cellsY=int((MaxY-MinY)/self.dy)
            #print cellsX,cellsY
            grid_y, grid_x = np.mgrid[MinY:MaxY:np.complex(0,cellsY),MinX:MaxX:np.complex(0,cellsX)]
            triang = tss.triangulation
            scalar_i=self.interpolateField(tss.s,grid_x, grid_y, triang, method=interpolationMethod, kind=kind)
            print 'adding scalar',varName
            self.data[varName]=np.flipud(scalar_i)
    def addFieldFromHdf5(self,
                         hdf5Parser,
                         key,
                         names=[],
                         projectedField=False):
        '''
        Add a list field stored in a hdf5 to the TriSurfaceContainer.
        
        Arguments:
            *hdf5Parser*: h5py parser object.
             Parser object of the source hdf5 file.
             
            *key*: python string.
             The time (as a key) to extract from the HDF5. If it does not exist, IOError
             is returned.
             
            *names*: python list of string.
             Name of the fields to extract from the HDF5. It can be a single
             field (names=['oneField']) or multiple. if names=[], all the
             fields are loaded. Default: names=[].
             
            *projectedField*: python bool
             Project the fields in the surface coordinate system.
             Default: projectedField=False.
        
        Usages:
            >>> import h5py
            >>> parser = h5py.File('myfile.h5','r')
            >>> tsc = TriSurfaceContainer.createFromHdf5(parser,xViewBasis=[1,0,0])
            >>> tsc.addFieldFromHdf5(parser,time=1.5,names=['U','S'])
        '''
        gTime = key
        time = float(key)
        if len(names) == 0:
            names = hdf5Parser[gTime].keys()
            names.pop(names.index('time'))

        for name in names:
            try:

                dataShape = hdf5Parser[gTime][name].value.shape
                if len(dataShape) == 1:  #data is a scalar
                    tss = TriSurfaceScalar.readFromHdf5(
                        hdf5Parser=hdf5Parser,
                        varName=name,
                        triSurfaceMesh=self.triSurfaceMesh,
                        key=key,
                        projectedField=projectedField)
                    self.fields[name] = tss
                elif len(dataShape
                         ) == 2 and dataShape[1] == 3:  #data is a vector
                    tsv = TriSurfaceVector.readFromHdf5(
                        hdf5Parser=hdf5Parser,
                        varName=name,
                        triSurfaceMesh=self.triSurfaceMesh,
                        key=key,
                        projectedField=projectedField)
                    self.fields[name] = tsv
                elif len(dataShape
                         ) == 2 and dataShape[1] == 6:  #data is a symmtensor
                    tsst = TriSurfaceSymmTensor.readFromHdf5(
                        hdf5Parser=hdf5Parser,
                        varName=name,
                        triSurfaceMesh=self.triSurfaceMesh,
                        key=key,
                        projectedField=projectedField)
                    self.fields[name] = tsst
                else:
                    raise IOError('variable of name "' + name + '" is not a'
                                  'scalar, not a vector, not a symmTensor.')

            except KeyError as e:
                'time "' + gTime + '" and/or name "' + name + '" does not exist as key in the HDF5 parser. Not read, but continuing'
                print e
            except KeyboardInterrupt as e:
                print e
                return False
            except MemoryError as e:
                print e
                return False
            except:
                print "Unexpected error:", sys.exc_info()[0]
                #raise IOError('time "'+gTime+'" and/or name "'+name+'" does not '
                #'exist as key in the HDF5 parser.')
                return False
        return True
    def addFieldFromHdf5(self, hdf5Parser, key, names=[], projectedField=False):
        """
        Add a list field stored in a hdf5 to the TriSurfaceContainer.
        
        Arguments:
            *hdf5Parser*: h5py parser object.
             Parser object of the source hdf5 file.
             
            *key*: python string.
             The time (as a key) to extract from the HDF5. If it does not exist, IOError
             is returned.
             
            *names*: python list of string.
             Name of the fields to extract from the HDF5. It can be a single
             field (names=['oneField']) or multiple. if names=[], all the
             fields are loaded. Default: names=[].
             
            *projectedField*: python bool
             Project the fields in the surface coordinate system.
             Default: projectedField=False.
        
        Usages:
            >>> import h5py
            >>> parser = h5py.File('myfile.h5','r')
            >>> tsc = TriSurfaceContainer.createFromHdf5(parser,xViewBasis=[1,0,0])
            >>> tsc.addFieldFromHdf5(parser,time=1.5,names=['U','S'])
        """
        gTime = key
        time = float(key)
        if len(names) == 0:
            names = hdf5Parser[gTime].keys()
            names.pop(names.index("time"))

        for name in names:
            try:

                dataShape = hdf5Parser[gTime][name].value.shape
                if len(dataShape) == 1:  # data is a scalar
                    tss = TriSurfaceScalar.readFromHdf5(
                        hdf5Parser=hdf5Parser,
                        varName=name,
                        triSurfaceMesh=self.triSurfaceMesh,
                        key=key,
                        projectedField=projectedField,
                    )
                    self.fields[name] = tss
                elif len(dataShape) == 2 and dataShape[1] == 3:  # data is a vector
                    tsv = TriSurfaceVector.readFromHdf5(
                        hdf5Parser=hdf5Parser,
                        varName=name,
                        triSurfaceMesh=self.triSurfaceMesh,
                        key=key,
                        projectedField=projectedField,
                    )
                    self.fields[name] = tsv
                elif len(dataShape) == 2 and dataShape[1] == 6:  # data is a symmtensor
                    tsst = TriSurfaceSymmTensor.readFromHdf5(
                        hdf5Parser=hdf5Parser,
                        varName=name,
                        triSurfaceMesh=self.triSurfaceMesh,
                        key=key,
                        projectedField=projectedField,
                    )
                    self.fields[name] = tsst
                else:
                    raise IOError('variable of name "' + name + '" is not a' "scalar, not a vector, not a symmTensor.")

            except KeyError as e:
                'time "' + gTime + '" and/or name "' + name + '" does not exist as key in the HDF5 parser. Not read, but continuing'
                print e
            except KeyboardInterrupt as e:
                print e
                return False
            except MemoryError as e:
                print e
                return False
            except:
                print "Unexpected error:", sys.exc_info()[0]
                # raise IOError('time "'+gTime+'" and/or name "'+name+'" does not '
                #'exist as key in the HDF5 parser.')
                return False
        return True