[[Image(ex46.png)]] {{{#!python # -*- coding: utf-8 -*- """ Created on Sun Mar 23 11:17:34 2014 @author: fuji """ import vtk import numpy as np # Strctured Grid sGrid = vtk.vtkStructuredGrid() with open('officeFlow.txt', 'r') as CFDfile: nx, ny, nz = CFDfile.readline().split() dims = [int(nx),int(ny),int(nz)] sGrid.SetDimensions(dims[0],dims[1],dims[2]) nodes = vtk.vtkPoints() nodes.SetNumberOfPoints(dims[0] * dims[1] * dims[2]) scalArray = vtk.vtkDoubleArray() scalArray.SetNumberOfComponents(1) scalArray.SetName("Pressure") scalArray.SetNumberOfTuples(dims[0]*dims[1]*dims[2]) velArray = vtk.vtkDoubleArray() velArray.SetNumberOfComponents(3) velArray.SetName("Velocity") velArray.SetNumberOfTuples(dims[0]*dims[1]*dims[2]) for k in range(dims[2]): kOffset = k * dims[0] * dims[1] for j in range(dims[1]): jOffset = j * dims[0] for i in range(dims[0]): offset = i + jOffset + kOffset x,y,z,u,v,w,p = CFDfile.readline().split() nodes.SetPoint(offset,[float(x), float(y), float(z)]) vel = np.array([float(u), float(v), float(w)]) scalArray.SetValue(offset,float(p)) velArray.SetTupleValue(offset,vel) sGrid.SetPoints(nodes) sGrid.GetPointData().SetVectors(velArray) sGrid.GetPointData().SetScalars(scalArray) # # writer = vtk.vtkXMLStructuredGridWriter() writer.SetInput(sGrid) writer.SetFileName("officeFlow.vts") writer.SetDataModeToAscii() writer.Write() }}} [http://wiki.ccs.tulane.edu/index.php5/Basics_of_VisIt]