addscalarbc.h

Go to the documentation of this file.

/*
 * SPDX-License-Identifier: GPL-3.0-or-later
 * 2008-2024 San Diego State University Research Foundation (SDSURF).
 * See LICENSE file or https://www.gnu.org/licenses/gpl-3.0.html for details.
 */

 
#ifndef ADDSCALARBC_H
#define ADDSCALARBC_H
 
#include "utils.h"
#include "gradient.h"
#include <vector>
#include <cassert>
 
using namespace arma;
 
namespace AddScalarBC {

struct BC1D {
    vec dc;  // Dirichlet coefficients a0 (2x1: left, right)
    vec nc;  // Neumann coefficients b0 (2x1: left, right)
    vec v;   // Boundary values g (2x1: left, right)
 
    BC1D() : dc(2, fill::zeros), nc(2, fill::zeros), v(2, fill::zeros) {}
};

struct BC2D {
    vec dc;              // Dirichlet coefficients a0 (4x1: left, right, bottom, top)
    vec nc;              // Neumann coefficients b0 (4x1: left, right, bottom, top)
    std::vector<vec> v;  // Boundary values g (4 vectors: left, right, bottom, top)
 
    BC2D() : dc(4, fill::zeros), nc(4, fill::zeros), v(4) {}
};

struct BC3D {
    vec dc;              // Dirichlet coefficients a0 (6x1: left, right, bottom, top, front, back)
    vec nc;              // Neumann coefficients b0 (6x1: left, right, bottom, top, front, back)
    std::vector<vec> v;  // Boundary values g (6 vectors: left, right, bottom, top, front, back)
 
    BC3D() : dc(6, fill::zeros), nc(6, fill::zeros), v(6) {}
};
 
// ============================================================================
// LHS: Boundary matrix construction (1D, 2D, 3D overloads)
// ============================================================================

void addScalarBClhs(u16 k, u32 m, Real dx,
                    const vec &dc, const vec &nc,
                    sp_mat &Al, sp_mat &Ar);

void addScalarBClhs(u16 k, u32 m, Real dx, u32 n, Real dy,
                    const vec &dc, const vec &nc,
                    sp_mat &Al, sp_mat &Ar, sp_mat &Ab, sp_mat &At);

void addScalarBClhs(u16 k, u32 m, Real dx, u32 n, Real dy, u32 o, Real dz,
                    const vec &dc, const vec &nc,
                    sp_mat &Al, sp_mat &Ar, sp_mat &Ab,
                    sp_mat &At, sp_mat &Af, sp_mat &Ak);
 
// ============================================================================
// RHS: Boundary value application (1D, 2D, 3D overloads)
// ============================================================================

void addScalarBCrhs(vec &b, const vec &v, const uvec &indices);

void addScalarBCrhs(vec &b, const vec &dc, const vec &nc,
                    const std::vector<vec> &v,
                    const uvec &rl, const uvec &rr,
                    const uvec &rb, const uvec &rt);

void addScalarBCrhs(vec &b, const vec &dc, const vec &nc,
                    const std::vector<vec> &v,
                    const uvec &rl, const uvec &rr, const uvec &rb,
                    const uvec &rt, const uvec &rf, const uvec &rk);
 
// ============================================================================
// Top-level BC application (1D, 2D, 3D overloads)
// ============================================================================

void addScalarBC(sp_mat &A, vec &b, u16 k, u32 m, Real dx, const BC1D &bc);

void addScalarBC(sp_mat &A, vec &b, u16 k, u32 m, Real dx,
                 u32 n, Real dy, const BC2D &bc);

void addScalarBC(sp_mat &A, vec &b, u16 k, u32 m, Real dx,
                 u32 n, Real dy, u32 o, Real dz, const BC3D &bc);
 
 
 
} // namespace AddScalarBC
 
#endif // ADDSCALARBC_H