checks.cpp

#include <vector>
#include <boost/type_index.hpp>
#include <boost/assert.hpp>
#include "gs.h"
#include "checks.h"

void checkTypes(double F, double k, const std::vector<std::vector<double>>& u, const std::vector<std::vector<double>>& v) {
    BOOST_ASSERT_MSG(boost::typeindex::type_id_with_cvr<decltype(F)>().pretty_name() == boost::typeindex::type_id_with_cvr<std::decay<decltype(u[0][0])>::type>().pretty_name(), "Type of F does not match type of elements in u");
    BOOST_ASSERT_MSG(boost::typeindex::type_id_with_cvr<decltype(k)>().pretty_name() == boost::typeindex::type_id_with_cvr<std::decay<decltype(u[0][0])>::type>().pretty_name(), "Type of k does not match type of elements in u");
    BOOST_ASSERT_MSG(boost::typeindex::type_id_with_cvr<decltype(F)>().pretty_name() == boost::typeindex::type_id_with_cvr<std::decay<decltype(v[0][0])>::type>().pretty_name(), "Type of F does not match type of elements in v");
    BOOST_ASSERT_MSG(boost::typeindex::type_id_with_cvr<decltype(k)>().pretty_name() == boost::typeindex::type_id_with_cvr<std::decay<decltype(v[0][0])>::type>().pretty_name(), "Type of k does not match type of elements in v");
}

void checkSizes(const std::vector<std::vector<double>>& u, const std::vector<std::vector<double>>& v) {
    BOOST_ASSERT_MSG(u.size() == v.size(), "u and v are not the same size");
}

void checkSimulation(const std::vector<std::vector<double>>& u, const std::vector<std::vector<double>>& v) {
    for (const auto& row : u) {
        for (const auto& element : row) {
            BOOST_ASSERT_MSG(abs(element - F * dt) < 1e-9, "Simulation does not produce correct result when u = 0");
        }
    }
    for (const auto& row : v) {
        for (const auto& element : row) {
            BOOST_ASSERT_MSG(abs(element - 0.0) < 1e-9, "Simulation does not produce correct result when v = 0");
        }
    }

    // Check the boundary elements
    for (int x = 0; x < width; ++x) {
        BOOST_ASSERT_MSG(u[x][0] == 0.0, "Boundary element in u is not zero");
        BOOST_ASSERT_MSG(u[x][height - 1] == 0.0, "Boundary element in u is not zero");
        BOOST_ASSERT_MSG(v[x][0] == 0.0, "Boundary element in v is not zero");
        BOOST_ASSERT_MSG(v[x][height - 1] == 0.0, "Boundary element in v is not zero");
    }
    for (int y = 0; y < height; ++y) {
        BOOST_ASSERT_MSG(u[0][y] == 0.0, "Boundary element in u is not zero");
        BOOST_ASSERT_MSG(u[width - 1][y] == 0.0, "Boundary element in u is not zero");
        BOOST_ASSERT_MSG(v[0][y] == 0.0, "Boundary element in v is not zero");
        BOOST_ASSERT_MSG(v[width - 1][y] == 0.0, "Boundary element in v is not zero");
    }
}

void setZero() {
    for (auto& row : u) {
        for (auto& element : row) {
            element = 0.0;
        }
    }
    for (auto& row : v) {
        for (auto& element : row) {
            element = 0.0;
        }
    }
}