Stokhos Package Browser (Single Doxygen Collection) Version of the Day
Loading...
Searching...
No Matches
Stokhos_InversePreconditioner.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_INVERSEPRECONDITIONER_HPP
46#define STOKHOS_INVERSEPRECONDITIONER_HPP
47
48#include "Teuchos_RCP.hpp"
49#include "Stokhos_Operator.hpp"
50#include "Teuchos_SerialDenseMatrix.hpp"
51#include "Teuchos_SerialDenseSolver.hpp"
52
53namespace Stokhos {
54
55 template <typename ordinal_type, typename value_type>
57 public Stokhos::Operator<ordinal_type,double> {
58 public:
59
62 const Teuchos::SerialDenseMatrix<ordinal_type,double> & 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 Teuchos::RCP< Teuchos::SerialDenseMatrix<ordinal_type, value_type> > AA, UU, RR;
72 AA = Teuchos::rcp(new Teuchos::SerialDenseMatrix<ordinal_type, value_type> (Teuchos::Copy,A));
73 UU = Teuchos::rcp(new Teuchos::SerialDenseMatrix<ordinal_type, value_type> (Teuchos::Copy,Result));
74 RR = Teuchos::rcp(new Teuchos::SerialDenseMatrix<ordinal_type, value_type> (Teuchos::Copy,Input));
75
76 // Setup solver
77 Teuchos::SerialDenseSolver<ordinal_type, value_type> solver;
78 solver.setMatrix(AA);
79 solver.setVectors(UU, RR);
80 //Solve A*Result=Input
81 if (solver.shouldEquilibrate()) {
82 solver.factorWithEquilibration(true);
83 solver.equilibrateMatrix();
84 }
85 solver.solve();
86
87 for (ordinal_type i=0; i<A.numRows(); i++)
88 Result(i,0)=(*UU)(i,0);
89
90 return 0;
91 }
92
93 protected:
94 const Teuchos::SerialDenseMatrix<ordinal_type,value_type> & A;
95 }; // class InversePreconditioner
96
97} // namespace Stokhos
98
99#endif // STOKHOS_INVERSEPRECONDITIONER_HPP
100
InversePreconditioner(const Teuchos::SerialDenseMatrix< ordinal_type, double > &A_)
Constructor.
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
Top-level namespace for Stokhos classes and functions.