dufort_frankel.cpp

Go to the documentation of this file.

#include "methods/dufort_frankel.hpp"
 
#include <algorithm> // std::copy
#include <cstddef>   // std::size_t
 
void DuFortFrankel::step(const Grid& g,
                         double D,
                         double dt,
                         const std::vector<double>& Tprev,
                         const std::vector<double>& Tcurr,
                         std::vector<double>& Tnext) const
{
    const double r = D * dt / (g.dx * g.dx);
    const std::size_t N = g.Nx;
 
    // Ensure the output vector is of the correct size
    if (Tnext.size() != N)
    {
        Tnext.resize(N);
    }
 
    // Start from the current state: this preserves the boundaries
    std::copy(Tcurr.begin(), Tcurr.end(), Tnext.begin());
 
    // Internal nodes
    for (std::size_t i = 1; i + 1 < N; ++i)
    {
        Tnext[i] = ((1.0 - 2.0 * r) * Tprev[i] + 2.0 * r * (Tcurr[i + 1] + Tcurr[i - 1])) /
                   (1.0 + 2.0 * r);
    }
}