Fixes for coverity.
[u/mrichter/AliRoot.git] / STEER / STEER / AliMinResSolve.h
1 #ifndef ALIMINRESSOLVE_H
2 #define ALIMINRESSOLVE_H
3
4 /**********************************************************************************************/
5 /* General class for solving large system of linear equations                                 */
6 /* Includes MINRES, FGMRES methods as well as a few precondiotiong methods                    */
7 /*                                                                                            */ 
8 /* Author: ruben.shahoyan@cern.ch                                                             */
9 /*                                                                                            */ 
10 /**********************************************************************************************/
11
12 #include <TObject.h>
13 #include <TVectorD.h>
14 class AliMatrixSq;
15 class AliMatrixSparse;
16 class AliSymBDMatrix;
17
18
19 class AliMinResSolve : public TObject {
20   //
21  public:
22   enum {kPreconBD=1,kPreconILU0=100,kPreconILU10=kPreconILU0+10,kPreconsTot};
23   enum {kSolMinRes,kSolFGMRes,kNSolvers};
24  public:
25   AliMinResSolve();
26   AliMinResSolve(const AliMatrixSq *mat, const TVectorD* rhs);
27   AliMinResSolve(const AliMatrixSq *mat, const double  * rhs);
28   AliMinResSolve(const AliMinResSolve& src);
29   ~AliMinResSolve();
30   AliMinResSolve& operator=(const AliMinResSolve& rhs);
31   //
32   // ---------  MINRES method (for symmetric matrices)
33   Bool_t SolveMinRes(Double_t* VecSol,Int_t precon=0,int itnlim=2000,double rtol=1e-12);
34   Bool_t SolveMinRes(TVectorD &VecSol,Int_t precon=0,int itnlim=2000,double rtol=1e-12);
35   //
36   // ---------  FGMRES method (for general symmetric matrices)
37   Bool_t SolveFGMRES(Double_t* VecSol,Int_t precon=0,int itnlim=2000,double rtol=1e-12, int nkrylov=60);
38   Bool_t SolveFGMRES(TVectorD &VecSol,Int_t precon=0,int itnlim=2000,double rtol=1e-12, int nkrylov=60);  
39   //
40   Bool_t InitAuxMinRes();
41   Bool_t InitAuxFGMRES(int nkrylov);
42   void   ApplyPrecon(const TVectorD& vecRHS, TVectorD& vecOut)     const;
43   void   ApplyPrecon(const double*   vecRHS, double*   vecOut)     const;
44   //
45   Int_t  BuildPrecon(Int_t val=0);
46   Int_t  GetPrecon()                                               const    {return fPrecon;} 
47   void   ClearAux();
48   //
49   Int_t  BuildPreconBD(Int_t hwidth);
50   Int_t  BuildPreconILUK(Int_t lofM);
51   Int_t  BuildPreconILUKDense(Int_t lofM);
52   Int_t  PreconILUKsymb(Int_t lofM);
53   Int_t  PreconILUKsymbDense(Int_t lofM);
54   //
55  protected:
56   //
57   Int_t               fSize;                             // dimension of the input matrix
58   Int_t               fPrecon;                           // preconditioner type
59   AliMatrixSq*        fMatrix;                           // matrix defining the equations
60   Double_t*           fRHS;                              // right hand side
61   //
62   Double_t            *fPVecY;                           // aux. space
63   Double_t            *fPVecR1;                          // aux. space
64   Double_t            *fPVecR2;                          // aux. space
65   Double_t            *fPVecV;                           // aux. space
66   Double_t            *fPVecW;                           // aux. space
67   Double_t            *fPVecW1;                          // aux. space
68   Double_t            *fPVecW2;                          // aux. space
69   Double_t            **fPvv;                            // aux. space
70   Double_t            **fPvz;                            // aux. space
71   Double_t            **fPhh;                            // aux. space
72   Double_t            *fDiagLU;                          // aux space
73   AliMatrixSparse     *fMatL;                            // aux. space
74   AliMatrixSparse     *fMatU;                            // aux. space
75   AliSymBDMatrix      *fMatBD;                           // aux. space
76   //
77   ClassDef(AliMinResSolve,0)
78 };
79
80 #endif