crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-26 05:05:30 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix a bug in ZernikeRadial

Browse source
This commit is contained in:
rockfool 2020-01-30 17:24:37 -05:00
parent 5b132d05bf
commit cf39a15ef7
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
openmc/polynomial.py
View file

@ -75,7 +75,9 @@ class ZernikeRadial(Polynomial):
def __call__(self, r):
import openmc.lib as lib
if isinstance(r, Iterable):
return [np.sum(self._norm_coef * lib.calc_zn_rad(self.order, r_i))
return [np.sum(self._norm_coef * lib.calc_zn_rad(self.order, r_i / self.radius))
for r_i in r]
else:
return np.sum(self._norm_coef * lib.calc_zn_rad(self.order, r))
return np.sum(self._norm_coef * lib.calc_zn_rad(self.order, r / self.radius))