ublas_wrapper.h

Go to the documentation of this file.

/* KRATOS  _     _                       ____        _
//        | |   (_)_ __   ___  __ _ _ __/ ___|  ___ | |_   _____ _ __ ___
//        | |   | | '_ \ / _ \/ _` | '__\___ \ / _ \| \ \ / / _ \ '__/ __|
//        | |___| | | | |  __/ (_| | |   ___) | (_) | |\ V /  __/ |  \__ |
//        |_____|_|_| |_|\___|\__,_|_|  |____/ \___/|_| \_/ \___|_|  |___/ Application
//
//  Authors: Thomas Oberbichler
*/
 
#if !defined(KRATOS_UBLAS_WRAPPER_H_INCLUDED)
#define KRATOS_UBLAS_WRAPPER_H_INCLUDED
 
// System includes
#include <vector>
 
// External includes
#include <Eigen/Core>
#include <Eigen/Sparse>
 
// Project includes
#include "includes/define.h"
#include "linear_solvers_define.h"
 
 
namespace Kratos
{
 
template <
    typename TScalar = double,
    typename TEigenSparseMatrix = Kratos::EigenSparseMatrix<TScalar>>
class UblasWrapper
{
    std::vector<int> m_index1;
    std::vector<int> m_index2;
    Eigen::Map<const TEigenSparseMatrix> m_map;
 
public:
    UblasWrapper()
    : m_index1(),
      m_index2(),
      m_map(0, 0, 0, nullptr, nullptr, nullptr)
    {
    }
 
    template <typename TUblasSparseMatrix>
    UblasWrapper(const TUblasSparseMatrix& matrix)
    : m_index1(matrix.index1_data().begin(), matrix.index1_data().end()),
      m_index2(matrix.index2_data().begin(), matrix.index2_data().end()),
      m_map(matrix.size1(), matrix.size2(), matrix.nnz(), m_index1.data(), m_index2.data(), matrix.value_data().begin())
    {
    }
 
    const Eigen::Map<const TEigenSparseMatrix>& matrix() const
    {
      return m_map;
    }
};
 
} // namespace Kratos
 
#endif // defined(KRATOS_UBLAS_WRAPPER_H_INCLUDED)