FM1DElectroStatic_fast.h

Go to the documentation of this file.

#ifndef CLASSIC_FIELDMAP1DELECTROSTATICFAST_HH
#define CLASSIC_FIELDMAP1DELECTROSTATICFAST_HH

#include "Fields/Fieldmap.h"

class _FM1DElectroStatic_fast: public _Fieldmap {

public:
    virtual ~_FM1DElectroStatic_fast();

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

    virtual bool isInside(const Vector_t &r) const;
private:
    _FM1DElectroStatic_fast(const std::string& filename);

    static FM1DElectroStatic_fast create(const std::string& filename);

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

    bool checkFileData(std::ifstream &fieldFile, bool parsingPassed);
    void computeFieldDerivatives(std::vector<double> fourierCoefs,
                                 double onAxisFieldP[], double onAxisFieldPP[],
                                 double onAxisFieldPPP[]);
    void computeFieldOffAxis(const Vector_t &R, Vector_t &E, Vector_t &B,
                             std::vector<double> fieldComponents) const;
    void computeFieldOnAxis(double z, std::vector<double> &fieldComponents) const;
    std::vector<double> computeFourierCoefficients(double fieldData[]);
    void computeInterpolationVectors(double onAxisFieldP[],
                                     double onAxisFieldPP[],
                                     double onAxisFieldPPP[]);
    void convertHeaderData();
    void normalizeField(double maxEz, std::vector<double> &fourierCoefs);
    double readFileData(std::ifstream &fieldFile, double fieldData[]);
    bool readFileHeader(std::ifstream &fieldFile);
    void stripFileHeader(std::ifstream &fieldFile);

    void prepareForMapCheck(std::vector<double> &fourierCoefs);

    double rBegin_m;                        
    double rEnd_m;                          
    double zBegin_m;                        
    double zEnd_m;                          
    double length_m;                        
    unsigned int numberOfGridPoints_m;      
    double deltaZ_m;                        

    unsigned int accuracy_m;

    double* onAxisField_m;                      
    gsl_spline *onAxisFieldInterpolants_m;      
    gsl_spline *onAxisFieldPInterpolants_m;     
    gsl_spline *onAxisFieldPPInterpolants_m;    
    gsl_spline *onAxisFieldPPPInterpolants_m;   

    gsl_interp_accel *onAxisFieldAccel_m;
    gsl_interp_accel *onAxisFieldPAccel_m;
    gsl_interp_accel *onAxisFieldPPAccel_m;
    gsl_interp_accel *onAxisFieldPPPAccel_m;

    friend class _Fieldmap;
};

inline bool _FM1DElectroStatic_fast::isInside(const Vector_t &r) const
{
    return r(2) >= zBegin_m && r(2) < zEnd_m;
}

using FM1DElectroStatic_fast = std::shared_ptr<_FM1DElectroStatic_fast>;

#endif