/home/brandes/workspace/LAMA/src/lama/solver/SimpleAMG.hpp

Go to the documentation of this file.

#ifndef LAMA_SIMPLEAMG_HPP_
#define LAMA_SIMPLEAMG_HPP_

// for dll_import
#include <lama/config.hpp>

// base classes
#include <lama/solver/IterativeSolver.hpp>

// others
#include <lama/solver/AMGSetup.hpp>

#include <vector>

namespace lama
{

class LAMA_DLL_IMPORTEXPORT SimpleAMG: public lama::IterativeSolver
{
public:

    SimpleAMG( const std::string& id );

    SimpleAMG( const std::string& id, LoggerPtr logger );

    SimpleAMG( const SimpleAMG& other );

    virtual ~SimpleAMG();

    virtual void initialize(const Matrix&  coefficients);

    virtual void iterate();

    void setMaxLevels( unsigned int levels );

    void setMinVarsCoarseLevel( unsigned int vars );

    unsigned int getNumLevels ();

    const Matrix& getGalerkin( unsigned int level );
    const Matrix& getRestriction( unsigned int level );
    const Matrix& getInterpolation( unsigned int level );

    Vector& getSolutionVector( unsigned int level );
    Vector& getRhsVector( unsigned int level );

    Solver& getSmoother( unsigned int level );
    Solver& getCoarseLevelSolver();

    void setSmootherContext( ContextPtr smootherContext );

    void setHostOnlyLevel( IndexType hostOnlyLevel );

    void setReplicatedLevel( IndexType replicatedLevel );

    void setCoarseLevelSolver( SolverPtr solver );

    void setSmoother( SolverPtr solver );

    struct SimpleAMGRuntime : IterativeSolverRuntime
    {
        SimpleAMGRuntime();
        virtual ~SimpleAMGRuntime();

        std::auto_ptr<AMGSetup> mSetup;
        unsigned int mCurrentLevel;
        void* mLibHandle;
        IndexType mHostOnlyLevel;
        IndexType mReplicatedLevel;

        LAMA_LOG_DECL_STATIC_LOGGER(logger);
    };

    virtual SimpleAMGRuntime& getRuntime();

    virtual const SimpleAMGRuntime& getConstRuntime() const;

    virtual SolverPtr copy();

protected:

    SimpleAMGRuntime mSimpleAMGRuntime;

    unsigned int mMaxLevels;
    unsigned int mMinVarsCoarseLevel;
    SolverPtr mCoarseLevelSolver;
    SolverPtr mSmoother;
    ContextPtr mSmootherContext;

private:

    void logSetupSettings();
    void logSetupInfo();
    void logSolverInfo();
    void logSetupDetails();

    LAMA_LOG_DECL_STATIC_LOGGER(logger);

    void cycle();
};

}

#endif /* LAMA_SIMPLEAMG_HPP_ */