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

#include "Epetra_ConfigDefs.h"
#include "Epetra_Object.h"
#include "Epetra_CompObject.h"
#include "Epetra_BLAS.h"

class Epetra_VbrMatrix;


//=========================================================================
class EPETRA_LIB_DLL_EXPORT 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;
 case 0:
   return D_[ColIndex];
   break;
 case 1:
   return DU_[RowIndex];
   break;
 default:
   throw ReportError("Row index = " +toString(RowIndex) + " differs from Col_Index " + toString(ColIndex) +" Out of Range -1 to 1",1);
   return D_[0]; // ultimate badness, should never get here
 }
 throw ReportError("Row index = " +toString(RowIndex) + " differs from Col_Index " + toString(ColIndex) + " Out of Range -1 to 1",1);
 return D_[0]; // ultimat badness, should never get here
}
//=========================================================================
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;
 case 0:
   return D_[ColIndex];
   break;
 case 1:
   return DU_[RowIndex];
   break;
 default:
   throw ReportError("Row index = " +toString(RowIndex) + " differs from Col_Index " + toString(ColIndex) + " Out of Range -1 to 1",-2);
  return D_[0]; // Ultimate badness, should never get here.
 }
 return D_[0]; // Crime against humanity, should never get here. 
}


#endif /* EPETRA_SERIALTRIDIMATRIX_H */