[PL] re-introduced VectorMatrixAssembler

ProcessLib/VectorMatrixAssembler.cpp

+/**
+ * \copyright
+ * Copyright (c) 2012-2016, 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"
+
+namespace ProcessLib
+{
+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)
+{
+    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();
+
+    local_assembler.assemble(t, local_x, _local_M_data, _local_K_data, _local_b_data);
+
+    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)
+{
+    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();
+
+    _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);
+
+    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!");
+    }
+}
+
+}  // ProcessLib

ProcessLib/VectorMatrixAssembler.h

+/**
+ * \copyright
+ * Copyright (c) 2012-2016, OpenGeoSys Community (http://www.opengeosys.org)
+ *            Distributed under a Modified BSD License.
+ *              See accompanying file LICENSE.txt or
+ *              http://www.opengeosys.org/project/license
+ *
+ */
+
+#ifndef PROCESSLIB_VECTORMATRIXASSEMBLER_H_
+#define PROCESSLIB_VECTORMATRIXASSEMBLER_H_
+
+#include <vector>
+#include "NumLib/NumericsConfig.h"
+#include "AbstractJacobianAssembler.h"
+
+namespace NumLib
+{
+class LocalToGlobalIndexMap;
+}  // NumLib
+
+namespace ProcessLib
+{
+class LocalAssemblerInterface;
+
+class VectorMatrixAssembler final
+{
+public:
+    VectorMatrixAssembler(
+        std::unique_ptr<AbstractJacobianAssembler>&& jacobian_assembler);
+
+    void assemble(std::size_t const mesh_item_id,
+                  LocalAssemblerInterface& local_assembler,
+                  NumLib::LocalToGlobalIndexMap const& dof_table,
+                  double const t, GlobalVector const& x, GlobalMatrix& M,
+                  GlobalMatrix& K, GlobalVector& b);
+
+    void 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);
+
+private:
+    std::vector<double> _local_M_data;
+    std::vector<double> _local_K_data;
+    std::vector<double> _local_b_data;
+    std::vector<double> _local_Jac_data;
+
+    std::unique_ptr<AbstractJacobianAssembler> _jacobian_assembler;
+};
+
+}   // namespace ProcessLib
+
+#endif  // PROCESSLIB_VECTORMATRIXASSEMBLER_H_