Ifpack_SerialTriDiMatrix.h

/*
//@HEADER
// ************************************************************************
//
//               Epetra: Linear Algebra Services Package
//                 Copyright 2011 Sandia Corporation
//
// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
// the U.S. Government retains certain rights in this software.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//
// 1. Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
//
// 2. Redistributions in binary form must reproduce the above copyright
// notice, this list of conditions and the following disclaimer in the
// documentation and/or other materials provided with the distribution.
//
// 3. Neither the name of the Corporation nor the names of the
// contributors may be used to endorse or promote products derived from
// this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//
// Questions? Contact Michael A. Heroux (maherou@sandia.gov)
//
// ************************************************************************
//@HEADER
*/
 
#ifndef IFPACK_SERIALTRIDIMATRIX_H
#define IFPACK_SERIALTRIDIMATRIX_H
 
#if defined(Ifpack_SHOW_DEPRECATED_WARNINGS)
#ifdef __GNUC__
#warning "The Ifpack package is deprecated"
#endif
#endif
 
#include "Epetra_ConfigDefs.h"
#include "Epetra_Object.h"
#include "Epetra_CompObject.h"
#include "Epetra_BLAS.h"
 
class Epetra_VbrMatrix;
 
 
//=========================================================================
class Ifpack_SerialTriDiMatrix : public Epetra_CompObject, public Epetra_Object, public Epetra_BLAS {
 
  public:
 
 
 
  Ifpack_SerialTriDiMatrix(bool set_object_label=true);
 
 
  Ifpack_SerialTriDiMatrix(int NumRowCol, bool set_object_label=true);
 
 
  Ifpack_SerialTriDiMatrix(Epetra_DataAccess CV, double* A_in, int NumRowCol,
                           bool set_object_label=true);
 
 
  Ifpack_SerialTriDiMatrix(const Ifpack_SerialTriDiMatrix& Source);
 
  virtual ~Ifpack_SerialTriDiMatrix ();
 
 
 
  int Shape(int NumRowCol);
 
  int Reshape(int, int);
 
 
 
 
  int Multiply(char TransA, char TransB, double ScalarAB,
               const Ifpack_SerialTriDiMatrix& A,
               const Ifpack_SerialTriDiMatrix& B,
               double ScalarThis);
 
  /* This method is intended to imitate the semantics of the matrix-vector
    multiplication provided by Epetra's sparse matrices. The 'vector' arguments
    are actually matrices; this method will return an error if the
    dimensions of 'x' are not compatible. 'y' will be reshaped if necessary.
  */
  /* int Multiply(bool transA, */
  /*              const Ifpack_SerialTriDiMatrix& x, */
  /*              Ifpack_SerialTriDiMatrix& y); */
 
 
  int Scale(double ScalarA);
 
 
  virtual double NormOne() const;
 
  virtual double NormInf() const;
 
 
 
 
 
    Ifpack_SerialTriDiMatrix & operator = (const Ifpack_SerialTriDiMatrix& Source);
 
 
    bool operator==(const Ifpack_SerialTriDiMatrix& rhs) const;
 
 
    bool operator!=(const Ifpack_SerialTriDiMatrix& rhs) const
    { return !(*this == rhs); }
 
 
    Ifpack_SerialTriDiMatrix & operator += (const Ifpack_SerialTriDiMatrix& Source);
 
 
            double& operator () (int RowIndex, int ColIndex);
 
 
    const double& operator () (int RowIndex, int ColIndex) const;
 
 
    //    double* operator [] (int ColIndex);
 
 
    //    const double* operator [] (int ColIndex) const;
 
 
  int Random();
 
  int N() const {return(N_);};
 
  int LDA() const {return(LDA_);};
 
  double* A() const {return(A_);};
 
  //  double* A() {return(A_);};
 
  double* DL() { return DL_;};
  double* DL() const { return DL_;};
  double* D() { return D_;};
  double* D() const { return D_;};
  double* DU() { return DU_;};
  double* DU() const { return DU_;};
  double* DU2() { return DU2_;};
  double* DU2() const { return DU2_;};
 
  Epetra_DataAccess CV() const {return(CV_);};
 
 
 
  virtual void Print(std::ostream& os) const;
 
 
 
 
  virtual double OneNorm() const {return(NormOne());};
 
