magfieldparam::rz_poly Class Reference

#include <rz_poly.h>

Public Member Functions
rz_poly	Diff (int nvar, bool keep_empty=false)
double *	Expand (double *C)
int	GetLength () const
int	GetMaxRPow () const
int	GetMaxZPow () const
int	GetNActive () const
double	GetSVal (double r, double z, const double *C) const
double *	GetVVal (double r, double z, double *rez_out=nullptr)
rz_poly	Int (int nvar)
rz_poly &	operator*= (double C)
rz_poly &	operator*= (double *C)
void	Print ()
	rz_poly ()
	rz_poly (int N)
	rz_poly (const rz_poly &S)
void	SetAllON (int npoly)
void	SetOFF (int npoly)
void	SetON (int npoly)
	~rz_poly ()

Private Attributes
poly_arr	data
bool *	is_off
int	max_nr
int	max_nz
int	n_active
double *	r_pow
double *	z_pow

Detailed Description

Author

V. Maroussov

Definition at line 26 of file rz_poly.h.

Constructor & Destructor Documentation

magfieldparam::rz_poly::rz_poly ( )

inline

Definition at line 36 of file rz_poly.h.

: data(), max_nr(0), max_nz(0), n_active(0), r_pow(nullptr), z_pow(nullptr), is_off(nullptr){};

rz_poly::rz_poly

(

int

N

)

Definition at line 8 of file rz_poly.cc.

References magfieldparam::poly_term::coeff, data, PVValHelper::fill(), visualization-live-secondInstance_cfg::m, N, magfieldparam::poly_term::np, and nt.

                      {
  int nz, nr = 0, nv, nt;
  poly_term v3;

  if (N < 2)
    N = 2;
  data.reserve(N);

  v3.coeff = 1.;
  v3.np[0] = 0;
  v3.np[1] = 1;

  data.push_back(poly_vect(1, v3));

  for (int m = 2; m <= N; ++m) {
    nz = m;
    nr = 0;
    nv = 0;
    v3.coeff = 1. / m;
    v3.np[0] = nr;
    v3.np[1] = nz;

    nt = (m + 2) / 2;
    poly_vect v3x(nt, v3);

    while (nz >= 2) {
      nz -= 2;
      nr += 2;
      nv += 1;
      v3x[nv].coeff = -v3x[nv - 1].coeff * (nz + 1) * (nz + 2) / (nr * nr);
      v3x[nv].np[0] = nr;
      v3x[nv].np[1] = nz;
    }
    data.push_back(v3x);
  }
  max_nr = nr + 1;
  max_nz = N + 1;
  r_pow = new double[max_nr];
  z_pow = new double[max_nz];
  fill(r_pow, r_pow + max_nr, 1.);
  fill(z_pow, z_pow + max_nz, 1.);

  n_active = data.size();
  is_off = new bool[n_active];
  fill(is_off, is_off + n_active, false);
}

rz_poly::rz_poly

(

const rz_poly &

S

)

Definition at line 56 of file rz_poly.cc.

References filterCSVwithJSON::copy, data, data, is_off, max_nr, max_nz, n_active, r_pow, and z_pow.

                                 {
  data = S.data;
  max_nr = S.max_nr;
  max_nz = S.max_nz;
  n_active = S.n_active;

  if (max_nr) {
    r_pow = new double[max_nr];
    copy(S.r_pow, S.r_pow + max_nr, r_pow);
  } else
    r_pow = nullptr;

  if (max_nz) {
    z_pow = new double[max_nz];
    copy(S.z_pow, S.z_pow + max_nz, z_pow);
  } else
    z_pow = nullptr;

  if (S.is_off) {
    is_off = new bool[data.size()];
    copy(S.is_off, S.is_off + data.size(), is_off);
  } else
    is_off = nullptr;
}

rz_poly::~rz_poly

(

)

Definition at line 82 of file rz_poly.cc.

                  {
  if (is_off)
    delete[] is_off;
  if (r_pow)
    delete[] r_pow;
  if (z_pow)
    delete[] z_pow;
}

