mulimoen
/
SummationByParts
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

apply -d1.rev() on right boundary

This commit is contained in:
Magnus Ulimoen 2020-09-21 16:24:31 +02:00
parent 3d66e48413
commit 939eae9af1
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
heat-equation/src/main.rs
View File

@ -47,7 +47,12 @@ fn dual_dirichlet(v: ArrayView1<Float>, v0: Float, vn: Float) {
for (d, fut) in d1.iter().zip(fut.iter_mut()) {
*fut += tau.0 / (h * h) * d * (prev[0] - v0);
}
for (d, fut) in d1.iter().zip(fut.iter_mut().rev().take(d1.len()).rev()) {
for (d, fut) in d1
.iter()
.rev()
.map(|d| -d)
.zip(fut.iter_mut().rev().take(d1.len()).rev())
{
*fut += tau.1 / (h * h) * d * (prev[nx - 1] - vn);
}
};
@ -109,7 +114,12 @@ fn neumann_dirichlet(v: ArrayView1<Float>, v0: Float, vn: Float) {
.map(|(x, y)| x * y)
.sum::<Float>()
- v0);
for (d, fut) in d1.iter().zip(fut.iter_mut().rev().take(d1.len()).rev()) {
for (d, fut) in d1
.iter()
.rev()
.map(|d| -d)
.zip(fut.iter_mut().rev().take(d1.len()).rev())
{
*fut += tau.1 / (h * h) * d * (prev[nx - 1] - vn);
}
};