Teko  Version of the Day
 All Classes Files Functions Variables Pages
Teko_ReorderedMappingStrategy.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_ReorderedMappingStrategy.hpp"
48 
49 #include "Teko_BlockedReordering.hpp"
50 
51 using Teuchos::RCP;
52 using Teuchos::rcp;
53 using Teuchos::rcp_dynamic_cast;
54 using Teuchos::rcpFromRef;
55 
56 namespace Teko {
57 namespace Epetra {
58 
59 ReorderedMappingStrategy::ReorderedMappingStrategy(const BlockReorderManager & brm,const Teuchos::RCP<const MappingStrategy> & map)
60  : reorderManager_(brm), mapStrategy_(map)
61 {
62  rangeMap_ = mapStrategy_->rangeMap();
63  domainMap_ = mapStrategy_->domainMap();
64 }
65 
66 void ReorderedMappingStrategy::copyEpetraIntoThyra(const Epetra_MultiVector& X,
67  const Teuchos::Ptr<Thyra::MultiVectorBase<double> > & thyra_X) const
68 {
69  using Teuchos::ptr_const_cast;
70  using Teuchos::rcp_const_cast;
71 
72  // first flatten the vector: notice this just works on the block structure
73  RCP<Thyra::ProductMultiVectorBase<double> > prod_X = rcp_dynamic_cast<Thyra::ProductMultiVectorBase<double> >(rcpFromRef(*thyra_X));
74  RCP<Thyra::MultiVectorBase<double> > flat_X = buildFlatMultiVector(reorderManager_,prod_X);
75 
76  // now use the underlying mapping strategy to copy the flat vector
77  mapStrategy_->copyEpetraIntoThyra(X,flat_X.ptr());
78 }
79 
80 void ReorderedMappingStrategy::copyThyraIntoEpetra(const RCP<const Thyra::MultiVectorBase<double> > & thyra_Y,
81  Epetra_MultiVector& Y) const
82 {
83  // first flatten the vector: notice this just works on the block structure
84  RCP<const Thyra::ProductMultiVectorBase<double> > prod_Y = rcp_dynamic_cast<const Thyra::ProductMultiVectorBase<double> >(rcpFromRef(*thyra_Y));
85  RCP<const Thyra::MultiVectorBase<double> > flat_Y = buildFlatMultiVector(reorderManager_,prod_Y);
86 
87  // now use the underlying mapping strategy to copy the flat vector
88  mapStrategy_->copyThyraIntoEpetra(flat_Y,Y);
89 }
90 
91 } // end namespace Epetra
92 } // end namespace Teko