Stokhos Package Browser (Single Doxygen Collection) Version of the Day
Loading...
Searching...
No Matches
Stokhos_JacobiPreconditioner.hpp
Go to the documentation of this file.
1/*
2// @HEADER
3// ***********************************************************************
4//
5// Stokhos Package
6// Copyright (2009) Sandia Corporation
7//
8// Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
9// license for use of this work by or on behalf of the U.S. Government.
10//
11// Redistribution and use in source and binary forms, with or without
12// modification, are permitted provided that the following conditions are
13// met:
14//
15// 1. Redistributions of source code must retain the above copyright
16// notice, this list of conditions and the following disclaimer.
17//
18// 2. Redistributions in binary form must reproduce the above copyright
19// notice, this list of conditions and the following disclaimer in the
20// documentation and/or other materials provided with the distribution.
21//
22// 3. Neither the name of the Corporation nor the names of the
23// contributors may be used to endorse or promote products derived from
24// this software without specific prior written permission.
25//
26// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
27// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
28// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
29// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
30// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
31// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
32// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
33// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
34// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
35// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
36// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
37//
38// Questions? Contact Eric T. Phipps (etphipp@sandia.gov).
39//
40// ***********************************************************************
41// @HEADER
42*/
43
44
45#ifndef STOKHOS_JACOBIPRECONDITIONER_HPP
46#define STOKHOS_JACOBIPRECONDITIONER_HPP
47
48#include "Teuchos_RCP.hpp"
49#include "Stokhos_Operator.hpp"
50#include "Teuchos_SerialDenseMatrix.hpp"
51
52namespace Stokhos {
53
54 template <typename ordinal_type, typename value_type>
56 public Stokhos::Operator<ordinal_type,value_type> {
57
58 public:
59
62 const Teuchos::SerialDenseMatrix<ordinal_type, value_type>& A_) : A(A_) {}
63
66
67 virtual ordinal_type ApplyInverse(
68 const Teuchos::SerialDenseMatrix<ordinal_type, value_type>& Input,
69 Teuchos::SerialDenseMatrix<ordinal_type, value_type>& Result,
70 ordinal_type m) const {
71 ordinal_type n=Input.numRows();
72 Teuchos::SerialDenseMatrix<ordinal_type, value_type> G(Teuchos::Copy,A);
73 Teuchos::SerialDenseMatrix<ordinal_type, value_type> z(n,1);
74 for (ordinal_type j=0; j<m; j++){
75 if (j==0){ // Compute z=D-1r
76 for (ordinal_type i=0; i<n; i++)
77 z(i,0)=Input(i,0)/A(i,i);
78 }
79 else {
80 //Compute G=invD(-L-U)=I-inv(D)A
81 for (ordinal_type i=0; i<n; i++){
82 for (ordinal_type j=0; j<n; j++){
83 if (j==i)
84 G(i,j)=0;
85 else
86 G(i,j)=-A(i,j)/A(i,i);
87 }
88 }
89
90 Result.assign(z);
91 //z=Gz+inv(D)r
92 Result.multiply(Teuchos::NO_TRANS,Teuchos::NO_TRANS,1.0, G, z, 1.0);
93
94 }
95 }
96
97 return 0;
98 }
99
100 protected:
101 const Teuchos::SerialDenseMatrix<ordinal_type, value_type>& A;
102 }; // class JacobiPreconditioner
103
104} // namespace Stokhos
105
106#endif // STOKHOS_JACOBIPRECONDITIONER_HPP
107
const Teuchos::SerialDenseMatrix< ordinal_type, value_type > & A
virtual ordinal_type ApplyInverse(const Teuchos::SerialDenseMatrix< ordinal_type, value_type > &Input, Teuchos::SerialDenseMatrix< ordinal_type, value_type > &Result, ordinal_type m) const
Returns the result of a Operator inverse applied to a Teuchos::SerialDenseMatrix Input in Result.
JacobiPreconditioner(const Teuchos::SerialDenseMatrix< ordinal_type, value_type > &A_)
Constructor.
Top-level namespace for Stokhos classes and functions.