trmd.h

Go to the documentation of this file.

// -*-C++-*- 
//      LAPACK++ (V. 1.1)
//      (C) 1992-1996 All Rights Reserved.


#include "arch.h"
#ifndef _LA_TRIDIAG_MAT_DOUBLE_
#define _LA_TRIDIAG_MAT_DOUBLE_

#include "lafnames.h"
#include LA_VECTOR_DOUBLE_H

class DLLIMPORT LaTridiagMatDouble
{   
      LaVectorDouble du2_;    /* second upper diag, N-2 */
      LaVectorDouble du_;     /* upper diag, N-1 */
      LaVectorDouble d_;      /* main diag, N */
      LaVectorDouble dl_;     /* lower diag, N-1 */
      int size_;

      static double outofbounds_; /* return this address, when addresing out
                                     of bounds */
      static int debug_;        // print debug info.
      static int *info_;        // print matrix info only, not values
      //   originally 0, set to 1, and then
      //   reset to 0 after use.

   public:

      LaTridiagMatDouble();
      LaTridiagMatDouble(int N);
      LaTridiagMatDouble(const LaTridiagMatDouble &);
      LaTridiagMatDouble(const LaVectorDouble& diag,
                         const LaVectorDouble& diaglower,
                         const LaVectorDouble& diagupper);

      ~LaTridiagMatDouble();

      int size() const { return size_;}

      inline double &operator()(int i, int j);

      inline double operator()(int i, int j) const;

      LaVectorDouble& diag(int diag_selection);

      const LaVectorDouble& diag(int diag_selection) const;


      LaTridiagMatDouble& copy(const LaTridiagMatDouble& s); 

      LaTridiagMatDouble& inject(const LaTridiagMatDouble& s);

      inline LaTridiagMatDouble& ref(LaTridiagMatDouble&); 

      const LaTridiagMatDouble& info() const {
         int *t = info_; *t = 1; return *this;}
      int debug() const { return debug_;}

      friend DLLIMPORT std::ostream& operator<<(std::ostream&,const LaTridiagMatDouble&);


};

DLLIMPORT std::ostream& operator<<(std::ostream& s, const LaTridiagMatDouble& td);


    // operators and member functions

inline double& LaTridiagMatDouble::operator()(int i,int j)
{
   switch (i-j)
   {
      case 0:   // main
#ifdef LA_BOUNDS_CHECK
         if (i>d_.size()-1)
            return outofbounds_;
         else
#endif
            return d_(i);
      case 1:  // lower
#ifdef LA_BOUNDS_CHECK
         if (i>dl_.size()-1)
            return outofbounds_;
         else
#endif
            // Before lapackpp-2.4.12 this was dl_(i) but that was WRONG!
            return dl_(j);
      case -1:   // upper
#ifdef LA_BOUNDS_CHECK
         if (i>du_.size()-1)
            return outofbounds_;
         else
#endif
            return du_(i);
      default:
         return outofbounds_;
   }
}


inline double LaTridiagMatDouble::operator()(int i,int j) const
{
   switch (i-j)
   {
      case 0:   // main
#ifdef LA_BOUNDS_CHECK
         if (i>d_.size()-1)
            return outofbounds_;
         else
#endif
            return d_(i);
      case 1:  // lower
#ifdef LA_BOUNDS_CHECK
         if (i>dl_.size()-1)
            return outofbounds_;
         else
#endif
            // Before lapackpp-2.4.12 this was dl_(i) but that was WRONG!
            return dl_(j);
      case -1:   // upper
#ifdef LA_BOUNDS_CHECK
         if (i>du_.size()-1)
            return outofbounds_;
         else
#endif
            return du_(i);
      default:
         return outofbounds_;
   }
}

inline LaTridiagMatDouble& LaTridiagMatDouble::ref(LaTridiagMatDouble&T) 
{
    du2_.ref(T.du2_);
    du_.ref(T.du_);
    d_.ref(T.d_);
    dl_.ref(T.dl_); 
    size_ = T.size_;

    return *this;
}







#endif 
// _LA_TRIDIAG_MAT_DOUBLE_

---