gsBlockOp.h

Go to the documentation of this file.

#pragma once

#include <gsCore/gsExport.h>
#include <gsCore/gsLinearAlgebra.h>
#include <gsSolver/gsLinearOperator.h>

namespace gismo
{

template<class T>
class gsBlockOp GISMO_FINAL : public gsLinearOperator<T>
{
public:

    typedef memory::shared_ptr< gsBlockOp<T> > Ptr;

    typedef memory::unique_ptr< gsBlockOp<T> > uPtr;

    typedef memory::shared_ptr< gsLinearOperator<T> > BasePtr;

    gsBlockOp(index_t nRows, index_t nCols);

    static uPtr make(index_t nRows, index_t nCols)
    { return memory::make_unique( new gsBlockOp(nRows,nCols) ); }

    void addOperator(index_t row, index_t col, const BasePtr& op);

    const BasePtr & getOperator(index_t row, index_t col) const {
        return m_blockPrec(row,col);
    }

    void apply(const gsMatrix<T> & input, gsMatrix<T> & result) const;

    index_t rowBlocks() const {return m_blockPrec.rows();}
    index_t colBlocks() const {return m_blockPrec.cols();}

    index_t rows() const {return m_blockTargetPositions.sum();}
    index_t cols() const {return m_blockInputPositions.sum() ;}

private:

    gsEigen::Array<BasePtr, Dynamic, Dynamic> m_blockPrec;

    //Contains the size of the target vector for each block
    gsVector<index_t> m_blockTargetPositions;
    //Contains the size of the input vector for each block
    gsVector<index_t> m_blockInputPositions;

};

} // namespace gismo

#ifndef GISMO_BUILD_LIB
#include GISMO_HPP_HEADER(gsBlockOp.hpp)
#endif