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
|
# Copyright (c) DataLab Platform Developers, BSD 3-Clause license, see LICENSE file.
"""
Signal extraction and ROI operations
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
"""
from __future__ import annotations
import numpy as np
from sigima.objects import ROI1DParam, SignalObj
from sigima.proc.base import dst_1_to_1
from sigima.proc.decorator import computation_function
@computation_function()
def extract_rois(src: SignalObj, params: list[ROI1DParam]) -> SignalObj:
"""Extract multiple regions of interest from data
Args:
src: source signal
params: list of ROI parameters
Returns:
Signal with multiple regions of interest
"""
suffix = None
if len(params) == 1:
p: ROI1DParam = params[0]
suffix = f"{p.xmin:.3g}≤x≤{p.xmax:.3g}"
dst = dst_1_to_1(src, "extract_rois", suffix)
x, y = src.get_data()
xout, yout = np.ones_like(x) * np.nan, np.ones_like(y) * np.nan
for p in params:
idx1, idx2 = np.searchsorted(x, p.xmin), np.searchsorted(x, p.xmax)
slice0 = slice(idx1, idx2)
xout[slice0], yout[slice0] = x[slice0], y[slice0]
nans = np.isnan(xout) | np.isnan(yout)
# TODO: Handle uncertainty data
dst.set_xydata(xout[~nans], yout[~nans])
# Remove ROI from destination signal: the extracted data no longer needs ROI
dst.roi = None
return dst
@computation_function()
def extract_roi(src: SignalObj, p: ROI1DParam) -> SignalObj:
"""Extract single region of interest from data
Args:
src: source signal
p: ROI parameters
Returns:
Signal with single region of interest
"""
dst = dst_1_to_1(src, "extract_roi", f"{p.xmin:.3g}≤x≤{p.xmax:.3g}")
x, y = p.get_data(src).copy()
# TODO: Handle uncertainty data
dst.set_xydata(x, y)
# Remove ROI from destination signal: the extracted data no longer needs ROI
dst.roi = None
return dst
|
