mirror of
https://github.com/OpenFOAM/ThirdParty-6.git
synced 2025-12-08 06:57:43 +00:00
84 lines
3.0 KiB
Python
84 lines
3.0 KiB
Python
|
|
try: paraview.simple
|
|
except: 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:
|
|
filename_4_vthb = coprocessor.CreateProducer( datadescription, "input" )
|
|
|
|
Slice1 = Slice( guiName="Slice1", Crinkleslice=0, SliceOffsetValues=[0.0], Triangulatetheslice=1, SliceType="Plane" )
|
|
Slice1.SliceType.Offset = 0.0
|
|
Slice1.SliceType.Origin = [20.0, 40.0, 40.0]
|
|
Slice1.SliceType.Normal = [1.0, 0.0, 0.0]
|
|
|
|
ParallelMultiBlockDataSetWriter1 = coprocessor.CreateWriter( XMLMultiBlockDataWriter, "slice_%t.vtm", 10 )
|
|
|
|
return Pipeline()
|
|
|
|
class CoProcessor(coprocessing.CoProcessor):
|
|
def CreatePipeline(self, datadescription):
|
|
self.Pipeline = _CreatePipeline(self, datadescription)
|
|
|
|
coprocessor = CoProcessor()
|
|
freqs = {'input': [10]}
|
|
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)
|