magfieldparam::poly2d_base Class Reference

#include <poly2d_base.h>

Inheritance diagram for magfieldparam::poly2d_base:

Public Member Functions
void	Collect ()
void	Compress ()
void	DecPow (int nvar)
void	Diff (int nvar)
double	Eval ()
double	GetVal ()
double	GetVal (const double r, const double z)
void	IncPow (int nvar)
void	Int (int nvar)
bool	IsOn ()
bool	IsRZSet ()
	poly2d_base ()
	poly2d_base (const poly2d_base &S)
void	Print (std::ostream &out=std::cout, const std::streamsize prec=5)
void	Scale (const double C)
virtual	~poly2d_base ()

Static Public Member Functions
static unsigned	Count ()
static int	GetMaxPow ()
static void	IncNPwr (const unsigned N)
static void	PrintTab (std::ostream &out=std::cout, const std::streamsize prec=5)
static void	SetPoint (const double r, const double z)

Static Protected Member Functions
static void	AdjustTab ()
static void	FillTable (const double r, const double z)
static void	SetTabSize (const unsigned N)

Protected Attributes
std::vector< poly2d_term >	data
unsigned	max_pwr

Static Protected Attributes
static const double	MIN_COEFF = DBL_EPSILON
static unsigned	NPwr = 0
static unsigned	NTab = 0
static std::set< poly2d_base * >	poly2d_base_set
static double	rval = 0.
static double **	rz_pow = nullptr
static bool	rz_set = false
static double	zval = 0.

Detailed Description

Definition at line 45 of file poly2d_base.h.

Constructor & Destructor Documentation

poly2d_base() [1/2]

magfieldparam::poly2d_base::poly2d_base ( )

inline

Definition at line 83 of file poly2d_base.h.

                  {
      max_pwr = 0;
      poly2d_base_set.insert(this);
    }

References max_pwr, and poly2d_base_set.

poly2d_base() [2/2]

magfieldparam::poly2d_base::poly2d_base ( const poly2d_base &  S )

inline

Definition at line 87 of file poly2d_base.h.

                                      {
      data = S.data;
      max_pwr = S.max_pwr;
      poly2d_base_set.insert(this);
    }

References data, max_pwr, and poly2d_base_set.

~poly2d_base()

poly2d_base::~poly2d_base ( )

virtual

Definition at line 53 of file poly2d_base.cc.

                          {
  poly2d_base_set.erase(poly2d_base_set.find(this));
  if (!poly2d_base_set.empty()) {  //some objects left
    if (max_pwr >= NPwr)
      NPwr = GetMaxPow();
  } else {
    if (rz_pow) {
      delete[] rz_pow[0];  //deleting the last instance -> memory cleanup
      delete[] rz_pow;
    }
    rz_pow = nullptr;
    rval = zval = 0.;
    NPwr = 0;
    NTab = 0;
    rz_set = false;
    //      poly2d_base_set.resize(0);
  }
}

References GetMaxPow(), max_pwr, NPwr, NTab, poly2d_base_set, rz_pow, rz_set, and zval.

Member Function Documentation

AdjustTab()

void poly2d_base::AdjustTab ( )

staticprotected

Definition at line 120 of file poly2d_base.cc.

                            {
  NPwr = GetMaxPow();
  if (NPwr >= NTab)
    SetTabSize(NPwr + 1);
}

References GetMaxPow(), NPwr, NTab, and SetTabSize().

Collect()

void poly2d_base::Collect

(

)

Definition at line 171 of file poly2d_base.cc.

                          {
  if (!(data.size()))
    return;
 
  unsigned j1, j2, rpow, zpow, noff = 0, jend = data.size();
  double C;
  std::vector<bool> mask(jend, false);
  max_pwr = 0;
 
  for (j1 = 0; j1 < jend; ++j1) {
    if (mask[j1])
      continue;
    C = data[j1].coeff;
    rpow = data[j1].np[0];
    zpow = data[j1].np[1];
    for (j2 = j1 + 1; j2 < jend; ++j2) {
      if (mask[j2])
        continue;
      if ((rpow == data[j2].np[0]) && (zpow == data[j2].np[1])) {
        C += data[j2].coeff;
        mask[j2] = true;
        ++noff;
      }
    }
    if (fabs(C) > MIN_COEFF) {
      data[j1].coeff = C;
      if ((rpow = rpow + zpow) > max_pwr)
        max_pwr = rpow;
    } else {
      mask[j1] = true;
      ++noff;
    }
  }
  std::vector<poly2d_term> newdata;
  newdata.reserve(jend - noff);
  for (j1 = 0; j1 < jend; ++j1) {
    if (!(mask[j1]))
      newdata.push_back(data[j1]);
  }
  data.swap(newdata);
}

