src/foam/matrices/DiagonalMatrix/DiagonalMatrix.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 "DiagonalMatrix.H"
  27
  28 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  29
  30 template<class Type>
  31 template<class Form>
  32 Foam::DiagonalMatrix<Type>::DiagonalMatrix(const Matrix<Form, Type>& a)
  33 :
  34     List<Type>(min(a.n(), a.m()))
  35 {
  36     forAll(*this, i)
  37     {
  38         this->operator[](i) = a[i][i];
  39     }
  40 }
  41
  42
  43 template<class Type>
  44 Foam::DiagonalMatrix<Type>::DiagonalMatrix(const label size)
  45 :
  46     List<Type>(size)
  47 {}
  48
  49
  50 template<class Type>
  51 Foam::DiagonalMatrix<Type>::DiagonalMatrix(const label size, const Type& val)
  52 :
  53     List<Type>(size, val)
  54 {}
  55
  56
  57 template<class Type>
  58 Foam::DiagonalMatrix<Type>& Foam::DiagonalMatrix<Type>::invert()
  59 {
  60     forAll(*this, i)
  61     {
  62         Type x = this->operator[](i);
  63         if (mag(x) < VSMALL)
  64         {
  65             this->operator[](i) = Type(0);
  66         }
  67         else
  68         {
  69             this->operator[](i) = Type(1)/x;
  70         }
  71     }
  72
  73     return this;
  74 }
  75
  76
  77 template<class Type>
  78 Foam::DiagonalMatrix<Type> Foam::inv(const DiagonalMatrix<Type>& A)
  79 {
  80     DiagonalMatrix<Type> Ainv = A;
  81
  82     forAll(A, i)
  83     {
  84         Type x = A[i];
  85         if (mag(x) < VSMALL)
  86         {
  87             Ainv[i] = Type(0);
  88         }
  89         else
  90         {
  91             Ainv[i] = Type(1)/x;
  92         }
  93     }
  94
  95     return Ainv;
  96 }
  97
  98
  99 // ************************************************************************* //