def __init__(
self,
filename="abl.nek5000",
arrays=(
"pressure",
"velocity_mag",
"x_velocity",
"y_velocity",
"z_velocity",
),
):
self.filename = filename
assert Path(filename).exists()
self.nek5000 = nek5000 = pv.VisItNek5000Reader(FileName=str(filename))
nek5000.Meshes = ["mesh"]
# get animation scene
self.animationScene1 = pv.GetAnimationScene()
# get the time-keeper
self.timeKeeper1 = pv.GetTimeKeeper()
# update animation scene based on data timesteps
self.animationScene1.UpdateAnimationUsingDataTimeSteps()
# Properties modified on ablnek5000
assert isinstance(arrays, (list, tuple))
nek5000.PointArrays = list(arrays)
data_info = nek5000.GetDataInformation()
self.dtype = data_info.GetDataSetTypeAsString()
def goToPrev(self):
oldTime = self.View.ViewTime
simple.GetAnimationScene().GoToPrevious()
if oldTime != self.View.ViewTime:
self.factory.dispatch("http://vtk.org/event#probeDataChanged", True)
return True
return False
def setupApplication(cls):
"""Setups the default application state."""
# read data directory.
root = {"name": "ROOT", "dirs": [], "files": []}
directory_map = {}
directory_map[_DataProber.DataPath] = root
for path, dirs, files in os.walk(_DataProber.DataPath):
element = directory_map[path]
for name in dirs:
item = {"name": name, "dirs": [], "files": []}
item["name"] = name
directory_map[os.path.join(path, name)] = item
element["dirs"].append(item)
element["files"] = []
for name in files:
relpath = os.path.relpath(os.path.join(path, name),
_DataProber.DataPath)
item = {"name": name, "itemValue": relpath}
element["files"].append(item)
cls.Database = root
cls.View = simple.CreateRenderView()
simple.Render()
# setup animation scene
scene = simple.GetAnimationScene()
simple.GetTimeTrack()
scene.PlayMode = "Snap To TimeSteps"
def goToPrev(self):
oldTime = self.View.ViewTime
simple.GetAnimationScene().GoToPrevious()
if oldTime != self.View.ViewTime:
self.publish("vtk.event.probe.data.changed", True)
print 'publish b'
return True
return False
def saveView(self, reader):
"""Save figure
"""
# Get animation scene
animationScene = pv.GetAnimationScene()
animationScene.PlayMode = "Snap To TimeSteps"
# Save animation images
print("[PNGViewer] ### Generating PNG images...")
for t in reader.TimestepValues.GetData()[:]:
animationScene.AnimationTime = t
pv.WriteImage(self.file_name + ".%f.png" % t)
print("[PNGViewer] ### All PNG images generated.")
def create_scene(self,
start,
end,
duration,
mode='Real Time',
loop=True):
"Inicializa a visualização da animação."
scene = pv.GetAnimationScene()
scene.EndTime = end
scene.Duration = duration
scene.StartTime = start
scene.PlayMode = mode
scene.Loop = int(loop)
self.objects_['scene'] = scene
def read(self):
path = self.path_run
if path.is_dir():
path = next(path.glob("*.nek5000"))
# TODO: needed or not?
# pvs._DisableFirstRenderCameraReset()
self.reader = reader = VisItNek5000Reader(FileName=str(path))
if not reader:
raise IOError("Error while loading file " + path)
reader.Meshes = ["mesh"]
# get animation scene
self.animationScene = pvs.GetAnimationScene()
# get the time-keeper
self.timeKeeper = pvs.GetTimeKeeper()
def updateTime(self, action):
view = simple.GetRenderView()
animationScene = simple.GetAnimationScene()
currentTime = view.ViewTime
if action == "next":
animationScene.GoToNext()
if currentTime == view.ViewTime:
animationScene.GoToFirst()
if action == "prev":
animationScene.GoToPrevious()
if currentTime == view.ViewTime:
animationScene.GoToLast()
if action == "first":
animationScene.GoToFirst()
if action == "last":
animationScene.GoToLast()
return view.ViewTime
def saveView(self, reader):
"""Save animation
"""
# Get animation scene
animationScene = pv.GetAnimationScene()
animationScene.PlayMode = "Snap To TimeSteps"
# Save animation images
for t in reader.TimestepValues.GetData()[:]:
animationScene.AnimationTime = t
# Save animation movie
print("[AnimationViewer] ### Generating AVI animation...")
pv.WriteAnimation(self.file_name + ".avi",
Magnification=1,
Quality=2,
FrameRate=1.0,
Compression=True)
print("[AnimationViewer] ### AVI animation generated.")
def updateActiveArgument(self, key, value):
if key == "time":
simple.GetAnimationScene().TimeKeeper.Time = float(value)
elif _PVCatalystManager.pipeline_handler:
_PVCatalystManager.pipeline_handler.update_argument(key, value)
def __init__(self):
super(ParaViewWebTimeHandler, self).__init__()
# setup animation scene
self.scene = simple.GetAnimationScene()
simple.GetTimeTrack()
self.scene.PlayMode = "Snap To TimeSteps"
from paraview import simple
from paraview.web.dataset_builder import *
# -----------------------------------------------------------------------------
# Pipeline creation
# -----------------------------------------------------------------------------
can = simple.OpenDataFile(inputFile)
can.ElementVariables = ['EQPS']
can.PointVariables = ['DISPL', 'VEL', 'ACCL']
can.GlobalVariables = ['KE', 'XMOM', 'YMOM', 'ZMOM', 'NSTEPS', 'TMSTEP']
can.ElementBlocks = [
'Unnamed block ID: 1 Type: HEX', 'Unnamed block ID: 2 Type: HEX'
]
anim = simple.GetAnimationScene()
anim.UpdateAnimationUsingDataTimeSteps()
timeValues = anim.TimeKeeper.TimestepValues
sceneDescription = {
'scene': [{
'name': 'Can',
'source': can,
'colors': {
'DISPL': {
'location': 'POINT_DATA'
},
'VEL': {
'location': 'POINT_DATA'
},
def render_frames(
scene,
frames_dir=None,
frame_window=None,
render_missing_frames=False,
save_state_to_file=None,
no_render=False,
show_preview=False,
show_progress=False,
job_id=None,
):
# Validate scene
if scene["View"]["ViewSize"][0] % 16 != 0:
logger.warning(
"The view width should be a multiple of 16 to be compatible with"
" QuickTime.")
if scene["View"]["ViewSize"][1] % 2 != 0:
logger.warning(
"The view height should be even to be compatible with QuickTime.")
render_start_time = time.time()
# Setup layout
layout = pv.CreateLayout("Layout")
# Setup view
if "Background" in scene["View"]:
bg_config = scene["View"]["Background"]
del scene["View"]["Background"]
if isinstance(bg_config, list):
if isinstance(bg_config[0], list):
assert len(bg_config) == 2, (
"When 'Background' is a list of colors, it must have 2"
" entries.")
bg_config = dict(
BackgroundColorMode="Gradient",
Background=parse_as.color(bg_config[0]),
Background2=parse_as.color(bg_config[1]),
)
else:
bg_config = dict(
BackgroundColorMode="Single Color",
Background=parse_as.color(bg_config),
)
bg_config["UseColorPaletteForBackground"] = 0
scene["View"].update(bg_config)
bg_config = None
else:
bg_config = None
view = pv.CreateRenderView(**scene["View"])
pv.AssignViewToLayout(view=view, layout=layout, hint=0)
# Set spherical background texture
if bg_config is not None:
bg_config["BackgroundColorMode"] = "Texture"
skybox_datasource = bg_config["Datasource"]
del bg_config["Datasource"]
background_texture = pvserver.rendering.ImageTexture(
FileName=parse_as.path(scene["Datasources"][skybox_datasource]))
background_sphere = pv.Sphere(Radius=bg_config["Radius"],
ThetaResolution=100,
PhiResolution=100)
background_texture_map = pv.TextureMaptoSphere(Input=background_sphere)
pv.Show(
background_texture_map,
view,
Texture=background_texture,
BackfaceRepresentation="Cull Frontface",
Ambient=1.0,
)
# Load the waveform data file
waveform_h5file, waveform_subfile = parse_as.file_and_subfile(
scene["Datasources"]["Waveform"])
waveform_data = WaveformDataReader(FileName=waveform_h5file,
Subfile=waveform_subfile)
pv.UpdatePipeline()
# Generate volume data from the waveform. Also sets the available time range.
# TODO: Pull KeepEveryNthTimestep out of datasource
waveform_to_volume_configs = scene["WaveformToVolume"]
if isinstance(waveform_to_volume_configs, dict):
waveform_to_volume_configs = [{
"Object": waveform_to_volume_configs,
}]
if "VolumeRepresentation" in scene:
waveform_to_volume_configs[0]["VolumeRepresentation"] = scene[
"VolumeRepresentation"]
waveform_to_volume_objects = []
for waveform_to_volume_config in waveform_to_volume_configs:
volume_data = WaveformToVolume(
WaveformData=waveform_data,
SwshCacheDirectory=parse_as.path(
scene["Datasources"]["SwshCache"]),
**waveform_to_volume_config["Object"],
)
if "Modes" in waveform_to_volume_config["Object"]:
volume_data.Modes = waveform_to_volume_config["Object"]["Modes"]
if "Polarizations" in waveform_to_volume_config["Object"]:
volume_data.Polarizations = waveform_to_volume_config["Object"][
"Polarizations"]
waveform_to_volume_objects.append(volume_data)
# Compute timing and frames information
time_range_in_M = (
volume_data.TimestepValues[0],
volume_data.TimestepValues[-1],
)
logger.debug(f"Full available data time range: {time_range_in_M} (in M)")
if "FreezeTime" in scene["Animation"]:
frozen_time = scene["Animation"]["FreezeTime"]
logger.info(f"Freezing time at {frozen_time}.")
view.ViewTime = frozen_time
animation = None
else:
if "Crop" in scene["Animation"]:
time_range_in_M = scene["Animation"]["Crop"]
logger.debug(f"Cropping time range to {time_range_in_M} (in M).")
animation_speed = scene["Animation"]["Speed"]
frame_rate = scene["Animation"]["FrameRate"]
num_frames = animate.num_frames(
max_animation_length=time_range_in_M[1] - time_range_in_M[0],
animation_speed=animation_speed,
frame_rate=frame_rate,
)
animation_length_in_seconds = num_frames / frame_rate
animation_length_in_M = animation_length_in_seconds * animation_speed
time_per_frame_in_M = animation_length_in_M / num_frames
logger.info(f"Rendering {animation_length_in_seconds:.2f}s movie with"
f" {num_frames} frames ({frame_rate} FPS or"
f" {animation_speed:.2e} M/s or"
f" {time_per_frame_in_M:.2e} M/frame)...")
if frame_window is not None:
animation_window_num_frames = frame_window[1] - frame_window[0]
animation_window_time_range = (
time_range_in_M[0] + frame_window[0] * time_per_frame_in_M,
time_range_in_M[0] +
(frame_window[1] - 1) * time_per_frame_in_M,
)
logger.info(
f"Restricting rendering to {animation_window_num_frames} frames"
f" (numbers {frame_window[0]} to {frame_window[1] - 1}).")
else:
animation_window_num_frames = num_frames
animation_window_time_range = time_range_in_M
frame_window = (0, num_frames)
# Setup animation so that sources can retrieve the `UPDATE_TIME_STEP`
animation = pv.GetAnimationScene()
# animation.UpdateAnimationUsingDataTimeSteps()
# Since the data can be evaluated at arbitrary times we define the time steps
# here by setting the number of frames within the full range
animation.PlayMode = "Sequence"
animation.StartTime = animation_window_time_range[0]
animation.EndTime = animation_window_time_range[1]
animation.NumberOfFrames = animation_window_num_frames
logger.debug(
f"Animating from scene time {animation.StartTime} to"
f" {animation.EndTime} in {animation.NumberOfFrames} frames.")
def scene_time_from_real(real_time):
return (real_time / animation_length_in_seconds *
animation_length_in_M)
# For some reason the keyframe time for animations is expected to be within
# (0, 1) so we need to transform back and forth from this "normalized" time
def scene_time_from_normalized(normalized_time):
return animation.StartTime + normalized_time * (
animation.EndTime - animation.StartTime)
def normalized_time_from_scene(scene_time):
return (scene_time - animation.StartTime) / (animation.EndTime -
animation.StartTime)
# Setup progress measuring already here so volume data computing for
# initial frame is measured
if show_progress and not no_render:
logging.getLogger().handlers = [TqdmLoggingHandler()]
animation_window_frame_range = tqdm.trange(
animation_window_num_frames,
desc="Rendering",
unit="frame",
miniters=1,
position=job_id,
)
else:
animation_window_frame_range = range(animation_window_num_frames)
# Set the initial time step
animation.GoToFirst()
# Display the volume data. This will trigger computing the volume data at the
# current time step.
for volume_data, waveform_to_volume_config in zip(
waveform_to_volume_objects, waveform_to_volume_configs):
vol_repr = (waveform_to_volume_config["VolumeRepresentation"]
if "VolumeRepresentation" in waveform_to_volume_config else
{})
volume_color_by = config_color.extract_color_by(vol_repr)
if (vol_repr["VolumeRenderingMode"] == "GPU Based"
and len(volume_color_by) > 2):
logger.warning(
"The 'GPU Based' volume renderer doesn't support multiple"
" components.")
volume = pv.Show(volume_data, view, **vol_repr)
pv.ColorBy(volume, value=volume_color_by)
if "Slices" in scene:
for slice_config in scene["Slices"]:
slice_obj_config = slice_config.get("Object", {})
slice = pv.Slice(Input=volume_data)
slice.SliceType = "Plane"
slice.SliceOffsetValues = [0.0]
slice.SliceType.Origin = slice_obj_config.get(
"Origin", [0.0, 0.0, -0.3])
slice.SliceType.Normal = slice_obj_config.get(
"Normal", [0.0, 0.0, 1.0])
slice_rep = pv.Show(slice, view,
**slice_config.get("Representation", {}))
pv.ColorBy(slice_rep, value=volume_color_by)
# Display the time
if "TimeAnnotation" in scene:
time_annotation = pv.AnnotateTimeFilter(volume_data,
**scene["TimeAnnotation"])
pv.Show(time_annotation, view, **scene["TimeAnnotationRepresentation"])
# Add spheres
if "Spheres" in scene:
for sphere_config in scene["Spheres"]:
sphere = pv.Sphere(**sphere_config["Object"])
pv.Show(sphere, view, **sphere_config["Representation"])
# Add trajectories and objects that follow them
if "Trajectories" in scene:
for trajectory_config in scene["Trajectories"]:
trajectory_name = trajectory_config["Name"]
radial_scale = (trajectory_config["RadialScale"]
if "RadialScale" in trajectory_config else 1.0)
# Load the trajectory data
traj_data_reader = TrajectoryDataReader(
RadialScale=radial_scale,
**scene["Datasources"]["Trajectories"][trajectory_name],
)
# Make sure the data is loaded so we can retrieve timesteps.
# TODO: This should be fixed in `TrajectoryDataReader` by
# communicating time range info down the pipeline, but we had issues
# with that (see also `WaveformDataReader`).
traj_data_reader.UpdatePipeline()
if "Objects" in trajectory_config:
with animate.restore_animation_state(animation):
follow_traj = FollowTrajectory(
TrajectoryData=traj_data_reader)
for traj_obj_config in trajectory_config["Objects"]:
for traj_obj_key in traj_obj_config:
if traj_obj_key in [
"Representation",
"Visibility",
"TimeShift",
"Glyph",
]:
continue
traj_obj_type = getattr(pv, traj_obj_key)
traj_obj_glyph = traj_obj_type(
**traj_obj_config[traj_obj_key])
follow_traj.UpdatePipeline()
traj_obj = pv.Glyph(Input=follow_traj,
GlyphType=traj_obj_glyph)
# Can't set this in the constructor for some reason
traj_obj.ScaleFactor = 1.0
for glyph_property in (traj_obj_config["Glyph"] if "Glyph"
in traj_obj_config else []):
setattr(
traj_obj,
glyph_property,
traj_obj_config["Glyph"][glyph_property],
)
traj_obj.UpdatePipeline()
if "TimeShift" in traj_obj_config:
traj_obj = animate.apply_time_shift(
traj_obj, traj_obj_config["TimeShift"])
pv.Show(traj_obj, view,
**traj_obj_config["Representation"])
if "Visibility" in traj_obj_config:
animate.apply_visibility(
traj_obj,
traj_obj_config["Visibility"],
normalized_time_from_scene,
scene_time_from_real,
)
if "Tail" in trajectory_config:
with animate.restore_animation_state(animation):
traj_tail = TrajectoryTail(TrajectoryData=traj_data_reader)
if "TimeShift" in trajectory_config:
traj_tail = animate.apply_time_shift(
traj_tail, trajectory_config["TimeShift"])
tail_config = trajectory_config["Tail"]
traj_color_by = config_color.extract_color_by(tail_config)
if "Visibility" in tail_config:
tail_visibility_config = tail_config["Visibility"]
del tail_config["Visibility"]
else:
tail_visibility_config = None
tail_rep = pv.Show(traj_tail, view, **tail_config)
pv.ColorBy(tail_rep, value=traj_color_by)
if tail_visibility_config is not None:
animate.apply_visibility(
traj_tail,
tail_visibility_config,
normalized_time_from_scene=normalized_time_from_scene,
scene_time_from_real=scene_time_from_real,
)
if "Move" in trajectory_config:
move_config = trajectory_config["Move"]
logger.debug(
f"Animating '{move_config['guiName']}' along trajectory.")
with h5py.File(trajectory_file, "r") as traj_data_file:
trajectory_data = np.array(
traj_data_file[trajectory_subfile])
if radial_scale != 1.0:
trajectory_data[:, 1:] *= radial_scale
logger.debug(f"Trajectory data shape: {trajectory_data.shape}")
animate.follow_path(
gui_name=move_config["guiName"],
trajectory_data=trajectory_data,
num_keyframes=move_config["NumKeyframes"],
scene_time_range=time_range_in_M,
normalized_time_from_scene=normalized_time_from_scene,
)
# Add non-spherical horizon shapes (instead of spherical objects following
# trajectories)
if "Horizons" in scene:
for horizon_config in scene["Horizons"]:
with animate.restore_animation_state(animation):
horizon = pv.PVDReader(FileName=scene["Datasources"]
["Horizons"][horizon_config["Name"]])
if horizon_config.get("InterpolateTime", False):
horizon = pv.TemporalInterpolator(
Input=horizon, DiscreteTimeStepInterval=0)
if "TimeShift" in horizon_config:
horizon = animate.apply_time_shift(horizon,
horizon_config["TimeShift"],
animation)
# Try to make horizon surfaces smooth. At low angular resoluton
# they still show artifacts, so perhaps more can be done.
horizon = pv.ExtractSurface(Input=horizon)
horizon = pv.GenerateSurfaceNormals(Input=horizon)
horizon_rep_config = horizon_config.get("Representation", {})
if "Representation" not in horizon_rep_config:
horizon_rep_config["Representation"] = "Surface"
if "AmbientColor" not in horizon_rep_config:
horizon_rep_config["AmbientColor"] = [0.0, 0.0, 0.0]
if "DiffuseColor" not in horizon_rep_config:
horizon_rep_config["DiffuseColor"] = [0.0, 0.0, 0.0]
if "Specular" not in horizon_rep_config:
horizon_rep_config["Specular"] = 0.2
if "SpecularPower" not in horizon_rep_config:
horizon_rep_config["SpecularPower"] = 10
if "SpecularColor" not in horizon_rep_config:
horizon_rep_config["SpecularColor"] = [1.0, 1.0, 1.0]
if "ColorBy" in horizon_rep_config:
horizon_color_by = config_color.extract_color_by(
horizon_rep_config)
else:
horizon_color_by = None
horizon_rep = pv.Show(horizon, view, **horizon_rep_config)
if horizon_color_by is not None:
pv.ColorBy(horizon_rep, value=horizon_color_by)
# Animate visibility
if "Visibility" in horizon_config:
animate.apply_visibility(
horizon,
horizon_config["Visibility"],
normalized_time_from_scene=normalized_time_from_scene,
scene_time_from_real=scene_time_from_real,
)
if "Contours" in horizon_config:
for contour_config in horizon_config["Contours"]:
contour = pv.Contour(Input=horizon,
**contour_config["Object"])
contour_rep = pv.Show(contour, view,
**contour_config["Representation"])
pv.ColorBy(contour_rep, None)
if "Visibility" in horizon_config:
animate.apply_visibility(
contour,
horizon_config["Visibility"],
normalized_time_from_scene=
normalized_time_from_scene,
scene_time_from_real=scene_time_from_real,
)
# Configure transfer functions
if "TransferFunctions" in scene:
for tf_config in scene["TransferFunctions"]:
colored_field = tf_config["Field"]
transfer_fctn = pv.GetColorTransferFunction(colored_field)
opacity_fctn = pv.GetOpacityTransferFunction(colored_field)
tf.configure_transfer_function(transfer_fctn, opacity_fctn,
tf_config["TransferFunction"])
# Save state file before configuring camera keyframes.
# TODO: Make camera keyframes work with statefile
if save_state_to_file is not None:
pv.SaveState(save_state_to_file + ".pvsm")
# Camera shots
# TODO: Make this work with freezing time while the camera is swinging
if animation is None:
for i, shot in enumerate(scene["CameraShots"]):
if (i == len(scene["CameraShots"]) - 1 or
(shot["Time"] if "Time" in shot else 0.0) >= view.ViewTime):
camera_motion.apply(shot)
break
else:
camera_motion.apply_swings(
scene["CameraShots"],
scene_time_range=time_range_in_M,
scene_time_from_real=scene_time_from_real,
normalized_time_from_scene=normalized_time_from_scene,
)
# Report time
if animation is not None:
report_time_cue = pv.PythonAnimationCue()
report_time_cue.Script = """
def start_cue(self): pass
def tick(self):
import paraview.simple as pv
import logging
logger = logging.getLogger('Animation')
scene_time = pv.GetActiveView().ViewTime
logger.info(f"Scene time: {scene_time}")
def end_cue(self): pass
"""
animation.Cues.append(report_time_cue)
if show_preview and animation is not None:
animation.PlayMode = "Real Time"
animation.Duration = 10
animation.Play()
animation.PlayMode = "Sequence"
if no_render:
logger.info("No rendering requested. Total time:"
f" {time.time() - render_start_time:.2f}s")
return
if frames_dir is None:
raise RuntimeError("Trying to render but `frames_dir` is not set.")
if os.path.exists(frames_dir):
logger.warning(
f"Output directory '{frames_dir}' exists, files may be overwritten."
)
else:
os.makedirs(frames_dir)
if animation is None:
pv.Render()
pv.SaveScreenshot(os.path.join(frames_dir, "frame.png"))
else:
# Iterate over frames manually to support filling in missing frames.
# If `pv.SaveAnimation` would support that, here's how it could be
# invoked:
# pv.SaveAnimation(
# os.path.join(frames_dir, 'frame.png'),
# view,
# animation,
# FrameWindow=frame_window,
# SuffixFormat='.%06d')
# Note that `FrameWindow` appears to be buggy, so we set up the
# `animation` according to the `frame_window` above so the frame files
# are numberd correctly.
for animation_window_frame_i in animation_window_frame_range:
frame_i = frame_window[0] + animation_window_frame_i
frame_file = os.path.join(frames_dir, f"frame.{frame_i:06d}.png")
if render_missing_frames and os.path.exists(frame_file):
continue
logger.debug(f"Rendering frame {frame_i}...")
animation.AnimationTime = (
animation.StartTime +
time_per_frame_in_M * animation_window_frame_i)
pv.Render()
pv.SaveScreenshot(frame_file)
logger.info(f"Rendered frame {frame_i}.")
logger.info(
f"Rendering done. Total time: {time.time() - render_start_time:.2f}s")
def pvd_to_mp4(sim_dir,
path_to_movies,
movie_name='movie',
representation='Surface',
num_regions=0):
###############################
# Validate directory and data #
###############################
if not (os.path.isdir(sim_dir)):
raise Exception('pvd_to_mp4: Invalid simulation directory')
if not (os.path.isdir(path_to_movies)):
raise Exception('pvd_to_mp4: Invalid movie directory')
if representation not in ['Surface', 'Points']:
raise Exception(
'pvd_to_mp4: Representation must be either Surface or Points')
if num_regions not in [0, 9]:
raise Exception(
'pvd_to_mp4: Currently there is only support for 9 node regions')
# sim_id = os.path.basename(os.path.normpath(sim_dir))
possible_data_directories = []
for directory in os.listdir(sim_dir):
if directory.startswith('results_from_time'):
possible_data_directories.append(os.path.join(sim_dir, directory))
if len(possible_data_directories) == 0:
raise Exception(
'pvd_to_mp4: Could not find a "results_from_time_X" directory')
# The last directory alphabetically will be the one after any initial relaxation simulation
data_directory = sorted(possible_data_directories)[-1]
# Get the location of the pvd file
pvd_file = os.path.join(data_directory, 'results.pvd')
if not (os.path.isfile(pvd_file)):
raise Exception('pvd_to_mp4: Could not find a pvd data file')
if os.path.getsize(pvd_file) < 1024:
raise Exception(
'pvd_to_mp4: pvd file exists but is < 1kb. Presumably simulation failed to finish as expected.'
)
full_movie_path = os.path.join(path_to_movies,
movie_name + '_' + representation + '.mp4')
##################################
# Set up scene with box and data #
##################################
# Get active view. This is the ParaView default view - blue background with cross hairs and orientation axes
render_view = pv.GetActiveViewOrCreate('RenderView')
# Change parameters to be how we want them for output
render_view.ViewSize = [1600, 900] # Size of output (pixels)
render_view.CenterAxesVisibility = 0 # Remove cross hairs
render_view.OrientationAxesVisibility = 0 # Remove orientation axes
render_view.Background = [0.5, 0.5,
0.5] # Set background colour to 50% grey
# Create a unit square (a 3D Box with ZLength set to zero)
unit_square = pv.Box()
unit_square.ZLength = 0.0
unit_square.YLength = 9.0 / 16.0
unit_square.Center = [0.5, 0.5, 0.0]
# Show the box in the current render view, and colour it black
unit_square_display = pv.Show(unit_square, render_view)
unit_square_display.DiffuseColor = [0.0, 0.0, 0.0]
# Read the relevant pvd file from the Chaste output
results_pvd = pv.PVDReader(FileName=pvd_file)
# Show the data in the current render view as a surface
results_pvd_display = pv.Show(results_pvd, render_view)
results_pvd_display.Representation = representation
if representation == 'Points' and num_regions == 9:
pv.ColorBy(results_pvd_display, ('POINTS', 'Node Regions'))
node_regions_lut = pv.GetColorTransferFunction('NodeRegions')
# Each four digits are: data value, r, g, b (colour)
node_regions_lut.RGBPoints = [
0.0,
1.00,
0.00,
0.00, # LEFT_APICAL_REGION, red
1.0,
1.00,
0.00,
0.00, # RIGHT_APICAL_REGION, red
2.0,
1.00,
0.00,
1.00, # LEFT_PERIAPICAL_REGION, purple
3.0,
1.00,
0.00,
1.00, # RIGHT_PERIAPICAL_REGION, purple
4.0,
0.00,
0.00,
1.00, # LEFT_LATERAL_REGION, blue
5.0,
0.00,
0.00,
1.00, # RIGHT_LATERAL_REGION, blue
6.0,
1.00,
1.00,
1.00, # LEFT_BASAL_REGION, white
7.0,
1.00,
1.00,
1.00, # RIGHT_BASAL_REGION, white
8.0,
1.00,
1.00,
1.00
] # LAMINA_REGION, white
cell_regions_lut = pv.GetColorTransferFunction('CellRegions')
cell_regions_lut.RGBPoints = [
-1.0,
0.0,
0.00,
0.00, # basal lamina, black
0.0,
0.00,
0.00,
1.00, # left region, blue
1.0,
1.00,
1.00,
1.00, # centre_region, white
2.0,
1.00,
0.00,
0.00
] # right_region, red
# Glyphs for cell region
cell_glyphs = pv.Glyph(Input=results_pvd, GlyphType='2D Glyph')
cell_glyphs.Scalars = ['CELLS', 'Cell Regions']
cell_glyphs.ScaleFactor = 0.01
cell_glyphs.GlyphMode = 'All Points'
cell_glyphs.GlyphType.GlyphType = 'Diamond'
cell_glyphs.GlyphType.Filled = 1
cell_glyphs.GlyphTransform.Scale = [1.0, 2.0, 1.0]
cell_glyphs_display = pv.Show(cell_glyphs, render_view)
cell_glyphs_display.Opacity = 0.5
# Glyphs for node region
node_glyphs = pv.Glyph(Input=results_pvd, GlyphType='2D Glyph')
node_glyphs.Scalars = ['POINTS', 'Node Regions']
node_glyphs.ScaleFactor = 0.002
node_glyphs.GlyphMode = 'All Points'
node_glyphs.GlyphType.GlyphType = 'Circle'
node_glyphs.GlyphType.Filled = 1
pv.Show(node_glyphs, render_view)
###################################
# Put the camera where we want it #
###################################
# This happens after all calls to Show, as Show may do some automatic camera re-setting
render_view.InteractionMode = '2D'
render_view.CameraPosition = [0.5, 0.5, 1.0]
render_view.CameraFocalPoint = [0.5, 0.5, 0.0]
# This parameter sets the 'zoom' and needs fine-tuning by the aspect ratio
render_view.CameraParallelScale = 0.5 * 9.0 / 16.0
##################################
# Set up and write the animation #
##################################
# Get an animation scene, which has parameters we need to change before output
animation_scene = pv.GetAnimationScene()
# Get a list of time step values from the pvd file. Typically this will look like [t_0, t1, t2, ..., t_end]
time_step_info = results_pvd.TimestepValues
num_time_steps = len(time_step_info)
# Set the animation parameters
animation_scene.NumberOfFrames = num_time_steps # If num frames != num time steps, some interpolation will be used
animation_scene.StartTime = time_step_info[
0] # Usually t_0, the first entry in time_step_info
animation_scene.EndTime = time_step_info[
-1] # Usually t_end, the final entry in time_step_info
# Write the animation as a series of uncompressed png files, with no magnification
pv.WriteAnimation(sim_dir + 'results_' + representation + '.png',
Magnification=1,
Compression=False)
# Raise exception if the png files are not generated as expected
if not (os.path.isfile(sim_dir + 'results_' + representation +
'.0000.png')):
raise Exception('pvd_to_mp4: png sequence not exported as expected')
#######################################
# Convert from png to mp4 and tidy up #
#######################################
# Ubuntu 14.04 (trusty) came bundled with avconv instead of ffmpeg, but they're nearly the same software
# so we don't have to change the command other than the name of the video converter to use
if platform.linux_distribution()[2] == 'trusty':
video_converter = 'avconv'
else:
video_converter = 'ffmpeg'
# Set how long you want the video to be (in seconds), and set the frame rate accordingly
video_duration = 15.0
frame_rate = str(num_time_steps / video_duration)
# Send the system command to run avconv/ffmpeg. Parameters:
# -v 0 Suppress console output so as not to clutter the terminal
# -r frame_rate Set the frame rate calculated above
# -f image2 Set the convert format (image sequence to video)
# -i dir/results.%04d.png Input expected as dir/results.####.png, the output from WriteAnimation above
# -c:v h264 Video codec to use is h264
# -crf 0 Set video quality: 0 best, 51 worst (https://trac.ffmpeg.org/wiki/Encode/H.264)
# -y dir/movie.mp4 Output directory and name
os.system(video_converter + ' -v 0 -r ' + frame_rate + ' -f image2 -i ' +
sim_dir + 'results_' + representation +
'.%04d.png -c:v h264 -crf 0 -y ' + full_movie_path)
# Raise exception if the mp4 file is not generated as expected
if not (os.path.isfile(full_movie_path)):
raise Exception('pvd_to_mp4: mp4 not generated as expected')
# Raise exception if the mp4 file file is created but is smaller than 1kb - ffmpeg sometimes
# generates an empty file even if an error occurs
if os.path.getsize(full_movie_path) < 1024:
raise Exception('pvd_to_mp4: mp4 not generated as expected')
# Clean up the png files created by WriteAnimation
os.system('rm ' + sim_dir + '*.png')
# import the simple module from the paraview
import paraview.simple as pv
# disable automatic camera reset on 'Show'
pv._DisableFirstRenderCameraReset()
# create a new 'VisItNek5000Reader'
ablnek5000 = pv.VisItNek5000Reader(
# FileName='/run/media/avmo/seagate/runs/abl_irrot_15x24x10_V1pix1.x1.571_2020-02-05_12-02-28/abl.nek5000'
FileName=
'/home/avmo/src/exabl/data/abl_irrot_15x24x10_V1pix1.x1.571_2020-02-05_12-02-28/abl.nek5000'
)
ablnek5000.Meshes = ['mesh']
ablnek5000.PointArrays = []
# get animation scene
animationScene1 = pv.GetAnimationScene()
# get the time-keeper
timeKeeper1 = pv.GetTimeKeeper()
# update animation scene based on data timesteps
animationScene1.UpdateAnimationUsingDataTimeSteps()
# Properties modified on ablnek5000
ablnek5000.PointArrays = [
'pressure', 'velocity_mag', 'x_velocity', 'y_velocity', 'z_velocity'
]
# get active view
renderView1 = pv.GetActiveViewOrCreate('RenderView')
# uncomment following to set a specific view size
def __init__(self, filepath='.'):
self.filepath = filepath
self.time = 0.0
self.surfaceColorMode = 0 # Local range
self.subSurfaceColorMode = 0 # Local range
# Surface View
self.viewSurface = simple.CreateRenderView(True)
self.viewSurface.EnableRenderOnInteraction = 0
self.viewSurface.OrientationAxesVisibility = 0
self.viewSurface.Background = [0.9, 0.9, 0.9]
self.viewSurface.InteractionMode = '2D'
self.viewSurface.CameraParallelProjection = 1
# SubSurface view
self.viewSubSurface = simple.CreateRenderView(True)
self.viewSubSurface.EnableRenderOnInteraction = 0
self.viewSubSurface.OrientationAxesVisibility = 0
self.viewSubSurface.Background = [0.9, 0.9, 0.9]
self.viewSubSurface.InteractionMode = '2D'
self.viewSubSurface.CameraParallelProjection = 1
# Read dataset
self.reader = simple.ParFlowReader(FileName=filepath, DeflectTerrain=1)
self.readerSurface = simple.OutputPort(self.reader, 1)
self.readerSubSurface = simple.OutputPort(self.reader, 0)
# Water table depth
self.waterTableDepth = simple.WaterTableDepth(
Subsurface=self.readerSubSurface, Surface=self.readerSurface)
self.cellCenter = simple.CellCenters(Input=self.waterTableDepth)
self.wtdVectCalc = simple.Calculator(Input=self.cellCenter)
self.wtdVectCalc.ResultArrayName = 'wtdVect'
self.wtdVectCalc.Function = 'iHat + jHat + kHat * water table depth'
self.waterTableDepthGlyph = simple.Glyph(
Input=self.wtdVectCalc,
GlyphType='Cylinder',
ScaleFactor=500,
GlyphMode='All Points',
GlyphTransform='Transform2',
ScaleArray=['POINTS', 'wtdVect'],
VectorScaleMode='Scale by Components',
)
self.waterTableDepthGlyph.GlyphTransform.Rotate = [90.0, 0.0, 0.0]
self.waterTableDepthGlyph.GlyphType.Resolution = 12
self.waterTableDepthGlyph.GlyphType.Radius = 0.25
self.waterTableRepresentation = simple.Show(self.waterTableDepthGlyph,
self.viewSubSurface)
self.waterTableRepresentation.Visibility = 0
# Water balance
self.waterBalance = simple.WaterBalance(
Subsurface=self.readerSubSurface, Surface=self.readerSurface)
self.waterBalanceOverTime = simple.PlotGlobalVariablesOverTime(
Input=self.waterBalance)
# Surface representation
self.surfaceRepresentation = simple.Show(self.readerSurface,
self.viewSurface)
self.surfaceRepresentation.SetScalarBarVisibility(
self.viewSurface, True)
# SubSurface representation + slice extract
self.reader.UpdatePipeline()
self.voi = self.reader.GetClientSideObject().GetOutputDataObject(
0).GetExtent()
self.extractSubset = simple.ExtractSubset(Input=self.readerSubSurface)
self.subSurfaceRepresentation = simple.Show(self.extractSubset,
self.viewSubSurface)
self.subSurfaceRepresentation.Representation = 'Surface'
# Reset camera + center of rotation
simple.Render(self.viewSurface)
simple.ResetCamera(self.viewSurface)
self.viewSurface.CenterOfRotation = self.viewSurface.CameraFocalPoint
simple.Render(self.viewSubSurface)
simple.ResetCamera(self.viewSubSurface)
self.viewSubSurface.CenterOfRotation = self.viewSubSurface.CameraFocalPoint
# Time management
self.animationScene = simple.GetAnimationScene()
self.animationScene.UpdateAnimationUsingDataTimeSteps()