Member Function Documentation

rz_poly rz_poly::Diff	(	int	nvar,
		bool	keep_empty = false
	)

Definition at line 132 of file rz_poly.cc.

References magfieldparam::poly_term::coeff, filterCSVwithJSON::copy, data, data, is_off, max_nr, max_nz, n_active, magfieldparam::poly_term::np, r_pow, findQualityFiles::size, and z_pow.

Referenced by magfieldparam::BFit::BFit().

                                               {
  //Return the derivative of original polynomial by variable nvar.
  //If keep_empty=true, resulting polynomial may contain zero basis terms,
  //otherwise the resulting polynomial will be compressed, so that corresponding
  //basis terms will be shifted relatively to the original polynomial.
  //
  poly_term v3;
  rz_poly p_out;
  p_out.data.reserve(data.size());

  bool *tmp_mask = new bool[data.size()];
  unsigned int ind_tmp = 0;
  int tmp_act = 0;

  for (unsigned int ip = 0; ip < data.size(); ++ip) {
    poly_vect v3x;
    v3x.reserve(data[ip].size());
    for (unsigned int it = 0; it < data[ip].size(); ++it) {
      v3 = data[ip][it];
      v3.coeff = data[ip][it].coeff * data[ip][it].np[nvar];
      if (v3.coeff != 0) {
        v3.np[nvar] = data[ip][it].np[nvar] - 1;
        v3x.push_back(v3);
      }
    }
    if (!v3x.empty() || keep_empty) {
      v3x.resize(v3x.size());
      p_out.data.push_back(v3x);
      tmp_mask[ind_tmp] = is_off[ip];
      ++ind_tmp;
      if (!is_off[ip])
        ++tmp_act;
    }
  }

  p_out.data.resize(p_out.data.size());
  p_out.max_nr = max_nr;
  p_out.max_nz = max_nz;

  if (nvar == 0)
    --p_out.max_nr;
  else
    --p_out.max_nz;

  p_out.r_pow = new double[p_out.max_nr];
  copy(r_pow, r_pow + p_out.max_nr, p_out.r_pow);
  p_out.z_pow = new double[p_out.max_nz];
  copy(z_pow, z_pow + p_out.max_nz, p_out.z_pow);

  p_out.n_active = tmp_act;
  p_out.is_off = new bool[p_out.data.size()];
  copy(tmp_mask, tmp_mask + p_out.data.size(), p_out.is_off);

  delete[] tmp_mask;

  return p_out;
}

double * rz_poly::Expand

(

double *

C

)

Definition at line 330 of file rz_poly.cc.

References gen::C, and data.

                                 {
  //Take C[n_active] - reduced vector of coefficients and return pointer to
  //expanded vector of coefficients of the data.size() length. It is in user's
  //responsibility to free the memory after all;
  //
  double *rez = new double[data.size()];
  for (int ip = 0; ip < (int)data.size(); ++ip) {
    if (is_off[ip])
      rez[ip] = 0.;
    else {
      rez[ip] = *C;
      ++C;
    }
  }
  return rez;
}

int magfieldparam::rz_poly::GetLength ( ) const

inline

Definition at line 60 of file rz_poly.h.

References data.

{ return (int)data.size(); }

int magfieldparam::rz_poly::GetMaxRPow ( ) const

inline

Definition at line 58 of file rz_poly.h.

References max_nr.

{ return max_nr - 1; }

int magfieldparam::rz_poly::GetMaxZPow ( ) const

inline

Definition at line 59 of file rz_poly.h.

References max_nz.

{ return max_nz - 1; }

int magfieldparam::rz_poly::GetNActive ( ) const

inline

Definition at line 61 of file rz_poly.h.

References n_active.

{ return n_active; }

double rz_poly::GetSVal	(	double	r,
		double	z,
		const double *	C
	)		const

Definition at line 257 of file rz_poly.cc.

