crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-28 14:15:42 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

add space next to math sign

Browse source
This commit is contained in:
rockfool 2020-02-05 23:25:50 -05:00
parent e74d6aa362
commit 93cb6120d1
1 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
openmc/polynomial.py
View file

@ -110,11 +110,11 @@ class Zernike(Polynomial):
"""
def __init__(self, coef, radius=1):
super().__init__(coef)
self._order = int(((np.sqrt(8 * len(self.coef) + 1)) -3) / 2)
self._order = int(((np.sqrt(8 * len(self.coef) + 1)) - 3) / 2)
self.radius = radius
norm_vec = np.ones((len(self.coef)))
for n in range(self._order + 1):
for m in range(-n,(n+1),2):
for m in range(-n, n + 1, 2):
j = int((n * (n + 2) + m) / 2)
if m==0:
norm_vec[j] = n + 1
@ -130,13 +130,13 @@ class Zernike(Polynomial):
def __call__(self, r, theta=0.0):
import openmc.lib as lib
if isinstance(r, Iterable) and isinstance(theta, Iterable):
return [[np.sum(self._norm_coef * lib.calc_zn(self.order, r_i/self.radius, theta_i))
return [[np.sum(self._norm_coef * lib.calc_zn(self.order, r_i / self.radius, theta_i))
for r_i in r] for theta_i in theta]
elif isinstance(r, Iterable) and not isinstance(theta, Iterable):
return [np.sum(self._norm_coef * lib.calc_zn(self.order, r_i/self.radius, theta))
return [np.sum(self._norm_coef * lib.calc_zn(self.order, r_i / self.radius, theta))
for r_i in r]
elif not isinstance(r, Iterable) and isinstance(theta, Iterable):
return [np.sum(self._norm_coef * lib.calc_zn(self.order, r/self.radius, theta_i))
return [np.sum(self._norm_coef * lib.calc_zn(self.order, r / self.radius, theta_i))
for theta_i in theta]
else:
return np.sum(self._norm_coef * lib.calc_zn(self.order, r/self.radius, theta))
return np.sum(self._norm_coef * lib.calc_zn(self.order, r / self.radius, theta))