Thyra_MultiVectorDefaultBase.hpp

// @HEADER
// ***********************************************************************
// 
//    Thyra: Interfaces and Support for Abstract Numerical Algorithms
//                 Copyright (2004) Sandia Corporation
// 
// Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
// license for use of this work by or on behalf of the U.S. Government.
// 
// This library is free software; you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as
// published by the Free Software Foundation; either version 2.1 of the
// License, or (at your option) any later version.
//  
// This library is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
//  
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
// USA
// Questions? Contact Michael A. Heroux (maherou@sandia.gov) 
// 
// ***********************************************************************
// @HEADER

#ifndef THYRA_MULTI_VECTOR_DEFAULT_BASE_HPP
#define THYRA_MULTI_VECTOR_DEFAULT_BASE_HPP


#include "Thyra_MultiVectorDefaultBaseDecl.hpp"
#include "Thyra_LinearOpDefaultBase.hpp"
#include "Thyra_MultiVectorBase.hpp"
#include "Thyra_MultiVectorStdOps.hpp"
#include "Thyra_VectorSpaceBase.hpp"
#include "Thyra_VectorBase.hpp"
#include "Thyra_AssertOp.hpp"
#include "Thyra_DefaultColumnwiseMultiVector.hpp"
#include "Teuchos_Workspace.hpp"
#include "Teuchos_TestForException.hpp"


