LSST Applications g1cfbe01172+01aa18f939,g20cdd03214+31e6b93548,g28da252d5a+ea8665a95b,g2bbee38e9b+9ec6cc348d,g2bc492864f+9ec6cc348d,g347aa1857d+9ec6cc348d,g3a166c0a6a+9ec6cc348d,g4322eb9e3a+65eff1e020,g461a3dce89+b86e4b8053,g50ff169b8f+f991eae79d,g52b1c1532d+b86e4b8053,g607f77f49a+31e6b93548,g78056777b3+8ae2798781,g858d7b2824+31e6b93548,g8cd86fa7b1+4851e61ca4,g9ddcbc5298+f24b38b85a,ga1e77700b3+3309dba821,gae0086650b+b86e4b8053,gb0e22166c9+6076c0b52b,gbb886bcc26+dccb771098,gbd462c55f0+dc07f8e65d,gc0c51c7ec2+31e6b93548,gc120e1dc64+a417ce3171,gc28159a63d+9ec6cc348d,gc2a6998b7e+f95f64aeae,gcdd4ae20e8+507450c4cd,gcf0d15dbbd+507450c4cd,gd1535ee943+bcf88ba65f,gd598c5cd71+66126f91fb,gdaeeff99f8+006e14e809,gdbce86181e+39d5515b1a,ge3d4d395c2+b12d4d6a95,ge79ae78c31+9ec6cc348d,gf048a9a2f4+d9c36e6b63,gfbcc870c63+ea41c4420b,w.2024.27
LSST Data Management Base Package
Loading...
Searching...
No Matches
CloughTocher2DInterpolatorUtils.cc
Go to the documentation of this file.
1// -*- LSST-C++ -*-
2/*
3 * This file is part of meas_algorithms.
4 *
5 * Developed for the LSST Data Management System.
6 * This product includes software developed by the LSST Project
7 * (https://www.lsst.org).
8 * See the COPYRIGHT file at the top-level directory of this distribution
9 * for details of code ownership.
10 *
11 * This program is free software: you can redistribute it and/or modify
12 * it under the terms of the GNU General Public License as published by
13 * the Free Software Foundation, either version 3 of the License, or
14 * (at your option) any later version.
15 *
16 * This program is distributed in the hope that it will be useful,
17 * but WITHOUT ANY WARRANTY; without even the implied warranty of
18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19 * GNU General Public License for more details.
20 *
21 * You should have received a copy of the GNU General Public License
22 * along with this program. If not, see <https://www.gnu.org/licenses/>.
23 */
24
35#include "lsst/geom/Point.h"
37
38namespace lsst {
39namespace meas {
40namespace algorithms {
41
42std::pair<ndarray::Array<float, 2, 2>, ndarray::Array<float, 2, 2>>
44 std::vector<std::string> const &maskPlanes,
45 int const buffer) {
46 auto bitMask = afw::image::Mask<>::getPlaneBitMask(maskPlanes);
47
48 auto badSpanSet = afw::geom::SpanSet::fromMask(
49 *mimage.getMask(), [bitMask](afw::image::MaskPixel pixel) { return bool(pixel & bitMask); });
50 ndarray::Array<float, 2, 2> badPixelArray = ndarray::allocate(badSpanSet->getArea(), 3);
51 badSpanSet->applyFunctor(
52 [](geom::Point2I pt, float &x, float &y, float &valueOut, float valueIn) {
53 x = pt.getX();
54 y = pt.getY();
55 valueOut = valueIn;
56 },
57 ndFlat(badPixelArray[ndarray::view()(0)].shallow()),
58 ndFlat(badPixelArray[ndarray::view()(1)].shallow()),
59 ndFlat(badPixelArray[ndarray::view()(2)].shallow()),
60 *mimage.getImage());
61
62 auto allSpanSet = badSpanSet->dilated(buffer, afw::geom::Stencil::BOX);
63 auto goodSpanSet = allSpanSet->intersectNot(*badSpanSet);
64 goodSpanSet = goodSpanSet->clippedTo(mimage.getBBox());
65
66 badSpanSet.reset();
67 allSpanSet.reset();
68
69 ndarray::Array<float, 2, 2> goodPixelArray = ndarray::allocate(goodSpanSet->getArea(), 3);
70 goodSpanSet->applyFunctor(
71 [](geom::Point2I pt, float &x, float &y, float &valueOut, float valueIn) {
72 x = pt.getX();
73 y = pt.getY();
74 valueOut = valueIn;
75 },
76 ndFlat(goodPixelArray[ndarray::view()(0)].shallow()),
77 ndFlat(goodPixelArray[ndarray::view()(1)].shallow()),
78 ndFlat(goodPixelArray[ndarray::view()(2)].shallow()),
79 *mimage.getImage());
80
81 goodSpanSet.reset();
82
83 return std::make_pair(badPixelArray, goodPixelArray);
84}
85
86void CloughTocher2DInterpolatorUtils::updateArrayFromImage(ndarray::Array<float, 2, 2> const &pixelArray,
88 afw::image::CheckIndices docheck(true);
89 auto x0 = image.getX0();
90 auto y0 = image.getY0();
91 for (auto row : pixelArray) {
92 int x = row[0] - x0;
93 int y = row[1] - y0;
94 row[2] = image(x, y, docheck);
95 }
96}
97
99 afw::image::Image<float> &image, ndarray::Array<float const, 2, 2> const &pixelArray) {
100 afw::image::CheckIndices docheck(true);
101 auto x0 = image.getX0();
102 auto y0 = image.getY0();
103 for (auto row : pixelArray) {
104 int x = row[0] - x0;
105 int y = row[1] - y0;
106 image(x, y, docheck) = row[2];
107 }
108}
109
110} // namespace algorithms
111} // namespace meas
112} // namespace lsst
Provide pixel-level utilities to support CloughTocher2DInterpolator.
afw::table::Key< afw::table::Array< ImagePixelT > > image
int y
Definition SpanSet.cc:48
static std::shared_ptr< geom::SpanSet > fromMask(image::Mask< T > const &mask, UnaryPredicate comparator=details::AnyBitSetFunctor< T >())
Create a SpanSet from a mask.
Definition SpanSet.h:644
A class used to request that array accesses be checked.
Definition ImageBase.h:74
A class to represent a 2-dimensional array of pixels.
Definition Image.h:51
static MaskPixelT getPlaneBitMask(const std::vector< std::string > &names)
Return the bitmask corresponding to a vector of plane names OR'd together.
Definition Mask.cc:376
A class to manipulate images, masks, and variance as a single object.
Definition MaskedImage.h:74
lsst::geom::Box2I getBBox(ImageOrigin const origin=PARENT) const
MaskPtr getMask() const
Return a (shared_ptr to) the MaskedImage's mask.
ImagePtr getImage() const
Return a (shared_ptr to) the MaskedImage's image.
static void updateArrayFromImage(ndarray::Array< float, 2, 2 > const &pixelArray, afw::image::Image< PixelT > const &image)
Update the values (third column) of the pixelArray from the image.
static std::pair< ndarray::Array< float, 2, 2 >, ndarray::Array< float, 2, 2 > > findGoodPixelsAroundBadPixels(afw::image::MaskedImage< PixelT, afw::image::MaskPixel, afw::image::VariancePixel > const &mimage, std::vector< std::string > const &maskPlanes, int const buffer)
Find the positions of bad pixels as defined by the masks, and also find the location and pixel value ...
static void updateImageFromArray(afw::image::Image< PixelT > &image, ndarray::Array< float const, 2, 2 > const &pixelArray)
Update the pixel values of the image from the pixelArray.
T make_pair(T... args)
int row
Definition CR.cc:145