Move vortex stuff to separate file
This commit is contained in:
120
euler/src/lib.rs
120
euler/src/lib.rs
@@ -6,6 +6,9 @@ use sbp::operators::{InterpolationOperator, SbpOperator2d, UpwindOperator2d};
|
|||||||
use sbp::utils::Direction;
|
use sbp::utils::Direction;
|
||||||
use sbp::Float;
|
use sbp::Float;
|
||||||
|
|
||||||
|
mod vortex;
|
||||||
|
pub use vortex::{vortex, VortexParameters, Vortice};
|
||||||
|
|
||||||
pub const GAMMA: Float = 1.4;
|
pub const GAMMA: Float = 1.4;
|
||||||
|
|
||||||
// A collection of buffers that allows one to efficiently
|
// A collection of buffers that allows one to efficiently
|
||||||
@@ -445,123 +448,6 @@ fn h2_diff() {
|
|||||||
assert!((field0.h2_err(&field1, &SBP8).powi(2) - 4.0).abs() < 1e-3);
|
assert!((field0.h2_err(&field1, &SBP8).powi(2) - 4.0).abs() < 1e-3);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Copy, Clone, Debug)]
|
|
||||||
#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]
|
|
||||||
pub struct Vortice {
|
|
||||||
pub x0: Float,
|
|
||||||
pub y0: Float,
|
|
||||||
pub rstar: Float,
|
|
||||||
pub eps: Float,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(feature = "serde1")]
|
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
|
|
||||||
#[derive(Clone, Debug)]
|
|
||||||
#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]
|
|
||||||
pub struct VortexParameters {
|
|
||||||
// The limit of 5 vortices can be bumped if needed...
|
|
||||||
pub vortices: ArrayVec<Vortice, 5>,
|
|
||||||
pub mach: Float,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[allow(clippy::too_many_arguments)]
|
|
||||||
#[allow(clippy::many_single_char_names)]
|
|
||||||
pub fn vortex(
|
|
||||||
rho: ArrayViewMut1<Float>,
|
|
||||||
rhou: ArrayViewMut1<Float>,
|
|
||||||
rhov: ArrayViewMut1<Float>,
|
|
||||||
e: ArrayViewMut1<Float>,
|
|
||||||
x: ArrayView1<Float>,
|
|
||||||
y: ArrayView1<Float>,
|
|
||||||
time: Float,
|
|
||||||
vortex_param: &VortexParameters,
|
|
||||||
) {
|
|
||||||
assert_eq!(rho.len(), rhou.len());
|
|
||||||
assert_eq!(rho.len(), rhov.len());
|
|
||||||
assert_eq!(rho.len(), e.len());
|
|
||||||
assert_eq!(rho.len(), x.len());
|
|
||||||
assert_eq!(rho.len(), y.len());
|
|
||||||
assert_eq!(x.shape(), y.shape());
|
|
||||||
|
|
||||||
let m = vortex_param.mach;
|
|
||||||
let p_inf = 1.0 / (GAMMA * m * m);
|
|
||||||
|
|
||||||
let rho_inf: Float = 1.0;
|
|
||||||
let u_inf: Float = 1.0;
|
|
||||||
let v_inf: Float = 0.0;
|
|
||||||
let e_inf = p_inf / (GAMMA - 1.0) + rho_inf * (u_inf.powi(2) + v_inf.powi(2)) / 2.0;
|
|
||||||
|
|
||||||
azip!((rho in rho,
|
|
||||||
rhou in rhou,
|
|
||||||
rhov in rhov,
|
|
||||||
e in e,
|
|
||||||
x in x,
|
|
||||||
y in y)
|
|
||||||
{
|
|
||||||
|
|
||||||
let mut iterator = vortex_param.vortices.iter();
|
|
||||||
|
|
||||||
match iterator.next() {
|
|
||||||
None => {
|
|
||||||
*rho = rho_inf;
|
|
||||||
*rhou = rho_inf*u_inf;
|
|
||||||
*rhou = rho_inf*v_inf;
|
|
||||||
*e = e_inf;
|
|
||||||
return;
|
|
||||||
},
|
|
||||||
Some(vortice) => {
|
|
||||||
use sbp::consts::PI;
|
|
||||||
|
|
||||||
let rstar = vortice.rstar;
|
|
||||||
let eps = vortice.eps;
|
|
||||||
|
|
||||||
let dx = (x - vortice.x0) - time;
|
|
||||||
let dy = y - vortice.y0;
|
|
||||||
let f = (1.0 - (dx*dx + dy*dy))/(rstar*rstar);
|
|
||||||
|
|
||||||
*rho = Float::powf(1.0 - eps*eps*(GAMMA - 1.0)*m*m/(8.0*PI*PI*p_inf*rstar*rstar)*f.exp(), 1.0/(GAMMA - 1.0));
|
|
||||||
assert!(*rho > 0.0);
|
|
||||||
let p = Float::powf(*rho, GAMMA)*p_inf;
|
|
||||||
let u = 1.0 - eps*dy/(2.0*PI*p_inf.sqrt()*rstar*rstar)*(f/2.0).exp();
|
|
||||||
let v = eps*dx/(2.0*PI*p_inf.sqrt()*rstar*rstar)*(f/2.0).exp();
|
|
||||||
|
|
||||||
assert!(p > 0.0);
|
|
||||||
|
|
||||||
assert!(*rho > 0.0);
|
|
||||||
*rhou = *rho*u;
|
|
||||||
*rhov = *rho*v;
|
|
||||||
*e = p/(GAMMA - 1.0) + *rho*(u*u + v*v)/2.0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for vortice in iterator {
|
|
||||||
use sbp::consts::PI;
|
|
||||||
|
|
||||||
let rstar = vortice.rstar;
|
|
||||||
let eps = vortice.eps;
|
|
||||||
|
|
||||||
let dx = (x - vortice.x0) - time;
|
|
||||||
let dy = y - vortice.y0;
|
|
||||||
let f = (1.0 - (dx*dx + dy*dy))/(rstar*rstar);
|
|
||||||
|
|
||||||
let rho_vortice = Float::powf(1.0 - eps*eps*(GAMMA - 1.0)*m*m/(8.0*PI*PI*p_inf*rstar*rstar)*f.exp(), 1.0/(GAMMA - 1.0));
|
|
||||||
let p = Float::powf(rho_vortice, GAMMA)*p_inf;
|
|
||||||
let u = 1.0 - eps*dy/(2.0*PI*p_inf.sqrt()*rstar*rstar)*(f/2.0).exp();
|
|
||||||
let v = eps*dx/(2.0*PI*p_inf.sqrt()*rstar*rstar)*(f/2.0).exp();
|
|
||||||
|
|
||||||
assert!(rho_vortice > 0.0);
|
|
||||||
assert!(p > 0.0);
|
|
||||||
|
|
||||||
*rho += rho_vortice - rho_inf;
|
|
||||||
assert!(*rho > 0.0);
|
|
||||||
*rhou += rho_vortice*u - rho_inf*u_inf;
|
|
||||||
*rhov += rho_vortice*v - rho_inf*v_inf;
|
|
||||||
*e += (p/(GAMMA - 1.0) + rho_vortice*(u*u + v*v)/2.0) - e_inf;
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
fn pressure(gamma: Float, rho: Float, rhou: Float, rhov: Float, e: Float) -> Float {
|
fn pressure(gamma: Float, rho: Float, rhou: Float, rhov: Float, e: Float) -> Float {
|
||||||
(gamma - 1.0) * (e - (rhou * rhou + rhov * rhov) / (2.0 * rho))
|
(gamma - 1.0) * (e - (rhou * rhou + rhov * rhov) / (2.0 * rho))
|
||||||
}
|
}
|
||||||
|
|||||||
118
euler/src/vortex.rs
Normal file
118
euler/src/vortex.rs
Normal file
@@ -0,0 +1,118 @@
|
|||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[derive(Copy, Clone, Debug)]
|
||||||
|
#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]
|
||||||
|
pub struct Vortice {
|
||||||
|
pub x0: Float,
|
||||||
|
pub y0: Float,
|
||||||
|
pub rstar: Float,
|
||||||
|
pub eps: Float,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "serde1")]
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
|
#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]
|
||||||
|
pub struct VortexParameters {
|
||||||
|
// The limit of 5 vortices can be bumped if needed...
|
||||||
|
pub vortices: ArrayVec<Vortice, 5>,
|
||||||
|
pub mach: Float,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[allow(clippy::too_many_arguments)]
|
||||||
|
#[allow(clippy::many_single_char_names)]
|
||||||
|
pub fn vortex(
|
||||||
|
rho: ArrayViewMut1<Float>,
|
||||||
|
rhou: ArrayViewMut1<Float>,
|
||||||
|
rhov: ArrayViewMut1<Float>,
|
||||||
|
e: ArrayViewMut1<Float>,
|
||||||
|
x: ArrayView1<Float>,
|
||||||
|
y: ArrayView1<Float>,
|
||||||
|
time: Float,
|
||||||
|
vortex_param: &VortexParameters,
|
||||||
|
) {
|
||||||
|
assert_eq!(rho.len(), rhou.len());
|
||||||
|
assert_eq!(rho.len(), rhov.len());
|
||||||
|
assert_eq!(rho.len(), e.len());
|
||||||
|
assert_eq!(rho.len(), x.len());
|
||||||
|
assert_eq!(rho.len(), y.len());
|
||||||
|
assert_eq!(x.shape(), y.shape());
|
||||||
|
|
||||||
|
let m = vortex_param.mach;
|
||||||
|
let p_inf = 1.0 / (GAMMA * m * m);
|
||||||
|
|
||||||
|
let rho_inf: Float = 1.0;
|
||||||
|
let u_inf: Float = 1.0;
|
||||||
|
let v_inf: Float = 0.0;
|
||||||
|
let e_inf = p_inf / (GAMMA - 1.0) + rho_inf * (u_inf.powi(2) + v_inf.powi(2)) / 2.0;
|
||||||
|
|
||||||
|
azip!((rho in rho,
|
||||||
|
rhou in rhou,
|
||||||
|
rhov in rhov,
|
||||||
|
e in e,
|
||||||
|
x in x,
|
||||||
|
y in y)
|
||||||
|
{
|
||||||
|
|
||||||
|
let mut iterator = vortex_param.vortices.iter();
|
||||||
|
|
||||||
|
match iterator.next() {
|
||||||
|
None => {
|
||||||
|
*rho = rho_inf;
|
||||||
|
*rhou = rho_inf*u_inf;
|
||||||
|
*rhou = rho_inf*v_inf;
|
||||||
|
*e = e_inf;
|
||||||
|
return;
|
||||||
|
},
|
||||||
|
Some(vortice) => {
|
||||||
|
use sbp::consts::PI;
|
||||||
|
|
||||||
|
let rstar = vortice.rstar;
|
||||||
|
let eps = vortice.eps;
|
||||||
|
|
||||||
|
let dx = (x - vortice.x0) - time;
|
||||||
|
let dy = y - vortice.y0;
|
||||||
|
let f = (1.0 - (dx*dx + dy*dy))/(rstar*rstar);
|
||||||
|
|
||||||
|
*rho = Float::powf(1.0 - eps*eps*(GAMMA - 1.0)*m*m/(8.0*PI*PI*p_inf*rstar*rstar)*f.exp(), 1.0/(GAMMA - 1.0));
|
||||||
|
assert!(*rho > 0.0);
|
||||||
|
let p = Float::powf(*rho, GAMMA)*p_inf;
|
||||||
|
let u = 1.0 - eps*dy/(2.0*PI*p_inf.sqrt()*rstar*rstar)*(f/2.0).exp();
|
||||||
|
let v = eps*dx/(2.0*PI*p_inf.sqrt()*rstar*rstar)*(f/2.0).exp();
|
||||||
|
|
||||||
|
assert!(p > 0.0);
|
||||||
|
|
||||||
|
assert!(*rho > 0.0);
|
||||||
|
*rhou = *rho*u;
|
||||||
|
*rhov = *rho*v;
|
||||||
|
*e = p/(GAMMA - 1.0) + *rho*(u*u + v*v)/2.0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for vortice in iterator {
|
||||||
|
use sbp::consts::PI;
|
||||||
|
|
||||||
|
let rstar = vortice.rstar;
|
||||||
|
let eps = vortice.eps;
|
||||||
|
|
||||||
|
let dx = (x - vortice.x0) - time;
|
||||||
|
let dy = y - vortice.y0;
|
||||||
|
let f = (1.0 - (dx*dx + dy*dy))/(rstar*rstar);
|
||||||
|
|
||||||
|
let rho_vortice = Float::powf(1.0 - eps*eps*(GAMMA - 1.0)*m*m/(8.0*PI*PI*p_inf*rstar*rstar)*f.exp(), 1.0/(GAMMA - 1.0));
|
||||||
|
let p = Float::powf(rho_vortice, GAMMA)*p_inf;
|
||||||
|
let u = 1.0 - eps*dy/(2.0*PI*p_inf.sqrt()*rstar*rstar)*(f/2.0).exp();
|
||||||
|
let v = eps*dx/(2.0*PI*p_inf.sqrt()*rstar*rstar)*(f/2.0).exp();
|
||||||
|
|
||||||
|
assert!(rho_vortice > 0.0);
|
||||||
|
assert!(p > 0.0);
|
||||||
|
|
||||||
|
*rho += rho_vortice - rho_inf;
|
||||||
|
assert!(*rho > 0.0);
|
||||||
|
*rhou += rho_vortice*u - rho_inf*u_inf;
|
||||||
|
*rhov += rho_vortice*v - rho_inf*v_inf;
|
||||||
|
*e += (p/(GAMMA - 1.0) + rho_vortice*(u*u + v*v)/2.0) - e_inf;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user