#include <rz_harm_poly.h>

Inheritance diagram for magfieldparam::rz_harm_poly:

Public Member Functions
void	CheatL (const unsigned newL)

double	GetCos ()

rz_harm_poly	GetDecPow (int nvar)

rz_harm_poly	GetDiff (int nvar)

rz_harm_poly	GetIncPow (int nvar)

rz_harm_poly	GetInt (int nvar)

unsigned	GetL ()

int	GetM ()

double	GetSin ()

bool	IsPhiSet ()

rz_harm_poly	LadderDwn ()

rz_harm_poly	LadderUp ()

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

	rz_harm_poly ()

	rz_harm_poly (const poly2d_base &S)

	rz_harm_poly (const rz_harm_poly &S)

	rz_harm_poly (const unsigned N)

	~rz_harm_poly () override

Public Member Functions inherited from magfieldparam::poly2d_base
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	GetMaxM ()

static unsigned	ParentCount ()

static void	PrintTrigArr (std::ostream &out=std::cout, const std::streamsize prec=5)

static void	SetPhi (const double phi)

static void	SetPoint (const double r, const double z, const double phi)

Static Public Member Functions inherited from magfieldparam::poly2d_base
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)

Private Member Functions
void	PrintLM (std::ostream &out=std::cout)

Static Private Member Functions
static void	FillTrigArr (const double phi)

static void	SetTrigArrSize (const unsigned N)

Private Attributes
unsigned	L

int	M

Static Private Attributes
static unsigned	Cnt = 0

static unsigned	MaxM = 0

static bool	phi_set = false

static double	phival = -11111.

static unsigned	TASize = 0

static trig_pair *	TrigArr = nullptr

Additional Inherited Members
Static Protected Member Functions inherited from magfieldparam::poly2d_base
static void	AdjustTab ()

static void	FillTable (const double r, const double z)

static void	SetTabSize (const unsigned N)

Protected Attributes inherited from magfieldparam::poly2d_base
std::vector< poly2d_term >	data

unsigned	max_pwr

Static Protected Attributes inherited from magfieldparam::poly2d_base
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 38 of file rz_harm_poly.h.

Constructor & Destructor Documentation

◆ rz_harm_poly() [1/4]

magfieldparam::rz_harm_poly::rz_harm_poly ( )

inline

Definition at line 68 of file rz_harm_poly.h.

68 : poly2d_base(), L(0), M(0) { ++Cnt; }

References Cnt.

◆ rz_harm_poly() [2/4]

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

inline

Definition at line 69 of file rz_harm_poly.h.

69 : poly2d_base(S), L(0), M(0) { ++Cnt; }

References Cnt.

◆ rz_harm_poly() [3/4]

magfieldparam::rz_harm_poly::rz_harm_poly ( const rz_harm_poly & S )

inline

Definition at line 70 of file rz_harm_poly.h.

70 : poly2d_base(S), L(S.L), M(S.M) { ++Cnt; }

References Cnt.

◆ rz_harm_poly() [4/4]

rz_harm_poly::rz_harm_poly ( const unsigned N )

Definition at line 23 of file rz_harm_poly.cc.

                                            {
   //Constructor for rz_harm_poly of length N. The polynomial P(r,z) is normalized
   //in such a way that dP/dz(r=0,z)=z^(N-1)
   //
   unsigned nz = N, nr = 0, nv = 0;
   poly2d_term v3(1. / N, nr, nz);
  
   data = std::vector<poly2d_term>((N + 2) / 2, v3);
  
   while (nz >= 2) {
     nz -= 2;
     nr += 2;
     nv += 1;
     data[nv].coeff = -data[nv - 1].coeff * (nz + 1) * (nz + 2) / (nr * nr);
     data[nv].np[0] = nr;
     data[nv].np[1] = nz;
   }
   max_pwr = N;
   if (max_pwr > NPwr) {
     NPwr = max_pwr;
     rz_set = false;
     phi_set = false;
   }
   L = N;
   M = 0;
   poly2d_base_set.insert(this);
   ++Cnt;
 }

