AbstractLinAlgPack_VectorMutableBlocked.hpp

// @HEADER
// ***********************************************************************
// 
// Moocho: Multi-functional Object-Oriented arCHitecture for Optimization
//                  Copyright (2003) 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 Roscoe A. Bartlett (rabartl@sandia.gov) 
// 
// ***********************************************************************
// @HEADER

#ifndef VECTOR_WITH_OP_MUTABLE_BLOCK_STD_H
#define VECTOR_WITH_OP_MUTABLE_BLOCK_STD_H

#include <vector>

#include "AbstractLinAlgPack_VectorMutable.hpp"
#include "AbstractLinAlgPack_VectorSpaceBlocked.hpp"

namespace AbstractLinAlgPack {

class VectorMutableBlocked : virtual public VectorMutable
{
public:

  typedef Teuchos::RCP<const VectorSpaceBlocked>  vec_space_comp_ptr_t;

  VectorMutableBlocked(
    VectorMutable::vec_mut_ptr_t*  vecs
    ,const vec_space_comp_ptr_t&         vec_space
    );

  void initialize(
    VectorMutable::vec_mut_ptr_t*  vecs
    ,const vec_space_comp_ptr_t&         vec_space
    );

  const VectorSpaceBlocked& block_space() const;

  const Vector& get_vector(int k) const;

  VectorMutable& get_vector(int k);



  index_type dim() const;
  const VectorSpace& space() const;
  void apply_op(
    const RTOpPack::RTOp& op
    ,const size_t num_vecs, const Vector* vecs[]
    ,const size_t num_targ_vecs, VectorMutable* targ_vecs[]
    ,RTOpPack::ReductTarget *reduct_obj
    ,const index_type first_ele, const index_type sub_dim, const index_type global_offset
    ) const;
  index_type nz() const;
  std::ostream& output(
    std::ostream& out, bool print_dim, bool newline
    ,index_type global_offset 
    ) const;
  value_type get_ele(index_type i) const;
  value_type norm_1() const;
  value_type norm_inf() const;
  value_type inner_product( const Vector& ) const;
  void get_sub_vector( const Range1D& rng, RTOpPack::SubVector* sub_vec ) const;
  void free_sub_vector( RTOpPack::SubVector* sub_vec ) const;



  vec_mut_ptr_t sub_view( const Range1D& rng );
  void axpy( value_type alpha, const Vector& x );
  VectorMutable& operator=(value_type);
  VectorMutable& operator=(const Vector&);
  void set_ele( index_type i, value_type val );
  void set_sub_vector( const RTOpPack::SparseSubVector& sub_vec );


protected:

  void has_changed() const;

private:

  // ////////////////////////////////////
  // Private types
  
  typedef std::vector<VectorMutable::vec_mut_ptr_t>   vecs_t; 

  // ////////////////////////////////////
  // Private data members

#ifdef DOXYGEN_COMPILE
  VectorMutable        *vectors;
  VectorSpaceBlocked    *block_vector_space;
#else
  vecs_t                 vecs_;        
  vec_space_comp_ptr_t   vec_space_;   
#endif
  mutable index_type     nz_; 
  mutable value_type     norm_1_, norm_inf_;   

  // ////////////////////////////////////
  // Private member functions

  void assert_in_range(int k) const;

  void assert_initialized() const;

  // not defined and not to be called!
  VectorMutableBlocked();
  VectorMutableBlocked(const VectorMutableBlocked&);
  VectorMutableBlocked& operator=(const VectorMutableBlocked&);

}; // end class VectorMutableBlocked

// ////////////////////////////////////
// Inline members

inline
const VectorSpaceBlocked&
VectorMutableBlocked::block_space() const
{
  assert_initialized();
  return *vec_space_;
}

inline
const Vector&
VectorMutableBlocked::get_vector(int k) const
{
  assert_in_range(k);
  return *vecs_[k];
}

inline
VectorMutable&
VectorMutableBlocked::get_vector(int k)
{
  assert_in_range(k);
  return *vecs_[k];
}

} // end namespace AbstractLinAlgPack

#endif  // VECTOR_WITH_OP_MUTABLE_BLOCK_STD_H