Skip to content

Commit 3448545

Browse files
Jammy2211Jammy2211
authored andcommitted
plotting bug?
1 parent b20d6b1 commit 3448545

18 files changed

Lines changed: 51 additions & 722 deletions

File tree

autoarray/inversion/pixelization/mesh/delaunay.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,14 @@
11
import numpy as np
2+
from typing import Optional
23

4+
from autoarray.inversion.pixelization.mappers.mapper_grids import MapperGrids
5+
from autoarray.inversion.pixelization.border_relocator import BorderRelocator
6+
from autoarray.inversion.pixelization.mesh.abstract import AbstractMesh
37
from autoarray.structures.mesh.delaunay_2d import Mesh2DDelaunay
4-
from autoarray.inversion.pixelization.mesh.triangulation import Triangulation
8+
from autoarray.structures.grids.uniform_2d import Grid2D
9+
from autoarray.structures.grids.irregular_2d import Grid2DIrregular
510

6-
7-
class Delaunay(Triangulation):
11+
class Delaunay(AbstractMesh):
812
def __init__(self):
913
"""
1014
An irregular mesh of Delaunay triangle pixels, which using linear barycentric interpolation are paired with

autoarray/plot/wrap/two_d/delaunay_drawer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import numpy as np
33
from typing import Optional
44

5+
from autoarray.inversion.pixelization.mappers.delaunay import MapperDelaunay
56
from autoarray.plot.wrap.two_d.abstract import AbstractMatWrap2D
67
from autoarray.plot.wrap.base.units import Units
78

autoarray/structures/mesh/delaunay_2d.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,7 @@ def find_simplex(self, query_points):
236236
return find_simplex_from(query_points, self.points, self.simplices)
237237

238238

239-
class Abstract2DMeshTriangulation(Abstract2DMesh):
239+
class Mesh2DDelaunay(Abstract2DMesh):
240240
def __init__(self, values: Union[np.ndarray, List], _xp=np):
241241
"""
242242
An irregular 2D grid of (y,x) coordinates which represents both a Delaunay triangulation and Voronoi mesh.

test_autoarray/config/visualize.yaml

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -267,12 +267,3 @@ mat_wrap_2d:
267267
linewidth: 51
268268
pivot: middle1
269269
units: xy1
270-
VoronoiDrawer:
271-
figure:
272-
alpha: 0.7
273-
edgecolor: k
274-
linewidth: 0.3
275-
subplot:
276-
alpha: 0.5
277-
edgecolor: r
278-
linewidth: 1.0

test_autoarray/conftest.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -256,11 +256,6 @@ def make_delaunay_mapper_9_3x3():
256256
return fixtures.make_delaunay_mapper_9_3x3()
257257

258258

259-
@pytest.fixture(name="voronoi_mapper_9_3x3")
260-
def make_voronoi_mapper_9_3x3():
261-
return fixtures.make_voronoi_mapper_9_3x3()
262-
263-
264259
@pytest.fixture(name="rectangular_inversion_7x7_3x3")
265260
def make_rectangular_inversion_7x7_3x3():
266261
return fixtures.make_rectangular_inversion_7x7_3x3()
@@ -271,11 +266,6 @@ def make_delaunay_inversion_9_3x3():
271266
return fixtures.make_delaunay_inversion_9_3x3()
272267

273268

274-
@pytest.fixture(name="voronoi_inversion_9_3x3")
275-
def make_voronoi_inversion_9_3x3():
276-
return fixtures.make_voronoi_inversion_9_3x3()
277-
278-
279269
@pytest.fixture(name="euclid_data")
280270
def make_euclid_data():
281271
return fixtures.make_euclid_data()

test_autoarray/inversion/inversion/test_factory.py

