itkFEMItpackSparseMatrix.h

Go to the documentation of this file.

/*=========================================================================

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkFEMItpackSparseMatrix.h,v $
  Language:  C++
  Date:      $Date: 2009-01-29 21:55:13 $
  Version:   $Revision: 1.14 $

  Copyright (c) Insight Software Consortium. All rights reserved.
  See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even 
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/

#ifndef __itkFEMItpackSparseMatrix_h
#define __itkFEMItpackSparseMatrix_h

#include "itkFEMException.h"


namespace itk {
namespace fem {

// Forward declaration of friend class
class ItpackLinearSystemWrapper;
  
class ItpackSparseMatrix
{
public:

  typedef long      integer;
  typedef double    doublereal;

  ItpackSparseMatrix();
  
  ItpackSparseMatrix(integer order);


  ItpackSparseMatrix(integer order, integer maxNonZeroValues);

  ~ItpackSparseMatrix();

  void SetOrder(integer order) { m_N = order; } 

  void SetMaxNonZeroValues(integer maxNonZeroValues) { m_NZ = maxNonZeroValues; }

  void Set(integer i, integer j, doublereal value);

  void Add(integer i, integer j, doublereal value);

  doublereal Get(integer i, integer j);

  integer*    GetN()   { return &m_N; }

  integer*     GetIA();

  void  SetCompressedRow(integer *ia, integer *ja, doublereal *a);
  integer*     GetJA();

  doublereal*  GetA();

  doublereal* GetValueArray()       { return GetA();  }

  integer*    GetColumnArray()      { return GetJA(); }

  integer*    GetRowArray()         { return GetIA(); }

  integer     GetOrder()       const     { return m_N; }

  integer     GetMaxNonZeroValues() const { return m_NZ; }

  void Clear();

  void mult(doublereal* vector, doublereal* result);

  void mult(ItpackSparseMatrix* rightMatrix, ItpackSparseMatrix* resultMatrix);

  void PrintCompressedRow();

private:

  friend class LinearSystemWrapperItpack;

  void Initialize();

  void UnFinalize();

  void Finalize();

  integer m_MatrixFinalized;

  integer m_MatrixInitialized;
  
  integer m_N;

  integer m_NZ;

  integer *m_IA;

  integer *m_JA;

  doublereal *m_A;

  integer *m_IWORK;

  integer m_MODE;

  integer m_NOUT;
  
  integer m_LEVEL;

  //integer m_IER;
};

class FEMExceptionItpackSparseMatrixSbagn : public FEMException
{
public:

  typedef long      integer;
  typedef double    doublereal;

  FEMExceptionItpackSparseMatrixSbagn(const char *file, unsigned int lineNumber, std::string location, integer errorCode);
 
  virtual ~FEMExceptionItpackSparseMatrixSbagn() throw() {}
  
  itkTypeMacro(FEMExceptionItpackSparseMatrixSbagn,FEMException);
  
};

class FEMExceptionItpackSparseMatrixSbsij : public FEMException
{
public:
  typedef long      integer;
  typedef double    doublereal;

  FEMExceptionItpackSparseMatrixSbsij(const char *file, unsigned int lineNumber, std::string location, integer errorCode);
 
  virtual ~FEMExceptionItpackSparseMatrixSbsij() throw() {}
  
  itkTypeMacro(FEMExceptionItpackSparseMatrixSbsij,FEMException);
  
};

}} // end namespace itk::fem

#endif