QMatrix.hh

Go to the documentation of this file.

 
#ifndef __QMATRIX_HH_
#define __QMATRIX_HH_
 
 
#include <gsl/gsl_matrix.h>
#include <map>
#include <iostream>
#include <QObject.hh>
 
Q_BEGIN_NAMESPACE
 
class QVector;
 
class QMatrix : public QObject {
public:
  QMatrix() ;
  QMatrix(UInt_t nrow,UInt_t ncol);
  QMatrix(const QVector& vec);
  QMatrix(const QMatrix& orig);
  virtual ~QMatrix();
  void Resize(UInt_t nrow,UInt_t ncol);
  
  void Clear();
  UInt_t GetNRow() const {return fNCol ? fSize/fNCol : 0;};
  UInt_t GetNCol() const {return fNCol;};
  void Initialize(double val=0);
  void SetToIdentity();
  void SetCol(UInt_t ncol,const QVector& vec);
  void SetRow(UInt_t nrow,const QVector& vec);
  void Set(const std::map<int,int>& p);
  QVector GetCol(UInt_t ncol);
  QVector GetCol(UInt_t ncol) const;
  QVector GetRow(UInt_t nrow);
  QVector GetRow(UInt_t nrow) const;
  QVector GetDiaganol();
  QVector GetDiaganol() const;
 
  QVector GetRowByColumnIntValue(UInt_t col,int val) const;
  double &operator()(UInt_t i,UInt_t j) { return GetElement(i,j); }
  const double &operator()(UInt_t i,UInt_t j) const { return GetElement(i,j); }
  const QMatrix& operator=(const QMatrix& orig);
  QMatrix operator-();
  const QMatrix & operator*=(double t);
  const QMatrix & operator*=(const QMatrix& other);
  const QMatrix & operator/=(double t);
  const QMatrix & operator+=(const QMatrix& mat);
  const QMatrix & operator-=(const QMatrix& mat);
  const QMatrix& Mult(const QMatrix&mat);
  const QMatrix& Div(const QMatrix& mat);
  QMatrix operator*(const QMatrix& mat) const;
  QMatrix operator*(double t) const;
  QMatrix operator/(double t) const;
  QMatrix operator+(const QMatrix& mat) const;
  QMatrix operator-(const QMatrix& mat) const;
  QVector operator*(const QVector& vec) const;
  QMatrix T() const; // 
  const QMatrix &Transpose(); //
  QMatrix Inv() const; // 
  const QMatrix &Invert(); 
  double Det() const;
  
  QMatrix GetCholeskyDecomposition() const;
  const QMatrix &CholeskyDecompose();
 
  void CholeskySolveX(const QVector &Y,QVector &X) const;
 
  QMatrix GetSubMatrix(UInt_t nI,UInt_t nJ,UInt_t startI=0,UInt_t startJ=0);
 
  const QMatrix GetSubMatrix(UInt_t nI,UInt_t nJ,UInt_t startI=0,
                             UInt_t startJ=0) const;
 
  QMatrix GetSummedAreaMatrix() const;
 
  
  void SetElement(UInt_t i,UInt_t j,Double_t val) { GetElement(i,j) = val; }
 
  
protected: 
  QMatrix (const gsl_matrix* mat);
  QMatrix(const double* orig);
  
  QMatrix(double *data,const UInt_t nrow,const UInt_t ncol,
          const UInt_t tda=0);
 
  
  void SetMat() const;
 
  double &GetElement(UInt_t i,UInt_t j);
 
  const double &GetElement(UInt_t i,UInt_t j) const;
  
  static void MatrixCopy(const double *orig,const UInt_t Nrow,
                         const UInt_t Ncol,double *dest,
                         UInt_t origTda=0,UInt_t destTda=0);
                    
 
  UInt_t MemRowElements() const { return (fTda>fNCol ? fTda : fNCol); }
 
  
private:
  Bool_t fDataOwner;
 
  UInt_t fSize;
  UInt_t fNCol;
  UInt_t fTda;
 
  UInt_t fAllocSize;
  Double_t *fData;//[fSize]
  mutable gsl_matrix* fMat;
  
  friend class QVector;
  friend class QMatrixC;
 
  QObjectDef(QMatrix,3);
};
 
// operations other than member functions
// formatted printout
 
Q_END_NAMESPACE
Diana::QMatrix operator*(double t, const Diana::QMatrix&mat);
std::ostream& operator<<(std::ostream&s,const Diana::QMatrix& m);
#endif