added append_with_circle_edge_points

Author: Jammy2211

autoarray/inversion/pixelization/image_mesh/__init__.py

@@ -1,3 +1,4 @@
 from .hilbert import Hilbert
 from .overlay import Overlay
 from .kmeans import KMeans
+from .abstract import append_with_circle_edge_points

autoarray/inversion/pixelization/image_mesh/abstract.py

@@ -1,16 +1,47 @@
 from typing import Optional
 
 import numpy as np
-import os
 
 from autoarray.mask.mask_2d import Mask2D
 from autoarray.structures.arrays.uniform_2d import Array2D
 from autoarray.structures.grids.irregular_2d import Grid2DIrregular
-from autoarray.inversion.inversion.settings import SettingsInversion
 
 from autoarray.structures.grids import grid_2d_util
 
-from autoarray import exc
+
+def append_with_circle_edge_points(image_plane_mesh_grid, centre, radius, n_points):
+    """
+    Generate N uniformly spaced (y, x) coordinates around a circle.
+
+    Parameters
+    ----------
+    centre : (float, float)
+        The (y, x) centre of the circle.
+    radius : float
+        Circle radius.
+    n_points : int
+        Number of points around the circle.
+    xp : array namespace (np or jnp)
+        Function will use NumPy or JAX depending on what is passed.
+
+    Returns
+    -------
+    coords : (n_points, 2) xp.ndarray
+        Array of (y, x) coordinates sampled uniformly around the circle.
+    """
+    y0, x0 = centre
+
+    # angles from 0 to 2π
+    theta = np.linspace(0, 2 * np.pi, n_points, endpoint=False)
+
+    # parametric circle
+    ys = y0 + radius * np.sin(theta)
+    xs = x0 + radius * np.cos(theta)
+
+    circle_edge_points = np.stack([ys, xs], axis=-1)
+
+    return Grid2DIrregular(np.vstack([image_plane_mesh_grid, circle_edge_points]))
+
 
 
 class AbstractImageMesh:

autoarray/inversion/pixelization/image_mesh/hilbert.py

@@ -9,7 +9,6 @@
 from autoarray.inversion.pixelization.image_mesh.abstract_weighted import (
     AbstractImageMeshWeighted,
 )
-from autoarray.inversion.inversion.settings import SettingsInversion
 from autoarray.structures.grids.irregular_2d import Grid2DIrregular
 
 from autoarray import exc

test_autoarray/inversion/pixelization/image_mesh/test_abstract.py

@@ -42,6 +42,27 @@ def test__mesh_pixels_per_image_pixels_from(mask, mesh_grid, image_mesh):
     )
 
 
+def test__append_with_circle_edge_points():
+
+    image_plane_mesh_grid = aa.Grid2DIrregular(
+        values=[(0.0, 0.0), (1.0, 1.0), (2.0, 2.0)]
+    )
+
+    grid_with_circle = aa.image_mesh.append_with_circle_edge_points(
+        image_plane_mesh_grid=image_plane_mesh_grid,
+        centre=(0.0, 0.0),
+        radius=3.0,
+        n_points=4,
+    )
+
+    assert grid_with_circle == pytest.approx(
+        np.array([[0.0, 0.0], [1.0, 1.0], [2.0, 2.0], [0.0, 3.0], [3.0, 0.0], [0.0, -3.0], [-3.0, 0.0]]), 1.0e-4
+    )
+
+
+
+
+
 def test__check_mesh_pixels_per_image_pixels(mask, mesh_grid, image_mesh):
     image_mesh.check_mesh_pixels_per_image_pixels(
         mask=mask,