References Cnt, magfieldparam::poly2d_base::data, L, M, magfieldparam::poly2d_base::max_pwr, N, magfieldparam::poly2d_base::NPwr, EgHLTOffHistBins_cfi::nr, phi_set, magfieldparam::poly2d_base::poly2d_base_set, and magfieldparam::poly2d_base::rz_set.

◆ ~rz_harm_poly()

rz_harm_poly::~rz_harm_poly ( )

override

Definition at line 53 of file rz_harm_poly.cc.

                             {
   if (--Cnt) {
     if (std::abs(M) >= int(MaxM)) {  //a number of objects still left
       M = 0;
       MaxM = GetMaxM();
     }
   } else {  //last instance -> memory cleanup
     if (TrigArr)
       delete[] TrigArr;
     TrigArr = nullptr;
     TASize = 0;
     MaxM = 0;
     phival = -11111.;
     phi_set = false;
   }
 }

References funct::abs(), Cnt, GetMaxM(), M, MaxM, phi_set, phival, TASize, and TrigArr.

Member Function Documentation

◆ CheatL()

void magfieldparam::rz_harm_poly::CheatL ( const unsigned newL )

inline

Definition at line 109 of file rz_harm_poly.h.

109 { L = newL; }

References L.

◆ Count()

static unsigned magfieldparam::rz_harm_poly::Count ( )

inlinestatic

Definition at line 61 of file rz_harm_poly.h.

61 { return Cnt; }

References Cnt.

◆ FillTrigArr()

void rz_harm_poly::FillTrigArr ( const double phi )

staticprivate

Definition at line 114 of file rz_harm_poly.cc.

                                                {
   //Fill TrigArr with trig_pair(jp*phi)
   if (!TrigArr)
     return;
   trig_pair tp(phi);
   TrigArr[1] = tp;
   for (unsigned jp = 2; jp <= MaxM; ++jp)
     TrigArr[jp] = TrigArr[jp - 1].Add(tp);
 }

References MaxM, cmsswSequenceInfo::tp, and TrigArr.

Referenced by SetPhi().

◆ GetCos()

double magfieldparam::rz_harm_poly::GetCos ( )

inline

Definition at line 106 of file rz_harm_poly.h.

106 { return TrigArr[M].CosPhi; }

References magfieldparam::trig_pair::CosPhi, M, and TrigArr.

◆ GetDecPow()

rz_harm_poly magfieldparam::rz_harm_poly::GetDecPow ( int nvar )

inline

Definition at line 91 of file rz_harm_poly.h.

                                      {
       rz_harm_poly R(*this);
       R.DecPow(nvar);
       return R;
     }

References dttmaxenums::R.

◆ GetDiff()

rz_harm_poly magfieldparam::rz_harm_poly::GetDiff ( int nvar )

inline

Definition at line 76 of file rz_harm_poly.h.

                                    {
       rz_harm_poly R(*this);
       R.Diff(nvar);
       return R;
     }

References dttmaxenums::R.

◆ GetIncPow()

rz_harm_poly magfieldparam::rz_harm_poly::GetIncPow ( int nvar )

inline

Definition at line 86 of file rz_harm_poly.h.

                                      {
       rz_harm_poly R(*this);
       R.IncPow(nvar);
       return R;
     }

References dttmaxenums::R.

◆ GetInt()

rz_harm_poly magfieldparam::rz_harm_poly::GetInt ( int nvar )

inline

Definition at line 81 of file rz_harm_poly.h.

                                   {
       rz_harm_poly R(*this);
       R.Int(nvar);
       return R;
     }

References dttmaxenums::R.

◆ GetL()

unsigned magfieldparam::rz_harm_poly::GetL ( )

inline

Definition at line 100 of file rz_harm_poly.h.

100 { return L; }

References L.

◆ GetM()

