ConcatenateOperator.h

Go to the documentation of this file.

#ifndef CONCATENATEOPERATOR_H
#define CONCATENATEOPERATOR_H
 
#include "MUQ/Modeling/LinearAlgebra/LinearOperator.h"
 
#include <vector>
 
namespace muq
{
namespace Modeling
{
 
    class ConcatenateOperator : public LinearOperator
    {
    public:
        ConcatenateOperator(std::vector<std::shared_ptr<LinearOperator>> const& opsIn,
                            const int                                           rowColIn);
 
        virtual ~ConcatenateOperator(){};
        
        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;
        
        static std::shared_ptr<ConcatenateOperator> VStack(std::shared_ptr<LinearOperator> Ain,
                                                           std::shared_ptr<LinearOperator> Bin);
 
        static std::shared_ptr<ConcatenateOperator> HStack(std::shared_ptr<LinearOperator> Ain,
                                                           std::shared_ptr<LinearOperator> Bin);
 
 
        
    private:
        std::vector<std::shared_ptr<LinearOperator>> ops;
        
        const int rowCol; // zero if stacked vertically along rows, 1 if stacked horizontally
 
        static int GetRows(std::vector<std::shared_ptr<LinearOperator>> const& opsIn,
                           const int                                           rowColIn);
        
        static int GetCols(std::vector<std::shared_ptr<LinearOperator>> const& opsIn,
                           const int                                           rowColIn);
 
        void CheckSizes();
        
    }; // class ConcatenateOperator
 
    
 
} // namespace Modeling
} // namespace muq
 
 
#endif // #ifndef CONCATENATEOPERATOR_H