src/foam/matrices/lduMatrix/smoothers/DILU/DILUSmoother.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 "DILUSmoother.H"
  27 #include "DILUPreconditioner.H"
  28
  29 // * * * * * * * * * * * * * * Static Data Members * * * * * * * * * * * * * //
  30
  31 namespace Foam
  32 {
  33     defineTypeNameAndDebug(DILUSmoother, 0);
  34
  35     lduSmoother::addasymMatrixConstructorToTable<DILUSmoother>
  36         addDILUSmootherAsymMatrixConstructorToTable_;
  37 }
  38
  39
  40 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
  41
  42 Foam::DILUSmoother::DILUSmoother
  43 (
  44     const lduMatrix& matrix,
  45     const FieldField<Field, scalar>& coupleBouCoeffs,
  46     const FieldField<Field, scalar>& coupleIntCoeffs,
  47     const lduInterfaceFieldPtrsList& interfaces
  48 )
  49 :
  50     lduSmoother
  51     (
  52         matrix,
  53         coupleBouCoeffs,
  54         coupleIntCoeffs,
  55         interfaces
  56     ),
  57     rD_(matrix_.diag())
  58 {
  59     DILUPreconditioner::calcReciprocalD(rD_, matrix_);
  60 }
  61
  62
  63 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
  64
  65 void Foam::DILUSmoother::smooth
  66 (
  67     scalarField& psi,
  68     const scalarField& source,
  69     const direction cmpt,
  70     const label nSweeps
  71 ) const
  72 {
  73     const scalar* const __restrict__ rDPtr = rD_.begin();
  74
  75     const label* const __restrict__ uPtr =
  76         matrix_.lduAddr().upperAddr().begin();
  77     const label* const __restrict__ lPtr =
  78         matrix_.lduAddr().lowerAddr().begin();
  79
  80     const scalar* const __restrict__ upperPtr = matrix_.upper().begin();
  81     const scalar* const __restrict__ lowerPtr = matrix_.lower().begin();
  82
  83     // Temporary storage for the residual
  84     scalarField rA(rD_.size());
  85     scalar* __restrict__ rAPtr = rA.begin();
  86
  87     for (label sweep=0; sweep<nSweeps; sweep++)
  88     {
  89         matrix_.residual
  90         (
  91             rA,
  92             psi,
  93             source,
  94             coupleBouCoeffs_,
  95             interfaces_,
  96             cmpt
  97         );
  98
  99         rA *= rD_;
 100
 101         register label nFaces = matrix_.upper().size();
 102         for (register label face=0; face<nFaces; face++)
 103         {
 104             register label u = uPtr[face];
 105             rAPtr[u] -= rDPtr[u]*lowerPtr[face]*rAPtr[lPtr[face]];
 106         }
 107
 108         register label nFacesM1 = nFaces - 1;
 109         for (register label face=nFacesM1; face>=0; face--)
 110         {
 111             register label l = lPtr[face];
 112             rAPtr[l] -= rDPtr[l]*upperPtr[face]*rAPtr[uPtr[face]];
 113         }
 114
 115         psi += rA;
 116     }
 117 }
 118
 119
 120 // ************************************************************************* //