CFD Online Discussion Forums - [OpenFOAM] Python script for Integrate Variable

Hi Mehdi,

I'm not sure it is the fastest way to deal with the csv files, but I did manage. Here is the interesting part of the script

Code:

try: paraview.simple

except: from paraview.simple import *

paraview.simple._DisableFirstRenderCameraReset()



import csv

surf = LegacyVTKReader( FileNames=['pathToImportFile.vtk'] )



RenderView1 = GetRenderView()

                                      

Clip1 = Clip( ClipType="Plane" )

Clip1.Scalars = ['POINTS', 'p']

Clip1.ClipType.Origin = [0.0, 0.0, 0.0] #can be a random origin point

Clip1.ClipType.Normal = [1.0, 0.0, 0.0]   #can be a random normal orientation

Clip1.ClipType = "Plane"              



Clip2 = Clip( ClipType="Plane" )

Clip2.Scalars = ['POINTS', 'p']  

Clip2.ClipType.Origin = [1.0, 0.0, 0.0]  #can be a random origin point

Clip2.ClipType.Normal = [-1.0, 0.0, 0.0]  #can be a random normal orientation

Clip2.ClipType = "Plane"

                                      

# we define the variable to be integrated on Clip2                  

Calculator1 = Calculator()       

Calculator1.AttributeMode = 'point_data'

Calculator1.Function = 'writeHereTheFunctionOfYourVariable'

Calculator1.ReplaceInvalidResults = 0

Calculator1.ResultArrayName = 'NameOfYourVariable'



IntegrateVariables1 = IntegrateVariables()   #this integrates the current variable over Clip2



finalFile = open('pathToCSVFile.csv','wb')

fd = csv.writer(finalFile)  

                                     

writer = CreateWriter('PathToATemporaryFile.csv', IntegrateVariables1) 

writer.FieldAssociation = "Points"  # or "Cells"



for x in range(-6, 6): #loop through the origin point of Clip1

    print "We integrate from %1.1f to %1.1f" % (x,x+1)

        

    Clip1.ClipType.Origin = [x,0.0,0.0] #moves the first clip

    Clip1.ClipType.Normal = [1.0,0.0,0.0]

      

    Clip2.ClipType.Origin = [x+1,0.0,0.0] # moves the second clip

    Clip2.ClipType.Normal = [-1.0,0.0,0.0]

                      

    writer.UpdatePipeline() # writes the integrated variable table to the tmp file

    

    # appends the integrated results of this Clip to the final csv file

    ftmp = open('PathToATemporaryFile.csv','rb')

    reader = csv.reader(ftmp, delimiter=',')

    row = reader.next()                                        

    row = reader.next()

    fd.writerow([str(row)]) 

    ftmp.close()           

    

finalFile.close()             

del(writer)                



Render()

If you know a way to make it faster, please share it :) I hope this will help you!

Eloïse