Thyra_VectorBaseDecl.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_VECTOR_BASE_DECL_HPP
#define THYRA_VECTOR_BASE_DECL_HPP


#include "Thyra_OperatorVectorTypes.hpp"
#include "Thyra_MultiVectorBaseDecl.hpp"
#include "RTOpPack_RTOpT.hpp"
#include "RTOpPack_SparseSubVectorT.hpp"


namespace Thyra {


template<class Scalar>
class VectorBase : virtual public MultiVectorBase<Scalar>
{
public:

#ifdef THYRA_INJECT_USING_DECLARATIONS
  using MultiVectorBase<Scalar>::apply;
#endif


  virtual RCP< const VectorSpaceBase<Scalar> > space() const = 0;



  void applyOp(
    const RTOpPack::RTOpT<Scalar> &op,
    const ArrayView<const Ptr<const VectorBase<Scalar> > > &vecs,
    const ArrayView<const Ptr<VectorBase<Scalar> > > &targ_vecs,
    const Ptr<RTOpPack::ReductTarget> &reduct_obj,
    const Index first_ele_offset,
    const Index sub_dim,
    const Index global_offset
    ) const
    { applyOpImpl(op,vecs,targ_vecs,reduct_obj,
        first_ele_offset,sub_dim,global_offset); }



  virtual RCP<VectorBase<Scalar> > clone_v() const = 0;



  void acquireDetachedView(
    const Range1D& rng, RTOpPack::ConstSubVectorView<Scalar>* sub_vec
    ) const
    { acquireDetachedVectorViewImpl(rng,sub_vec); }

  void releaseDetachedView(
    RTOpPack::ConstSubVectorView<Scalar>* sub_vec
    ) const
    { releaseDetachedVectorViewImpl(sub_vec); }

  void acquireDetachedView(
    const Range1D& rng, RTOpPack::SubVectorView<Scalar>* sub_vec
    )
    { acquireNonconstDetachedVectorViewImpl(rng,sub_vec); }

  void commitDetachedView(
    RTOpPack::SubVectorView<Scalar>* sub_vec
    )
    { commitNonconstDetachedVectorViewImpl(sub_vec); }

  void setSubVector(
    const RTOpPack::SparseSubVectorT<Scalar>& sub_vec
    )
    { setSubVectorImpl(sub_vec); }


protected:


  virtual void applyOpImpl(
    const RTOpPack::RTOpT<Scalar> &op,
    const ArrayView<const Ptr<const VectorBase<Scalar> > > &vecs,
    const ArrayView<const Ptr<VectorBase<Scalar> > > &targ_vecs,
    const Ptr<RTOpPack::ReductTarget> &reduct_obj,
    const Index first_ele_offset,
    const Index sub_dim,
    const Index global_offset
    ) const = 0;
/*
    {
      // 2008/02/19: rabartl: ToDo: Remove this default implemenation once the
      // refactoring is done.
      using Teuchos::as;
      Array<const VectorBase<Scalar>*> raw_vecs(vecs.size());
      for ( int k = 0; k < as<int>(vecs.size()); ++k )
        raw_vecs[k] = &*vecs[k];
      Array<VectorBase<Scalar>*> raw_targ_vecs(targ_vecs.size());
      for ( int k = 0; k < as<int>(targ_vecs.size()); ++k )
        raw_targ_vecs[k] = &*targ_vecs[k];
      this->applyOp(op, raw_vecs.size(), raw_vecs.getRawPtr(),
        raw_targ_vecs.size(), raw_targ_vecs.getRawPtr(),
        reduct_obj.get(), first_ele_offset, sub_dim, global_offset );
    }
*/
  
  virtual void acquireDetachedVectorViewImpl(
    const Range1D& rng, RTOpPack::ConstSubVectorView<Scalar>* sub_vec
    ) const = 0;

  virtual void releaseDetachedVectorViewImpl(
    RTOpPack::ConstSubVectorView<Scalar>* sub_vec
    ) const = 0;

  virtual void acquireNonconstDetachedVectorViewImpl(
    const Range1D& rng, RTOpPack::SubVectorView<Scalar>* sub_vec
    ) = 0;

  virtual void commitNonconstDetachedVectorViewImpl(
    RTOpPack::SubVectorView<Scalar>* sub_vec
    ) = 0;

  virtual void setSubVectorImpl(
    const RTOpPack::SparseSubVectorT<Scalar>& sub_vec
    ) = 0;


public:

/*
  virtual void applyOp(
    const RTOpPack::RTOpT<Scalar> &op,
    const int num_vecs,
    const VectorBase<Scalar>*const vecs[],
    const int num_targ_vecs,
    VectorBase<Scalar>*const targ_vecs[],
    RTOpPack::ReductTarget *reduct_obj,
    const Index first_ele_offset,
    const Index sub_dim,
    const Index global_offset
    ) const
    {
      TEST_FOR_EXCEPT(true);
      // ToDo: remove this function once the refactoring is finished!
    }
*/

};


template<class Scalar>
inline
void applyOp(
  const RTOpPack::RTOpT<Scalar> &op,
  const ArrayView<const Ptr<const VectorBase<Scalar> > > &vecs,
  const ArrayView<const Ptr<VectorBase<Scalar> > > &targ_vecs,
  const Ptr<RTOpPack::ReductTarget> &reduct_obj,
  const Index first_ele_offset = 0,
  const Index sub_dim = -1,
  const Index global_offset = 0
  )
{
  if (vecs.size())
    vecs[0]->applyOp(
      op, vecs, targ_vecs, reduct_obj, first_ele_offset, sub_dim, global_offset);
  else if (targ_vecs.size())
    targ_vecs[0]->applyOp(
      op, vecs, targ_vecs, reduct_obj, first_ele_offset, sub_dim, global_offset);
}


template<class Scalar>
inline
void applyOp(
  const RTOpPack::RTOpT<Scalar> &op,
  const int num_vecs,
  const VectorBase<Scalar>*const vecs_in[],
  const int num_targ_vecs,
  VectorBase<Scalar>*const targ_vecs_inout[],
  RTOpPack::ReductTarget *reduct_obj,
  const Index first_ele_offset = 0,
  const Index sub_dim = -1,
  const Index global_offset = 0
  )
{
  using Teuchos::ptr;
  Array<Ptr<const VectorBase<Scalar> > > vecs(num_vecs);
  for ( int k = 0; k < num_vecs; ++k )
    vecs[k] = ptr(vecs_in[k]);
  Array<Ptr<VectorBase<Scalar> > > targ_vecs(num_targ_vecs);
  for ( int k = 0; k < num_targ_vecs; ++k )
    targ_vecs[k] = ptr(targ_vecs_inout[k]);
  applyOp<Scalar>(op, vecs(), targ_vecs(), ptr(reduct_obj),
    first_ele_offset, sub_dim, global_offset );
}


} // end namespace Thyra


#endif  // THYRA_VECTOR_BASE_DECL_HPP

---