References gen::C, data, max_pwr, MIN_COEFF, and np.

Referenced by Compress(), magfieldparam::rz_harm_poly::LadderDwn(), and magfieldparam::rz_harm_poly::LadderUp().

Compress()

void magfieldparam::poly2d_base::Compress ( )

inline

Definition at line 99 of file poly2d_base.h.

{ Collect(); }

References Collect().

Count()

static unsigned magfieldparam::poly2d_base::Count ( )

inlinestatic

Definition at line 78 of file poly2d_base.h.

{ return poly2d_base_set.size(); }

References poly2d_base_set.

Referenced by magfieldparam::rz_harm_poly::ParentCount(), and SetPoint().

DecPow()

void poly2d_base::DecPow

(

int

nvar

)

Definition at line 282 of file poly2d_base.cc.

                                 {
  //Divide the polynomial by variable# nvar. Remove terms with zero coefficients
  //and also terms where the initial power of nvar is equal zero
  //
  poly2d_term v3;
  std::vector<poly2d_term> newdata;
  newdata.reserve(data.size());
  unsigned cur_pwr = 0, maxp = 0, oldp = max_pwr;
  for (unsigned it = 0; it < data.size(); ++it) {
    v3 = data[it];
    if ((v3.coeff != 0.) && (v3.np[nvar] > 0)) {
      --v3.np[nvar];
      newdata.push_back(v3);
      if ((cur_pwr = v3.np[0] + v3.np[1]) > maxp)
        maxp = cur_pwr;
    }
  }
  newdata.resize(newdata.size());
  max_pwr = maxp;
  data.swap(newdata);
  if (oldp >= NPwr)
    NPwr = GetMaxPow();
}

References magfieldparam::poly2d_term::coeff, data, GetMaxPow(), max_pwr, magfieldparam::poly2d_term::np, and NPwr.

Diff()

void poly2d_base::Diff

(

int

nvar

)

Definition at line 231 of file poly2d_base.cc.

                               {
  //differentiate the polynomial by variable nvar.
  //
  poly2d_term v3;
  std::vector<poly2d_term> newdata;
  newdata.reserve(data.size());
  unsigned cur_pwr = 0, maxp = 0, oldp = max_pwr;
  for (unsigned it = 0; it < data.size(); ++it) {
    v3 = data[it];
    v3.coeff *= v3.np[nvar];
    if (v3.coeff != 0.) {
      --v3.np[nvar];
      newdata.push_back(v3);
      if ((cur_pwr = v3.np[0] + v3.np[1]) > maxp)
        maxp = cur_pwr;
    }
  }
  newdata.resize(newdata.size());
  max_pwr = maxp;
  data.swap(newdata);
  if (oldp >= NPwr)
    NPwr = GetMaxPow();
}

References magfieldparam::poly2d_term::coeff, data, GetMaxPow(), max_pwr, magfieldparam::poly2d_term::np, and NPwr.

Eval()

double poly2d_base::Eval

(

)

Definition at line 163 of file poly2d_base.cc.

                         {
  double S = 0.;
  for (unsigned j = 0; j < data.size(); ++j)
    S += data[j].coeff * rz_pow[data[j].np[0]][data[j].np[1]];
  return S;
}

References data, dqmiolumiharvest::j, np, rz_pow, and S().

Referenced by GetVal().

FillTable()

void poly2d_base::FillTable ( const double  r, const double  z  )

staticprotected

Definition at line 93 of file poly2d_base.cc.

                                                          {
  if (!rz_pow)
    return;
  unsigned jr, jz;
  for (jz = 1; jz <= NPwr; ++jz)
    rz_pow[0][jz] = z * rz_pow[0][jz - 1];
  for (jr = 1; jr <= NPwr; ++jr) {
    for (jz = 0; jz <= (NPwr - jr); ++jz) {
      rz_pow[jr][jz] = r * rz_pow[jr - 1][jz];
    }
  }
}

References NPwr, alignCSCRings::r, and rz_pow.

Referenced by SetPoint().

GetMaxPow()

int poly2d_base::GetMaxPow ( )

static

Definition at line 107 of file poly2d_base.cc.

                           {
  int curp, maxp = 0;
  std::set<poly2d_base *>::iterator it;
 
  for (it = poly2d_base_set.begin(); it != poly2d_base_set.end(); ++it) {
    curp = (*it)->max_pwr;
    if (curp > maxp)
      maxp = curp;
  }
  return maxp;
}

References poly2d_base_set.

Referenced by AdjustTab(), DecPow(), Diff(), IncPow(), Int(), and ~poly2d_base().

