AbstractLinAlgPack_COOMatrixPartitionViewSubclass.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 COO_MATRIX_PARTITION_VIEW_SUBCLASS_H
#define COO_MATRIX_PARTITION_VIEW_SUBCLASS_H

#include "AbstractLinAlgPack_MatrixOp.hpp"
#include "AbstractLinAlgPack_COOMatrixWithPartitionedView.hpp"

namespace AbstractLinAlgPack {

// Could not derive this class form MatrixWithOpConcreteEncap because the assignment
// operator is not defined for the partition class.

class COOMatrixPartitionViewSubclass : public MatrixOp
{
public:

  typedef COOMatrixWithPartitionedView::partition_type  M;

  COOMatrixPartitionViewSubclass()
    : trans_(BLAS_Cpp::no_trans)
  {}

  COOMatrixPartitionViewSubclass(BLAS_Cpp::Transp trans)
    : trans_(trans)
  {}

  COOMatrixPartitionViewSubclass(const M& m)
    : m_(m), trans_(BLAS_Cpp::no_trans)
  {}

  COOMatrixPartitionViewSubclass(const M& m, BLAS_Cpp::Transp trans)
    : m_(m), trans_(trans)
  {}

  void set_trans(BLAS_Cpp::Transp trans) {
    trans_ = trans;
  }

  // /////////////////////////////////////////////////////

  M& m() {
    return m_;
  }

  const M& m() const {
    return m_;
  }

  //    end Representation access

  // /////////////////////////////////////////////////////
  // Overridden from Matrix

  size_type rows() const;

  size_type cols() const;

  // /////////////////////////////////////////////////////
  // Overridden from MatrixOp

  MatrixOp& operator=(const MatrixOp& m);

  // /////////////////////////////////////////////////////

  void Mp_StM(DMatrixSlice* gms_lhs, value_type alpha, BLAS_Cpp::Transp trans_rhs) const;

  //    end Level-1 BLAS

  // ////////////////////////////////////////////////////

  void Vp_StMtV(DVectorSlice* vs_lhs, value_type alpha, BLAS_Cpp::Transp trans_rhs1
    , const DVectorSlice& vs_rhs2, value_type beta) const;

  void Vp_StMtV(DVectorSlice* vs_lhs, value_type alpha, BLAS_Cpp::Transp trans_rhs1
    , const SpVectorSlice& sv_rhs2, value_type beta) const;

  value_type transVtMtV(const DVectorSlice& vs_rhs1, BLAS_Cpp::Transp trans_rhs2
    , const DVectorSlice& vs_rhs3) const;

  value_type transVtMtV(const SpVectorSlice& sv_rhs1, BLAS_Cpp::Transp trans_rhs2
    , const SpVectorSlice& sv_rhs3) const;

  //    end Level-2 BLAS

  // ////////////////////////////////////////////////////

  void Mp_StMtM(DMatrixSlice* gms_lhs, value_type alpha
    , BLAS_Cpp::Transp trans_rhs1, const DMatrixSlice& gms_rhs2
    , BLAS_Cpp::Transp trans_rhs2, value_type beta) const;

  void Mp_StMtM(DMatrixSlice* gms_lhs, value_type alpha, const DMatrixSlice& gms_rhs1
    , BLAS_Cpp::Transp trans_rhs1, BLAS_Cpp::Transp trans_rhs2, value_type beta) const;

  //    end Level-3 BLAS

private:
  M m_;
  BLAS_Cpp::Transp trans_;  // for how the matrix if viewed as.

  BLAS_Cpp::Transp op(BLAS_Cpp::Transp trans) const {
    using BLAS_Cpp::trans_not;
    return trans_ == BLAS_Cpp::no_trans ? trans : trans_not(trans);
  }

};  // end class COOMatrixPartitionViewSubclass

} // end namespace AbstractLinAlgPack 

#endif  // COO_MATRIX_PARTITION_VIEW_SUBCLASS_H