// @(#)root/matrix:$Id$
// Authors: Fons Rademakers, Eddy Offermann   Dec 2003

/*************************************************************************
 * Copyright (C) 1995-2000, Rene Brun and Fons Rademakers.               *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/

#ifndef ROOT_TDecompQRH
#define ROOT_TDecompQRH

///////////////////////////////////////////////////////////////////////////
//                                                                       //
// QR Decomposition class                                                //
//                                                                       //
///////////////////////////////////////////////////////////////////////////

#ifndef ROOT_TDecompBase
#include "TDecompBase.h"
#endif

class TDecompQRH : public TDecompBase
{
protected :

   //  A = fQ fR H (m x n) matrix
   TMatrixD fQ;  // (m x n) - orthogonal matrix
   TMatrixD fR;  // (n x n) - upper triangular matrix
   TVectorD fUp; // (n) - vector with Householder up's
   TVectorD fW;  // (n) - vector with Householder beta's

   static Bool_t QRH(TMatrixD &q,TVectorD &diagR,TVectorD &up,TVectorD &w,Double_t tol);

   virtual const TMatrixDBase &GetDecompMatrix() const { return fR; }

public :

   enum {kWorkMax = 100}; // size of work array

   TDecompQRH() {}
   TDecompQRH(Int_t nrows,Int_t ncols);
   TDecompQRH(Int_t row_lwb,Int_t row_upb,Int_t col_lwb,Int_t col_upb);
   TDecompQRH(const TMatrixD &m,Double_t tol = 0.0); // be careful for slicing in operator=
   TDecompQRH(const TDecompQRH &another);
   virtual ~TDecompQRH() {}

   virtual       Int_t     GetNrows () const { return fQ.GetNrows(); }
   virtual       Int_t     GetNcols () const { return fQ.GetNcols(); }
   virtual const TMatrixD &GetQ     ()       { if ( !TestBit(kDecomposed) ) Decompose();
                                               return fQ; }
   virtual const TMatrixD &GetR     ()       { if ( !TestBit(kDecomposed) ) Decompose();
                                               return fR; }
   virtual const TVectorD &GetUp    ()       { if ( !TestBit(kDecomposed) ) Decompose();
                                               return fUp; }
   virtual const TVectorD &GetW     ()       { if ( !TestBit(kDecomposed) ) Decompose();
                                               return fW; }

   virtual       void      SetMatrix(const TMatrixD &a);

   virtual Bool_t   Decompose  ();
   virtual Bool_t   Solve      (      TVectorD &b);
   virtual TVectorD Solve      (const TVectorD& b,Bool_t &ok) { TVectorD x = b; ok = Solve(x); return x; }
   virtual Bool_t   Solve      (      TMatrixDColumn &b);
   virtual Bool_t   TransSolve (      TVectorD &b);
   virtual TVectorD TransSolve (const TVectorD& b,Bool_t &ok) { TVectorD x = b; ok = TransSolve(x); return x; }
   virtual Bool_t   TransSolve (      TMatrixDColumn &b);
   virtual void     Det        (Double_t &d1,Double_t &d2);

           Bool_t   Invert     (TMatrixD &inv);
           TMatrixD Invert     (Bool_t &status);
           TMatrixD Invert     () { Bool_t status; return Invert(status); }

   void Print(Option_t *opt ="") const; // *MENU*

   TDecompQRH &operator= (const TDecompQRH &source);

   ClassDef(TDecompQRH,1) // Matrix Decompositition QRH
};

#endif
 TDecompQRH.h:1
 TDecompQRH.h:2
 TDecompQRH.h:3
 TDecompQRH.h:4
 TDecompQRH.h:5
 TDecompQRH.h:6
 TDecompQRH.h:7
 TDecompQRH.h:8
 TDecompQRH.h:9
 TDecompQRH.h:10
 TDecompQRH.h:11
 TDecompQRH.h:12
 TDecompQRH.h:13
 TDecompQRH.h:14
 TDecompQRH.h:15
 TDecompQRH.h:16
 TDecompQRH.h:17
 TDecompQRH.h:18
 TDecompQRH.h:19
 TDecompQRH.h:20
 TDecompQRH.h:21
 TDecompQRH.h:22
 TDecompQRH.h:23
 TDecompQRH.h:24
 TDecompQRH.h:25
 TDecompQRH.h:26
 TDecompQRH.h:27
 TDecompQRH.h:28
 TDecompQRH.h:29
 TDecompQRH.h:30
 TDecompQRH.h:31
 TDecompQRH.h:32
 TDecompQRH.h:33
 TDecompQRH.h:34
 TDecompQRH.h:35
 TDecompQRH.h:36
 TDecompQRH.h:37
 TDecompQRH.h:38
 TDecompQRH.h:39
 TDecompQRH.h:40
 TDecompQRH.h:41
 TDecompQRH.h:42
 TDecompQRH.h:43
 TDecompQRH.h:44
 TDecompQRH.h:45
 TDecompQRH.h:46
 TDecompQRH.h:47
 TDecompQRH.h:48
 TDecompQRH.h:49
 TDecompQRH.h:50
 TDecompQRH.h:51
 TDecompQRH.h:52
 TDecompQRH.h:53
 TDecompQRH.h:54
 TDecompQRH.h:55
 TDecompQRH.h:56
 TDecompQRH.h:57
 TDecompQRH.h:58
 TDecompQRH.h:59
 TDecompQRH.h:60
 TDecompQRH.h:61
 TDecompQRH.h:62
 TDecompQRH.h:63
 TDecompQRH.h:64
 TDecompQRH.h:65
 TDecompQRH.h:66
 TDecompQRH.h:67
 TDecompQRH.h:68
 TDecompQRH.h:69
 TDecompQRH.h:70
 TDecompQRH.h:71
 TDecompQRH.h:72
 TDecompQRH.h:73
 TDecompQRH.h:74
 TDecompQRH.h:75
 TDecompQRH.h:76
 TDecompQRH.h:77
 TDecompQRH.h:78
 TDecompQRH.h:79
 TDecompQRH.h:80
 TDecompQRH.h:81
 TDecompQRH.h:82
 TDecompQRH.h:83