2da4528add
in a sequence of VTK files. Instead of loading a group of VTK which uses their index as time, the respective .vtk.series file can be loaded which relates each file to the correct time.
207 lines
6.0 KiB
Bash
Executable File
207 lines
6.0 KiB
Bash
Executable File
#!/bin/sh
|
|
#------------------------------------------------------------------------------
|
|
# ========= |
|
|
# \\ / F ield | OpenFOAM: The Open Source CFD Toolbox
|
|
# \\ / O peration | Website: https://openfoam.org
|
|
# \\ / A nd | Copyright (C) 2015-2024 OpenFOAM Foundation
|
|
# \\/ M anipulation |
|
|
#------------------------------------------------------------------------------
|
|
# License
|
|
# This file is part of OpenFOAM.
|
|
#
|
|
# OpenFOAM is free software: you can redistribute it and/or modify it
|
|
# under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# OpenFOAM is distributed in the hope that it will be useful, but WITHOUT
|
|
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
# for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with OpenFOAM. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
# Script
|
|
# foamSequenceVTKFiles
|
|
#
|
|
# Description
|
|
# Creates symbolic links to all VTK files in a post-processing directory
|
|
# Links form a sequence like 'patch.0000.vtk', 'patch.0001.vtk', etc.
|
|
#
|
|
# If files of the same name exist in different sub-directories of the main
|
|
# directory, e.g. 'inletPatch/0/patch.vtk' and 'outletPatch/0/patch.vtk',
|
|
# a further index is appended to the name in the links, e.g.
|
|
# 'patch0.0000.vtk' and 'patch1.0000.vtk'
|
|
#
|
|
# Paraview recognises the link names as a sequence which can be opened and
|
|
# played. The sequence of links to images can be used to create a video
|
|
# from the images.
|
|
# - Default directory name for VTK files is 'postProcessing'
|
|
#------------------------------------------------------------------------------
|
|
error() {
|
|
exec 1>&2
|
|
while [ "$#" -ge 1 ]; do echo "$1"; shift; done
|
|
usage
|
|
exit 1
|
|
}
|
|
|
|
usage() {
|
|
cat <<USAGE
|
|
|
|
Usage: ${0##*/} [OPTIONS] ...
|
|
options:
|
|
-case | -c <dir> specify case directory (default = local dir)
|
|
-dir | -d <dir> post-processing directory <dir> (default = postProcessing)
|
|
-help | -h print the usage
|
|
-out | -o <dir> output links directory <dir> (default = sequencedVTK)
|
|
|
|
Creates symbolic links to all VTK files in a post-processing directory
|
|
Links form a sequence like 'patch.0000.vtk', 'patch.0001.vtk', etc.
|
|
|
|
If files of the same name exist in different sub-directories of the main
|
|
directory, e.g. 'inletPatch/0/patch.vtk' and 'outletPatch/0/patch.vtk', a
|
|
further index is appended to the name in the links, e.g. 'patch0.0000.vtk' and
|
|
'patch1.0000.vtk'
|
|
|
|
Paraview recognises the link names as a sequence which can be opened and played.
|
|
The sequence of links to images can be used to create a video from the images.
|
|
- Default directory name for VTK files is 'postProcessing'
|
|
USAGE
|
|
}
|
|
|
|
# If image files of the same name exist in multiple sub-directories,
|
|
# list the sub-directory names
|
|
multipleSeries () {
|
|
_file_list="$1"
|
|
_depth="$(echo "$file_list" | awk -F'/' '{print NF}' | sort -un | head -1)"
|
|
|
|
# List no sub-directories if there are insufficient directory levels
|
|
[ "$_depth" -lt 3 ] && return 1
|
|
|
|
# Unique sub-directories
|
|
_dirs="$(echo "$_file_list" | \
|
|
awk -F'/' '{for(i=1; i<NF-1; i++) printf "%s/", $i; printf "\n"}' | \
|
|
sort -u | xargs -n 1)"
|
|
|
|
# Only return sub-directories if there is more than one
|
|
[ "$(echo "$_dirs" | wc -l)" -gt 1 ] || return 1
|
|
|
|
echo "$_dirs"
|
|
}
|
|
|
|
seriesFileHead () {
|
|
cat<<EOF
|
|
{
|
|
"file-series-version": "1.0",
|
|
"files": [
|
|
EOF
|
|
}
|
|
|
|
seriesFileFoot () {
|
|
cat<<EOF
|
|
|
|
]
|
|
}
|
|
EOF
|
|
}
|
|
|
|
dir=postProcessing
|
|
out=sequencedVTK
|
|
|
|
while [ "$#" -gt 0 ]
|
|
do
|
|
case "$1" in
|
|
-c | -case)
|
|
[ "$#" -ge 2 ] || error "'$1' option requires an argument"
|
|
cd "$2" 2>/dev/null || error "directory does not exist: '$2'"
|
|
shift 2
|
|
;;
|
|
-d | -dir)
|
|
[ "$#" -ge 2 ] || error "'$1' option requires an argument"
|
|
dir=$2
|
|
shift 2
|
|
;;
|
|
-h | -help)
|
|
usage && exit 0
|
|
;;
|
|
-o | -out)
|
|
[ "$#" -ge 2 ] || error "'$1' option requires an argument"
|
|
out=$2
|
|
shift 2
|
|
;;
|
|
-*)
|
|
error "invalid option '$1'"
|
|
;;
|
|
*)
|
|
break
|
|
;;
|
|
esac
|
|
done
|
|
|
|
[ ! -d "$dir" ] && error "Cannot find postProcessing directory, exiting."
|
|
|
|
files=$(find "$dir" -type f -name "*vtk" -print0 | xargs -0 -n 1)
|
|
names=$(for f in $files; do basename "$f" .vtk; done | sort -u)
|
|
|
|
if [ -d "$out" ]; then
|
|
echo "$out directory already exists. Deleting links within it..."
|
|
rm "${out:?}"/* 2>/dev/null
|
|
else
|
|
echo "Creating $out directory..."
|
|
mkdir "$out"
|
|
fi
|
|
|
|
for n in $names
|
|
do
|
|
echo "Sequencing all VTK files named $n.vtk"
|
|
|
|
# Create list of VTK files, ordered by time step
|
|
file_list=$(echo "$files" | \
|
|
grep -w "$n\.vtk" | \
|
|
awk -F'/' '{print($(NF-1)" "$0)}' | \
|
|
LC_ALL=C sort -k 1 -g | \
|
|
cut -d' ' -f2)
|
|
|
|
# sub-dirs of files with same name
|
|
dirs="$(multipleSeries "$file_list")" || dirs=SINGLE
|
|
|
|
# Series index, required if files with same name exist in multiple sub-dirs
|
|
s=0
|
|
|
|
for d in $dirs
|
|
do
|
|
# Image index
|
|
i=0
|
|
|
|
[ "$d" = SINGLE ] && unset s
|
|
|
|
series_file="$out/$n$s.vtk.series"
|
|
seriesFileHead > "$series_file"
|
|
|
|
for f in $file_list
|
|
do
|
|
! [ "$d" = SINGLE ] && \
|
|
! echo "$f" | grep -q "$d" && \
|
|
continue
|
|
|
|
link=$(printf "${n}${s}.%04d.vtk" $i)
|
|
ln -s "../$f" "$out/$link"
|
|
|
|
# Write series file
|
|
[ "$i" -eq 0 ] || printf ",\n" >> "$series_file"
|
|
time="$(dirname "$f" | awk -F/ '{print $NF}')"
|
|
printf "%17s\"%s\", \"time\": %s }" \
|
|
'{"name": ' \
|
|
"$link" \
|
|
"$time" >> "$series_file"
|
|
|
|
i=$(( i + 1 )) # Relies on ordered list of files
|
|
done
|
|
|
|
seriesFileFoot >> "$series_file"
|
|
|
|
s=$(( s + 1 ))
|
|
done
|
|
done
|