VectorMatrixAssembler.cpp

/**
 * \copyright
 * Copyright (c) 2012-2017, OpenGeoSys Community (http://www.opengeosys.org)
 *            Distributed under a Modified BSD License.
 *              See accompanying file LICENSE.txt or
 *              http://www.opengeosys.org/project/license
 *
 */

#include "VectorMatrixAssembler.h"

#include <cassert>

#include "NumLib/DOF/DOFTableUtil.h"
#include "MathLib/LinAlg/Eigen/EigenMapTools.h"
#include "LocalAssemblerInterface.h"

#include "Process.h"

namespace ProcessLib
{
static std::map<ProcessLib::ProcessType, const std::vector<double>>
getPreviousLocalSolutionsOfCoupledProcesses(
    const StaggeredCouplingTerm& coupled_term,
    const std::vector<GlobalIndexType>& indices)
{
    std::map<ProcessLib::ProcessType, const std::vector<double>>
        local_coupled_xs0;
    auto it = coupled_term.coupled_processes.begin();
    while (it != coupled_term.coupled_processes.end())
    {
        auto const& coupled_pcs = it->second;
        auto const prevous_time_x = coupled_pcs.getPreviousTimeStepSolution();
        if (prevous_time_x)
        {
            auto const local_coupled_x0 = prevous_time_x->get(indices);
            BaseLib::insertMapIfKeyUniqueElseError(local_coupled_xs0, it->first,
                                                   local_coupled_x0,
                                                   "local_coupled_x0");
        }
        else
        {
            const std::vector<double> local_coupled_x0;
            BaseLib::insertMapIfKeyUniqueElseError(local_coupled_xs0, it->first,
                                                   local_coupled_x0,
                                                   "local_coupled_x0");
        }
        it++;
    }
    return local_coupled_xs0;
}

static std::map<ProcessLib::ProcessType, const std::vector<double>>
getCurrentLocalSolutionsOfCoupledProcesses(
    const std::map<ProcessType, GlobalVector const&>& global_coupled_xs,
    const std::vector<GlobalIndexType>& indices)
{
    std::map<ProcessLib::ProcessType, const std::vector<double>>
        local_coupled_xs;
    auto it = global_coupled_xs.begin();
    while (it != global_coupled_xs.end())
    {
        auto const coupled_x = it->second;
        auto const local_coupled_x = coupled_x.get(indices);
        BaseLib::insertMapIfKeyUniqueElseError(
            local_coupled_xs, it->first, local_coupled_x, "local_coupled_x");
        it++;
    }
    return local_coupled_xs;
}

VectorMatrixAssembler::VectorMatrixAssembler(
    std::unique_ptr<AbstractJacobianAssembler>&& jacobian_assembler)
    : _jacobian_assembler(std::move(jacobian_assembler))
{
}

void VectorMatrixAssembler::assemble(
    const std::size_t mesh_item_id, LocalAssemblerInterface& local_assembler,
    const NumLib::LocalToGlobalIndexMap& dof_table, const double t,
    const GlobalVector& x, GlobalMatrix& M, GlobalMatrix& K, GlobalVector& b,
    const StaggeredCouplingTerm& coupled_term)
{
    auto const indices = NumLib::getIndices(mesh_item_id, dof_table);
    auto const local_x = x.get(indices);

    _local_M_data.clear();
    _local_K_data.clear();
    _local_b_data.clear();

    if (coupled_term.empty)
    {
        local_assembler.assemble(t, local_x, _local_M_data, _local_K_data,
                                 _local_b_data);
    }
    else
    {
        auto local_coupled_xs0 =
            getPreviousLocalSolutionsOfCoupledProcesses(coupled_term, indices);
        auto local_coupled_xs = getCurrentLocalSolutionsOfCoupledProcesses(
            coupled_term.coupled_xs, indices);
        ProcessLib::LocalCouplingTerm local_coupling_term(
            coupled_term.dt, coupled_term.coupled_processes,
            std::move(local_coupled_xs0), std::move(local_coupled_xs));

        local_assembler.coupling_assemble(t, local_x, _local_M_data,
                                          _local_K_data, _local_b_data,
                                          local_coupling_term);
    }

    auto const num_r_c = indices.size();
    auto const r_c_indices =
        NumLib::LocalToGlobalIndexMap::RowColumnIndices(indices, indices);

    if (!_local_M_data.empty())
    {
        auto const local_M = MathLib::toMatrix(_local_M_data, num_r_c, num_r_c);
        M.add(r_c_indices, local_M);
    }
    if (!_local_K_data.empty())
    {
        auto const local_K = MathLib::toMatrix(_local_K_data, num_r_c, num_r_c);
        K.add(r_c_indices, local_K);
    }
    if (!_local_b_data.empty())
    {
        assert(_local_b_data.size() == num_r_c);
        b.add(indices, _local_b_data);
    }
}

void VectorMatrixAssembler::assembleWithJacobian(
    std::size_t const mesh_item_id, LocalAssemblerInterface& local_assembler,
    NumLib::LocalToGlobalIndexMap const& dof_table, const double t,
    GlobalVector const& x, GlobalVector const& xdot, const double dxdot_dx,
    const double dx_dx, GlobalMatrix& M, GlobalMatrix& K, GlobalVector& b,
    GlobalMatrix& Jac, const StaggeredCouplingTerm& coupled_term)
{
    auto const indices = NumLib::getIndices(mesh_item_id, dof_table);
    auto const local_x = x.get(indices);
    auto const local_xdot = xdot.get(indices);

    _local_M_data.clear();
    _local_K_data.clear();
    _local_b_data.clear();
    _local_Jac_data.clear();

    if (coupled_term.empty)
    {
        _jacobian_assembler->assembleWithJacobian(
            local_assembler, t, local_x, local_xdot, dxdot_dx, dx_dx,
            _local_M_data, _local_K_data, _local_b_data, _local_Jac_data);
    }
    else
    {
        auto local_coupled_xs0 =
            getPreviousLocalSolutionsOfCoupledProcesses(coupled_term, indices);
        auto local_coupled_xs = getCurrentLocalSolutionsOfCoupledProcesses(
            coupled_term.coupled_xs, indices);
        ProcessLib::LocalCouplingTerm local_coupling_term(
            coupled_term.dt, coupled_term.coupled_processes,
            std::move(local_coupled_xs0), std::move(local_coupled_xs));

        _jacobian_assembler->coupling_assembleWithJacobian(
            local_assembler, t, local_x, local_xdot, dxdot_dx, dx_dx,
            _local_M_data, _local_K_data, _local_b_data, _local_Jac_data,
            local_coupling_term);
    }

    auto const num_r_c = indices.size();
    auto const r_c_indices =
        NumLib::LocalToGlobalIndexMap::RowColumnIndices(indices, indices);

    if (!_local_M_data.empty())
    {
        auto const local_M = MathLib::toMatrix(_local_M_data, num_r_c, num_r_c);
        M.add(r_c_indices, local_M);
    }
    if (!_local_K_data.empty())
    {
        auto const local_K = MathLib::toMatrix(_local_K_data, num_r_c, num_r_c);
        K.add(r_c_indices, local_K);
    }
    if (!_local_b_data.empty())
    {
        assert(_local_b_data.size() == num_r_c);
        b.add(indices, _local_b_data);
    }
    if (!_local_Jac_data.empty())
    {
        auto const local_Jac =
            MathLib::toMatrix(_local_Jac_data, num_r_c, num_r_c);
        Jac.add(r_c_indices, local_Jac);
    }
    else
    {
        OGS_FATAL(
            "No Jacobian has been assembled! This might be due to programming "
            "errors in the local assembler of the current process.");
    }
}

}  // ProcessLib