move generate indices and grid to public method

Author: chris-langfield

src/hexfft/array.py

@@ -2,7 +2,7 @@
 from hexfft.grids import heshgrid, skew_heshgrid
 
 
-def _generate_indices(shape, pattern):
+def generate_indices(shape, pattern):
     N1, N2 = shape
     n1, n2 = np.meshgrid(np.arange(N1), np.arange(N2))
 
@@ -14,7 +14,7 @@ def _generate_indices(shape, pattern):
     return n1, n2
 
 
-def _generate_grid(shape, pattern):
+def generate_grid(shape, pattern):
     if pattern == "oblique":
         return skew_heshgrid(shape)
     elif pattern == "offset":
@@ -53,8 +53,8 @@ def __new__(cls, arr, pattern="offset"):
             )
 
         obj.pattern = pattern
-        obj.indices = _generate_indices(arr.shape[-2:], pattern)
-        obj.grid = _generate_grid(arr.shape[-2:], pattern)
+        obj.indices = generate_indices(arr.shape[-2:], pattern)
+        obj.grid = generate_grid(arr.shape[-2:], pattern)
 
         return obj
 

src/hexfft/utils.py

@@ -1,5 +1,5 @@
 import numpy as np
-from hexfft.array import HexArray, _generate_indices
+from hexfft.array import HexArray, generate_indices
 from hexfft.grids import heshgrid, skew_heshgrid
 
 
@@ -14,7 +14,7 @@ def hsupport(N, pattern="oblique"):
     :param h: a HexArray
     """
     # assert N1 % 2 == 0, "Side length must be even."
-    n1, n2 = _generate_indices((N, N), pattern)
+    n1, n2 = generate_indices((N, N), pattern)
     M = N // 2
     if pattern == "offset":
         n2 = n2 - N // 4