AbstractLinAlgPack_MatrixCOORTmplItfc.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 SLAP_MATRIX_COOR_TMPL_ITFC_H
#define SLAP_MATRIX_COOR_TMPL_ITFC_H

#include <stdexcept>

#include "AbstractLinAlgPack_Types.hpp"
#include "Teuchos_TestForException.hpp"

namespace AbstractLinAlgPack {

template<class T_Scalar, class T_Index>
class MatrixCOORTmplItfcItrEleView;

template<class T_Scalar, class T_Index>
class MatrixCOORTmplItfcItr;

template<class T_Scalar, class T_Index>
class MatrixCOORTmplItfc {
public:
  typedef T_Index                                          size_type;
  typedef ptrdiff_t                                        difference_type;
  typedef MatrixCOORTmplItfcItrEleView<T_Scalar,T_Index>   element_type;
  typedef T_Scalar                                         value_type;
  typedef T_Index                                          index_type;
  typedef MatrixCOORTmplItfcItr<T_Scalar,T_Index>          const_iterator;
  MatrixCOORTmplItfc(
    size_type rows, size_type cols, size_type nz
    ,difference_type row_offset, difference_type col_offset
    ,const T_Scalar *values, const T_Index* row_i, const T_Index* col_j
    )
    :rows_(rows), cols_(cols), nz_(nz), row_offset_(row_offset), col_offset_(col_offset)
    ,values_(values), row_i_(row_i), col_j_(col_j)
  {}
  size_type       rows()       const { return rows_;       }
  size_type       cols()       const { return cols_;       }
  size_type       nz()         const { return nz_;         }
  difference_type row_offset() const { return row_offset_; }  
  difference_type col_offset() const { return col_offset_; }
  const_iterator  begin() const;
  const_iterator  end() const;
private:
  size_type        rows_;
    size_type        cols_;
  size_type        nz_;
  difference_type  row_offset_;
  difference_type  col_offset_;
  const T_Scalar   *values_;
  const T_Index    *row_i_;
  const T_Index    *col_j_;
  // Not defined and not to be called
  MatrixCOORTmplItfc();
}; // end class MatrixCOORTmplItfc


// ///////////////////////////////////////////////
// Implementatioins, not of the user to look at!

template<class T_Scalar, class T_Index>
class MatrixCOORTmplItfcItrEleView {
public:
  typedef T_Scalar   value_type;
  typedef T_Index    index_type;
  MatrixCOORTmplItfcItrEleView(
    const T_Scalar* value, const T_Index* row_i, const T_Index* col_j
    )
    : value_(value), row_i_(row_i), col_j_(col_j)
  {}
  void increment()       { ++value_; ++row_i_; ++col_j_; }
  bool operator!=(const MatrixCOORTmplItfcItrEleView<T_Scalar,T_Index>& ele) const
  {   return value_ != ele.value_ || row_i_ != ele.row_i_ || col_j_ != ele.col_j_; }
  T_Scalar value() const { return *value_;  }
  T_Index  row_i() const { return *row_i_;  }
  T_Index  col_j() const { return *col_j_;  }
private:
  const T_Scalar  *value_;
  const T_Index   *row_i_;
  const T_Index   *col_j_;
  // Not defined and not to be called
  MatrixCOORTmplItfcItrEleView();
}; // end class MatrixCOORTmplItfcItrEleView

template<class T_Scalar, class T_Index>
class MatrixCOORTmplItfcItr {
public:
  MatrixCOORTmplItfcItr( const T_Scalar* value, const T_Index* row_i, const T_Index* col_j, T_Index nz )
    : ele_(value,row_i,col_j)
#ifdef TEUCHOS_DEBUG
    , nz_left_(nz)
#endif
  {}
  void operator++() {
#ifdef TEUCHOS_DEBUG
    --nz_left_;
#endif
    ele_.increment();
  }
  bool operator!=(const MatrixCOORTmplItfcItr<T_Scalar,T_Index>& itr) const
  {   return ele_ != itr.ele_; }
  const MatrixCOORTmplItfcItrEleView<T_Scalar,T_Index>* operator->() const
  { assert_nz(); return &ele_; }
private:
  MatrixCOORTmplItfcItrEleView<T_Scalar,T_Index>  ele_;
#ifdef TEUCHOS_DEBUG
  T_Index   nz_left_;
  void assert_nz() const
  {
    TEST_FOR_EXCEPTION(
      nz_left_ <= 0, std::logic_error
      ,"MatrixCOORTmplItfcItr<>::assert_nz: Error, trying to access past storage!" );
  }
#else
  void assert_nz() const {}
#endif
  // Not defined and not to be called
  MatrixCOORTmplItfcItr();
}; // end class MatrixCOORTmplItfcItr

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

// MatrixCOORTmplItfc

template<class T_Scalar, class T_Index>
inline
typename MatrixCOORTmplItfc<T_Scalar,T_Index>::const_iterator
MatrixCOORTmplItfc<T_Scalar,T_Index>::begin() const
{
  return MatrixCOORTmplItfcItr<T_Scalar,T_Index>(values_,row_i_,col_j_,nz_);
}

template<class T_Scalar, class T_Index>
inline
typename MatrixCOORTmplItfc<T_Scalar,T_Index>::const_iterator
MatrixCOORTmplItfc<T_Scalar,T_Index>::end() const
{
  return MatrixCOORTmplItfcItr<T_Scalar,T_Index>(values_+nz_,row_i_+nz_,col_j_+nz_,0);
}

} // end namespace AbstractLinAlgPack

#endif // SLAP_MATRIX_COOR_TMPL_ITFC_H