SMatrix.h

Go to the documentation of this file.

/***************************************************************************
 *   Copyright (C) 2004 by Claudio Attaccalite                             *
 *   claudio@freescience.info                                              *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program 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 General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/


#ifndef SMatrix_H
#define SMatrix_H
#if HAVE_CONFIG
#include <config.h>
#endif

#include <iostream>
#include "SVector.h"
#include "lapack_wrapper.h"
#include "blas_wrapper.h"


using namespace std;


class SMatrix
{
   public:
          friend class TMatrix;
  private:
  int n_1;


  int n_2;

  float* pArray;

  float* tmpArray;

  float temp;

  int one;

public:
  int dim1(){return n_1;}

  int dim2(){return n_2;}

  int size(){return n_1*n_2;}

  float* array(){return pArray;}

 void transpose()
 {
 if(n_1 !=n_2)
 {
         cerr << " Transpose only for square matrix ! " << endl;
         exit(1);
 }
 for(int i=0;i<n_1;i++)
        for(int j=0;j<i;j++)
        {
                temp=pArray[j+i*n_1];
                pArray[j+i*n_1]=pArray[i+j*n_1];
                pArray[i+j*n_1]=temp;
        }
 }


  void allocate(int i, int j)
    {
      if( n_1 != i || n_2 != j )
        {
          deallocate();
          
          n_1 = i;
          n_2 = j;
          
          if(n_1 >= 1 && n_2 >= 1)
              pArray = new float[n_1*n_2];
          else 
              pArray = 0;
        }
    }


  void deallocate()
    {
      if(n_2>0 && n_1>0)
      {
       delete [] pArray;
      }
       pArray = 0;

      n_1 = 0;
      n_2 = 0;
    }


  void operator=(const SMatrix & rhs)
    {
     int tot=n_1*n_2;
#ifdef DEBUG
     if(n_1 != rhs.n_1 || n_2 != rhs.n_2) allocate(rhs.n_1,rhs.n_2);
#endif
     
         F77_FUNC(scopy,SCOPY)(&tot,rhs.pArray,&one,pArray,&one);
    }


  void operator=(const float C)
    {
      for(int i=0; i<n_1*n_2;i++) pArray[i] = C;
    }

  void operator*=(float C)
    {
        int tot=n_1*n_2;
        F77_FUNC(sscal,SSCAL)(&tot,&C,pArray,&one);
    }

  void operator/=(const float C)
    {
        int tot=n_1*n_2;
        float inv=1.0/C;
        F77_FUNC(sscal,SSCAL)(&tot,&inv,pArray,&one);
    }


  SMatrix(){pArray = 0; n_1 = 0; n_2 = 0; one = 1;}


  SMatrix(int i, int j) {pArray = 0; n_1 = 0; n_2 = 0; one=1; allocate(i,j);}


  SMatrix( const SMatrix & rhs)
    {
      n_1 = 0;
      n_2 = 0;
      pArray = 0;
      allocate(rhs.n_1,rhs.n_2);
      int tot=n_1*n_2;
      one=1;
      F77_FUNC(scopy,SCOPY)(&tot,rhs.pArray,&one,pArray,&one);
    }


  ~SMatrix(){deallocate();}

  float SumAll()
  {
   float sum=0;
   for(int i=0;i<n_1*n_2;i++) sum+=pArray[i];
   return sum;
  }
 
  float Trace()
  {
#ifdef DEBUG
   if(n_1 !=n_2)
   {
         cerr << " Trace only for Squared SMatrix ! " << endl;
         exit(1);
   } 
#endif
   float sum=0;
   for(int i=0;i<n_1;i++) sum+=pArray[i*n_1+i];
   return sum;
  }
  
  void setrow(int irow,SVector &vector)
  {
#ifdef DEBUG
          if(vector.n_1!=n_2 || irow>=n_1)
          { 
                  cerr << " ERROR! dimension array copying row! " << endl;
                  exit(1);
          }
#endif
   F77_FUNC(scopy,SCOPY)(&n_2,vector.pArray,&one,pArray+irow,&n_1);
  }

  void setrow(int irow,float *vector)
  {
   F77_FUNC(scopy,SCOPY)(&n_2,vector,&one,pArray+irow,&n_1);
  }

  void setrow(int irow,float *vector,int lda)
  {
   F77_FUNC(scopy,SCOPY)(&n_2,vector,&lda,pArray+irow,&n_1);
  }
  
  void setcol(int icol, const SVector & vector)
  {
#ifdef DEBUG
          if(vector.n_1!=n_1 || icol>=n_2)
          { 
                  cerr << " ERROR! dimension array copying coloumn! " << endl;
                  exit(1);
          }
#endif
   F77_FUNC(scopy,SCOPY)(&n_1,vector.pArray,&one,pArray+icol*n_1,&one);
  }

  void setcol(int icol,float *vector)
  {
#ifdef DEBUG
          if(icol>=n_2)
          { 
                  cerr << " ERROR! dimension array copying coloumn! " << endl;
                  exit(1);
          }
#endif
   F77_FUNC(scopy,SCOPY)(&n_1,vector,&one,pArray+icol*n_1,&one);
  }
 
  void row_scale(int irow, float C)
  {
#ifdef DEBUG
          if(irow>=n_1)
          { 
                  cerr << " ERROR! dimension array in row_scale! " << endl;
                  exit(1);
          }
#endif
   F77_FUNC(sscal,SSCAL)(&n_2,&C,pArray+irow,&n_1);
  }



  
  float rowdot(int irow, const SVector & vector)
  {
#ifdef DEBUG
          if(vector.n_1!=n_2 || irow>=n_1)
          { 
                  cerr << " ERROR! dimension array in row dot! " << endl;
                  exit(1);
          }
#endif
   return F77_FUNC(sdot,SDOT)(&n_2,pArray+irow,&n_1,vector.pArray,&one);
  }

  float coldot(int icol, const SVector &vector)
  {
#ifdef DEBUG
          if(vector.n_1!=n_1 || icol>=n_2)
          { 
                  cerr << " ERROR! dimension array in coloumn dot! " << endl;
                  exit(1);
          }
#endif
   return F77_FUNC(sdot,SDOT)(&n_1,pArray+icol*n_1,&one,vector.pArray,&one);
  }
  
  void col_scale(int icol, float C)
  {
#ifdef DEBUG
          if(icol>=n_2)
          { 
                  cerr << " ERROR! dimension array in col_rescale! " << endl;
                  exit(1);
          }
#endif
   F77_FUNC(sscal,SSCAL)(&n_1,&C,pArray+icol*n_1,&one);
  }


  float rowdot(int irow, float *vector)
  {
   return F77_FUNC(sdot,SDOT)(&n_2,pArray+irow,&n_1,vector,&one);
  }

  float coldot(int icol, float *vector)
  {
   return F77_FUNC(sdot,SDOT)(&n_1,pArray+icol*n_1,&one,vector,&one);
  }



  float *row_pointer(int irow)
  {
#ifdef DEBUG
          if(irow>=n_1)
          { 
                  cerr << " ERROR! dimension array getting a row! " << endl;
                  exit(1);
          }
#endif
   return(pArray+irow);
  }
  
  SVector row(int irow)
  {
#ifdef DEBUG
          if(irow>=n_1)
          { 
                  cerr << " ERROR! dimension array getting a row! " << endl;
                  exit(1);
          }
#endif
   SVector TEMP(n_2);
   F77_FUNC(scopy,SCOPY)(&n_2,pArray+irow,&n_1,TEMP.pArray,&one);
   return TEMP;
  }




  float *col_pointer(int icol)
  {
#ifdef DEBUG
          if(icol>=n_2)
          { 
                  cerr << " ERROR! dimension array getting a coloumn! " << endl;
                  exit(1);
          }
#endif
   return(pArray+icol*n_1);
  }


  
  SVector col(int icol)
  {
#ifdef DEBUG
          if(icol>=n_2)
          { 
                  cerr << " ERROR! dimension array getting a coloumn! " << endl;
                  exit(1);
          }
#endif
   SVector TEMP(n_1);
   F77_FUNC(scopy,SCOPY)(&n_1,pArray+icol*n_1,&one,TEMP.pArray,&one);
   return TEMP;
  }

  float SumRow(const int irow)
  {
#ifdef DEBUG
          if(irow>=n_1)
          { 
                  cerr << " ERROR! dimension array getting a row! " << endl;
                  exit(1);
          }
#endif
   float sum=0;
   for(int i=0;i<n_2;i++) sum+=pArray[irow+i*n_1];
   return sum;
  }


  float SumCol(const int icol)
  {
#ifdef DEBUG
          if(icol>=n_2)
          { 
                  cerr << " ERROR! dimension array getting a row! " << endl;
                  exit(1);
          }
#endif
   float sum=0;
   for(int i=0;i<n_1;i++) sum+=pArray[i+icol*n_1];
   return sum;
  }





  SMatrix operator+( const SMatrix & rhs)
    {
#ifdef DEBUG
      if( n_1 != rhs.n_1 || n_2 !=rhs.n_2 )
        {
          cerr << "ERROR: incorrect sizes for two SMatrix's in SMatrix+"
               << "SMatrix" << endl;
          exit(1);
        }
      if( n_1 <= 0 || n_2 <=0 )
        {
          cerr << "ERROR: SMatrix of size 0 used in SMatrix+SMatrix" << endl;
          exit(1);
        }
#endif
      int tot=n_1*n_2;
      float d_one=1.0;
      SMatrix  A(n_1,n_2);
      F77_FUNC(scopy,SCOPY)(&tot,pArray,&one,A.pArray,&one);
      F77_FUNC(saxpy,SAXPY)(&tot,&d_one,rhs.pArray,&one,A.pArray,&one); 
      return A;
    }


  SMatrix operator-( const SMatrix & rhs)
    {
#ifdef DEBUG
      if( n_1 != rhs.n_1 || n_2 !=rhs.n_2 )
        {
          cerr << "ERROR: incorrect sizes for two SMatrix's in SMatrix-"
               << "SMatrix" << endl;
          exit(1);
        }
      if( n_1 <= 0 || n_2 <=0)
        {
          cerr << "ERROR: SMatrix of size 0 used in SMatrix-SMatrix" << endl;
          exit(1);
        }
#endif
      SMatrix A(n_1,n_2);
      int tot=n_1*n_2;
      float d_m_one=-1.0;
      F77_FUNC(scopy,SCOPY)(&tot,pArray,&one,A.pArray,&one);
      F77_FUNC(saxpy,SAXPY)(&tot,&d_m_one,rhs.pArray,&one,A.pArray,&one); 
      return A;
    }

  
  float & operator()(int i,int j)
  {
#ifdef DEBUG
          if(i>=n_1 || j>=n_2)
          { 
                  cerr << " ERROR! index out of range! " 
                       << "("<<i<<","<<j<<")" <<  endl;
                  exit(1);
          }
#endif
          return pArray[i+j*n_1];
  }


  friend ostream& operator<<(ostream & strm, const SMatrix & rhs)
    {
      for(int i=0; i<rhs.n_1;i++)
        {
          for(int j=0; j<rhs.n_2; j++)
            {
              strm << rhs.pArray[i+j*rhs.n_1] << "\t";
            }
          strm << endl;
        }
      return strm;
    }

};

#endif

---