int magfieldparam::rz_harm_poly::GetM ( )

inline

Definition at line 101 of file rz_harm_poly.h.

101 { return M; }

References M.

◆ GetMaxM()

int rz_harm_poly::GetMaxM ( )

static

Definition at line 71 of file rz_harm_poly.cc.

                           {
   //Return max abs(M) for all rz_harm_poly objects created
   //
   int M_cur, M_max = 0;
   for (auto const &poly : poly2d_base_set) {
     if (typeid(*poly) == typeid(rz_harm_poly)) {
       M_cur = std::abs(static_cast<rz_harm_poly const *>(poly)->M);
       if (M_cur > M_max)
         M_max = M_cur;
     }
   }
   return M_max;
 }

References funct::abs(), M, and magfieldparam::poly2d_base::poly2d_base_set.

Referenced by ~rz_harm_poly().

◆ GetSin()

double magfieldparam::rz_harm_poly::GetSin ( )

inline

Definition at line 107 of file rz_harm_poly.h.

107 { return TrigArr[M].SinPhi; }

References M, magfieldparam::trig_pair::SinPhi, and TrigArr.

◆ IsPhiSet()

bool magfieldparam::rz_harm_poly::IsPhiSet ( )

inline

Definition at line 74 of file rz_harm_poly.h.

74 { return phi_set; }

References phi_set.

◆ LadderDwn()

rz_harm_poly rz_harm_poly::LadderDwn ( )

Definition at line 194 of file rz_harm_poly.cc.

                                      {
   //Return a polynomial with decreased M
   //
   rz_harm_poly p_out;
   p_out.data.reserve(2 * L);
   unsigned it;
   poly2d_term term;
  
   //In 2 passes (for-cycles) to get terms in z-descending order
   for (it = 0; it < data.size(); ++it) {
     term = data[it];
     if (term.np[0]) {
       term.coeff *= -int(term.np[0]) - M;
       --term.np[0];
       ++term.np[1];
       p_out.data.push_back(term);
     }
   }
   for (it = 0; it < data.size(); ++it) {
     term = data[it];
     if (term.np[1]) {
       term.coeff *= term.np[1];
       --term.np[1];
       ++term.np[0];
       p_out.data.push_back(term);
     }
   }
   p_out.Collect();
   if (!p_out.data.empty()) {
     p_out.L = L;
     p_out.M = M - 1;
     if (std::abs(p_out.M) > int(MaxM))
       MaxM = std::abs(p_out.M);
     p_out.Scale(1. / sqrt(double((L + M) * (L - M + 1))));
   }
   return p_out;
 }

References funct::abs(), magfieldparam::poly2d_term::coeff, magfieldparam::poly2d_base::Collect(), magfieldparam::poly2d_base::data, createfilelist::int, L, M, MaxM, magfieldparam::poly2d_term::np, magfieldparam::poly2d_base::Scale(), and mathSSE::sqrt().

◆ LadderUp()

rz_harm_poly rz_harm_poly::LadderUp ( )

Definition at line 155 of file rz_harm_poly.cc.

                                     {
   //Return a polynomial with increased M
   //
   rz_harm_poly p_out;
   p_out.data.reserve(2 * L);
   unsigned it;
   poly2d_term term;
  
   //In 2 passes (for-cycles) to get terms in z-descending order
   for (it = 0; it < data.size(); ++it) {
     term = data[it];
     if (term.np[0]) {
       term.coeff *= int(term.np[0]) - M;
       --term.np[0];
       ++term.np[1];
       p_out.data.push_back(term);
     }
   }
   for (it = 0; it < data.size(); ++it) {
     term = data[it];
     if (term.np[1]) {
       term.coeff *= -(int)term.np[1];
       --term.np[1];
       ++term.np[0];
       p_out.data.push_back(term);
     }
   }
   p_out.Collect();
   if (!p_out.data.empty()) {
     p_out.L = L;
     p_out.M = M + 1;
     if (std::abs(p_out.M) > int(MaxM))
       MaxM = std::abs(p_out.M);
     p_out.Scale(1. / sqrt(double((L - M) * (L + M + 1))));
   }
   return p_out;
 }

