diagpre.h

Go to the documentation of this file.

/*+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++*/
/*             ********   ***                                 SparseLib++    */
/*          *******  **  ***       ***      ***               v. 1.5c        */
/*           *****      ***     ******** ********                            */
/*            *****    ***     ******** ********              R. Pozo        */
/*       **  *******  ***   **   ***      ***                 K. Remington   */
/*        ********   ********                                 A. Lumsdaine   */
/*+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++*/
/*                                                                           */
/*                                                                           */
/*                     SparseLib++ : Sparse Matrix Library                   */
/*                                                                           */
/*               National Institute of Standards and Technology              */
/*                        University of Notre Dame                           */
/*              Authors: R. Pozo, K. Remington, A. Lumsdaine                 */
/*                                                                           */
/*                                 NOTICE                                    */
/*                                                                           */
/* Permission to use, copy, modify, and distribute this software and         */
/* its documentation for any purpose and without fee is hereby granted       */
/* provided that the above notice appear in all copies and supporting        */
/* documentation.                                                            */
/*                                                                           */
/* Neither the Institutions (National Institute of Standards and Technology, */
/* University of Notre Dame) nor the Authors make any representations about  */
/* the suitability of this software for any purpose.  This software is       */
/* provided ``as is'' without expressed or implied warranty.                 */
/*                                                                           */
/*+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++*/

#ifndef diagpre_h
#define diagpre_h

#include "floatarray.h"
#include "sparsemtrx.h"
#include "precond.h"

namespace oofem {
class OOFEM_EXPORT DiagPreconditioner : public Preconditioner
{
private:
    FloatArray diag_;

public:
    DiagPreconditioner(const SparseMtrx &, InputRecord & attributes);
    DiagPreconditioner() : Preconditioner() { }
    virtual ~DiagPreconditioner(void) { }

    virtual void init(const SparseMtrx &a);

    void solve(const FloatArray &rhs, FloatArray &solution) const;
    void trans_solve(const FloatArray &rhs, FloatArray &solution) const;

    virtual const char *giveClassName() const { return "DiagPre"; }

    const double &diag(int i) const { return diag_(i); }
    double &diag(int i) { return diag_(i); }
};
} // end namespace oofem
#endif // diagpre_h