Coverage for python/lsst/meas/extensions/piff/piffPsf.py: 42%
60 statements
« prev ^ index » next coverage.py v7.5.1, created at 2024-05-11 04:13 -0700
« prev ^ index » next coverage.py v7.5.1, created at 2024-05-11 04:13 -0700
1# This file is part of meas_extensions_piff.
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/>.
22__all__ = ["PiffPsf"]
24import pickle
25import numpy as np
26from lsst.afw.typehandling import StorableHelperFactory
27from lsst.meas.algorithms import ImagePsf
28from lsst.afw.image import Image
29from lsst.geom import Box2I, Point2I, Extent2I, Point2D
32class PiffPsf(ImagePsf):
33 _factory = StorableHelperFactory(
34 "lsst.meas.extensions.piff.piffPsf",
35 "PiffPsf"
36 )
38 def __init__(self, width, height, piffResult):
39 assert width == height
40 ImagePsf.__init__(self)
41 self.width = width
42 self.height = height
43 self.dimensions = Extent2I(width, height)
44 self._piffResult = piffResult
45 self._averagePosition = None
47 @property
48 def piffResult(self):
49 return self._piffResult
51 # Storable overrides
53 def isPersistable(self):
54 return True
56 def _getPersistenceName(self):
57 return "PiffPsf"
59 def _getPythonModule(self):
60 return "lsst.meas.extensions.piff.piffPsf"
62 def _write(self):
63 return pickle.dumps((self.width, self.height, self._piffResult))
65 @staticmethod
66 def _read(pkl):
67 return PiffPsf(*pickle.loads(pkl))
69 # ImagePsf overrides
71 def __deepcopy__(self, meta=None):
72 return PiffPsf(self.width, self.height, self._piffResult)
74 def resized(self, width, height):
75 assert width == height
76 return PiffPsf(width, height, self._piffResult)
78 def _doComputeImage(self, position, color):
79 return self._doImage(position, center=None)
81 def _doComputeKernelImage(self, position, color):
82 return self._doImage(position, center=True)
84 def _doComputeBBox(self, position, color):
85 return self._doBBox(Point2I(0, 0), center=True)
87 def getAveragePosition(self):
88 if self._averagePosition is None:
89 x = np.mean([star.image_pos.x for star in self._piffResult.stars])
90 y = np.mean([star.image_pos.y for star in self._piffResult.stars])
91 self._averagePosition = Point2D(x, y)
92 return self._averagePosition
94 # Internal private methods
96 def _doImage(self, position, center):
97 # Follow Piff conventions for center.
98 # None => draw as if star at position
99 # True => draw in center of image
100 gsimg = self._piffResult.draw(
101 position.x, position.y, stamp_size=self.width, center=center
102 )
103 bbox = self._doBBox(position, center)
104 img = Image(bbox, dtype=np.float64)
105 img.array[:] = gsimg.array
106 img.array /= np.sum(img.array)
107 return img
109 def _doBBox(self, position, center):
110 origin = -(self.dimensions//2)
111 if center is None:
112 origin = Point2I(position) + origin
113 return Box2I(Point2I(origin), self.dimensions)