Teko Version of the Day
Loading...
Searching...
No Matches
Teko_TpetraReorderedMappingStrategy.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 "Teko_TpetraReorderedMappingStrategy.hpp"
48
49#include "Teko_BlockedReordering.hpp"
50
51using Teuchos::RCP;
52using Teuchos::rcp;
53using Teuchos::rcp_dynamic_cast;
54using Teuchos::rcpFromRef;
55
56namespace Teko {
57namespace TpetraHelpers {
58
59TpetraReorderedMappingStrategy::TpetraReorderedMappingStrategy(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
66void TpetraReorderedMappingStrategy::copyTpetraIntoThyra(const Tpetra::MultiVector<ST,LO,GO,NT> & X,
67 const Teuchos::Ptr<Thyra::MultiVectorBase<ST> > & 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<ST> > prod_X = rcp_dynamic_cast<Thyra::ProductMultiVectorBase<ST> >(rcpFromRef(*thyra_X));
74 RCP<Thyra::MultiVectorBase<ST> > flat_X = buildFlatMultiVector(reorderManager_,prod_X);
75
76 // now use the underlying mapping strategy to copy the flat vector
77 mapStrategy_->copyTpetraIntoThyra(X,flat_X.ptr());
78}
79
80void TpetraReorderedMappingStrategy::copyThyraIntoTpetra(const RCP<const Thyra::MultiVectorBase<ST> > & thyra_Y,
81 Tpetra::MultiVector<ST,LO,GO,NT> & Y) const
82{
83 // first flatten the vector: notice this just works on the block structure
84 RCP<const Thyra::ProductMultiVectorBase<ST> > prod_Y = rcp_dynamic_cast<const Thyra::ProductMultiVectorBase<ST> >(rcpFromRef(*thyra_Y));
85 RCP<const Thyra::MultiVectorBase<ST> > flat_Y = buildFlatMultiVector(reorderManager_,prod_Y);
86
87 // now use the underlying mapping strategy to copy the flat vector
88 mapStrategy_->copyThyraIntoTpetra(flat_Y,Y);
89}
90
91} // end namespace TpetraHelpers
92} // end namespace Teko