  virtual double InfNorm() const {return(NormInf());};
 
 
 
 
    virtual int SetUseTranspose(bool UseTranspose_in) { UseTranspose_ = UseTranspose_in; return (0); }
 
 
    virtual int ApplyInverse(const Ifpack_SerialTriDiMatrix & X, Ifpack_SerialTriDiMatrix & Y)
    {
      (void)X;//prevents unused variable compiler warning
      (void)Y;
      return (-1);
    }
 
    virtual const char * Label() const { return Epetra_Object::Label(); }
 
    virtual bool UseTranspose() const { return UseTranspose_; }
 
    virtual bool HasNormInf() const { return true; }
 
    virtual int RowColDim() const { return N(); }
 
 protected:
 
  void CopyMat(const double* Source, int NumRowCol,
               double* Target, int NRC2, bool add=false);
  void CleanupData();
 
  int N_;
  int LDA_;
  bool A_Copied_;
  Epetra_DataAccess CV_;
 
  //For performance reasons, it's better if Epetra_VbrMatrix can access the
  //A_ members of this class directly without going through an
  //accessor method. Rather than making them public members, we'll make
  //Epetra_VbrMatrix a friend class.
 
  friend class Epetra_VbrMatrix;
 
  double* A_;
  double* DL_;
  double* D_;
  double* DU_;
  double* DU2_;
 
  bool UseTranspose_;
};
 
// inlined definitions of op() and op[]
//=========================================================================
inline double& Ifpack_SerialTriDiMatrix::operator () (int RowIndex, int ColIndex) {
 
 int diff = ColIndex - RowIndex;
 
#ifdef HAVE_EPETRA_ARRAY_BOUNDS_CHECK
 if (ColIndex >= N_ || ColIndex < 0)
                throw ReportError("Column index = " +toString(ColIndex) +
                                  " Out of Range 0 - " + toString(N_-1),-2);
 if (RowIndex >= N_ || RowIndex < 0)
                throw ReportError("Row index = " +toString(RowIndex) +
                                  " Out of Range 0 - " + toString(N_-1),-2);
 
 if ( diff > 1 || diff < -1 )
   throw ReportError("Row index = " +toString(RowIndex) + " differs from Col_Index " + toString(ColIndex) +
                     " Out of Range -1 to 1",-2);
#endif
 
 switch (diff) {
 case -1:
   // DL
   return DL_[ColIndex];
   // break; // unreachable
 case 0:
   return D_[ColIndex];
   // break; // unreachable
 case 1:
   return DU_[RowIndex];
   // break; // unreachable
 default:
   throw ReportError("Row index = " +toString(RowIndex) + " differs from Col_Index " + toString(ColIndex) +" Out of Range -1 to 1",1);
   // return D_[0]; // unreachable
 }
 //throw ReportError("Row index = " +toString(RowIndex) + " differs from Col_Index " + toString(ColIndex) + " Out of Range -1 to 1",1); // unreachable
 // return D_[0]; // unreachable
}
//=========================================================================
inline const double& Ifpack_SerialTriDiMatrix::operator () (int RowIndex, int ColIndex) const {
 int diff = ColIndex - RowIndex;
 
#ifdef HAVE_EPETRA_ARRAY_BOUNDS_CHECK
 if (ColIndex >= N_ || ColIndex < 0)
                throw ReportError("Column index = " +toString(ColIndex) +
                                  " Out of Range 0 - " + toString(N_-1),-2);
 if (RowIndex >= N_ || RowIndex < 0)
                throw ReportError("Row index = " +toString(RowIndex) +
                                  " Out of Range 0 - " + toString(N_-1),-2);
 if ( diff > 1 || diff < -1 )
   throw ReportError("Row index = " +toString(RowIndex) + " differs from Col_Index " + toString(ColIndex) + " Out of Range -1 to 1",-2);
#endif
 switch (diff) {
 case -1:
   // DL
   return DL_[ColIndex];
   // break; // unreachable
 case 0:
   return D_[ColIndex];
   // break; // unreachable
 case 1:
   return DU_[RowIndex];
   // break; // unreachable
 default:
   throw ReportError("Row index = " +toString(RowIndex) + " differs from Col_Index " + toString(ColIndex) + " Out of Range -1 to 1",-2);
   // return D_[0]; // unreachable
 }
 // return D_[0]; // unreachable
}
 
 
#endif /* EPETRA_SERIALTRIDIMATRIX_H */