FM3DMagnetoStaticExtended.h

Go to the documentation of this file.

#ifndef CLASSIC_FM2DMAGNETOSTATICEXTENDED_HH
#define CLASSIC_FM2DMAGNETOSTATICEXTENDED_HH

#include "Fields/Fieldmap.h"

class FM3DMagnetoStaticExtended: public Fieldmap {

public:
    virtual bool getFieldstrength(const Vector_t &R, Vector_t &E, Vector_t &B) const;
    virtual bool getFieldDerivative(const Vector_t &R, Vector_t &E, Vector_t &B, const DiffDirection &dir) const;
    virtual void getFieldDimensions(double &zBegin, double &zEnd, double &rBegin, double &rEnd) const;
    virtual void getFieldDimensions(double &xIni, double &xFinal, double &yIni, double &yFinal, double &zIni, double &zFinal) const;
    virtual void swap();
    virtual void getInfo(Inform *msg);
    virtual double getFrequency() const;
    virtual void setFrequency(double freq);

    virtual bool isInside(const Vector_t &r) const;
private:
    FM3DMagnetoStaticExtended(std::string aFilename);
    ~FM3DMagnetoStaticExtended();

    virtual void readMap();
    virtual void freeMap();

    struct IndexTriplet {
        unsigned int i;
        unsigned int j;
        unsigned int k;
        Vector_t weight;
        IndexTriplet():
            i(0),
            j(0),
            k(0),
            weight(0.0)
        { }
    };

    IndexTriplet getIndex(const Vector_t &X) const;
    unsigned long getIndex(unsigned int i, unsigned int j, unsigned int k) const;
    Vector_t interpolateTrilinearly(const Vector_t &X) const;
    double getWeightedData(double *data, const IndexTriplet &idx, unsigned short corner) const;

    enum { LX = 0,
           LY = 0,
           LZ = 0,
           HX = 4,
           HY = 2,
           HZ = 1};

    void integrateBx(unsigned j);
    void integrateBz(unsigned j);
    void integrateBy(unsigned j);

    void smoothData(double * data, unsigned j);

    void saveField(const std::string &fname, unsigned int j) const;

    double *FieldstrengthBx_m;    
    double *FieldstrengthBy_m;    
    double *FieldstrengthBz_m;    
    double xbegin_m;
    double xend_m;
    double ybegin_m;
    double yend_m;
    double zbegin_m;
    double zend_m;
    double length_m;
    double hx_m;                   
    double hy_m;                   
    double hz_m;                   
    unsigned int num_gridpx_m;     
    unsigned int num_gridpy_m;     
    unsigned int num_gridpz_m;     
    friend class Fieldmap;
};

inline
bool FM3DMagnetoStaticExtended::isInside(const Vector_t &r) const
{
    return r(2) >= 0.0 && r(2) < length_m && r(0) >= xbegin_m && r(0) < xend_m && std::abs(r(1)) < yend_m;
    // return r(2) >= zbegin_m && r(2) < zend_m && r(0) >= xbegin_m && r(0) < xend_m && std::abs(r(1)) < yend_m;
}

inline
unsigned long FM3DMagnetoStaticExtended::getIndex(unsigned int i, unsigned int j, unsigned int k) const
{
    unsigned long result = i + j * num_gridpx_m;
    result = k + result * num_gridpz_m;
    PAssert_LT(result, num_gridpx_m * num_gridpy_m * num_gridpz_m);
    return result;
}

inline
FM3DMagnetoStaticExtended::IndexTriplet FM3DMagnetoStaticExtended::getIndex(const Vector_t &X) const {
    IndexTriplet idx;
    idx.i = std::floor((X(0) - xbegin_m) / hx_m);
    idx.j = std::floor(std::abs(X(1)) / hy_m);
    idx.k = std::floor((X(2)) / hz_m);
    // idx.k = std::floor((X(2) - zbegin_m) / hz_m);

    idx.weight(0) = (X(0) - xbegin_m) / hx_m - idx.i;
    idx.weight(1) = std::abs(X(1)) / hy_m - idx.j;
    idx.weight(2) = X(2) / hz_m - idx.k;

    return idx;
}

#endif