BlockDiagonalOperator.h

Go to the documentation of this file.

#ifndef BLOCKDIAGONALOPERATOR_H
#define BLOCKDIAGONALOPERATOR_H
 
#include "MUQ/Modeling/LinearAlgebra/LinearOperator.h"
 
#include <vector>
 
namespace muq{
namespace Modeling{
 
    class BlockDiagonalOperator : public LinearOperator
    {
 
    public:
 
        BlockDiagonalOperator(std::vector<std::shared_ptr<LinearOperator>> const& blocksIn);
 
        virtual Eigen::MatrixXd Apply(Eigen::Ref<const Eigen::MatrixXd> const& x) override;
        
        virtual Eigen::MatrixXd ApplyTranspose(Eigen::Ref<const Eigen::MatrixXd> const& x) override;
        
        virtual Eigen::MatrixXd GetMatrix() override;
 
        std::shared_ptr<LinearOperator> GetBlock(int i) const{return blocks.at(i);};
        std::vector<std::shared_ptr<LinearOperator>> const& GetBlocks() const{return blocks;};
        
    private:
        std::vector<std::shared_ptr<LinearOperator>> blocks;
 
        static int SumRows(std::vector<std::shared_ptr<LinearOperator>> const& blocksIn);
 
        static int SumCols(std::vector<std::shared_ptr<LinearOperator>> const& blocksIn);
        
    };
}
}
 
 
 
        
 
 
 
#endif // BLOCKDIAGONALOPERATOR_H