"""
Plots the vorticity field of a 2D cuIBM simulation.
"""
import os
from snake.cuibm.simulation import CuIBMSimulation
from snake.body import Body
simulation = CuIBMSimulation()
simulation.read_grid()
for time_step in simulation.get_time_steps():
all_bodies = Body(
file_path=os.path.join('{:0>7}'.format(time_step), 'bodies'))
n_total = all_bodies.x.size
bodies = [Body(), Body()]
bodies[0].x = all_bodies.x[:n_total // 2]
bodies[0].y = all_bodies.y[:n_total // 2]
bodies[1].x = all_bodies.x[n_total // 2:]
bodies[1].y = all_bodies.y[n_total // 2:]
simulation.read_fields('vorticity', time_step)
simulation.plot_contour('vorticity',
field_range=(-2.0, 2.0, 40),
filled_contour=True,
bodies=bodies,
view=(-3.0, -5.0, 15.0, 5.0),
style='seaborn-dark',
cmap='viridis',
width=8.0)
"""
Computes, plots, and saves the 2D vorticity field from a cuIBM simulation at
saved time-steps.
"""
from snake.cuibm.simulation import CuIBMSimulation
simulation = CuIBMSimulation()
simulation.read_grid()
for time_step in simulation.get_time_steps():
simulation.read_fields('vorticity', time_step)
simulation.plot_contour('vorticity',
field_range=(-5.0, 5.0, 101),
view=(-1.0, -5.0, 15.0, 5.0),
cmap='viridis',
style='seaborn-dark',
width=8.0)
import os
from matplotlib import pyplot
from snake.cuibm.simulation import CuIBMSimulation
from snake.solutions.ghiaEtAl1982 import GhiaEtAl1982
directory = os.getcwd()
# Reads the velocity fields from files.
simulation = CuIBMSimulation(directory=directory,
description='cuIBM')
simulation.read_grid(file_path=os.path.join(directory, 'grid'))
time_step = simulation.get_time_steps()[-1]
simulation.read_fields(['x-velocity', 'y-velocity'], time_step)
# Grabs the mid-cavity velocity values.
y, u = simulation.fields['x-velocity'].get_vertical_gridline_values(0.5)
x, v = simulation.fields['y-velocity'].get_horizontal_gridline_values(0.5)
# Gets the centerline velocities at Re=100 reported in Ghia et al. (1982).
file_path = os.path.join(os.environ['CUIBM_DIR'],
'data',
'ghia_et_al_1982_lid_driven_cavity.dat')
ghia = GhiaEtAl1982(file_path=file_path, Re=100)
# Plots the instantaneous drag coefficients.
images_directory = os.path.join(directory, 'images')
if not os.path.isdir(images_directory):
os.makedirs(images_directory)
pyplot.style.use('seaborn-dark')
"""
import os
from matplotlib import pyplot
from snake.cuibm.simulation import CuIBMSimulation
from snake.solutions.ghiaEtAl1982 import GhiaEtAl1982
directory = os.getcwd()
# Reads the velocity fields from files.
simulation = CuIBMSimulation(directory=directory, description='cuIBM')
simulation.read_grid(file_path=os.path.join(directory, 'grid'))
time_step = simulation.get_time_steps()[-1]
simulation.read_fields(['x-velocity', 'y-velocity'], time_step)
# Grabs the mid-cavity velocity values.
y, u = simulation.fields['x-velocity'].get_vertical_gridline_values(0.5)
x, v = simulation.fields['y-velocity'].get_horizontal_gridline_values(0.5)
# Gets the centerline velocities at Re=1000 reported in Ghia et al. (1982).
file_path = os.path.join(os.environ['CUIBM_DIR'], 'data',
'ghia_et_al_1982_lid_driven_cavity.dat')
ghia = GhiaEtAl1982(file_path=file_path, Re=1000)
# Plots the instantaneous drag coefficients.
images_directory = os.path.join(directory, 'images')
if not os.path.isdir(images_directory):
os.makedirs(images_directory)
pyplot.style.use('seaborn-dark')
simu_kwargs = {
# file: plotVorticitySnake.py # author: Olivier Mesnard ([email protected]) # description: Plots the 2D vorticity field near the snake. # Run this script from the simulation directory. from snake.cuibm.simulation import CuIBMSimulation simulation = CuIBMSimulation() simulation.read_grid() for time_step in simulation.get_time_steps(): simulation.read_fields("vorticity", time_step) simulation.plot_contour( "vorticity", field_range=[-5.0, 5.0, 101], filled_contour=True, view=[-0.75, -1.0, 1.50, 1.0], width=8.0 )
'linear'),
'folders': ['20', '60', '180', '540']},
'vquad': {'directory': os.path.join('vquad_T00.25_20_0.00050',
'linear'),
'folders': ['20', '60', '180', '540']}}
time_step = 500
field_names = ['x-velocity', 'y-velocity']
for key, series in data.items():
series['cases'] = []
for folder in series['folders']:
case = CuIBMSimulation(directory=os.path.join(series['directory'], folder),
description=folder)
case.read_grid()
case.read_fields(field_names, time_step)
nx, ny = case.grid[0].size - 1, case.grid[1].size - 1
masks = read_mask(os.path.join(case.directory, 'mask.txt'), nx, ny)
for i, name in enumerate(field_names):
case.fields[name].values = numpy.multiply(case.fields[name].values,
masks[i])
series['cases'].append(case)
series['first'] = get_observed_orders(series['cases'][:-1],
field_names,
series['cases'][0],
save_name=None)
series['last'] = get_observed_orders(series['cases'][1:],
field_names,
series['cases'][0],
save_name=None)