CFD Online Discussion Forums - [OpenFOAM] Get cell center coordinates in a python script

Hi all,

I've made a python script that create a slice and extract the points coordinates of the vertex as well as the velocity components.

But I'm unable to obtain the cell center coordinates, howto do that ?

You will see below my script which can extract the points coordinates.

Code:

import numpy as np



try: paraview.simple

except: from paraview.simple import *

paraview.simple._DisableFirstRenderCameraReset()



from vtk.util import numpy_support as npvtk 



# Define the OpenFOAM data source

case_OpenFOAM = OpenDataFile('./profile4.OpenFOAM')

case_OpenFOAM.MeshParts = ['internalMesh']

case_OpenFOAM.VolumeFields = ['p', 'U']

print case_OpenFOAM.TimestepValues



mergedcase_OpenFOAM = MergeBlocks(case_OpenFOAM)

dataOpenFOAM = servermanager.Fetch(mergedcase_OpenFOAM)



# Select the active source

SetActiveSource(mergedcase_OpenFOAM)



# Define the plane

Slice1 = Slice(SliceType="Plane")

Slice1.SliceType = 'Plane'

Slice1.SliceType.Origin = [0.01,0,0]

Slice1.SliceType.Normal = [1,0,0]

Slice1.Triangulatetheslice = 0 # Don't triangulate the slice



dataPlane = servermanager.Fetch(Slice1)

nbCells = dataPlane.GetNumberOfCells()

nbPoints = dataPlane.GetNumberOfPoints()



print 'Number of cells in the slice', nbCells

print 'Number of points in the slice', nbPoints 



# Put the velocity vector U into a numpy array

U = npvtk.vtk_to_numpy(dataPlane.GetCellData().GetArray('U'))



# Put the points coordinates in x, y and z arrays

x=[]

y=[]

z=[]



for i in range(nbPoints):

    coord = dataPlane.GetPoint(i)

    xx, yy, zz = coord[:3]

    x.append(xx)

    y.append(yy)

    z.append(zz)



x=np.array(x)

y=np.array(y)

z=np.array(z)



print 'x=',x

print 'y=',y

print 'z=',z

Any help will be warmly welcome. :D
Thanks !

François