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

only translate when origin is not 0, 0, 0

Browse source
This commit is contained in:
RemDelaporteMathurin 2022-10-21 15:55:08 +02:00
parent b0743dfdea
commit b874d83634
1 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

14
openmc/mesh.py
View file

@ -1322,9 +1322,10 @@ class CylindricalMesh(StructuredMesh):
pts_cartesian[:, 1] = r * np.sin(phi)
# offset with 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]
if self.origin != [0.0, 0.0, 0.0]:
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]
return super().write_data_to_vtk(
points=pts_cartesian,
@ -1576,9 +1577,10 @@ class SphericalMesh(StructuredMesh):
pts_cartesian[:, 2] = r * np.cos(phi)
# offset with 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]
if self.origin != [0.0, 0.0, 0.0]:
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]
return super().write_data_to_vtk(
points=pts_cartesian,