References funct::abs(), magfieldparam::poly2d_term::coeff, magfieldparam::poly2d_base::Collect(), magfieldparam::poly2d_base::data, createfilelist::int, L, M, MaxM, magfieldparam::poly2d_term::np, magfieldparam::poly2d_base::Scale(), and mathSSE::sqrt().

◆ ParentCount()

static unsigned magfieldparam::rz_harm_poly::ParentCount ( )

inlinestatic

Definition at line 60 of file rz_harm_poly.h.

60 { return poly2d_base::Count(); }

References magfieldparam::poly2d_base::Count().

◆ Print()

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

inline

Definition at line 110 of file rz_harm_poly.h.

                                                                         {
       PrintLM(out);
       poly2d_base::Print(out, prec);
     }

References MillePedeFileConverter_cfg::out, magfieldparam::poly2d_base::Print(), and PrintLM().

◆ PrintLM()

void magfieldparam::rz_harm_poly::PrintLM ( std::ostream & out = std::cout )

inlineprivate

Definition at line 54 of file rz_harm_poly.h.

                                             {
       out << "L=" << std::setw(3) << std::left << L << ", M=" << std::setw(3) << std::left << M << "; ";
     }

References L, M, and MillePedeFileConverter_cfg::out.

Referenced by Print().

◆ PrintTrigArr()

void rz_harm_poly::PrintTrigArr	(	std::ostream &	out = `std::cout`,
		const std::streamsize	prec = `5`
	)

static

Definition at line 125 of file rz_harm_poly.cc.

                                                                          {
   out << "TrigArr: TASize = " << TASize << "\tMaxM = " << MaxM << std::endl;
   if (TrigArr) {
     if (MaxM < TASize) {
       unsigned jm;
       std::streamsize old_prec = out.precision(), wdt = prec + 7;
       out.precision(prec);
       out << "M:     ";
       for (jm = 0; jm <= MaxM; ++jm) {
         out << std::setw(wdt) << std::left << jm;
       }
       out << "|\nCos_M: ";
       for (jm = 0; jm <= MaxM; ++jm) {
         out << std::setw(wdt) << std::left << TrigArr[jm].CosPhi;
       }
       out << "|\nSin_M: ";
       for (jm = 0; jm <= MaxM; ++jm) {
         out << std::setw(wdt) << std::left << TrigArr[jm].SinPhi;
       }
       out << "|" << std::endl;
       out.precision(old_prec);
     } else {
       out << "\tTrigArr size is not adjusted." << std::endl;
     }
   } else {
     out << "\tTrigArr is not allocated." << std::endl;
   }
 }

References magfieldparam::trig_pair::CosPhi, MaxM, MillePedeFileConverter_cfg::out, magfieldparam::trig_pair::SinPhi, TASize, and TrigArr.

◆ SetPhi()

void rz_harm_poly::SetPhi ( const double phi )

static

Definition at line 86 of file rz_harm_poly.cc.

                                           {
   //Set value of the angle argument, adjust the TrigArr size if neccessary
   //and fill TrigArr if the phi value is changed
   //
   if (MaxM >= TASize) {
     SetTrigArrSize(MaxM + 1);
     FillTrigArr(phi);
   } else if (phi != phival)
     FillTrigArr(phi);
   phival = phi;
   phi_set = true;
 }

References FillTrigArr(), MaxM, phi_set, phival, SetTrigArrSize(), and TASize.

Referenced by SetPoint().

◆ SetPoint()

static void magfieldparam::rz_harm_poly::SetPoint	(	const double	r,
		const double	z,
		const double	phi
	)

inlinestatic

Definition at line 63 of file rz_harm_poly.h.

                                                                            {
       poly2d_base::SetPoint(r, z);
       SetPhi(phi);
     }

