RectangularDomain.h

Go to the documentation of this file.

#ifndef RECTANGULAR_DOMAIN_H
#define RECTANGULAR_DOMAIN_H
#ifdef HAVE_SAAMG_SOLVER

#include <vector>
#include <string>
#include "IrregularDomain.h"

class RectangularDomain : public IrregularDomain {

public:

    RectangularDomain(Vector_t nr, Vector_t hr);
    RectangularDomain(double a, double b, Vector_t nr, Vector_t hr);

    void compute(Vector_t hr);

    int getNumXY(int z);
    void getBoundaryStencil(int x, int y, int z, double &W, double &E, double &S, double &N, double &F, double &B, double &C, double &scaleFactor);
    void getBoundaryStencil(int idx, double &W, double &E, double &S, double &N, double &F, double &B, double &C, double &scaleFactor);
    using IrregularDomain::getNeighbours;
    void getNeighbours(int x, int y, int z, double &W, double &E, double &S, double &N, double &F, double &B);
    void getNeighbours(int idx, double &W, double &E, double &S, double &N, double &F, double &B);
    std::string getType() {return "Rectangular";}
    inline bool isInside(int x, int y, int z) {
        double xx = (x - (nr[0] - 1) / 2.0) * hr[0];
        double yy = (y - (nr[1] - 1) / 2.0) * hr[1];
        return (xx <= a_m && yy < b_m);
    }

    void setB_m(double b) {b_m = b;}
    void setA_m(double a) {a_m = a;}

    double getXRangeMin() { return -a_m; }
    double getXRangeMax() { return a_m; }
    double getYRangeMin() { return -b_m; }
    double getYRangeMax() { return b_m; }
    double getZRangeMin() { return getMinZ(); }
    double getZRangeMax() { return getMaxZ(); }


    int getStartIdx() {return 0;}

private:

    double a_m;
    double b_m;
    int nxy_m;

    inline int getIdx(int x, int y, int z) {
        if(isInside(x, y, z) && x >= 0 && y >= 0 && z >= 0)
            return y * nr[0] + x + z * nxy_m;
        else
            return -1;
    }
    inline void getCoord(int idx, int &x, int &y, int &z) {
        int ixy = idx % nxy_m;
        int inr = nr[0];
        x = ixy % inr;
        y = (ixy - x) / nr[0];
        z = (idx - ixy) / nxy_m;
    }

};

#endif //#ifdef HAVE_SAAMG_SOLVER
#endif //#ifdef RECTANGULAR_DOMAIN_H