Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
SWIFT
SWIFTsim
Commits
f6f80792
Commit
f6f80792
authored
May 21, 2018
by
Josh Borrow
Browse files
Updated movie script and added it to automatically be ran after the production of snapshots
parent
69c00a44
Changes
2
Hide whitespace changes
Inline
Side-by-side
examples/KelvinHelmholtz_2D/makeMovie.py
View file @
f6f80792
...
...
@@ -9,15 +9,10 @@ Edit this file near the bottom with the number of snaps you have.
Written by Josh Borrow (joshua.borrow@durham.ac.uk)
"""
import
matplotlib
matplotlib
.
use
(
"Agg"
)
import
os
import
h5py
as
h5
import
numpy
as
np
import
scipy
as
sp
import
scipy.interpolate
as
si
import
matplotlib.pyplot
as
plt
def
load_and_extract
(
filename
):
...
...
@@ -67,14 +62,32 @@ def frame(n, *args):
if
__name__
==
"__main__"
:
import
matplotlib
matplotlib
.
use
(
"Agg"
)
from
tqdm
import
tqdm
from
matplotlib.animation
import
FuncAnimation
from
scipy.stats
import
gaussian_kde
import
matplotlib.pyplot
as
plt
filename
=
"kelvinhelmholtz"
dpi
=
512
frames
=
tqdm
(
np
.
arange
(
0
,
448
))
# Look for the number of files in the directory.
i
=
0
while
True
:
if
os
.
path
.
isfile
(
"{}_{:04d}.hdf5"
.
format
(
filename
,
i
)):
i
+=
1
else
:
break
if
i
>
10000
:
raise
FileNotFoundError
(
"Could not find the snapshots in the directory"
)
frames
=
tqdm
(
np
.
arange
(
0
,
i
))
# Creation of first frame
fig
,
ax
=
plt
.
subplots
(
1
,
1
,
figsize
=
(
1
,
1
),
frameon
=
False
)
...
...
examples/KelvinHelmholtz_2D/run.sh
View file @
f6f80792
...
...
@@ -8,7 +8,8 @@ then
fi
# Run SWIFT
../swift
-s
-t
1
kelvinHelmholtz.yml 2>&1 |
tee
output.log
../swift
-s
-t
4
kelvinHelmholtz.yml 2>&1 |
tee
output.log
# Plot the solution
python plotSolution.py 6
python makeMovie.py
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment