File: catalyst_pipeline.py

package info (click to toggle)
paraview 5.13.2%2Bdfsg-3
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 544,220 kB
  • sloc: cpp: 3,374,605; ansic: 1,332,409; python: 150,381; xml: 122,166; sql: 65,887; sh: 7,317; javascript: 5,262; yacc: 4,417; java: 3,977; perl: 2,363; lex: 1,929; f90: 1,397; makefile: 170; objc: 153; tcl: 59; pascal: 50; fortran: 29
file content (55 lines) | stat: -rw-r--r-- 2,111 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
from paraview.simple import *

import paraview.catalyst as catalyst

from mpi4py import MPI  # we only use it to make the output neater

comm = MPI.COMM_WORLD
rank = comm.Get_rank()
size = comm.Get_size()

# print values for parameters passed via adaptor (note these don't change,
# and hence must be created as command line params)
print("executing catalyst_pipeline")
print("===================================")
print("script args={}".format(catalyst.get_args()))
print("===================================")

# registrationName must match the channel name used in the
# 'CatalystAdaptor'.
producer = TrivialProducer(registrationName="${args.channel-name}")

extractor = CreateExtractor("VTPD", producer, registrationName="VTPD")
extractor.Writer.FileName = "data_{timestep:06d}.vtpd"

# this is only for testing purposes. never directly import `catalyst.detail`
from paraview.catalyst.detail import IsInsituInput, _transform_registration_name

assert IsInsituInput("${args.channel-name}")
assert _transform_registration_name("${args.channel-name}") == "grid"

# ------------------------------------------------------------------------------
# Catalyst options
options = catalyst.Options()
# the pipeline with be executed evey two timesteps
options.GlobalTrigger.Frequency = 2
# Enable live visualization. To use it in ParaView click Catalyst > Connect prior to running
# this script.
options.EnableCatalystLive = 1


def catalyst_execute(info):
    global producer

    producer.UpdatePipeline()

    output = f"------------------- rank {rank}/{size} ----------------------\n"
    output += "executing (cycle={}, time={})\n".format(info.cycle, info.time)
    output += "bounds: {}\n".format(producer.GetDataInformation().GetBounds())
    output += "psi01-u-range: {}\n".format(producer.PointData["psi01"].GetRange(0))
    output += "psi01-v-range: {}\n".format(producer.PointData["psi01"].GetRange(1))
    output += "psi01-magnitute-range: {} \n".format(
        producer.PointData["psi01"].GetRange(-1)
    )
    output += "------------------------------------------------------------\n"
    print(output)