From a90094689066b4acd72ade68df88ca71d61b75c3 Mon Sep 17 00:00:00 2001 From: Jan Petykiewicz Date: Fri, 13 Dec 2019 01:22:00 -0800 Subject: [PATCH] fix crash in shifted_exyz --- gridlock/grid.py | 2 +- gridlock/position.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/gridlock/grid.py b/gridlock/grid.py index 06be4b0..2d716a9 100644 --- a/gridlock/grid.py +++ b/gridlock/grid.py @@ -139,7 +139,7 @@ class Grid(object): # If shift is negative, use left cell's dx to determine shift for a in range(3): 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)] diff --git a/gridlock/position.py b/gridlock/position.py index ae58b8b..c9aa43b 100644 --- a/gridlock/position.py +++ b/gridlock/position.py @@ -99,7 +99,7 @@ def pos2ind(self, x = self.shifted_xyz(which_shifts)[a][xi] dx = self.shifted_dxyz(which_shifts)[a][xi] f = (r[a] - x) / dx - + # Clip to centers grid_pos[a] = numpy.clip(xi + f, 0, self.shape[a] - 1) return grid_pos