Thyra_DetachedMultiVectorView.hpp

// @HEADER
// *****************************************************************************
//    Thyra: Interfaces and Support for Abstract Numerical Algorithms
//
// Copyright 2004 NTESS and the Thyra contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#include "Thyra_MultiVectorBase.hpp"
#include "Thyra_AssertOp.hpp"

#ifndef THYRA_EXPLICIT_MULTI_VECTOR_VIEW_HPP
#define THYRA_EXPLICIT_MULTI_VECTOR_VIEW_HPP

namespace Thyra {


template<class Scalar>
class ConstDetachedMultiVectorView {
public:
  ConstDetachedMultiVectorView(
    const RCP<const MultiVectorBase<Scalar> > &mv,
    const Range1D &rowRng = Range1D(), const Range1D &colRng = Range1D()
    )
    : mv_(mv) { mv_->acquireDetachedView(rowRng, colRng, &smv_); }
  ConstDetachedMultiVectorView(
    const MultiVectorBase<Scalar>& mv,
    const Range1D &rowRng = Range1D(), const Range1D &colRng = Range1D()
    )
    : mv_(rcpFromRef(mv)) { mv_->acquireDetachedView(rowRng, colRng, &smv_); }
  ~ConstDetachedMultiVectorView() { mv_->releaseDetachedView(&smv_); }
  const RTOpPack::ConstSubMultiVectorView<Scalar>& smv() const { return smv_; }
  Ordinal globalOffset() const { return smv_.globalOffset(); }
  Ordinal subDim() const { return smv_.subDim(); }
  Ordinal colOffset() const { return smv_.colOffset(); }
  Ordinal numSubCols() const { return smv_.numSubCols(); }
  const Scalar* values() const { return smv_.values().get(); }
  Ordinal leadingDim() const { return smv_.leadingDim(); }
  const Scalar& operator()(Ordinal i, Ordinal j) const { return smv_(i,j); }
private:
  RCP<const MultiVectorBase<Scalar> > mv_;
  RTOpPack::ConstSubMultiVectorView<Scalar> smv_;
  // Not defined and not to be called
  ConstDetachedMultiVectorView();
  ConstDetachedMultiVectorView(const ConstDetachedMultiVectorView<Scalar>&);
  ConstDetachedMultiVectorView<Scalar>& operator==(
    const ConstDetachedMultiVectorView<Scalar>&);
};
 

template<class Scalar>
class DetachedMultiVectorView {
public:
  DetachedMultiVectorView(
    const RCP<MultiVectorBase<Scalar> >& mv,
    const Range1D &rowRng = Range1D(), const Range1D &colRng = Range1D()
    )
    : mv_(mv) { mv_->acquireDetachedView(rowRng, colRng, &smv_); }
  DetachedMultiVectorView(
    MultiVectorBase<Scalar>& mv,
    const Range1D &rowRng = Range1D(), const Range1D &colRng = Range1D()
    )
    : mv_(rcpFromRef(mv)) { mv_->acquireDetachedView(rowRng,colRng,&smv_); }
  ~DetachedMultiVectorView() { mv_->commitDetachedView(&smv_); }
  const RTOpPack::SubMultiVectorView<Scalar>& smv() const { return smv_; }
  Ordinal globalOffset() const { return smv_.globalOffset(); }
  Ordinal subDim() const { return smv_.subDim(); }
  Ordinal colOffset() const { return smv_.colOffset(); }
  Ordinal numSubCols() const { return smv_.numSubCols(); }
  Scalar* values() const { return smv_.values().get(); }
  Ordinal leadingDim() const { return smv_.leadingDim(); }
  Scalar& operator()(Ordinal i, Ordinal j) { return smv_(i,j); }
private:
  RCP<MultiVectorBase<Scalar> > mv_;
  RTOpPack::SubMultiVectorView<Scalar> smv_;
  // Not defined and not to be called
  DetachedMultiVectorView();
  DetachedMultiVectorView(const DetachedMultiVectorView<Scalar>&);
  DetachedMultiVectorView<Scalar>& operator==(
    const DetachedMultiVectorView<Scalar>&);
};


template<class Scalar>
void doExplicitMultiVectorAdjoint(
  const MultiVectorBase<Scalar>& mvIn, MultiVectorBase<Scalar>* mvTransOut
  )
{
  typedef Teuchos::ScalarTraits<Scalar> ST;
#ifdef TEUCHOS_DEBUG
  TEUCHOS_TEST_FOR_EXCEPT(0==mvTransOut);
  THYRA_ASSERT_VEC_SPACES("doExplicitMultiVectorAdjoint(...)",
    *mvIn.domain(), *mvTransOut->range()
    );
  THYRA_ASSERT_VEC_SPACES("doExplicitMultiVectorAdjoint(...)",
    *mvIn.range(), *mvTransOut->domain()
    );
#endif
  ConstDetachedMultiVectorView<Scalar> dMvIn(mvIn);
  DetachedMultiVectorView<Scalar> dMvTransOut(*mvTransOut);
  const int m = dMvIn.subDim();
  const int n = dMvIn.numSubCols();
  for ( int j = 0; j < n; ++j ) {
    for ( int i = 0; i < m; ++i ) {
      dMvTransOut(j,i) = ST::conjugate(dMvIn(i,j));
    }
  }
}


} // namespace Thyra


#endif // THYRA_EXPLICIT_MULTI_VECTOR_VIEW_HPP