LSST Applications 27.0.0,g0265f82a02+469cd937ee,g02d81e74bb+21ad69e7e1,g1470d8bcf6+cbe83ee85a,g2079a07aa2+e67c6346a6,g212a7c68fe+04a9158687,g2305ad1205+94392ce272,g295015adf3+81dd352a9d,g2bbee38e9b+469cd937ee,g337abbeb29+469cd937ee,g3939d97d7f+72a9f7b576,g487adcacf7+71499e7cba,g50ff169b8f+5929b3527e,g52b1c1532d+a6fc98d2e7,g591dd9f2cf+df404f777f,g5a732f18d5+be83d3ecdb,g64a986408d+21ad69e7e1,g858d7b2824+21ad69e7e1,g8a8a8dda67+a6fc98d2e7,g99cad8db69+f62e5b0af5,g9ddcbc5298+d4bad12328,ga1e77700b3+9c366c4306,ga8c6da7877+71e4819109,gb0e22166c9+25ba2f69a1,gb6a65358fc+469cd937ee,gbb8dafda3b+69d3c0e320,gc07e1c2157+a98bf949bb,gc120e1dc64+615ec43309,gc28159a63d+469cd937ee,gcf0d15dbbd+72a9f7b576,gdaeeff99f8+a38ce5ea23,ge6526c86ff+3a7c1ac5f1,ge79ae78c31+469cd937ee,gee10cc3b42+a6fc98d2e7,gf1cff7945b+21ad69e7e1,gfbcc870c63+9a11dc8c8f
LSST Data Management Base Package
Loading...
Searching...
No Matches
source.py
Go to the documentation of this file.
1# This file is part of scarlet_lite.
2#
3# Developed for the LSST Data Management System.
4# This product includes software developed by the LSST Project
5# (https://www.lsst.org).
6# See the COPYRIGHT file at the top-level directory of this distribution
7# for details of code ownership.
8#
9# This program is free software: you can redistribute it and/or modify
10# it under the terms of the GNU General Public License as published by
11# the Free Software Foundation, either version 3 of the License, or
12# (at your option) any later version.
13#
14# This program is distributed in the hope that it will be useful,
15# but WITHOUT ANY WARRANTY; without even the implied warranty of
16# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17# GNU General Public License for more details.
18#
19# You should have received a copy of the GNU General Public License
20# along with this program. If not, see <https://www.gnu.org/licenses/>.
21
22__all__ = ["Source"]
23
24from typing import Callable
25
26from .bbox import Box
27from .component import Component
28from .image import Image
29
30
31class Source:
32 """A container for components associated with the same astrophysical object
33
34 A source can have a single component, or multiple components,
35 and each can be contained in different bounding boxes.
36
37 Parameters
38 ----------
39 components:
40 The components contained in the source.
41 """
42
43 def __init__(self, components: list[Component]):
44 self.components = components
45 self.flux_weighted_image: Image | None = None
46
47 @property
48 def n_components(self) -> int:
49 """The number of components in this source"""
50 return len(self.components)
51
52 @property
53 def center(self) -> tuple[int, int] | None:
54 """The center of the source in the full Blend."""
55 if not self.is_null and hasattr(self.components[0], "peak"):
56 return self.components[0].peak # type: ignore
57 return None
58
59 @property
60 def source_center(self) -> tuple[int, int] | None:
61 """The center of the source in its local bounding box."""
62 _center = self.center
63 _origin = self.bbox.origin
64 if _center is not None:
65 center = (
66 _center[0] - _origin[0],
67 _center[1] - _origin[1],
68 )
69 return center
70 return None
71
72 @property
73 def is_null(self) -> bool:
74 """True if the source does not have any components"""
75 return self.n_componentsn_components == 0
76
77 @property
78 def bbox(self) -> Box:
79 """The minimal bounding box to contain all of this sources components
80
81 Null sources have a bounding box with shape `(0,0,0)`
82 """
83 if self.n_componentsn_components == 0:
84 return Box((0, 0))
85 bbox = self.components[0].bbox
86 for component in self.components[1:]:
87 bbox = bbox | component.bbox
88 return bbox
89
90 @property
91 def bands(self) -> tuple:
92 """The bands in the full source model."""
93 if self.is_null:
94 return ()
95 return self.components[0].bands
96
97 def get_model(self, use_flux: bool = False) -> Image:
98 """Build the model for the source
99
100 This is never called during optimization and is only used
101 to generate a model of the source for investigative purposes.
102
103 Parameters
104 ----------
105 use_flux:
106 Whether to use the re-distributed flux associated with the source
107 instead of the component models.
108
109 Returns
110 -------
111 model:
112 The full-color model.
113 """
114 if self.n_componentsn_components == 0:
115 return 0 # type: ignore
116
117 if use_flux:
118 # Return the redistributed flux
119 # (calculated by scarlet.lite.measure.weight_sources)
120 return self.flux_weighted_image # type: ignore
121
122 model = self.components[0].get_model()
123 for component in self.components[1:]:
124 model = model + component.get_model()
125 return model
126
127 def parameterize(self, parameterization: Callable):
128 """Convert the component parameter arrays into Parameter instances
129
130 Parameters
131 ----------
132 parameterization:
133 A function to use to convert parameters of a given type into
134 a `Parameter` in place. It should take a single argument that
135 is the `Component` or `Source` that is to be parameterized.
136 """
137 for component in self.components:
138 component.parameterize(parameterization)
139
140 def __str__(self):
141 return f"Source<{len(self.components)}>"
142
143 def __repr__(self):
144 return f"Source(components={repr(self.components)})>"
tuple[int, int]|None center(self)
Definition source.py:53
Image get_model(self, bool use_flux=False)
Definition source.py:97
tuple[int, int]|None source_center(self)
Definition source.py:60
__init__(self, list[Component] components)
Definition source.py:43
parameterize(self, Callable parameterization)
Definition source.py:127