pardiso.hh

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
#ifndef DUNE_ISTL_PARDISO_HH
#define DUNE_ISTL_PARDISO_HH
 
#include <dune/istl/preconditioners.hh>
#include <dune/istl/solvertype.hh>
/* Change this, if your Fortran compiler does not append underscores. */
/* e.g. the AIX compiler:  #define F77_FUNC(func) func                */
 
#ifdef AIX
#define F77_FUNC(func)  func
#else
#define F77_FUNC(func)  func ## _
#endif
 
 
#ifdef HAVE_PARDISO
/* PARDISO prototype. */
extern "C" int F77_FUNC(pardisoinit)
  (void *, int *, int *);
 
extern "C" int F77_FUNC(pardiso)
  (void *, int *, int *, int *, int *, int *,
  double *, int *, int *, int *, int *, int *,
  int *, double *, double *, int *);
#endif
 
namespace Dune {
 
 
  template<class M, class X, class Y>
  class SeqPardiso : public Preconditioner<X,Y> {
  public:
    typedef M matrix_type;
    typedef X domain_type;
    typedef Y range_type;
    typedef typename X::field_type field_type;
 
    typedef typename M::RowIterator RowIterator;
    typedef typename M::ColIterator ColIterator;
 
    // define the category
    enum {
      category=SolverCategory::sequential
    };
 
    SeqPardiso (const M& A)
      : A_(A)
    {
#ifdef HAVE_PARDISO
 
      mtype_ = 11;
      nrhs_ = 1;
      num_procs_ = 1;
      maxfct_ = 1;
      mnum_   = 1;
      msglvl_ = 0;
      error_  = 0;
 
      n_ = A_.rowdim();
      int nnz = 0;
      RowIterator endi = A_.end();
      for (RowIterator i = A_.begin(); i != endi; ++i)
      {
        if (A_.rowdim(i.index()) != 1)
          DUNE_THROW(NotImplemented, "SeqPardiso: row blocksize != 1.");
        ColIterator endj = (*i).end();
        for (ColIterator j = (*i).begin(); j != endj; ++j) {
          if (A_.coldim(j.index()) != 1)
            DUNE_THROW(NotImplemented, "SeqPardiso: column blocksize != 1.");
          nnz++;
        }
      }
 
      std::cout << "dimension = " << n_ << ", number of nonzeros = " << nnz << std::endl;
 
      a_ = new double[nnz];
      ia_ = new int[n_+1];
      ja_ = new int[nnz];
 
      int count = 0;
      for (RowIterator i = A_.begin(); i != endi; ++i)
      {
        ia_[i.index()] = count+1;
        ColIterator endj = (*i).end();
        for (ColIterator j = (*i).begin(); j != endj; ++j) {
          a_[count] = *j;
          ja_[count] = j.index()+1;
 
          count++;
        }
      }
      ia_[n_] = count+1;
 
      F77_FUNC(pardisoinit) (pt_,  &mtype_, iparm_);
 
      int phase = 11;
      int idum;
      double ddum;
      iparm_[2]  = num_procs_;
 
      F77_FUNC(pardiso) (pt_, &maxfct_, &mnum_, &mtype_, &phase,
                         &n_, a_, ia_, ja_, &idum, &nrhs_,
                         iparm_, &msglvl_, &ddum, &ddum, &error_);
 
      if (error_ != 0)
        DUNE_THROW(MathError, "Constructor SeqPardiso: Factorization failed. Error code " << error_);
 
      std::cout << "Constructor SeqPardiso: Factorization completed." << std::endl;
 
#else
      DUNE_THROW(NotImplemented, "no Pardiso library available, reconfigure with correct --with-pardiso options");
#endif
    }
 
    virtual void pre (X& x, Y& b) {}
 
    virtual void apply (X& v, const Y& d)
    {
#ifdef HAVE_PARDISO
      int phase = 33;
 
      iparm_[7] = 1;         /* Max numbers of iterative refinement steps. */
      int idum;
 
      double x[n_];
      double b[n_];
      for (int i = 0; i < n_; i++) {
        x[i] = v[i];
        b[i] = d[i];
      }
 
      F77_FUNC(pardiso) (pt_, &maxfct_, &mnum_, &mtype_, &phase,
                         &n_, a_, ia_, ja_, &idum, &nrhs_,
                         iparm_, &msglvl_, b, x, &error_);
 
      if (error_ != 0)
        DUNE_THROW(MathError, "SeqPardiso.apply: Backsolve failed. Error code " << error_);
 
      for (int i = 0; i < n_; i++)
        v[i] = x[i];
 
      std::cout << "SeqPardiso: Backsolve completed." << std::endl;
#endif
    }
 
    virtual void post (X& x) {}
 
    ~SeqPardiso()
    {
#ifdef HAVE_PARDISO
      int phase = -1;                   // Release internal memory.
      int idum;
      double ddum;
 
      F77_FUNC(pardiso) (pt_, &maxfct_, &mnum_, &mtype_, &phase,
                         &n_, &ddum, ia_, ja_, &idum, &nrhs_,
                         iparm_, &msglvl_, &ddum, &ddum, &error_);
      delete[] a_;
      delete[] ia_;
      delete[] ja_;
#endif
    }
 
  private:
    M A_; 
    int n_; 
    double *a_; 
    int *ia_; 
    int *ja_; 
    int mtype_; 
    int nrhs_; 
    void *pt_[64]; 
    int iparm_[64]; 
    int maxfct_;        
    int mnum_;  
    int msglvl_;    
    int error_;      
    int num_procs_; 
  };
 
  template<class M, class X, class Y>
  struct IsDirectSolver<SeqPardiso<M,X,Y> >
  {
    enum { value=true};
  };
 
 
} // end namespace Dune
#endif