References gen::C, data, and findQualityFiles::size.

                                                                 {
  //Return value of a polynomial, ignoring terms, that are switched off

  if (r_pow == nullptr)
    return 0.;

  double term, rez = 0.;
  int ip, it;

  if (r != r_pow[1])
    for (ip = 1; ip < max_nr; ++ip)
      r_pow[ip] = r * r_pow[ip - 1];
  if (z != z_pow[1])
    for (ip = 1; ip < max_nz; ++ip)
      z_pow[ip] = z * z_pow[ip - 1];

  for (ip = 0; ip < (int)data.size(); ++ip) {
    if (is_off[ip])
      continue;
    term = 0.;
    for (it = 0; it < (int)data[ip].size(); ++it) {
      term += data[ip][it].coeff * r_pow[data[ip][it].np[0]] * z_pow[data[ip][it].np[1]];
    }
    rez += *C * term;
    ++C;
  }

  return rez;
}

double * rz_poly::GetVVal	(	double	r,
		double	z,
		double *	rez_out = nullptr
	)

Definition at line 288 of file rz_poly.cc.

References data, and findQualityFiles::size.

                                                            {
  //return an array of doubleype with values of the basis functions in the point
  //(r,z). In a case if rez_out != 0, the rez_out array must be long enough to fit
  //in n_active elements. In a case if rez_out == 0, a new array of n_active length
  //is created; it is in user's responsibility to free the memory after all;
  //
  if (r_pow == nullptr)
    return nullptr;

  double term;
  int ip, it;

  double *rez;
  if (rez_out)
    rez = rez_out;
  else
    rez = new double[n_active];

  double *pnt = rez;

  if (r != r_pow[1])
    for (ip = 1; ip < max_nr; ++ip)
      r_pow[ip] = r * r_pow[ip - 1];
  if (z != z_pow[1])
    for (ip = 1; ip < max_nz; ++ip)
      z_pow[ip] = z * z_pow[ip - 1];

  for (ip = 0; ip < (int)data.size(); ++ip) {
    if (is_off[ip])
      continue;
    term = 0.;
    for (it = 0; it < (int)data[ip].size(); ++it) {
      term += data[ip][it].coeff * r_pow[data[ip][it].np[0]] * z_pow[data[ip][it].np[1]];
    }
    *pnt = term;
    ++pnt;
  }

  return rez;
}

rz_poly rz_poly::Int

(

int

nvar

)

Definition at line 191 of file rz_poly.cc.

References filterCSVwithJSON::copy, data, data, is_off, max_nr, max_nz, n_active, r_pow, and z_pow.

                             {
  //Return the integral of original polynomial by variable nvar
  //
  rz_poly p_out;
  p_out.data = data;
  for (unsigned int ip = 0; ip < data.size(); ++ip) {
    for (unsigned int it = 0; it < data[ip].size(); ++it) {
      p_out.data[ip][it].coeff /= ++p_out.data[ip][it].np[nvar];
    }
  }

  p_out.max_nr = max_nr;
  p_out.max_nz = max_nz;

  if (nvar == 0)
    ++p_out.max_nr;
  else
    ++p_out.max_nz;

  p_out.r_pow = new double[p_out.max_nr];
  copy(r_pow, r_pow + max_nr, p_out.r_pow);
  p_out.z_pow = new double[p_out.max_nz];
  copy(z_pow, z_pow + max_nz, p_out.z_pow);

  if (nvar == 0)
    p_out.r_pow[max_nr] = p_out.r_pow[max_nr - 1] * r_pow[1];
  else
    p_out.z_pow[max_nz] = p_out.z_pow[max_nz - 1] * z_pow[1];

  p_out.n_active = n_active;
  p_out.is_off = new bool[data.size()];
  copy(is_off, is_off + data.size(), p_out.is_off);

  return p_out;
}

rz_poly & rz_poly::operator*=

(

double

C

)

Definition at line 228 of file rz_poly.cc.

