1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
# Copyright (c) DataLab Platform Developers, BSD 3-Clause license, see LICENSE file.
"""
Image offset correction unit test.
"""
# pylint: disable=invalid-name # Allows short reference names like x, y, ...
# pylint: disable=duplicate-code
from __future__ import annotations
import numpy as np
import pytest
import sigima.objects
import sigima.params
import sigima.proc.image
from sigima.tests import guiutils
from sigima.tests.data import create_noisy_gaussian_image
@pytest.mark.gui
def test_image_offset_correction_interactive() -> None:
"""Image offset correction interactive test."""
with guiutils.lazy_qt_app_context(force=True):
# pylint: disable=import-outside-toplevel
from plotpy.builder import make
from plotpy.items import RectangleShape
from plotpy.tools import RectangleTool
from plotpy.widgets.selectdialog import SelectDialog, select_with_shape_tool
from sigima.tests import vistools
i1 = create_noisy_gaussian_image()
shape: RectangleShape = select_with_shape_tool(
None,
RectangleTool,
make.image(i1.data, interpolation="nearest", eliminate_outliers=1.0),
"Select background area",
tooldialogclass=SelectDialog,
)
if shape is not None:
param = sigima.objects.ROI2DParam()
# pylint: disable=unbalanced-tuple-unpacking
ix0, iy0, ix1, iy1 = i1.physical_to_indices(shape.get_rect())
param.x0, param.y0, param.dx, param.dy = ix0, iy0, ix1 - ix0, iy1 - iy0
i2 = sigima.proc.image.offset_correction(i1, param)
i3 = sigima.proc.image.clip(i2, sigima.params.ClipParam.create(lower=0))
vistools.view_images_side_by_side(
[i1, i3],
titles=["Original image", "Corrected image"],
title="Image offset correction and thresholding",
)
@pytest.mark.validation
def test_image_offset_correction() -> None:
"""Image offset correction validation test."""
i1 = create_noisy_gaussian_image()
p = sigima.objects.ROI2DParam.create(x0=0, y0=0, dx=10, dy=10)
i2 = sigima.proc.image.offset_correction(i1, p)
# Check that the offset correction has been applied
ix0, iy0 = int(p.x0), int(p.y0)
ix1, iy1 = int(p.x0 + p.dx), int(p.y0 + p.dy)
offset = np.mean(i1.data[iy0:iy1, ix0:ix1])
assert np.allclose(i2.data, i1.data - offset), "Offset correction failed"
if __name__ == "__main__":
test_image_offset_correction_interactive()
test_image_offset_correction()
|
