Teko  Version of the Day
 All Classes Files Functions Variables Pages
Teko_BasicMappingStrategy.cpp
1 /*
2 // @HEADER
3 //
4 // ***********************************************************************
5 //
6 // Teko: A package for block and physics based preconditioning
7 // Copyright 2010 Sandia Corporation
8 //
9 // Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
10 // the U.S. Government retains certain rights in this software.
11 //
12 // Redistribution and use in source and binary forms, with or without
13 // modification, are permitted provided that the following conditions are
14 // met:
15 //
16 // 1. Redistributions of source code must retain the above copyright
17 // notice, this list of conditions and the following disclaimer.
18 //
19 // 2. Redistributions in binary form must reproduce the above copyright
20 // notice, this list of conditions and the following disclaimer in the
21 // documentation and/or other materials provided with the distribution.
22 //
23 // 3. Neither the name of the Corporation nor the names of the
24 // contributors may be used to endorse or promote products derived from
25 // this software without specific prior written permission.
26 //
27 // THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
28 // EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
29 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
30 // PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
31 // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
32 // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
33 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
34 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
35 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
36 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
37 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
38 //
39 // Questions? Contact Eric C. Cyr (eccyr@sandia.gov)
40 //
41 // ***********************************************************************
42 //
43 // @HEADER
44 
45 */
46 
47 #include "Epetra/Teko_BasicMappingStrategy.hpp"
48 #include "Epetra/Teko_EpetraHelpers.hpp"
49 
50 #include "Thyra_EpetraThyraWrappers.hpp"
51 #include "Thyra_DefaultSpmdMultiVector.hpp"
52 
53 using Teuchos::RCP;
54 using Teuchos::rcp;
55 using Teuchos::rcp_dynamic_cast;
56 
57 namespace Teko {
58 namespace Epetra {
59 
60 // Creates a strided mapping strategy. This class is useful
61 // for breaking up nodally ordered matrices (i.e. the unknowns
62 // in a FEM problem are ordered [u0,v0,p0,u1,v1,p1,...]). Current
63 // implimentation only supports a fixed number of variables
64 //
65 // arguments:
66 // vars - Number of different variables
67 // map - original Epetra_Map to be broken up
68 // comm - Epetra_Comm object related to the map
69 //
70 BasicMappingStrategy::BasicMappingStrategy(const Teuchos::RCP<const Epetra_Map> & rMap,
71  const Teuchos::RCP<const Epetra_Map> & dMap, const Epetra_Comm & /* comm */)
72 {
73  rangeMap_ = rMap;
74  domainMap_ = dMap;
75 }
76 
77 // Virtual function defined in MappingStrategy. This copies
78 // an Epetra_MultiVector into a Thyra::MultiVectorBase with
79 // blocking handled by the strides defined in the constructor.
80 //
81 // arguments:
82 // X - source Epetra_MultiVector
83 // thyra_X - destination Thyra::MultiVectorBase
84 //
85 void BasicMappingStrategy::copyEpetraIntoThyra(const Epetra_MultiVector& X,
86  const Teuchos::Ptr<Thyra::MultiVectorBase<double> > & thyra_X) const
87 {
88  // perform a simple copy
89  RCP<Thyra::DefaultSpmdMultiVector<double> > vec
90  = rcp_dynamic_cast<Thyra::DefaultSpmdMultiVector<double> >(Teuchos::rcpFromRef(*thyra_X));
91  Teuchos::RCP<Epetra_MultiVector> ptrX = Teuchos::rcp_const_cast<Epetra_MultiVector>(Teuchos::rcpFromRef(X));
92  fillDefaultSpmdMultiVector(vec,ptrX);
93 }
94 
95 // Virtual function defined in MappingStrategy. This copies
96 // an Epetra_MultiVector into a Thyra::MultiVectorBase with
97 // blocking handled by the strides defined in the constructor.
98 //
99 // arguments:
100 // thyra_Y - source Thyra::MultiVectorBase
101 // Y - destination Epetra_MultiVector
102 //
103 void BasicMappingStrategy::copyThyraIntoEpetra(const RCP<const Thyra::MultiVectorBase<double> > & thyra_Y,
104  Epetra_MultiVector& Y) const
105 {
106  RCP<const Epetra_MultiVector> eSrc = Thyra::get_Epetra_MultiVector(*rangeMap(),thyra_Y);
107 
108  Y = *eSrc;
109 }
110 
111 } // end namespace Epetra
112 } // end namespace Teko