References gen::C, and data.

                                     {
  //Multiply the polynomial by a constant. Skips terms that are switched off

  for (unsigned int ip = 0; ip < data.size(); ++ip) {
    if (is_off[ip])
      continue;
    for (unsigned int it = 0; it < data[ip].size(); ++it) {
      data[ip][it].coeff *= C;
    }
  }
  return *this;
}

rz_poly & rz_poly::operator*=

(

double *

C

)

Definition at line 242 of file rz_poly.cc.

References gen::C, and data.

                                      {
  //Multiply the polynomial by an array. Skips terms that are switched off

  for (unsigned int ip = 0; ip < data.size(); ++ip) {
    if (is_off[ip])
      continue;
    for (unsigned int it = 0; it < data[ip].size(); ++it) {
      data[ip][it].coeff *= *C;
    }
    ++C;
  }
  return *this;
}

void rz_poly::Print

(

void

)

Definition at line 112 of file rz_poly.cc.

References gather_cfg::cout, and data.

                    {
  if (data.empty()) {
    cout << "The \"rz_poly\" object is NOT initialized!" << endl;
    return;
  }

  for (unsigned int ip = 0; ip < data.size(); ++ip) {
    cout << "Polynomial " << ip << " (size=" << data[ip].size() << ", status=";
    if (is_off[ip])
      cout << "OFF):" << endl;
    else
      cout << "ON):" << endl;
    for (unsigned int it = 0; it < data[ip].size(); ++it) {
      cout << "\tnr=" << data[ip][it].np[0] << "\tnz=" << data[ip][it].np[1] << "\tcoeff=" << data[ip][it].coeff
           << endl;
    }
  }
}

void magfieldparam::rz_poly::SetAllON ( int  npoly )

inline

Definition at line 44 of file rz_poly.h.

References data, PVValHelper::fill(), and is_off.

                             {
      if (is_off)
        std::fill(is_off, is_off + data.size(), false);
    }

void rz_poly::SetOFF

(

int

npoly

)

Definition at line 92 of file rz_poly.cc.

References data.

Referenced by magfieldparam::BFit::BFit().

                              {
  if ((npoly < 0) || (npoly >= (int)data.size()))
    return;
  if (is_off[npoly])
    return;
  is_off[npoly] = true;
  --n_active;
}

void rz_poly::SetON

(

int

npoly

)

Definition at line 102 of file rz_poly.cc.

References data.

                             {
  if ((npoly < 0) || (npoly >= (int)data.size()))
    return;
  if (is_off[npoly]) {
    is_off[npoly] = false;
    ++n_active;
  }
}

Member Data Documentation

poly_arr magfieldparam::rz_poly::data

private

Definition at line 29 of file rz_poly.h.

Referenced by data_sources.node::__str__(), data_sources.json_list::as_dicts(), data_sources.json_list::as_table(), Diff(), data_sources.json_list::get_members(), GetLength(), Int(), data_sources.json_list::last(), rz_poly(), SetAllON(), and cuy.FindIssue::startElement().

bool* magfieldparam::rz_poly::is_off

private

Definition at line 33 of file rz_poly.h.

Referenced by Diff(), Int(), rz_poly(), and SetAllON().

int magfieldparam::rz_poly::max_nr

private

Definition at line 30 of file rz_poly.h.

Referenced by Diff(), GetMaxRPow(), Int(), and rz_poly().

int magfieldparam::rz_poly::max_nz

private

Definition at line 30 of file rz_poly.h.

Referenced by Diff(), GetMaxZPow(), Int(), and rz_poly().

int magfieldparam::rz_poly::n_active

private

Definition at line 30 of file rz_poly.h.

Referenced by Diff(), GetNActive(), Int(), and rz_poly().

double* magfieldparam::rz_poly::r_pow

private

Definition at line 31 of file rz_poly.h.

Referenced by Diff(), Int(), and rz_poly().

double* magfieldparam::rz_poly::z_pow

private

Definition at line 32 of file rz_poly.h.

Referenced by Diff(), Int(), and rz_poly().