From 893d8bcb151f044fd105efe7f996f236e1d64192 Mon Sep 17 00:00:00 2001 From: jan Date: Sun, 7 Apr 2019 17:18:45 -0700 Subject: [PATCH] Cleaner approach to indexing --- gridlock/draw.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/gridlock/draw.py b/gridlock/draw.py index 8ba6f52..63ac508 100644 --- a/gridlock/draw.py +++ b/gridlock/draw.py @@ -165,8 +165,8 @@ def draw_polygons(self, w = (w_xy[:, :, newaxis] * w_z).transpose(numpy.insert([0, 1], surface_normal, (2,))) # ## Modify the grid - g_slice = tuple(numpy.s_[bdi_min[a]:bdi_max[a] + 1] for a in range(3)) - self.grids[i][g_slice] = (1 - w) * self.grids[i][g_slice] + w * eps[i] + g_slice = (i,) + tuple(numpy.s_[bdi_min[a]:bdi_max[a] + 1] for a in range(3)) + self.grids[g_slice] = (1 - w) * self.grids[g_slice] + w * eps[i] def draw_polygon(self,