42 #ifndef _TEUCHOS_SERIALBANDDENSEMATRIX_HPP_
43 #define _TEUCHOS_SERIALBANDDENSEMATRIX_HPP_
132 template<
typename OrdinalType,
typename ScalarType>
293 ScalarType&
operator () (OrdinalType rowIndex, OrdinalType colIndex);
303 const ScalarType&
operator () (OrdinalType rowIndex, OrdinalType colIndex)
const;
323 const ScalarType*
operator [] (OrdinalType colIndex)
const;
357 int scale (
const ScalarType alpha );
424 virtual void print(std::ostream& os)
const;
429 void copyMat(ScalarType* inputMatrix, OrdinalType strideInput,
430 OrdinalType
numRows, OrdinalType
numCols, ScalarType* outputMatrix,
433 void checkIndex( OrdinalType rowIndex, OrdinalType colIndex = 0 )
const;
448 template<
typename OrdinalType,
typename ScalarType>
451 BLAS<OrdinalType,ScalarType>(),
457 valuesCopied_ (false),
461 template<
typename OrdinalType,
typename ScalarType>
464 OrdinalType numCols_in,
469 BLAS<OrdinalType,ScalarType>(),
470 numRows_ (numRows_in),
471 numCols_ (numCols_in),
472 stride_ (kl_in+ku_in+1),
475 valuesCopied_ (true),
484 template<
typename OrdinalType,
typename ScalarType>
487 ScalarType* values_in,
488 OrdinalType stride_in,
489 OrdinalType numRows_in,
490 OrdinalType numCols_in,
494 BLAS<OrdinalType,ScalarType>(),
495 numRows_ (numRows_in),
496 numCols_ (numCols_in),
500 valuesCopied_ (false),
511 template<
typename OrdinalType,
typename ScalarType>
515 BLAS<OrdinalType,ScalarType>(),
521 valuesCopied_ (true),
541 for (OrdinalType j = 0; j <
numCols_; ++j) {
556 for (OrdinalType j=0; j<
numCols_; j++) {
565 template<
typename OrdinalType,
typename ScalarType>
569 OrdinalType numRows_in,
570 OrdinalType numCols_in,
571 OrdinalType startCol)
573 BLAS<OrdinalType,ScalarType>(),
574 numRows_ (numRows_in),
575 numCols_ (numCols_in),
576 stride_ (Source.stride_),
579 valuesCopied_ (false),
580 values_ (Source.values_)
592 template<
typename OrdinalType,
typename ScalarType>
602 template<
typename OrdinalType,
typename ScalarType>
604 OrdinalType numRows_in, OrdinalType numCols_in, OrdinalType kl_in, OrdinalType ku_in
608 numRows_ = numRows_in;
609 numCols_ = numCols_in;
613 values_ =
new ScalarType[stride_*numCols_];
615 valuesCopied_ =
true;
620 template<
typename OrdinalType,
typename ScalarType>
622 OrdinalType numRows_in, OrdinalType numCols_in, OrdinalType kl_in, OrdinalType ku_in
626 numRows_ = numRows_in;
627 numCols_ = numCols_in;
631 values_ =
new ScalarType[stride_*numCols_];
632 valuesCopied_ =
true;
637 template<
typename OrdinalType,
typename ScalarType>
639 OrdinalType numRows_in, OrdinalType numCols_in, OrdinalType kl_in, OrdinalType ku_in
644 ScalarType* values_tmp =
new ScalarType[(kl_in+ku_in+1) * numCols_in];
646 for(OrdinalType k = 0; k < (kl_in+ku_in+1) * numCols_in; k++) {
647 values_tmp[k] = zero;
649 OrdinalType numRows_tmp =
TEUCHOS_MIN(numRows_, numRows_in);
650 OrdinalType numCols_tmp =
TEUCHOS_MIN(numCols_, numCols_in);
652 copyMat(values_, stride_, numRows_tmp, numCols_tmp, values_tmp,
656 numRows_ = numRows_in;
657 numCols_ = numCols_in;
661 values_ = values_tmp;
662 valuesCopied_ =
true;
671 template<
typename OrdinalType,
typename ScalarType>
676 for(OrdinalType j = 0; j < numCols_; j++) {
678 values_[(ku_+i-j) + j*stride_] = value_in;
685 template<
typename OrdinalType,
typename ScalarType>
690 for(OrdinalType j = 0; j < numCols_; j++) {
699 template<
typename OrdinalType,
typename ScalarType>
731 const OrdinalType newsize = stride_ * numCols_;
733 values_ =
new ScalarType[newsize];
734 valuesCopied_ =
true;
753 const OrdinalType newsize = stride_ * numCols_;
755 values_ =
new ScalarType[newsize];
756 valuesCopied_ =
true;
760 copyMat(Source.
values_, Source.
stride_, numRows_, numCols_, values_, stride_, 0);
766 template<
typename OrdinalType,
typename ScalarType>
771 if ((numRows_ != Source.
numRows_) || (numCols_ != Source.
numCols_) || (kl_ != Source.
kl_) || (ku_ != Source.
ku_)) {
779 template<
typename OrdinalType,
typename ScalarType>
784 if ((numRows_ != Source.
numRows_) || (numCols_ != Source.
numCols_) || (kl_ != Source.
kl_) || (ku_ != Source.
ku_)) {
792 template<
typename OrdinalType,
typename ScalarType>
801 if ((numRows_ != Source.
numRows_) || (numCols_ != Source.
numCols_) || (kl_ != Source.
kl_) || (ku_ != Source.
ku_)) {
804 copyMat(Source.
values_, Source.
stride_, numRows_, numCols_, values_, stride_, 0);
813 template<
typename OrdinalType,
typename ScalarType>
816 #ifdef HAVE_TEUCHOS_ARRAY_BOUNDSCHECK
817 checkIndex( rowIndex, colIndex );
819 return(values_[colIndex * stride_ + ku_+rowIndex-colIndex]);
822 template<
typename OrdinalType,
typename ScalarType>
825 #ifdef HAVE_TEUCHOS_ARRAY_BOUNDSCHECK
826 checkIndex( rowIndex, colIndex );
828 return(values_[colIndex * stride_ + ku_+rowIndex-colIndex]);
831 template<
typename OrdinalType,
typename ScalarType>
834 #ifdef HAVE_TEUCHOS_ARRAY_BOUNDSCHECK
835 checkIndex( 0, colIndex );
837 return(values_ + colIndex * stride_);
840 template<
typename OrdinalType,
typename ScalarType>
843 #ifdef HAVE_TEUCHOS_ARRAY_BOUNDSCHECK
844 checkIndex( 0, colIndex );
846 return(values_ + colIndex * stride_);
853 template<
typename OrdinalType,
typename ScalarType>
861 for(j = 0; j < numCols_; j++) {
863 ptr = values_ + j * stride_ +
TEUCHOS_MAX(0, ku_-j);
872 updateFlops((kl_+ku_+1) * numCols_);
877 template<
typename OrdinalType,
typename ScalarType>
883 for (i = 0; i < numRows_; i++) {
890 updateFlops((kl_+ku_+1) * numCols_);
895 template<
typename OrdinalType,
typename ScalarType>
901 for (j = 0; j < numCols_; j++) {
907 updateFlops((kl_+ku_+1) * numCols_);
916 template<
typename OrdinalType,
typename ScalarType>
921 if((numRows_ != Operand.
numRows_) || (numCols_ != Operand.
numCols_) || (kl_ != Operand.
kl_) || (ku_ != Operand.
ku_)) {
925 for(j = 0; j < numCols_; j++) {
927 if((*
this)(i, j) != Operand(i, j)) {
937 template<
typename OrdinalType,
typename ScalarType>
940 return !((*this) == Operand);
947 template<
typename OrdinalType,
typename ScalarType>
950 this->scale( alpha );
954 template<
typename OrdinalType,
typename ScalarType>
961 for (j=0; j<numCols_; j++) {
964 *ptr = alpha * (*ptr); ptr++;
967 updateFlops( (kl_+ku_+1)*numCols_ );
972 template<
typename OrdinalType,
typename ScalarType>
983 for (j=0; j<numCols_; j++) {
986 *ptr = A(i,j) * (*ptr); ptr++;
989 updateFlops( (kl_+ku_+1)*numCols_ );
994 template<
typename OrdinalType,
typename ScalarType>
999 os <<
"Values_copied : yes" << std::endl;
1001 os <<
"Values_copied : no" << std::endl;
1002 os <<
"Rows : " << numRows_ << std::endl;
1003 os <<
"Columns : " << numCols_ << std::endl;
1004 os <<
"Lower Bandwidth : " << kl_ << std::endl;
1005 os <<
"Upper Bandwidth : " << ku_ << std::endl;
1006 os <<
"LDA : " << stride_ << std::endl;
1007 if(numRows_ == 0 || numCols_ == 0) {
1008 os <<
"(matrix is empty, no values to display)" << std::endl;
1011 for(OrdinalType i = 0; i < numRows_; i++) {
1013 os << (*this)(i,j) <<
" ";
1024 template<
typename OrdinalType,
typename ScalarType>
1030 "SerialBandDenseMatrix<T>::checkIndex: "
1031 "Row index " << rowIndex <<
" out of range [0, "<< numRows_ <<
")");
1033 "SerialBandDenseMatrix<T>::checkIndex: "
1034 "Col index " << colIndex <<
" out of range [0, "<< numCols_ <<
")");
1038 template<
typename OrdinalType,
typename ScalarType>
1041 if (valuesCopied_) {
1044 valuesCopied_ =
false;
1048 template<
typename OrdinalType,
typename ScalarType>
1050 ScalarType* inputMatrix, OrdinalType strideInput, OrdinalType numRows_in,
1051 OrdinalType numCols_in, ScalarType* outputMatrix, OrdinalType strideOutput, OrdinalType startCol, ScalarType alpha
1055 ScalarType* ptr1 = 0;
1056 ScalarType* ptr2 = 0;
1058 for(j = 0; j < numCols_in; j++) {
1059 ptr1 = outputMatrix + (j * strideOutput) +
TEUCHOS_MAX(0, ku_-j);
1060 ptr2 = inputMatrix + (j + startCol) * strideInput +
TEUCHOS_MAX(0, ku_-j);
1063 *ptr1++ += alpha*(*ptr2++);
1073 #ifndef TEUCHOS_HIDE_DEPRECATED_CODE
1074 template<
typename OrdinalType,
typename ScalarType>
1076 std::ostream& operator<< (std::ostream& os, const Teuchos::SerialBandDenseMatrix<OrdinalType, ScalarType>& obj)
bool empty() const
Returns whether this matrix is empty.
OrdinalType numRows() const
Returns the row dimension of this matrix.
SerialBandDenseMatrix< OrdinalType, ScalarType > & operator+=(const SerialBandDenseMatrix< OrdinalType, ScalarType > &Source)
Add another matrix to this matrix.
ScalarType & operator()(OrdinalType rowIndex, OrdinalType colIndex)
Element access method (non-const).
int scale(const ScalarType alpha)
Scale this matrix by alpha; *this = alpha**this.
int shape(OrdinalType numRows, OrdinalType numCols, OrdinalType kl, OrdinalType ku)
Shape method for changing the size of a SerialBandDenseMatrix, initializing entries to zero...
ScalarTraits< ScalarType >::magnitudeType normInf() const
Returns the Infinity-norm of the matrix.
Templated interface class to BLAS routines.
virtual ~SerialBandDenseMatrix()
Destructor.
#define TEUCHOS_TEST_FOR_EXCEPTION(throw_exception_test, Exception, msg)
Macro for throwing an exception with breakpointing to ease debugging.
int reshape(OrdinalType numRows, OrdinalType numCols, OrdinalType kl, OrdinalType ku)
Reshaping method for changing the size of a SerialBandDenseMatrix, keeping the entries.
Teuchos header file which uses auto-configuration information to include necessary C++ headers...
ScalarTraits< ScalarType >::magnitudeType normOne() const
Returns the 1-norm of the matrix.
SerialBandDenseMatrix< OrdinalType, ScalarType > & assign(const SerialBandDenseMatrix< OrdinalType, ScalarType > &Source)
Copies values from one matrix to another.
Object for storing data and providing functionality that is common to all computational classes...
This structure defines some basic traits for a scalar field type.
OrdinalType stride() const
Returns the stride between the columns of this matrix in memory.
virtual void print(std::ostream &os) const
Print method. Defines the behavior of the std::ostream << operator.
#define TEUCHOS_CHK_ERR(a)
ScalarType * values() const
Data array access method.
Functionality and data that is common to all computational classes.
bool operator!=(const SerialBandDenseMatrix< OrdinalType, ScalarType > &Operand) const
Inequality of two matrices.
ScalarTraits< ScalarType >::magnitudeType normFrobenius() const
Returns the Frobenius-norm of the matrix.
This class creates and provides basic support for banded dense matrices of templated type...
SerialBandDenseMatrix< OrdinalType, ScalarType > & operator*=(const ScalarType alpha)
Scale this matrix by alpha; *this = alpha**this.
#define TEUCHOS_MAX(x, y)
SerialBandDenseMatrix< OrdinalType, ScalarType > & operator-=(const SerialBandDenseMatrix< OrdinalType, ScalarType > &Source)
Subtract another matrix from this matrix.
int putScalar(const ScalarType value=Teuchos::ScalarTraits< ScalarType >::zero())
Set all values in the matrix to a constant value.
int random()
Set all values in the matrix to be random numbers.
SerialBandDenseMatrix()
Default Constructor.
bool operator==(const SerialBandDenseMatrix< OrdinalType, ScalarType > &Operand) const
Equality of two matrices.
static magnitudeType magnitude(T a)
Returns the magnitudeType of the scalar type a.
OrdinalType ordinalType
Typedef for ordinal type.
SerialBandDenseMatrix< OrdinalType, ScalarType > & operator=(const SerialBandDenseMatrix< OrdinalType, ScalarType > &Source)
Copies values from one matrix to another.
OrdinalType upperBandwidth() const
Returns the upper bandwidth of this matrix.
void copyMat(ScalarType *inputMatrix, OrdinalType strideInput, OrdinalType numRows, OrdinalType numCols, ScalarType *outputMatrix, OrdinalType strideOutput, OrdinalType startCol, ScalarType alpha=ScalarTraits< ScalarType >::zero())
Defines basic traits for the scalar field type.
static T zero()
Returns representation of zero for this scalar type.
OrdinalType lowerBandwidth() const
Returns the lower bandwidth of this matrix.
void checkIndex(OrdinalType rowIndex, OrdinalType colIndex=0) const
ScalarType * operator[](OrdinalType colIndex)
Column access method (non-const).
ScalarType scalarType
Typedef for scalar type.
#define TEUCHOS_CHK_REF(a)
OrdinalType numCols() const
Returns the column dimension of this matrix.
#define TEUCHOS_MIN(x, y)
Reference-counted pointer class and non-member templated function implementations.
int shapeUninitialized(OrdinalType numRows, OrdinalType numCols, OrdinalType kl, OrdinalType ku)
Same as shape() except leaves uninitialized.
Teuchos::DataAccess Mode enumerable type.