Exemplo n.º 1
0
 def addFoamSymmTensorField(self, fieldpath, name, time, projectedField=False):
     """
     """
     tSurf = TriSurfaceSymmTensor.readFromFoamFile(
         fieldpath, self.triSurfaceMesh, time=time, projectedField=projectedField
     )
     self.fields[name] = tSurf
Exemplo n.º 2
0
 def addFoamSymmTensorField(self,
                            fieldpath,
                            name,
                            time,
                            projectedField=False):
     '''
     '''
     tSurf = TriSurfaceSymmTensor.readFromFoamFile(
         fieldpath,
         self.triSurfaceMesh,
         time=time,
         projectedField=projectedField)
     self.fields[name] = tSurf
Exemplo n.º 3
0
    def readReStressFromFoamFile(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'):
        '''
        '''
        tsm = TriSurfaceMesh.readFromFoamFile(pointsFile=pointsFile,
                                              facesFile=facesFile,
                                              viewAnchor=viewAnchor,
                                              xViewBasis=xViewBasis,
                                              yViewBasis=yViewBasis)
                                              
        tsst = TriSurfaceSymmTensor.readFromFoamFile(varsFile=varsFile,
                                                     triSurfaceMesh=tsm,
                                                     time=0,
                                                     projectedField=False)                  

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

        #if not hasattr(self,'data'):
        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 = tsst.triangulation            
            uu_bar=self.interpolateField(tsst.txx, grid_x, grid_y, triang, method=interpolationMethod, kind=kind)
            uv_bar=self.interpolateField(tsst.txy, grid_x, grid_y, triang, method=interpolationMethod, kind=kind)
            uw_bar=self.interpolateField(tsst.tyy, grid_x, grid_y, triang, method=interpolationMethod, kind=kind)
            vv_bar=self.interpolateField(tsst.tyy, grid_x, grid_y, triang, method=interpolationMethod, kind=kind)
            vw_bar=self.interpolateField(tsst.tyz, grid_x, grid_y, triang, method=interpolationMethod, kind=kind)
            ww_bar=self.interpolateField(tsst.tzz, grid_x, grid_y, triang, method=interpolationMethod, kind=kind)
            vx_i=np.empty(uu_bar.shape)
            vy_i=np.empty(uu_bar.shape)
            vz_i=np.empty(uu_bar.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.dx=dx
            self.dy=dy
            self.minX=MinX
            self.maxX=MaxX
            self.minY=MinY
            self.maxY=MaxY
            self.createDataDict()

            print 'adding Tensor'
            self.data['uu_bar']=np.flipud(uu_bar)
            self.data['uv_bar']=np.flipud(uv_bar)
            self.data['uw_bar']=np.flipud(uw_bar)
            self.data['vv_bar']=np.flipud(vv_bar)
            self.data['vw_bar']=np.flipud(vw_bar)
            self.data['ww_bar']=np.flipud(ww_bar)
            self.data['TKE_bar']=0.5*(self.data['uu_bar']+self.data['vv_bar']+self.data['ww_bar'])

        else:
            print 'dict exists'
            MaxX=self.maxX
            MinX=self.minX
            MaxY=self.maxY
            MinY=self.minY

            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 = tsst.triangulation
            uu_bar=self.interpolateField(tsst.txx, grid_x, grid_y, triang, method=interpolationMethod, kind=kind)
            uv_bar=self.interpolateField(tsst.txy, grid_x, grid_y, triang, method=interpolationMethod, kind=kind)
            uw_bar=self.interpolateField(tsst.tyy, grid_x, grid_y, triang, method=interpolationMethod, kind=kind)
            vv_bar=self.interpolateField(tsst.tyy, grid_x, grid_y, triang, method=interpolationMethod, kind=kind)
            vw_bar=self.interpolateField(tsst.tyz, grid_x, grid_y, triang, method=interpolationMethod, kind=kind)
            ww_bar=self.interpolateField(tsst.tzz, grid_x, grid_y, triang, method=interpolationMethod, kind=kind)
            print 'adding Tensor'
            self.data['uu_bar']=np.flipud(uu_bar)
            self.data['uv_bar']=np.flipud(uv_bar)
            self.data['uw_bar']=np.flipud(uw_bar)
            self.data['vv_bar']=np.flipud(vv_bar)
            self.data['vw_bar']=np.flipud(vw_bar)
            self.data['ww_bar']=np.flipud(ww_bar)
            self.data['TKE_bar']=0.5*(self.data['uu_bar']+self.data['vv_bar']+self.data['ww_bar'])
Exemplo n.º 4
0
    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
Exemplo n.º 5
0
    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