mirror of
https://github.com/OpenFOAM/ThirdParty-6.git
synced 2025-12-08 06:57:43 +00:00
92 lines
3.7 KiB
Python
92 lines
3.7 KiB
Python
from paraview.simple import *
|
|
|
|
from paraview import coprocessing
|
|
|
|
|
|
#--------------------------------------------------------------
|
|
# Code generated from cpstate.py to create the CoProcessor.
|
|
|
|
|
|
# ----------------------- CoProcessor definition -----------------------
|
|
|
|
def CreateCoProcessor():
|
|
def _CreatePipeline(coprocessor, datadescription):
|
|
class Pipeline:
|
|
adaptorinput = coprocessor.CreateProducer( datadescription, "input" )
|
|
grid = adaptorinput.GetClientSideObject().GetOutputDataObject(0)
|
|
if grid.IsA('vtkImageData') or grid.IsA('vtkUniformGrid'):
|
|
writer = coprocessor.CreateWriter( XMLPImageDataWriter, "filename_%t.pvti", 1 )
|
|
elif grid.IsA('vtkRectilinearGrid'):
|
|
writer = coprocessor.CreateWriter( XMLPRectilinearGridWriter, "filename_%t.pvtr", 1 )
|
|
elif grid.IsA('vtkStructuredGrid'):
|
|
writer = coprocessor.CreateWriter( XMLPStructuredGridWriter, "filename_%t.pvts", 1 )
|
|
elif grid.IsA('vtkPolyData'):
|
|
writer = coprocessor.CreateWriter( XMLPPolyDataWriter, "filename_%t.pvtp", 1 )
|
|
elif grid.IsA('vtkUnstructuredGrid'):
|
|
writer = coprocessor.CreateWriter( XMLPUnstructuredGridWriter, "filename_%t.pvtu", 1 )
|
|
elif grid.IsA('vtkUniformGridAMR'):
|
|
writer = coprocessor.CreateWriter( XMLHierarchicalBoxDataWriter, "filename_%t.vthb", 1 )
|
|
elif grid.IsA('vtkMultiBlockDataSet'):
|
|
writer = coprocessor.CreateWriter( XMLMultiBlockDataWriter, "filename_%t.vtm", 1 )
|
|
else:
|
|
print "Don't know how to create a writer for a ", grid.GetClassName()
|
|
|
|
return Pipeline()
|
|
|
|
class CoProcessor(coprocessing.CoProcessor):
|
|
def CreatePipeline(self, datadescription):
|
|
self.Pipeline = _CreatePipeline(self, datadescription)
|
|
|
|
coprocessor = CoProcessor()
|
|
freqs = {'input': [1]}
|
|
coprocessor.SetUpdateFrequencies(freqs)
|
|
return coprocessor
|
|
|
|
#--------------------------------------------------------------
|
|
# Global variables that will hold the pipeline for each timestep
|
|
# Creating the CoProcessor object, doesn't actually create the ParaView pipeline.
|
|
# It will be automatically setup when coprocessor.UpdateProducers() is called the
|
|
# first time.
|
|
coprocessor = CreateCoProcessor()
|
|
|
|
#--------------------------------------------------------------
|
|
# Enable Live-Visualizaton with ParaView
|
|
coprocessor.EnableLiveVisualization(False)
|
|
|
|
|
|
# ---------------------- Data Selection method ----------------------
|
|
|
|
def RequestDataDescription(datadescription):
|
|
"Callback to populate the request for current timestep"
|
|
global coprocessor
|
|
if datadescription.GetForceOutput() == True:
|
|
# We are just going to request all fields and meshes from the simulation
|
|
# code/adaptor.
|
|
for i in range(datadescription.GetNumberOfInputDescriptions()):
|
|
datadescription.GetInputDescription(i).AllFieldsOn()
|
|
datadescription.GetInputDescription(i).GenerateMeshOn()
|
|
return
|
|
|
|
# setup requests for all inputs based on the requirements of the
|
|
# pipeline.
|
|
coprocessor.LoadRequestedData(datadescription)
|
|
|
|
# ------------------------ Processing method ------------------------
|
|
|
|
def DoCoProcessing(datadescription):
|
|
"Callback to do co-processing for current timestep"
|
|
global coprocessor
|
|
|
|
# Update the coprocessor by providing it the newly generated simulation data.
|
|
# If the pipeline hasn't been setup yet, this will setup the pipeline.
|
|
coprocessor.UpdateProducers(datadescription)
|
|
|
|
# Write output data, if appropriate.
|
|
coprocessor.WriteData(datadescription);
|
|
|
|
# Write image capture (Last arg: rescale lookup table), if appropriate.
|
|
coprocessor.WriteImages(datadescription, rescale_lookuptable=False)
|
|
|
|
# Live Visualization, if enabled.
|
|
coprocessor.DoLiveVisualization(datadescription, "localhost", 22222)
|