MinResSolve.h

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#ifndef ALICEO2_FWDALIGN_MINRESSOLVE_H
#define ALICEO2_FWDALIGN_MINRESSOLVE_H
 
#include <TObject.h>
#include <TVectorD.h>
#include <TString.h>
 
namespace o2
{
namespace fwdalign
{
 
class MatrixSq;
class MatrixSparse;
class SymBDMatrix;
 
class MinResSolve : public TObject
{
 
 public:
  enum { kPreconBD = 1,
         kPreconILU0 = 100,
         kPreconILU10 = kPreconILU0 + 10,
         kPreconsTot };
  enum { kSolMinRes,
         kSolFGMRes,
         kNSolvers };
 
 public:
  MinResSolve();
 
  MinResSolve(const MatrixSq* mat, const TVectorD* rhs);
 
  MinResSolve(const MatrixSq* mat, const double* rhs);
 
  MinResSolve(const MinResSolve& src);
 
  ~MinResSolve() override;
 
  MinResSolve& operator=(const MinResSolve& rhs);
 
  Bool_t SolveMinRes(Double_t* VecSol, Int_t precon = 0, int itnlim = 2000, double rtol = 1e-12);
 
  Bool_t SolveMinRes(TVectorD& VecSol, Int_t precon = 0, int itnlim = 2000, double rtol = 1e-12);
 
  Bool_t SolveFGMRES(Double_t* VecSol, Int_t precon = 0, int itnlim = 2000, double rtol = 1e-12, int nkrylov = 60);
 
  Bool_t SolveFGMRES(TVectorD& VecSol, Int_t precon = 0, int itnlim = 2000, double rtol = 1e-12, int nkrylov = 60);
 
  Bool_t InitAuxMinRes();
 
  Bool_t InitAuxFGMRES(int nkrylov);
 
  void ApplyPrecon(const TVectorD& vecRHS, TVectorD& vecOut) const;
 
  void ApplyPrecon(const double* vecRHS, double* vecOut) const;
 
  Int_t BuildPrecon(Int_t val = 0);
 
  Int_t GetPrecon() const { return fPrecon; }
 
  void ClearAux();
 
  Int_t BuildPreconBD(Int_t hwidth);
 
  Int_t BuildPreconILUK(Int_t lofM);
 
  Int_t BuildPreconILUKDense(Int_t lofM);
 
  Int_t PreconILUKsymb(Int_t lofM);
 
  Int_t PreconILUKsymbDense(Int_t lofM);
 
 protected:
  Int_t fSize;       
  Int_t fPrecon;     
  MatrixSq* fMatrix; 
  Double_t* fRHS;    
 
  Double_t* fPVecY;    
  Double_t* fPVecR1;   // aux. space
  Double_t* fPVecR2;   // aux. space
  Double_t* fPVecV;    // aux. space
  Double_t* fPVecW;    // aux. space
  Double_t* fPVecW1;   // aux. space
  Double_t* fPVecW2;   // aux. space
  Double_t** fPvv;     // aux. space
  Double_t** fPvz;     // aux. space
  Double_t** fPhh;     // aux. space
  Double_t* fDiagLU;   // aux space
  MatrixSparse* fMatL; // aux. space
  MatrixSparse* fMatU; // aux. space
  SymBDMatrix* fMatBD; // aux. space
 
  ClassDefOverride(MinResSolve, 0);
};
 
} // namespace fwdalign
} // namespace o2
 
#endif