Teko_ReorderedMappingStrategy.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 "Epetra/Teko_ReorderedMappingStrategy.hpp"
 
#include "Teko_BlockedReordering.hpp"
 
using Teuchos::RCP;
using Teuchos::rcp;
using Teuchos::rcp_dynamic_cast;
using Teuchos::rcpFromRef;
 
namespace Teko {
namespace Epetra {
 
ReorderedMappingStrategy::ReorderedMappingStrategy(const BlockReorderManager& brm,
                                                   const Teuchos::RCP<const MappingStrategy>& map)
    : reorderManager_(brm), mapStrategy_(map) {
  rangeMap_  = mapStrategy_->rangeMap();
  domainMap_ = mapStrategy_->domainMap();
}
 
void ReorderedMappingStrategy::copyEpetraIntoThyra(
    const Epetra_MultiVector& X,
    const Teuchos::Ptr<Thyra::MultiVectorBase<double> >& 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<double> > prod_X =
      rcp_dynamic_cast<Thyra::ProductMultiVectorBase<double> >(rcpFromRef(*thyra_X));
  RCP<Thyra::MultiVectorBase<double> > flat_X = buildFlatMultiVector(reorderManager_, prod_X);
 
  // now use the underlying mapping strategy to copy the flat vector
  mapStrategy_->copyEpetraIntoThyra(X, flat_X.ptr());
}
 
void ReorderedMappingStrategy::copyThyraIntoEpetra(
    const RCP<const Thyra::MultiVectorBase<double> >& thyra_Y, Epetra_MultiVector& Y) const {
  // first flatten the vector: notice this just works on the block structure
  RCP<const Thyra::ProductMultiVectorBase<double> > prod_Y =
      rcp_dynamic_cast<const Thyra::ProductMultiVectorBase<double> >(rcpFromRef(*thyra_Y));
  RCP<const Thyra::MultiVectorBase<double> > flat_Y = buildFlatMultiVector(reorderManager_, prod_Y);
 
  // now use the underlying mapping strategy to copy the flat vector
  mapStrategy_->copyThyraIntoEpetra(flat_Y, Y);
}
 
}  // end namespace Epetra
}  // end namespace Teko