rz_poly.h

Go to the documentation of this file.

#ifndef rz_poly_h
#define rz_poly_h
 
#include <vector>
 
namespace magfieldparam {
 
  //_______________________________________________________________________________
  struct poly_term {
    double coeff;
    int np[2];
  };
 
  //_______________________________________________________________________________
  typedef std::vector<poly_term> poly_vect;
  //_______________________________________________________________________________
  typedef std::vector<poly_vect> poly_arr;
 
  //_______________________________________________________________________________
  class rz_poly {  // represent a set of homogeneous polynomials
 
  private:
    poly_arr data;
    int max_nr, max_nz, n_active;
    double *r_pow;
    double *z_pow;
    bool *is_off;
 
  public:
    rz_poly() : data(), max_nr(0), max_nz(0), n_active(0), r_pow(nullptr), z_pow(nullptr), is_off(nullptr){};
 
    rz_poly(int N);
    rz_poly(const rz_poly &S);
    ~rz_poly();
 
    void SetOFF(int npoly);
    void SetON(int npoly);
    void SetAllON(int npoly) {
      if (is_off)
        std::fill(is_off, is_off + data.size(), false);
    }
 
    rz_poly Diff(int nvar, bool keep_empty = false);
    rz_poly Int(int nvar);
 
    rz_poly &operator*=(double C);
    rz_poly &operator*=(double *C);
 
    double GetSVal(double r, double z, const double *C) const;
    double *GetVVal(double r, double z, double *rez_out = nullptr);
 
    int GetMaxRPow() const { return max_nr - 1; }
    int GetMaxZPow() const { return max_nz - 1; }
    int GetLength() const { return (int)data.size(); }
    int GetNActive() const { return n_active; }
 
    double *Expand(double *C);
 
    void Print();
  };
}  // namespace magfieldparam
 
#endif