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