References alignCSCRings::r, SetPhi(), and magfieldparam::poly2d_base::SetPoint().

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

◆ SetTrigArrSize()

void rz_harm_poly::SetTrigArrSize ( const unsigned N )

staticprivate

Definition at line 100 of file rz_harm_poly.cc.

                                                   {
   //Increase TrigArr size if neccessary
   //
   if (N <= TASize)
     return;
   if (TrigArr)
     delete[] TrigArr;
   TrigArr = new trig_pair[N];
   (*TrigArr) = trig_pair(1., 0.);
   TASize = N;
   phi_set = false;
 }

References N, phi_set, TASize, and TrigArr.

Referenced by SetPhi().

Member Data Documentation

◆ Cnt

unsigned rz_harm_poly::Cnt = 0

staticprivate

Definition at line 43 of file rz_harm_poly.h.

Referenced by Count(), rz_harm_poly(), and ~rz_harm_poly().

◆ L

unsigned magfieldparam::rz_harm_poly::L

private

Definition at line 40 of file rz_harm_poly.h.

Referenced by CheatL(), GetL(), LadderDwn(), LadderUp(), PrintLM(), and rz_harm_poly().

◆ M

int magfieldparam::rz_harm_poly::M

private

Definition at line 41 of file rz_harm_poly.h.

Referenced by GetCos(), GetM(), GetMaxM(), GetSin(), LadderDwn(), LadderUp(), PrintLM(), rz_harm_poly(), and ~rz_harm_poly().

◆ MaxM

unsigned rz_harm_poly::MaxM = 0

staticprivate

Definition at line 46 of file rz_harm_poly.h.

Referenced by FillTrigArr(), LadderDwn(), LadderUp(), PrintTrigArr(), SetPhi(), and ~rz_harm_poly().

◆ phi_set

bool rz_harm_poly::phi_set = false

staticprivate

Definition at line 45 of file rz_harm_poly.h.

Referenced by IsPhiSet(), rz_harm_poly(), SetPhi(), SetTrigArrSize(), and ~rz_harm_poly().

◆ phival

double rz_harm_poly::phival = -11111.

staticprivate

Definition at line 44 of file rz_harm_poly.h.

Referenced by SetPhi(), and ~rz_harm_poly().

◆ TASize

unsigned rz_harm_poly::TASize = 0

staticprivate

Definition at line 48 of file rz_harm_poly.h.

Referenced by PrintTrigArr(), SetPhi(), SetTrigArrSize(), and ~rz_harm_poly().

◆ TrigArr

trig_pair * rz_harm_poly::TrigArr = nullptr

staticprivate

Definition at line 49 of file rz_harm_poly.h.

Referenced by FillTrigArr(), GetCos(), GetSin(), PrintTrigArr(), SetTrigArrSize(), and ~rz_harm_poly().

Public Member Functions

Static Public Member Functions

Private Member Functions

Static Private Member Functions

Private Attributes

Static Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ rz_harm_poly() [1/4]

◆ rz_harm_poly() [2/4]

◆ rz_harm_poly() [3/4]

◆ rz_harm_poly() [4/4]

◆ ~rz_harm_poly()

Member Function Documentation

◆ CheatL()

◆ Count()

◆ FillTrigArr()

◆ GetCos()

◆ GetDecPow()

◆ GetDiff()

◆ GetIncPow()

◆ GetInt()

◆ GetL()

◆ GetM()

◆ GetMaxM()

◆ GetSin()

◆ IsPhiSet()

◆ LadderDwn()

◆ LadderUp()

◆ ParentCount()

◆ Print()

◆ PrintLM()

◆ PrintTrigArr()

◆ SetPhi()

◆ SetPoint()

◆ SetTrigArrSize()

Member Data Documentation

◆ Cnt

◆ L

◆ M

◆ MaxM

◆ phi_set

◆ phival

◆ TASize

◆ TrigArr