simple.hpp

Go to the documentation of this file.

#ifndef IRLBA_MATRIX_SIMPLE_HPP
#define IRLBA_MATRIX_SIMPLE_HPP
 
#include <memory>
#include <type_traits>
 
#include "../utils.hpp"
#include "interface.hpp"
 
#include "Eigen/Dense"
 
namespace irlba {
 
template<class EigenVector_, class Simple_>
class SimpleWorkspace final : public Workspace<EigenVector_> {
public:
    SimpleWorkspace(const Simple_& matrix) : my_matrix(matrix) {}
 
private:
    const Simple_& my_matrix;
 
public:
    void multiply(const EigenVector_& right, EigenVector_& output) {
        output.noalias() = my_matrix * right;
    }
};
 
template<class EigenVector_, class Simple_>
class SimpleAdjointWorkspace final : public AdjointWorkspace<EigenVector_> {
public:
    SimpleAdjointWorkspace(const Simple_& matrix) : my_matrix(matrix) {}
 
private:
    const Simple_& my_matrix;
 
public:
    void multiply(const EigenVector_& right, EigenVector_& output) {
        output.noalias() = my_matrix.adjoint() * right;
    }
};
 
template<class EigenMatrix_, class Simple_>
class SimpleRealizeWorkspace final : public RealizeWorkspace<EigenMatrix_> {
public:
    SimpleRealizeWorkspace(const Simple_& matrix) : my_matrix(matrix) {}
 
private:
    const Simple_& my_matrix;
    static constexpr bool is_same = std::is_same<EigenMatrix_, I<Simple_> >::value;
    typename std::conditional<is_same, bool, EigenMatrix_>::type buffer;
 
public:
    const EigenMatrix_& realize(EigenMatrix_& buffer) {
        if constexpr(is_same) {
            return my_matrix;
        } else {
            buffer = my_matrix;
            return buffer;
        }
    }
};
template<class EigenVector_, class EigenMatrix_, class SimplePointer_>
class SimpleMatrix final : public Matrix<EigenVector_, EigenMatrix_> {
public:
    SimpleMatrix(SimplePointer_ matrix) : my_matrix(std::move(matrix)) {}
 
private:
    SimplePointer_ my_matrix;
 
public:
    Eigen::Index rows() const {
        return my_matrix->rows();
    }
 
    Eigen::Index cols() const {
        return my_matrix->cols();
    }
 
public:
    std::unique_ptr<Workspace<EigenVector_> > new_workspace() const {
        return new_known_workspace();
    }
 
    std::unique_ptr<AdjointWorkspace<EigenVector_> > new_adjoint_workspace() const {
        return new_known_adjoint_workspace();
    }
 
    std::unique_ptr<RealizeWorkspace<EigenMatrix_> > new_realize_workspace() const {
        return new_known_realize_workspace();
    }
 
public:
    auto new_known_workspace() const {
        return std::make_unique<SimpleWorkspace<EigenVector_, I<decltype(*my_matrix)> > >(*my_matrix);
    }
 
    auto new_known_adjoint_workspace() const {
        return std::make_unique<SimpleAdjointWorkspace<EigenVector_, I<decltype(*my_matrix)> > >(*my_matrix);
    }
 
    auto new_known_realize_workspace() const {
        return std::make_unique<SimpleRealizeWorkspace<EigenMatrix_, I<decltype(*my_matrix)> > >(*my_matrix);
    }
};
 
}
 
#endif