src/OpenFOAM/matrices/DiagonalMatrix/DiagonalMatrix.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright held by original author
   6      \\/     M anipulation  |
   7 -------------------------------------------------------------------------------
   8 License
   9     This file is part of OpenFOAM.
  10
  11     OpenFOAM 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 2 of the License, or (at your
  14     option) any later version.
  15
  16     OpenFOAM is distributed in the hope that it will be useful, but WITHOUT
  17     ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  18     FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  19     for more details.
  20
  21     You should have received a copy of the GNU General Public License
  22     along with OpenFOAM; if not, write to the Free Software Foundation,
  23     Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
  24
  25 \*---------------------------------------------------------------------------*/
  26
  27 #include "DiagonalMatrix.H"
  28
  29 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  30
  31 template<class Type>
  32 template<class Form>
  33 Foam::DiagonalMatrix<Type>::DiagonalMatrix(const Matrix<Form, Type>& a)
  34 :
  35     List<Type>(min(a.n(), a.m()))
  36 {
  37     forAll(*this, i)
  38     {
  39         this->operator[](i) = a[i][i];
  40     }
  41 }
  42
  43
  44 template<class Type>
  45 Foam::DiagonalMatrix<Type>::DiagonalMatrix(const label size)
  46 :
  47     List<Type>(size)
  48 {}
  49
  50
  51 template<class Type>
  52 Foam::DiagonalMatrix<Type>::DiagonalMatrix(const label size, const Type& val)
  53 :
  54     List<Type>(size, val)
  55 {}
  56
  57
  58 template<class Type>
  59 Foam::DiagonalMatrix<Type>& Foam::DiagonalMatrix<Type>::invert()
  60 {
  61     forAll(*this, i)
  62     {
  63         Type x = this->operator[](i);
  64         if (mag(x) < VSMALL)
  65         {
  66             this->operator[](i) = Type(0);
  67         }
  68         else
  69         {
  70             this->operator[](i) = Type(1)/x;
  71         }
  72     }
  73
  74     return this;
  75 }
  76
  77
  78 template<class Type>
  79 Foam::DiagonalMatrix<Type> Foam::inv(const DiagonalMatrix<Type>& A)
  80 {
  81     DiagonalMatrix<Type> Ainv = A;
  82
  83     forAll(A, i)
  84     {
  85         Type x = A[i];
  86         if (mag(x) < VSMALL)
  87         {
  88             Ainv[i] = Type(0);
  89         }
  90         else
  91         {
  92             Ainv[i] = Type(1)/x;
  93         }
  94     }
  95
  96     return Ainv;
  97 }
  98
  99
 100 // ************************************************************************* //