namespace Thyra {


// Cloning


template<class Scalar>
Teuchos::RCP<MultiVectorBase<Scalar> >
MultiVectorDefaultBase<Scalar>::clone_mv() const
{
  const VectorSpaceBase<Scalar>
    &domain = *this->domain(),
    &range  = *this->range();
  Teuchos::RCP<MultiVectorBase<Scalar> >
    copy = createMembers(range,domain.dim());
  assign( &*copy, *this );
  return copy;
}


// Collective applyOp() methods


template<class Scalar>
void MultiVectorDefaultBase<Scalar>::mvMultiReductApplyOpImpl(
  const RTOpPack::RTOpT<Scalar>         &prim_op
  ,const int                            num_multi_vecs
  ,const MultiVectorBase<Scalar>*const  multi_vecs[]
  ,const int                            num_targ_multi_vecs
  ,MultiVectorBase<Scalar>*const        targ_multi_vecs[]
  ,RTOpPack::ReductTarget*const         reduct_objs[]
  ,const Index                          prim_first_ele_offset_in
  ,const Index                          prim_sub_dim_in
  ,const Index                          prim_global_offset_in
  ,const Index                          sec_first_ele_offset_in
  ,const Index                          sec_sub_dim_in
  ) const
{
  using Teuchos::Workspace;
  Teuchos::WorkspaceStore* wss = Teuchos::get_default_workspace_store().get();

  // ToDo: Validate the input!

  const VectorSpaceBase<Scalar>  &domain = *this->domain();

  // Get the primary and secondary dimensions.

  const Index sec_dim = domain.dim();
  const Index sec_sub_dim  = ( sec_sub_dim_in >= 0 ? sec_sub_dim_in : sec_dim - sec_first_ele_offset_in  );
#ifdef TEUCHOS_DEBUG
  const VectorSpaceBase<Scalar>  &range  = *this->range();
  const Index prim_dim = range.dim();
  const Index prim_sub_dim = ( prim_sub_dim_in >= 0 ? prim_sub_dim_in : prim_dim - prim_first_ele_offset_in );
  const char err_msg[] = "MultiVectorDefaultBase<Scalar>::mvMultiReductApplyOpImpl(...): Error!";
  TEST_FOR_EXCEPTION( !(0 < prim_sub_dim && prim_sub_dim <= prim_dim), std::invalid_argument, err_msg );
  TEST_FOR_EXCEPTION( !(0 < sec_sub_dim  && sec_sub_dim  <= sec_dim),  std::invalid_argument, err_msg );
#endif

  //
  // Apply the reduction/transformation operator and transform the
  // target vectors and reduce each of the reduction objects.
  //

  Workspace< Teuchos::RCP<const VectorBase<Scalar> > >   vecs_s(wss,num_multi_vecs);
  Workspace<const VectorBase<Scalar>*>                           vecs(wss,num_multi_vecs,false);
  Workspace< Teuchos::RCP<VectorBase<Scalar> > >         targ_vecs_s(wss,num_targ_multi_vecs);
  Workspace<VectorBase<Scalar>*>                                 targ_vecs(wss,num_targ_multi_vecs,false);

  for(Index j = sec_first_ele_offset_in; j < sec_first_ele_offset_in + sec_sub_dim; ++j) {
    // Fill the arrays of vector arguments
    {for(Index k = 0; k < static_cast<Index>(num_multi_vecs); ++k) {
      vecs_s[k] = multi_vecs[k]->col(j);
      vecs[k] = vecs_s[k].get();
    }}
    {for(Index k = 0; k < static_cast<Index>(num_targ_multi_vecs); ++k) {
      targ_vecs_s[k] = targ_multi_vecs[k]->col(j);
      targ_vecs[k] = targ_vecs_s[k].get();
    }}
    // Apply the reduction/transformation operator
    Thyra::applyOp(
      prim_op
      ,num_multi_vecs,      (num_multi_vecs      ? &vecs[0]      : NULL)
      ,num_targ_multi_vecs, (num_targ_multi_vecs ? &targ_vecs[0] : NULL)
      ,reduct_objs ? reduct_objs[j] : NULL
      ,prim_first_ele_offset_in, prim_sub_dim_in, prim_global_offset_in
      );
  }
  // At this point all of the designated targ vectors in the target multi-vectors have
  // been transformed and all the reduction objects in reduct_obj[] have accumulated
  // the reductions.
}


template<class Scalar>
void MultiVectorDefaultBase<Scalar>::mvSingleReductApplyOpImpl(
  const RTOpPack::RTOpT<Scalar>         &prim_op
  ,const RTOpPack::RTOpT<Scalar>        &sec_op
  ,const int                            num_multi_vecs
  ,const MultiVectorBase<Scalar>*const  multi_vecs[]
  ,const int                            num_targ_multi_vecs
  ,MultiVectorBase<Scalar>*const        targ_multi_vecs[]
  ,RTOpPack::ReductTarget               *reduct_obj
  ,const Index                          prim_first_ele_offset_in
  ,const Index                          prim_sub_dim_in
  ,const Index                          prim_global_offset_in
  ,const Index                          sec_first_ele_offset_in
  ,const Index                          sec_sub_dim_in
  ) const
{
  using Teuchos::Workspace;
  Teuchos::WorkspaceStore* wss = Teuchos::get_default_workspace_store().get();

  // ToDo: Validate the input!

  const VectorSpaceBase<Scalar> &domain = *this->domain();

  // Get the primary and secondary dimensions.
  const Index sec_dim = domain.dim();
  const Index sec_sub_dim  = ( sec_sub_dim_in >= 0 ? sec_sub_dim_in : sec_dim - sec_first_ele_offset_in );
#ifdef TEUCHOS_DEBUG
  const VectorSpaceBase<Scalar> &range = *this->range();
  const Index prim_dim = range.dim();
  const Index prim_sub_dim = ( prim_sub_dim_in >= 0 ? prim_sub_dim_in : prim_dim - prim_first_ele_offset_in );
  const char err_msg[] = "MultiVectorDefaultBase<Scalar>::mvSingleReductApplyOpImpl(...): Error!";
  TEST_FOR_EXCEPTION( !(0 < prim_sub_dim && prim_sub_dim <= prim_dim), std::invalid_argument, err_msg );
  TEST_FOR_EXCEPTION( !(0 < sec_sub_dim  && sec_sub_dim  <= sec_dim),  std::invalid_argument, err_msg );
#endif

  // Create a temporary buffer for the reduction objects of the primary reduction
  // so that we can call the companion version of this method.
  Workspace<Teuchos::RCP<RTOpPack::ReductTarget> >
    rcp_reduct_objs(wss,reduct_obj!=NULL?sec_sub_dim:0);
  Workspace<RTOpPack::ReductTarget*>
    reduct_objs(wss,reduct_obj!=NULL?sec_sub_dim:0,false);
  if(reduct_obj) {
    for(Index k = 0; k < sec_sub_dim; ++k) {
      rcp_reduct_objs[k] = prim_op.reduct_obj_create();
      reduct_objs[k] = &*rcp_reduct_objs[k];
    }
  }
  
  // Call the companion version that accepts an array of reduction objects
  this->applyOp(
    prim_op
    ,num_multi_vecs,       multi_vecs
    ,num_targ_multi_vecs,  targ_multi_vecs
    ,reduct_obj ? &reduct_objs[0] : NULL
    ,prim_first_ele_offset_in, prim_sub_dim_in, prim_global_offset_in
    ,sec_first_ele_offset_in,  sec_sub_dim_in
    );

  // Reduce all the reduction objects using the secondary reduction operator
  // into one reduction object and free the intermediate reduction objects.
  if(reduct_obj) {
    for(Index k = 0; k < sec_sub_dim; ++k) {
      sec_op.reduce_reduct_objs( *reduct_objs[k], reduct_obj );
    }
  }
}


// Explicit sub-multi-vector access


template<class Scalar>
void MultiVectorDefaultBase<Scalar>::acquireDetachedMultiVectorViewImpl(
  const Range1D &rowRng_in,
  const Range1D &colRng_in,
  RTOpPack::ConstSubMultiVectorView<Scalar> *sub_mv
  ) const
{
  const Index
    rangeDim  = this->range()->dim(),
    domainDim = this->domain()->dim();
  const Range1D
    rowRng = rowRng_in.full_range() ? Range1D(0,rangeDim-1)  : rowRng_in,
    colRng = colRng_in.full_range() ? Range1D(0,domainDim-1) : colRng_in;
#ifdef TEUCHOS_DEBUG
  TEST_FOR_EXCEPTION(
    !(rowRng.ubound() < rangeDim), std::out_of_range
    ,"MultiVectorDefaultBase<Scalar>::acquireDetachedMultiVectorViewImpl(...): Error, rowRng = ["
    <<rowRng.lbound()<<","<<rowRng.ubound()<<"] is not in the range = [0,"
    <<(rangeDim-1)<<"]!"
    );
  TEST_FOR_EXCEPTION(
    !(colRng.ubound() < domainDim), std::out_of_range
    ,"MultiVectorDefaultBase<Scalar>::acquireDetachedMultiVectorViewImpl(...): Error, colRng = ["
    <<colRng.lbound()<<","<<colRng.ubound()<<"] is not in the range = [0,"
    <<(domainDim-1)<<"]!"
    );
#endif
  // Allocate storage for the multi-vector (stored column-major)
  Scalar *values = new Scalar[ rowRng.size() * colRng.size() ];
  // Extract multi-vector values column by column
  RTOpPack::ConstSubVectorView<Scalar> sv; // uninitialized by default
  for( int k = colRng.lbound(); k <= colRng.ubound(); ++k ) {
    Teuchos::RCP<const VectorBase<Scalar> > col_k = this->col(k);
    col_k->acquireDetachedView( rowRng, &sv );
    for( int i = 0; i < rowRng.size(); ++i )
      values[ i + k*rowRng.size() ] = sv[i];
    col_k->releaseDetachedView( &sv );
  }
  // Initialize the multi-vector view object
  sub_mv->initialize(
    rowRng.lbound()              // globalOffset
    ,rowRng.size()               // subDim
    ,colRng.lbound()             // colOffset
    ,colRng.size()               // numSubCols
    ,values                      // values
    ,rowRng.size()               // leadingDim
    );
}


template<class Scalar>
void MultiVectorDefaultBase<Scalar>::releaseDetachedMultiVectorViewImpl(
  RTOpPack::ConstSubMultiVectorView<Scalar>* sub_mv
  ) const
{
  // Here we just need to free the view and that is it!
  delete [] const_cast<Scalar*>(sub_mv->values());
  sub_mv->set_uninitialized();
}


template<class Scalar>
void MultiVectorDefaultBase<Scalar>::acquireNonconstDetachedMultiVectorViewImpl(
  const Range1D &rowRng,
  const Range1D &colRng,
  RTOpPack::SubMultiVectorView<Scalar> *sub_mv
  )
{
  // Use the non-const implementation since it does exactly the
  // correct thing in this case also!
  MultiVectorDefaultBase<Scalar>::acquireDetachedMultiVectorViewImpl(
    rowRng, colRng,
    static_cast<RTOpPack::ConstSubMultiVectorView<Scalar>*>(sub_mv)
    // This cast will work as long as SubMultiVectorView
    // maintains no extra state over ConstSubMultiVectorView (which it
    // currently does not) but this is something that I should
    // technically check for some how.
    );
}


template<class Scalar>
void MultiVectorDefaultBase<Scalar>::commitNonconstDetachedMultiVectorViewImpl(
  RTOpPack::SubMultiVectorView<Scalar>* sub_mv
  )
{
#ifdef TEUCHOS_DEBUG
  TEST_FOR_EXCEPTION(
    sub_mv==NULL, std::logic_error,
    "MultiVectorDefaultBase<Scalar>::commitNonconstDetachedMultiVectorViewImpl(...): Error!"
    );
#endif
  // Set back the multi-vector values column by column
  const Range1D rowRng(sub_mv->globalOffset(),sub_mv->globalOffset()+sub_mv->subDim()-1);
  RTOpPack::SubVectorView<Scalar> msv; // uninitialized by default
  for( int k = sub_mv->colOffset(); k < sub_mv->numSubCols(); ++k ) {
    Teuchos::RCP<VectorBase<Scalar> > col_k = this->col(k);
    col_k->acquireDetachedView( rowRng, &msv );
    for( int i = 0; i < rowRng.size(); ++i )
      msv[i] = sub_mv->values()[ i + k*rowRng.size() ];
    col_k->commitDetachedView( &msv );
  }
  // Free the memory
  delete [] const_cast<Scalar*>(sub_mv->values());
  // Zero out the view
  sub_mv->set_uninitialized();
}


// Sub-view methods


template<class Scalar>
Teuchos::RCP<const MultiVectorBase<Scalar> >
MultiVectorDefaultBase<Scalar>::subView( const Range1D& colRng_in ) const
{
  using Teuchos::Workspace;
  Teuchos::WorkspaceStore        *wss      = Teuchos::get_default_workspace_store().get();
  const VectorSpaceBase<Scalar>  &domain   = *this->domain();
  const VectorSpaceBase<Scalar>  &range    = *this->range();
  const Index                    dimDomain = domain.dim();
  const Range1D                  colRng    = Teuchos::full_range(colRng_in,0,dimDomain-1);
  if( colRng.lbound() == 0 && static_cast<Index>(colRng.ubound()) == dimDomain-1 )
    return Teuchos::rcp(this,false); // Takes all of the columns!
  if( colRng.size() ) {
    // We have to create a view of a subset of the columns
    Workspace< Teuchos::RCP< VectorBase<Scalar> > >  col_vecs(wss,colRng.size());
    for( Index j = colRng.lbound(); j <= colRng.ubound(); ++j )
      col_vecs[j-colRng.lbound()] = Teuchos::rcp_const_cast<VectorBase<Scalar> >(this->col(j));
    return Teuchos::rcp(
      new DefaultColumnwiseMultiVector<Scalar>(
        this->range(),range.smallVecSpcFcty()->createVecSpc(colRng.size()),&col_vecs[0]
        )
      );
  }
  return Teuchos::null; // There was an empty set in colRng_in!
}


template<class Scalar>
Teuchos::RCP<MultiVectorBase<Scalar> >
MultiVectorDefaultBase<Scalar>::subView( const Range1D& colRng_in )
{
  using Teuchos::Workspace;
  Teuchos::WorkspaceStore        *wss      = Teuchos::get_default_workspace_store().get();
  const VectorSpaceBase<Scalar>  &domain   = *this->domain();
  const VectorSpaceBase<Scalar>  &range    = *this->range();
  const Index                    dimDomain = domain.dim();
  const Range1D                  colRng    = Teuchos::full_range(colRng_in,0,dimDomain-1);
  if( colRng.lbound() == 0 && static_cast<Index>(colRng.ubound()) == dimDomain-1 )
    return Teuchos::rcp(this,false); // Takes all of the columns!
  if( colRng.size() ) {
    // We have to create a view of a subset of the columns
    Workspace< Teuchos::RCP< VectorBase<Scalar> > >  col_vecs(wss,colRng.size());
    for( Index j = colRng.lbound(); j <= colRng.ubound(); ++j )
      col_vecs[j-colRng.lbound()] = this->col(j);
    return Teuchos::rcp(
      new DefaultColumnwiseMultiVector<Scalar>(
        this->range(),range.smallVecSpcFcty()->createVecSpc(colRng.size()),&col_vecs[0]
        )
      );
  }
  return Teuchos::null; // There was an empty set in colRng_in!
}


template<class Scalar>
Teuchos::RCP<const MultiVectorBase<Scalar> >
MultiVectorDefaultBase<Scalar>::subView( const int numCols, const int cols[] ) const
{
  using Teuchos::Workspace;
  Teuchos::WorkspaceStore        *wss      = Teuchos::get_default_workspace_store().get();
  const VectorSpaceBase<Scalar>  &range    = *this->range();
#ifdef TEUCHOS_DEBUG
  const VectorSpaceBase<Scalar>  &domain   = *this->domain();
  const Index                    dimDomain = domain.dim();
  const char msg_err[] = "MultiVectorDefaultBase<Scalar>::subView(numCols,cols[]): Error!";
   TEST_FOR_EXCEPTION( numCols < 1 || dimDomain < numCols, std::invalid_argument, msg_err );
#endif
  // We have to create a view of a subset of the columns
  Workspace< Teuchos::RCP< VectorBase<Scalar> > > col_vecs(wss,numCols);
  for( int k = 0; k < numCols; ++k ) {
    const int col_k = cols[k];
#ifdef TEUCHOS_DEBUG
    TEST_FOR_EXCEPTION(
      !( 0 <= col_k && col_k < dimDomain ), std::invalid_argument
      ,msg_err << " col["<<k<<"] = " << col_k << " is not in the range [0,"<<(dimDomain-1)<<"]!"
      );
#endif
    col_vecs[k] = Teuchos::rcp_const_cast<VectorBase<Scalar> >(this->col(col_k));
  }
  return Teuchos::rcp(
    new DefaultColumnwiseMultiVector<Scalar>(
      this->range(),range.smallVecSpcFcty()->createVecSpc(numCols),&col_vecs[0]
      )
    );
}


template<class Scalar>
Teuchos::RCP<MultiVectorBase<Scalar> >
MultiVectorDefaultBase<Scalar>::subView( const int numCols, const int cols[] )
{
  using Teuchos::Workspace;
  Teuchos::WorkspaceStore        *wss      = Teuchos::get_default_workspace_store().get();
  const VectorSpaceBase<Scalar>  &range    = *this->range();
#ifdef TEUCHOS_DEBUG
  const VectorSpaceBase<Scalar>  &domain   = *this->domain();
  const Index                    dimDomain = domain.dim();
  const char msg_err[] = "MultiVectorDefaultBase<Scalar>::subView(numCols,cols[]): Error!";
   TEST_FOR_EXCEPTION( numCols < 1 || dimDomain < numCols, std::invalid_argument, msg_err );
#endif
  // We have to create a view of a subset of the columns
  Workspace< Teuchos::RCP< VectorBase<Scalar> > > col_vecs(wss,numCols);
  for( int k = 0; k < numCols; ++k ) {
    const int col_k = cols[k];
#ifdef TEUCHOS_DEBUG
    TEST_FOR_EXCEPTION(
      !( 0 <= col_k && col_k < dimDomain ), std::invalid_argument
      ,msg_err << " col["<<k<<"] = " << col_k << " is not in the range [0,"<<(dimDomain-1)<<"]!"
      );
#endif
    col_vecs[k] = this->col(col_k);
  }
  return Teuchos::rcp(
    new DefaultColumnwiseMultiVector<Scalar>(
      this->range(),range.smallVecSpcFcty()->createVecSpc(numCols),&col_vecs[0]
      )
    );
}


} // end namespace Thyra


#endif // THYRA_MULTI_VECTOR_DEFAULT_BASE_HPP

---