amg.hpp

Go to the documentation of this file.

#ifndef PARALUTION_AMG_HPP_
#define PARALUTION_AMG_HPP_

#include "../solver.hpp"
#include "base_amg.hpp"

#include <vector>

namespace paralution {

enum _interp {
  Aggregation,
  SmoothedAggregation
};

template <class OperatorType, class VectorType, typename ValueType>
class AMG : public BaseAMG<OperatorType, VectorType, ValueType> {
  
public:

  AMG();
  virtual ~AMG();

  virtual void Print(void) const;

  virtual void BuildSmoothers(void);

  virtual void SetCouplingStrength(const ValueType eps);
  virtual void SetInterpolation(_interp interpType);
  virtual void SetInterpRelax(const ValueType relax);
  virtual void SetOverInterp(const ValueType overInterp);

protected:

  virtual void Aggregate(const OperatorType &op,
                         Operator<ValueType> *pro,
                         Operator<ValueType> *res,
                         OperatorType *coarse);

  virtual void PrintStart_(void) const;
  virtual void PrintEnd_(void) const;

private:

  ValueType eps_;

  ValueType relax_;

  ValueType over_interp_;
  
  _interp interp_type_;

};


}

#endif // PARALUTION_AMG_HPP_