src/foam/matrices/lduMatrix/preconditioners/diagonalPreconditioner/diagonalPreconditioner.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | foam-extend: Open Source CFD
   4    \\    /   O peration     | Version:     3.2
   5     \\  /    A nd           | Web:         http://www.foam-extend.org
   6      \\/     M anipulation  | For copyright notice see file Copyright
   7 -------------------------------------------------------------------------------
   8 License
   9     This file is part of foam-extend.
  10
  11     foam-extend is free software: you can redistribute it and/or modify it
  12     under the terms of the GNU General Public License as published by the
  13     Free Software Foundation, either version 3 of the License, or (at your
  14     option) any later version.
  15
  16     foam-extend is distributed in the hope that it will be useful, but
  17     WITHOUT ANY WARRANTY; without even the implied warranty of
  18     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  19     General Public License for more details.
  20
  21     You should have received a copy of the GNU General Public License
  22     along with foam-extend.  If not, see <http://www.gnu.org/licenses/>.
  23
  24 \*---------------------------------------------------------------------------*/
  25
  26 #include "diagonalPreconditioner.H"
  27
  28 // * * * * * * * * * * * * * * Static Data Members * * * * * * * * * * * * * //
  29
  30 namespace Foam
  31 {
  32     defineTypeNameAndDebug(diagonalPreconditioner, 0);
  33
  34     lduPreconditioner::
  35         addsymMatrixConstructorToTable<diagonalPreconditioner>
  36         adddiagonalPreconditionerSymMatrixConstructorToTable_;
  37
  38     lduPreconditioner::
  39         addasymMatrixConstructorToTable<diagonalPreconditioner>
  40         adddiagonalPreconditionerAsymMatrixConstructorToTable_;
  41 }
  42
  43
  44 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
  45
  46 Foam::diagonalPreconditioner::diagonalPreconditioner
  47 (
  48     const lduMatrix& matrix,
  49     const FieldField<Field, scalar>& coupleBouCoeffs,
  50     const FieldField<Field, scalar>& coupleIntCoeffs,
  51     const lduInterfaceFieldPtrsList& interfaces,
  52     const dictionary&
  53 )
  54 :
  55     lduPreconditioner
  56     (
  57         matrix,
  58         coupleBouCoeffs,
  59         coupleIntCoeffs,
  60         interfaces
  61     ),
  62     rD(matrix.diag().size())
  63 {
  64     scalar* __restrict__ rDPtr = rD.begin();
  65     const scalar* __restrict__ DPtr = matrix_.diag().begin();
  66
  67     register label nCells = rD.size();
  68
  69     // Generate reciprocal diagonal
  70     for (register label cell=0; cell<nCells; cell++)
  71     {
  72         rDPtr[cell] = 1.0/DPtr[cell];
  73     }
  74 }
  75
  76
  77 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
  78
  79 void Foam::diagonalPreconditioner::precondition
  80 (
  81     scalarField& wA,
  82     const scalarField& rA,
  83     const direction
  84 ) const
  85 {
  86     scalar* __restrict__ wAPtr = wA.begin();
  87     const scalar* __restrict__ rAPtr = rA.begin();
  88     const scalar* __restrict__ rDPtr = rD.begin();
  89
  90     register label nCells = wA.size();
  91
  92     for (register label cell=0; cell<nCells; cell++)
  93     {
  94         wAPtr[cell] = rDPtr[cell]*rAPtr[cell];
  95     }
  96 }
  97
  98
  99 // ************************************************************************* //