jan/gridlock
2
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix crash in shifted_exyz

Browse Source
This commit is contained in:
Jan Petykiewicz 2019-12-13 01:22:00 -08:00
parent 43936d1415
commit a900946890
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gridlock/grid.py
View File

@ -139,7 +139,7 @@ class Grid(object):
# If shift is negative, use left cell's dx to determine shift # If shift is negative, use left cell's dx to determine shift
for a in range(3): for a in range(3):
if shifts[a] < 0: if shifts[a] < 0:
dxyz[a] = numpy.roll(dxyz, 1) dxyz[a] = numpy.roll(dxyz[a], 1)
return [self.exyz[a] + dxyz[a] * shifts[a] for a in range(3)] return [self.exyz[a] + dxyz[a] * shifts[a] for a in range(3)]

2
gridlock/position.py
View File

@ -99,7 +99,7 @@ def pos2ind(self,
x = self.shifted_xyz(which_shifts)[a][xi] x = self.shifted_xyz(which_shifts)[a][xi]
dx = self.shifted_dxyz(which_shifts)[a][xi] dx = self.shifted_dxyz(which_shifts)[a][xi]
f = (r[a] - x) / dx f = (r[a] - x) / dx
# Clip to centers # Clip to centers
grid_pos[a] = numpy.clip(xi + f, 0, self.shape[a] - 1) grid_pos[a] = numpy.clip(xi + f, 0, self.shape[a] - 1)
return grid_pos return grid_pos