GetVal() [1/2]

double magfieldparam::poly2d_base::GetVal ( )

inline

Definition at line 110 of file poly2d_base.h.

                    {
      if (rz_set)
        return Eval();
      else
        return 0.;
    }

References Eval(), and rz_set.

GetVal() [2/2]

double magfieldparam::poly2d_base::GetVal ( const double  r, const double  z  )

inline

Definition at line 116 of file poly2d_base.h.

                                                  {
      SetPoint(r, z);
      return Eval();
    }

References Eval(), alignCSCRings::r, and SetPoint().

IncNPwr()

static void magfieldparam::poly2d_base::IncNPwr ( const unsigned  N )

inlinestatic

Definition at line 73 of file poly2d_base.h.

                                          {
      if (N > NPwr)
        NPwr = N;
    }

References N, and NPwr.

Referenced by magfieldparam::HarmBasis3DCyl::HarmBasis3DCyl().

IncPow()

void poly2d_base::IncPow

(

int

nvar

)

Definition at line 270 of file poly2d_base.cc.

                                 {
  //Multiply the polynomial by variable# nvar
  //
  for (unsigned it = 0; it < data.size(); ++it) {
    ++data[it].np[nvar];
  }
  ++max_pwr;
  if (max_pwr > NPwr)
    NPwr = GetMaxPow();
}

References data, GetMaxPow(), max_pwr, and NPwr.

Int()

void poly2d_base::Int

(

int

nvar

)

Definition at line 256 of file poly2d_base.cc.

                              {
  //Integrate the polynomial by variable# nvar. Doesn't remove terms
  //with zero coefficients; if you suspect they can appear, use Compress()
  //after the integration.
  //
  for (unsigned it = 0; it < data.size(); ++it) {
    data[it].coeff /= ++data[it].np[nvar];
  }
  ++max_pwr;
  if (max_pwr > NPwr)
    NPwr = GetMaxPow();
}

References data, GetMaxPow(), max_pwr, and NPwr.

IsOn()

bool magfieldparam::poly2d_base::IsOn ( )

inline

Definition at line 95 of file poly2d_base.h.

{ return bool(!data.empty()); }

References electrons_cff::bool, and data.

IsRZSet()

bool magfieldparam::poly2d_base::IsRZSet ( )

inline

Definition at line 96 of file poly2d_base.h.

{ return rz_set; }

References rz_set.

Print()

void poly2d_base::Print	(	std::ostream &	out = std::cout,
		const std::streamsize	prec = 5
	)

Definition at line 214 of file poly2d_base.cc.

                                                                 {
  if (data.empty()) {
    out << "\"poly2d_base\" object contains no terms." << std::endl;
    return;
  }
  out << data.size() << " terms; max. degree = " << max_pwr << ":" << std::endl;
  std::streamsize old_prec = out.precision();
  out.precision(prec);
  data[0].Print(out);
  for (unsigned it = 1; it < data.size(); ++it) {
    data[it].Print(out, false);
  }
  out << std::endl;
  out.precision(old_prec);
}

References data, max_pwr, and MillePedeFileConverter_cfg::out.

Referenced by magfieldparam::rz_harm_poly::Print().

PrintTab()

void poly2d_base::PrintTab ( std::ostream &  out = std::cout, const std::streamsize  prec = 5  )

static

Definition at line 127 of file poly2d_base.cc.

                                                                    {
  out << "poly2d_base table size NTab = " << NTab << "\tmax. power NPwr = " << NPwr << std::endl;
  if (rz_pow) {
    if (NPwr < NTab) {
      std::streamsize old_prec = out.precision(), wdt = prec + 7;
      out.precision(prec);
      out << "Table content:" << std::endl;
      unsigned jr, jz;
      for (jr = 0; jr <= NPwr; ++jr) {
        for (jz = 0; jz <= (NPwr - jr); ++jz) {
          out << std::setw(wdt) << std::left << rz_pow[jr][jz];
        }
        out << "|" << std::endl;
      }
      out.precision(old_prec);
    } else {
      out << "\tTable size is not adjusted." << std::endl;
    }
  } else {
    out << "\tTable is not allocated." << std::endl;
  }
}

References NPwr, NTab, MillePedeFileConverter_cfg::out, and rz_pow.

Scale()

void poly2d_base::Scale

(

const double

C

)

Definition at line 307 of file poly2d_base.cc.

                                      {
  //Multiply the polynomial by a constant.
  //
  if (C != 0.) {
    for (unsigned it = 0; it < data.size(); ++it) {
      data[it].coeff *= C;
    }
  } else
    data.resize(0);
}

