d9568a4b56
- SIGFPE handling, cmake configuration CONFIG: adjust for simplified Catalyst channel naming - no sub-channels, select volField output based in the (internal | boundary) dictionary switches instead
81 lines
3.1 KiB
Python
81 lines
3.1 KiB
Python
from paraview.simple import *
|
|
from paraview import coprocessing
|
|
|
|
# The frequency to output everything
|
|
outputfrequency = 5
|
|
|
|
# Padding for files
|
|
fileNamePadding = 0
|
|
|
|
# This is largely identical to the Catalyst allinputsgridwriter.py example
|
|
# but only handle vtkMultiBlockDataSet, since that is what we generate
|
|
|
|
# ----------------------- CoProcessor definition -----------------------
|
|
|
|
def CreateCoProcessor():
|
|
def _CreatePipeline(coprocessor, datadescription):
|
|
class Pipeline:
|
|
for i in range(datadescription.GetNumberOfInputDescriptions()):
|
|
name = datadescription.GetInputDescriptionName(i)
|
|
input = coprocessor.CreateProducer(datadescription, name)
|
|
grid = input.GetClientSideObject().GetOutputDataObject(0)
|
|
if grid.IsA('vtkMultiBlockDataSet'):
|
|
writer = servermanager.writers.XMLMultiBlockDataWriter(Input=input)
|
|
coprocessor.RegisterWriter(writer, filename=name+'_%t.vtm', freq=outputfrequency, paddingamount=fileNamePadding)
|
|
|
|
return Pipeline()
|
|
|
|
class CoProcessor(coprocessing.CoProcessor):
|
|
def CreatePipeline(self, datadescription):
|
|
self.Pipeline = _CreatePipeline(self, datadescription)
|
|
|
|
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 or datadescription.GetTimeStep() % outputfrequency == 0:
|
|
# 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)
|