CMatrix.hpp

/*
 * CMatrix.hpp
 * 
 * Copyright 2018 Fernando Pujaico Rivera <fernando.pujaico.rivera@gmail.com>
 * 
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 * MA 02110-1301, USA.
 * 
 */
 
#ifndef __PDS_CMATRIX_HPP__
#define __PDS_CMATRIX_HPP__
 
#include <string>
#include <Pds/Ra>
#include <Pds/Size>
#include <Pds/Complex>
 
namespace Pds{
 
class CVector;
 
class CMatrix
{
 
private:
    // Parte real de una matriz compleja
    Pds::Matrix Real;
    // Parte imaginaria de una matriz compleja
    Pds::Matrix Imag;
    
    friend class CVector; 
 
public:
 
    CMatrix(void);
    
    CMatrix(unsigned int Nlin,unsigned int Ncol);
 
    
    CMatrix(unsigned int Nlin,unsigned int Ncol, double val);
    
    CMatrix(unsigned int Nlin,unsigned int Ncol, double valr, double vali);
     
    CMatrix(unsigned int Nlin,unsigned int Ncol, const Pds::Complex &C);
 
    CMatrix(const Pds::Matrix &Real,const Pds::Matrix &Imag);
 
 
    CMatrix(const Pds::CMatrix &B);
 
 
    CMatrix(Pds::Complex func(const Pds::Complex &a,const Pds::Complex &b),const Pds::CMatrix &A,const Pds::CMatrix &B);
    
    ~CMatrix(); 
    
public:
    Pds::Complex operator[](const unsigned int &id) const
    {
        return Pds::Complex(this->Real.Get(id),this->Imag.Get(id));
    }
 
    Pds::Complex GetRaw(unsigned int lin,unsigned int col) const
    {
        return Pds::Complex(this->Real.GetRaw(lin,col),this->Imag.GetRaw(lin,col));
    }
 
    void SetRaw(unsigned int lin,unsigned int col,const Pds::Complex &val)
    {
        this->Real.SetRaw(lin,col,val.Real);
        this->Imag.SetRaw(lin,col,val.Imag);
    }
    
    
public:
    bool SetRowVector(unsigned int lin,const Pds::CVector &X);   
  
public: 
    bool Copy(const Pds::CMatrix &A);
    
 
    bool Copy(const Pds::Matrix &A);
    
    Pds::CMatrix& operator = (const Pds::CMatrix &B);
    
    
    Pds::CMatrix& operator = (const Pds::Matrix &B);
    
    
    void MakeEmpty(void);
    
    bool Reshape(unsigned int Nlin,unsigned int Ncol);
 
public:
    bool Rect(const Pds::Matrix &Real,const Pds::Matrix &Imag);
 
    bool Polar(const Pds::Matrix &Module,const Pds::Matrix &Angle);
    
public:
 
    Pds::CMatrix operator + (void) const;
 
    Pds::CMatrix Plus (void) const;
 
    Pds::CMatrix operator - (void) const;
 
    Pds::CMatrix Minus (void) const;
public:
    Pds::CMatrix operator + (const Pds::CMatrix &B) const;
    
    Pds::CMatrix operator + (double b) const;
    
    Pds::CMatrix operator + (const Pds::Complex &c) const;
 
    Pds::CMatrix Add(const Pds::CMatrix &B) const;
 
    Pds::CMatrix Add(double b) const;
 
    Pds::CMatrix Add(const Pds::Complex &c) const;
 
    Pds::CMatrix operator - (const Pds::CMatrix &B) const;
    
    Pds::CMatrix Sub(const Pds::CMatrix &B) const;
    
    Pds::CMatrix operator - (double b) const;
    
    Pds::CMatrix operator - (const Pds::Complex &c) const;
    
    Pds::CMatrix Sub(double b) const;
    
    Pds::CMatrix Sub(const Pds::Complex &c) const;
 
    Pds::CMatrix operator * (const Pds::CMatrix &B) const;
 
    Pds::CMatrix Mul(const Pds::CMatrix &B) const;
 
    Pds::CMatrix operator * (double b) const;
 
    Pds::CMatrix operator * (const Pds::Complex &c) const;
    
    Pds::CMatrix Mul(double b) const;
 
    Pds::CMatrix Mul(const Pds::Complex &c) const;
 
    Pds::CMatrix Product(const Pds::CMatrix &B) const;
 
    Pds::CMatrix operator / (const Pds::CMatrix &B) const;
 
    Pds::CMatrix Div(const Pds::CMatrix &B) const;
 
 
    Pds::CMatrix operator / (const Pds::Matrix &B) const;
 
    Pds::CMatrix Div(const Pds::Matrix &B) const;
 
 
    Pds::CMatrix operator / (double b) const;
 
 
    Pds::CMatrix operator / (const Pds::Complex &c) const;
 
 
    Pds::CMatrix Div(double b) const;
 
    Pds::CMatrix Div(const Pds::Complex &c) const;
 
 
    Pds::CMatrix Pow(const Pds::CMatrix &B) const;
 
    Pds::CMatrix Pow(double b) const;
public:
    Pds::CMatrix Conj(void) const;
    
    Pds::Matrix Module(void) const;
 
    
    Pds::Matrix Module2(void) const;
 
    
    Pds::Matrix Angle(void) const;
 
public:
    Pds::Size Size(void) const;
 
    unsigned int Nlin(void) const;
 
    unsigned int Ncol(void) const;
 
    unsigned int Nel(void) const;
public:
 
    bool IsEmpty(void) const;
 
    bool IsNotSimilarTo(const Pds::CMatrix &B) const;
 
    bool HasNan(void) const;
public:
    void Print(std::string str="") const;
 
}; // Class CMatrix
 
} // namespace Pds
 
 
Pds::CMatrix operator + (double a, const Pds::CMatrix &B);
 
Pds::CMatrix operator + (const Pds::Complex &c, const Pds::CMatrix &B);
 
Pds::CMatrix operator - (double a, const Pds::CMatrix &B);
 
Pds::CMatrix operator - (const Pds::Complex &c, const Pds::CMatrix &B);
 
Pds::CMatrix operator * (double a, const Pds::CMatrix &B);
 
Pds::CMatrix operator * (const Pds::Complex &c, const Pds::CMatrix &B);
 
Pds::CMatrix operator / (double a, const Pds::CMatrix &B);
 
 
Pds::CMatrix operator / (const Pds::Complex &c, const Pds::CMatrix &B);
 
#endif