References gen::C, and data.

Referenced by magfieldparam::rz_harm_poly::LadderDwn(), and magfieldparam::rz_harm_poly::LadderUp().

SetPoint()

void poly2d_base::SetPoint ( const double  r, const double  z  )

static

Definition at line 151 of file poly2d_base.cc.

                                                         {
  if (!Count())
    return;
  if (NPwr >= NTab) {
    SetTabSize(NPwr + 1);
    FillTable(r, z);
  } else if ((r != rval) || (z != zval))
    FillTable(r, z);
  rz_set = true;
}

References Count(), FillTable(), NPwr, NTab, alignCSCRings::r, rz_set, SetTabSize(), and zval.

Referenced by GetVal(), and magfieldparam::rz_harm_poly::SetPoint().

SetTabSize()

void poly2d_base::SetTabSize ( const unsigned  N )

staticprotected

Definition at line 73 of file poly2d_base.cc.

                                             {
  if (N <= NTab)
    return;
  if (rz_pow) {
    delete[] rz_pow[0];
    delete[] rz_pow;
  }
  rz_pow = new double *[N];
  unsigned jr, dN = N * (N + 1) / 2;
  rz_pow[0] = new double[dN];
  memset(rz_pow[0], 0, dN * sizeof(double));
  rz_pow[0][0] = 1.;
  for (jr = 1, dN = N; jr < N; ++jr, --dN) {
    rz_pow[jr] = rz_pow[jr - 1] + dN;
  }
  rval = zval = 0.;
  NTab = N;
}

References N, NTab, rz_pow, and zval.

Referenced by AdjustTab(), and SetPoint().

Member Data Documentation

data

std::vector<poly2d_term> magfieldparam::poly2d_base::data

protected

Definition at line 69 of file poly2d_base.h.

Referenced by data_sources.node::__str__(), data_sources.json_list::as_dicts(), data_sources.json_list::as_table(), Collect(), DecPow(), Diff(), Eval(), data_sources.json_list::get_members(), IncPow(), Int(), IsOn(), magfieldparam::rz_harm_poly::LadderDwn(), magfieldparam::rz_harm_poly::LadderUp(), data_sources.json_list::last(), poly2d_base(), Print(), magfieldparam::rz_harm_poly::rz_harm_poly(), Scale(), and cuy.FindIssue::startElement().

max_pwr

unsigned magfieldparam::poly2d_base::max_pwr

protected

Definition at line 70 of file poly2d_base.h.

Referenced by Collect(), DecPow(), Diff(), IncPow(), Int(), poly2d_base(), Print(), magfieldparam::rz_harm_poly::rz_harm_poly(), and ~poly2d_base().

MIN_COEFF

const double poly2d_base::MIN_COEFF = DBL_EPSILON

staticprotected

Definition at line 58 of file poly2d_base.h.

Referenced by Collect().

NPwr

unsigned poly2d_base::NPwr = 0

staticprotected

Definition at line 55 of file poly2d_base.h.

Referenced by AdjustTab(), DecPow(), Diff(), FillTable(), IncNPwr(), IncPow(), Int(), PrintTab(), magfieldparam::rz_harm_poly::rz_harm_poly(), SetPoint(), and ~poly2d_base().

NTab

unsigned poly2d_base::NTab = 0

staticprotected

Definition at line 54 of file poly2d_base.h.

Referenced by AdjustTab(), PrintTab(), SetPoint(), SetTabSize(), and ~poly2d_base().

poly2d_base_set

std::set< poly2d_base * > poly2d_base::poly2d_base_set

staticprotected

Definition at line 60 of file poly2d_base.h.

Referenced by Count(), magfieldparam::rz_harm_poly::GetMaxM(), GetMaxPow(), poly2d_base(), magfieldparam::rz_harm_poly::rz_harm_poly(), and ~poly2d_base().

rval

double poly2d_base::rval = 0.

staticprotected

Definition at line 50 of file poly2d_base.h.

rz_pow

double ** poly2d_base::rz_pow = nullptr

staticprotected

Definition at line 53 of file poly2d_base.h.

Referenced by Eval(), FillTable(), PrintTab(), SetTabSize(), and ~poly2d_base().

rz_set

bool poly2d_base::rz_set = false

staticprotected

Definition at line 56 of file poly2d_base.h.

Referenced by GetVal(), IsRZSet(), magfieldparam::rz_harm_poly::rz_harm_poly(), SetPoint(), and ~poly2d_base().

zval

double poly2d_base::zval = 0.

staticprotected

Definition at line 51 of file poly2d_base.h.

Referenced by SetPoint(), SetTabSize(), and ~poly2d_base().