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