Teko_TpetraReorderedMappingStrategy.cpp

// @HEADER
// *****************************************************************************
//      Teko: A package for block and physics based preconditioning
//
// Copyright 2010 NTESS and the Teko contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER
 
#include "Teko_TpetraReorderedMappingStrategy.hpp"
 
#include "Teko_BlockedReordering.hpp"
 
using Teuchos::RCP;
using Teuchos::rcp;
using Teuchos::rcp_dynamic_cast;
using Teuchos::rcpFromRef;
 
namespace Teko {
namespace TpetraHelpers {
 
TpetraReorderedMappingStrategy::TpetraReorderedMappingStrategy(
    const BlockReorderManager& brm, const Teuchos::RCP<const MappingStrategy>& map)
    : reorderManager_(brm), mapStrategy_(map) {
  rangeMap_  = mapStrategy_->rangeMap();
  domainMap_ = mapStrategy_->domainMap();
}
 
void TpetraReorderedMappingStrategy::copyTpetraIntoThyra(
    const Tpetra::MultiVector<ST, LO, GO, NT>& X,
    const Teuchos::Ptr<Thyra::MultiVectorBase<ST> >& thyra_X) const {
  using Teuchos::ptr_const_cast;
  using Teuchos::rcp_const_cast;
 
  // first flatten the vector: notice this just works on the block structure
  RCP<Thyra::ProductMultiVectorBase<ST> > prod_X =
      rcp_dynamic_cast<Thyra::ProductMultiVectorBase<ST> >(rcpFromRef(*thyra_X));
  RCP<Thyra::MultiVectorBase<ST> > flat_X = buildFlatMultiVector(reorderManager_, prod_X);
 
  // now use the underlying mapping strategy to copy the flat vector
  mapStrategy_->copyTpetraIntoThyra(X, flat_X.ptr());
}
 
void TpetraReorderedMappingStrategy::copyThyraIntoTpetra(
    const RCP<const Thyra::MultiVectorBase<ST> >& thyra_Y,
    Tpetra::MultiVector<ST, LO, GO, NT>& Y) const {
  // first flatten the vector: notice this just works on the block structure
  RCP<const Thyra::ProductMultiVectorBase<ST> > prod_Y =
      rcp_dynamic_cast<const Thyra::ProductMultiVectorBase<ST> >(rcpFromRef(*thyra_Y));
  RCP<const Thyra::MultiVectorBase<ST> > flat_Y = buildFlatMultiVector(reorderManager_, prod_Y);
 
  // now use the underlying mapping strategy to copy the flat vector
  mapStrategy_->copyThyraIntoTpetra(flat_Y, Y);
}
 
}  // end namespace TpetraHelpers
}  // end namespace Teko