Lines changed: 0 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,6 @@ def test__inversion_imaging__via_mapper(
116116
def test__inversion_imaging__via_regularizations(
117117
masked_imaging_7x7_no_blur,
118118
delaunay_mapper_9_3x3,
119-
voronoi_mapper_9_3x3,
120119
regularization_constant,
121120
regularization_constant_split,
122121
regularization_adaptive_brightness,
@@ -154,40 +153,6 @@ def test__inversion_imaging__via_regularizations(
154153
)
155154
assert inversion.mapped_reconstructed_image == pytest.approx(np.ones(9), 1.0e-4)
156155

157-
pytest.importorskip(
158-
"autoarray.util.nn.nn_py",
159-
reason="Voronoi C library not installed, see util.nn README.md",
160-
)
161-
162-
mapper = copy.copy(voronoi_mapper_9_3x3)
163-
mapper.regularization = regularization_constant
164-
165-
inversion = aa.Inversion(
166-
dataset=masked_imaging_7x7_no_blur_w_tilde,
167-
linear_obj_list=[mapper],
168-
)
169-
170-
assert isinstance(inversion.linear_obj_list[0], aa.MapperVoronoi)
171-
assert inversion.log_det_curvature_reg_matrix_term == pytest.approx(
172-
10.66505, 1.0e-4
173-
)
174-
assert inversion.mapped_reconstructed_image == pytest.approx(np.ones(9), 1.0e-4)
175-
176-
mapper = copy.copy(voronoi_mapper_9_3x3)
177-
mapper.regularization = regularization_constant_split
178-
179-
inversion = aa.Inversion(
180-
dataset=masked_imaging_7x7_no_blur_w_tilde,
181-
linear_obj_list=[mapper],
182-
)
183-
184-
assert isinstance(inversion.linear_obj_list[0], aa.MapperVoronoi)
185-
assert inversion.log_det_curvature_reg_matrix_term == pytest.approx(
186-
10.37955, 1.0e-4
187-
)
188-
assert inversion.mapped_reconstructed_image == pytest.approx(np.ones(9), 1.0e-4)
189-
190-
191156
def test__inversion_imaging__source_pixel_zeroed_indices(
192157
masked_imaging_7x7_no_blur,
193158
rectangular_mapper_7x7_3x3,

test_autoarray/inversion/inversion/test_mapper_valued.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
def test__max_pixel_list_from_and_centre():
88
mapper = aa.m.MockMapper(
9-
source_plane_mesh_grid=aa.Mesh2DVoronoi(
9+
source_plane_mesh_grid=aa.Mesh2DDelaunay(
1010
[[1.0, 2.0], [3.0, 4.0], [5.0, 6.0], [5.0, 0.0]]
1111
)
1212
)
@@ -25,7 +25,7 @@ def test__max_pixel_list_from_and_centre():
2525

2626
def test__max_pixel_list_from__filter_neighbors():
2727
mapper = aa.m.MockMapper(
28-
source_plane_mesh_grid=aa.Mesh2DVoronoi(
28+
source_plane_mesh_grid=aa.Mesh2DDelaunay(
2929
[
3030
[1.0, 1.0],
3131
[1.0, 2.0],
@@ -101,7 +101,7 @@ def test__magnification_via_mesh_from():
101101

102102
magnification = [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0]
103103

104-
source_plane_mesh_grid = aa.Mesh2DVoronoi(
104+
source_plane_mesh_grid = aa.Mesh2DDelaunay(
105105
values=np.array(
106106
[
107107
[0.0, 0.0],
@@ -147,7 +147,7 @@ def test__magnification_via_mesh_from__with_pixel_mask():
147147

148148
magnification = np.array([1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0])
149149

150-
source_plane_mesh_grid = aa.Mesh2DVoronoi(
150+
source_plane_mesh_grid = aa.Mesh2DDelaunay(
151151
values=np.array(
152152
[
153153
[0.0, 0.0],

test_autoarray/inversion/pixelization/mappers/test_delaunay.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44
import autoarray as aa
55

6-
from autoarray.structures.mesh.triangulation_2d import find_simplex_from
6+
from autoarray.structures.mesh.delaunay_2d import find_simplex_from
77
from autoarray.inversion.pixelization.mappers.delaunay import (
88
pix_indexes_for_sub_slim_index_delaunay_from,
99
)

test_autoarray/inversion/pixelization/mappers/test_factory.py

Lines changed: 0 additions & 53 deletions
Original file line numberDiff line numberDiff line change
@@ -107,56 +107,3 @@ def test__delaunay_mapper():
107107
1.0e-2,
108108
)
109109

110-
111-
def test__voronoi_mapper():
112-
pytest.importorskip(
113-
"autoarray.util.nn.nn_py",
114-
reason="Voronoi C library not installed, see util.nn README.md",
115-
)
116-
117-
mask = aa.Mask2D(
118-
mask=[
119-
[True, True, True, True, True],
120-
[True, True, False, True, True],
121-
[True, False, False, False, True],
122-
[True, True, False, True, True],
123-
[True, True, True, True, True],
124-
],
125-
pixel_scales=1.0,
126-
)
127-
128-
# Slightly manipulate input grid so sub gridding is evidence in first source pixel.
129-
grid = aa.Grid2D.from_mask(mask=mask, over_sample_size=2)
130-
131-
grid.over_sampled[0, 0] = -2.0
132-
grid.over_sampled[0, 1] = 2.0
133-
134-
mesh = aa.mesh.Voronoi()
135-
image_mesh = aa.image_mesh.Overlay(shape=(3, 3))
136-
image_plane_mesh_grid = image_mesh.image_plane_mesh_grid_from(
137-
mask=mask, adapt_data=None
138-
)
139-
140-
mapper_grids = mesh.mapper_grids_from(
141-
mask=mask,
142-
border_relocator=None,
143-
source_plane_data_grid=grid,
144-
source_plane_mesh_grid=image_plane_mesh_grid,
145-
)
146-
147-
mapper = aa.Mapper(mapper_grids=mapper_grids, regularization=None)
148-
149-
assert (mapper.source_plane_mesh_grid == image_plane_mesh_grid).all()
150-
assert mapper.source_plane_mesh_grid.origin == pytest.approx((0.0, 0.0), 1.0e-4)
151-
152-
# assert mapper.mapping_matrix == pytest.approx(
153-
# np.array(
154-
# [
155-
# [0.6875, 0.0, 0.0, 0.3125, 0.0],
156-
# [0.0, 1.0, 0.0, 0.0, 0.0],
157-
# [0.125, 0.125, 0.5, 0.125, 0.125],
158-
# [0.0, 0.0, 0.0, 0.9375, 0.0625],
159-
# [0.0, 0.0, 0.0, 0.0, 1.0],
160-
# ]
161-
# )
162-
# )

test_autoarray/inversion/pixelization/mappers/test_voronoi.py

Lines changed: 0 additions & 44 deletions
This file was deleted.

0 commit comments

Comments
 (0)