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

Add some comments

Browse source
This commit is contained in:
Zhuoran Han 2018-08-16 14:14:05 -05:00
parent 847e2cc280
commit f7d098f075
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
openmc/polynomials.py
View file

@ -19,6 +19,7 @@ class Legendre(Polynomials):
self.norm_coeff = np.multiply(self.norm_vec,self.coeff)
def __call__(self, x):
# x is the normalized position on [-1,1]
pn = capi.calc_pn(self.order, x)
return np.sum(np.multiply(self.norm_coeff, pn))
@ -27,9 +28,11 @@ class ZernikeRadial(Polynomials):
super().__init__(coeff)
self.order = 2 * (self.n_coeff - 1)
self.radius = radius
self.norm_vec = (2 * np.arange(self.n_coeff) + 1)/(np.pi * radius ** 2)
self.norm_vec = (2 * np.arange(self.n_coeff) + 1) / (np.pi *
radius ** 2)
self.norm_coeff = np.multiply(self.norm_vec,self.coeff)
def __call__(self, r):
# r is the normalized position on radius [0,1]
zn_rad = capi.calc_zn_rad(self.order, r)
return np.sum(np.multiply(self.norm_coeff, zn_rad))