Bugfix in dense optimized solution.

kyleaoman · kyleaoman · commit b1aaa4ae9b33 · 2024-11-12T16:19:58.000Z
diff --git a/swiftgalaxy/iterator.py b/swiftgalaxy/iterator.py
@@ -453,13 +453,13 @@ def _eval_dense_optimized_solution(self):
             np.array(
                 np.meshgrid(*[np.arange(gdi) + 0.5 for gdi in grid_dim], indexing="ij")
             )
-            * sm.cell_size[:, np.newaxis, np.newaxis, np.newaxis]
+            * (sm.cell_size * grid_element_dim)[:, np.newaxis, np.newaxis, np.newaxis]
         )
         centres_offset = (
             np.array(
                 np.meshgrid(*[np.arange(gdi) + 1.0 for gdi in grid_dim], indexing="ij")
             )
-            * sm.cell_size[:, np.newaxis, np.newaxis, np.newaxis]
+            * (sm.cell_size * grid_element_dim)[:, np.newaxis, np.newaxis, np.newaxis]
         )
         unique_region_centres = cosmo_array(
             (centres_aligned, centres_offset)

Original file line number	Diff line number	Diff line change
`@@ -453,13 +453,13 @@ def _eval_dense_optimized_solution(self):`
`453`	`453`	`np.array(`
`454`	`454`	`np.meshgrid(*[np.arange(gdi) + 0.5 for gdi in grid_dim], indexing="ij")`
`455`	`455`	`)`
`456`		`- * sm.cell_size[:, np.newaxis, np.newaxis, np.newaxis]`
	`456`	`+ * (sm.cell_size * grid_element_dim)[:, np.newaxis, np.newaxis, np.newaxis]`
`457`	`457`	`)`
`458`	`458`	`centres_offset = (`
`459`	`459`	`np.array(`
`460`	`460`	`np.meshgrid(*[np.arange(gdi) + 1.0 for gdi in grid_dim], indexing="ij")`
`461`	`461`	`)`
`462`		`- * sm.cell_size[:, np.newaxis, np.newaxis, np.newaxis]`
	`462`	`+ * (sm.cell_size * grid_element_dim)[:, np.newaxis, np.newaxis, np.newaxis]`
`463`	`463`	`)`
`464`	`464`	`unique_region_centres = cosmo_array(`
`465`	`465`	`(centres_aligned, centres_offset)`