EpetraExt Development
Loading...
Searching...
No Matches
EpetraExt_RestrictedCrsMatrixWrapper.h
Go to the documentation of this file.
1//@HEADER
2// ***********************************************************************
3//
4// EpetraExt: Epetra Extended - Linear Algebra Services Package
5// Copyright (2011) Sandia Corporation
6//
7// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
8// the U.S. Government retains certain rights in this software.
9//
10// Redistribution and use in source and binary forms, with or without
11// modification, are permitted provided that the following conditions are
12// met:
13//
14// 1. Redistributions of source code must retain the above copyright
15// notice, this list of conditions and the following disclaimer.
16//
17// 2. Redistributions in binary form must reproduce the above copyright
18// notice, this list of conditions and the following disclaimer in the
19// documentation and/or other materials provided with the distribution.
20//
21// 3. Neither the name of the Corporation nor the names of the
22// contributors may be used to endorse or promote products derived from
23// this software without specific prior written permission.
24//
25// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
26// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
28// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
29// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
30// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
31// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
32// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
33// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
34// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
35// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
36//
37// Questions? Contact Michael A. Heroux (maherou@sandia.gov)
38//
39// ***********************************************************************
40//@HEADER
41
42#ifndef EPETRAEXT_RESTRICTEDCRSMATRIXWRAPPER_H
43#define EPETRAEXT_RESTRICTEDCRSMATRIXWRAPPER_H
45#include <Teuchos_RCP.hpp>
46#include <mpi.h>
47
48#ifdef HAVE_MPI
49class Epetra_Comm;
50class Epetra_MpiComm;
51class Epetra_Map;
53
54namespace EpetraExt {
55class RestrictedCrsMatrixWrapper{
56
57public:
58 RestrictedCrsMatrixWrapper();
59
60 ~RestrictedCrsMatrixWrapper();
61
63
67 int SetMPISubComm(MPI_Comm MPI_SubComm);
68
70 MPI_Comm GetMPISubComm(){return MPI_SubComm_;}
71
73 const Epetra_MpiComm & RestrictedComm(){return *RestrictedComm_;}
74
76 bool RestrictedProcIsActive(){return proc_is_active;}
77
79 Teuchos::RCP<Epetra_CrsMatrix> InputMatrix(){return input_matrix_;}
80
82 Teuchos::RCP<Epetra_CrsMatrix> RestrictedMatrix(){return restricted_matrix_;}
83
85
94 int restrict_comm(Teuchos::RCP<Epetra_CrsMatrix> input_matrix);
95
96private:
97 bool proc_is_active;
98 bool subcomm_is_set;
99
100 MPI_Comm MPI_SubComm_;
101 Epetra_MpiComm *RestrictedComm_;
102 Epetra_Map *ResRowMap_;
103 Epetra_Map *ResColMap_;
104
105 Teuchos::RCP<Epetra_CrsMatrix> input_matrix_;
106 Teuchos::RCP<Epetra_CrsMatrix> restricted_matrix_;
107
108 template<typename int_type>
109 int Trestrict_comm(Teuchos::RCP<Epetra_CrsMatrix> input_matrix);
110};
111
112
113} // namespace EpetraExt
114
115
116#endif
117#endif /* EPETRAEXT_RESTRICTEDCRSMATRIXWRAPPER_H */
EpetraExt::BlockCrsMatrix: A class for constructing a distributed block matrix.