crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-27 21:55:41 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

origin is tuple + fixed any

Browse source
This commit is contained in:
RemDelaporteMathurin 2022-11-21 14:07:45 -05:00
parent 3325667c51
commit 10a95186e3
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
openmc/mesh.py
View file

@ -1052,7 +1052,7 @@ class CylindricalMesh(StructuredMesh):
self._r_grid = None
self._phi_grid = [0.0, 2*pi]
self._z_grid = None
self._origin = [0., 0., 0.]
self._origin = (0., 0., 0.)
@property
def dimension(self):
@ -1325,7 +1325,7 @@ class CylindricalMesh(StructuredMesh):
pts_cartesian[:, 1] = r * np.sin(phi)
# offset with origin
if any(self.origin != (0.0, 0.0, 0.0)):
if any([coord != 0 for coord in self.origin]):
pts_cartesian[:, 0] = pts_cartesian[:, 0] + self.origin[0]
pts_cartesian[:, 1] = pts_cartesian[:, 1] + self.origin[1]
pts_cartesian[:, 2] = pts_cartesian[:, 2] + self.origin[2]
@ -1583,7 +1583,7 @@ class SphericalMesh(StructuredMesh):
pts_cartesian[:, 2] = r * np.cos(phi)
# offset with origin
if any(self.origin != (0.0, 0.0, 0.0)):
if any([coord != 0 for coord in self.origin]):
pts_cartesian[:, 0] = pts_cartesian[:, 0] + self.origin[0]
pts_cartesian[:, 1] = pts_cartesian[:, 1] + self.origin[1]
pts_cartesian[:, 2] = pts_cartesian[:, 2] + self.origin[2]