Dissipation.h

#ifndef gridripper_amr1d_Dissipation_h
#define gridripper_amr1d_Dissipation_h

#include <gridripper/config.h>
#include <gridripper/lang/IllegalArgumentException.h>
#include <gridripper/tvalarray.h>
#include <string>
#include "Sigma.h"

namespace gridripper { namespace amr1d {

class Grid;
class PDE;
class FieldWrapper;

using namespace std;

class Dissipation {

private:
    const Sigma* sigma;

public:
    Dissipation() {
        sigma = NULL;
    }

    virtual ~Dissipation() { }

    virtual void addDissipation(const Grid& g, const PDE* pde, int i,
                                tvalarray<GReal_t>& dfdt, int exclude) =0;

    const Sigma& getSigma() const {
        return *sigma;
    }

    void setSigma(const Sigma& s) {
        sigma = &s;
    }

    virtual string toString() const =0;

protected:
    GReal_t evalSigma(const Grid& g, int i) const;
};

class O2Dissipation: public Dissipation
{
private:
    FieldWrapper& fw0;
    FieldWrapper& fw1;
    FieldWrapper& fw2;

public:
    O2Dissipation(const PDE& pde, int level);
    ~O2Dissipation();

    void addDissipation(const Grid& g, const PDE* pde, int i,
                        tvalarray<GReal_t>& dfdt, int exclude)
        throw(IllegalArgumentException&);

    string toString() const {
        return "Dissipation.O2";
    }

private:
    void addExtendedFields(const Grid& g, const PDE* pde, int i,
                           tvalarray<GReal_t>& dfdt, GReal_t eps, int exclude);
};

class O4Dissipation: public Dissipation
{
private:
    FieldWrapper& fw0;
    FieldWrapper& fw1;
    FieldWrapper& fw2;
    FieldWrapper& fw3;
    FieldWrapper& fw4;

public:
    O4Dissipation(const PDE& pde, int level);
    ~O4Dissipation();

    void addDissipation(const Grid& g, const PDE* pde, int i,
                        tvalarray<GReal_t>& dfdt, int exclude)
        throw(IllegalArgumentException&);

    string toString() const {
        return "Dissipation.O4";
    }

private:
    void addExtendedFields(const Grid& g, const PDE* pde, int i,
                           tvalarray<GReal_t>& dfdt, GReal_t eps, int exclude);
};

class O6Dissipation: public Dissipation
{
private:
    FieldWrapper& fw0;
    FieldWrapper& fw1;
    FieldWrapper& fw2;
    FieldWrapper& fw3;
    FieldWrapper& fw4;
    FieldWrapper& fw5;
    FieldWrapper& fw6;

public:
    O6Dissipation(const PDE& pde, int level);
    ~O6Dissipation();

    void addDissipation(const Grid& g, const PDE* pde, int i,
                        tvalarray<GReal_t>& dfdt, int exclude)
        throw(IllegalArgumentException&);

    string toString() const {
        return "Dissipation.O6";
    }

private:
    void addExtendedFields(const Grid& g, const PDE* pde, int i,
                           tvalarray<GReal_t>& dfdt, GReal_t eps, int exclude);
};

} } // namespace gridripper::amr1d

#endif /* gridripper_amr1d_Dissipation_h */