LSST Applications g0f08755f38+82efc23009,g12f32b3c4e+e7bdf1200e,g1653933729+a8ce1bb630,g1a0ca8cf93+50eff2b06f,g28da252d5a+52db39f6a5,g2bbee38e9b+37c5a29d61,g2bc492864f+37c5a29d61,g2cdde0e794+c05ff076ad,g3156d2b45e+41e33cbcdc,g347aa1857d+37c5a29d61,g35bb328faa+a8ce1bb630,g3a166c0a6a+37c5a29d61,g3e281a1b8c+fb992f5633,g414038480c+7f03dfc1b0,g41af890bb2+11b950c980,g5fbc88fb19+17cd334064,g6b1c1869cb+12dd639c9a,g781aacb6e4+a8ce1bb630,g80478fca09+72e9651da0,g82479be7b0+04c31367b4,g858d7b2824+82efc23009,g9125e01d80+a8ce1bb630,g9726552aa6+8047e3811d,ga5288a1d22+e532dc0a0b,gae0086650b+a8ce1bb630,gb58c049af0+d64f4d3760,gc28159a63d+37c5a29d61,gcf0d15dbbd+2acd6d4d48,gd7358e8bfb+778a810b6e,gda3e153d99+82efc23009,gda6a2b7d83+2acd6d4d48,gdaeeff99f8+1711a396fd,ge2409df99d+6b12de1076,ge79ae78c31+37c5a29d61,gf0baf85859+d0a5978c5a,gf3967379c6+4954f8c433,gfb92a5be7c+82efc23009,gfec2e1e490+2aaed99252,w.2024.46
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