LSST Applications  21.0.0-147-g0e635eb1+1acddb5be5,22.0.0+052faf71bd,22.0.0+1ea9a8b2b2,22.0.0+6312710a6c,22.0.0+729191ecac,22.0.0+7589c3a021,22.0.0+9f079a9461,22.0.1-1-g7d6de66+b8044ec9de,22.0.1-1-g87000a6+536b1ee016,22.0.1-1-g8e32f31+6312710a6c,22.0.1-10-gd060f87+016f7cdc03,22.0.1-12-g9c3108e+df145f6f68,22.0.1-16-g314fa6d+c825727ab8,22.0.1-19-g93a5c75+d23f2fb6d8,22.0.1-19-gb93eaa13+aab3ef7709,22.0.1-2-g8ef0a89+b8044ec9de,22.0.1-2-g92698f7+9f079a9461,22.0.1-2-ga9b0f51+052faf71bd,22.0.1-2-gac51dbf+052faf71bd,22.0.1-2-gb66926d+6312710a6c,22.0.1-2-gcb770ba+09e3807989,22.0.1-20-g32debb5+b8044ec9de,22.0.1-23-gc2439a9a+fb0756638e,22.0.1-3-g496fd5d+09117f784f,22.0.1-3-g59f966b+1e6ba2c031,22.0.1-3-g849a1b8+f8b568069f,22.0.1-3-gaaec9c0+c5c846a8b1,22.0.1-32-g5ddfab5d3+60ce4897b0,22.0.1-4-g037fbe1+64e601228d,22.0.1-4-g8623105+b8044ec9de,22.0.1-5-g096abc9+d18c45d440,22.0.1-5-g15c806e+57f5c03693,22.0.1-7-gba73697+57f5c03693,master-g6e05de7fdc+c1283a92b8,master-g72cdda8301+729191ecac,w.2021.39
LSST Data Management Base Package
Mask.h
Go to the documentation of this file.
1 // -*- lsst-c++ -*-
2 
3 /*
4  * LSST Data Management System
5  * Copyright 2008, 2009, 2010 LSST Corporation.
6  *
7  * This product includes software developed by the
8  * LSST Project (http://www.lsst.org/).
9  *
10  * This program is free software: you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License as published by
12  * the Free Software Foundation, either version 3 of the License, or
13  * (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the LSST License Statement and
21  * the GNU General Public License along with this program. If not,
22  * see <http://www.lsstcorp.org/LegalNotices/>.
23  */
24 
25 /*
26  * LSST bitmasks
27  */
28 
29 #ifndef LSST_AFW_IMAGE_MASK_H
30 #define LSST_AFW_IMAGE_MASK_H
31 
32 #include <list>
33 #include <map>
34 #include <string>
35 
36 #include <memory>
37 
38 #include "lsst/base.h"
40 #include "lsst/pex/exceptions.h"
43 #include "lsst/afw/fitsDefaults.h"
44 
45 namespace lsst {
46 namespace afw {
47 namespace image {
48 
49 namespace detail {
50 class MaskDict; // forward declaration
51 }
52 
53 // all masks will initially be instantiated with the same pixel type
54 namespace detail {
56 struct Mask_tag : public detail::basic_tag {};
57 
59 } // namespace detail
60 
76 template <typename MaskPixelT = lsst::afw::image::MaskPixel>
77 class Mask : public ImageBase<MaskPixelT> {
78 public:
80 
82 
84  template <typename MaskPT = MaskPixelT>
87  using type = Mask<MaskPT>;
88  };
89 
90  // Constructors
98  explicit Mask(unsigned int width, unsigned int height, MaskPlaneDict const& planeDefs = MaskPlaneDict());
107  explicit Mask(unsigned int width, unsigned int height, MaskPixelT initialValue,
108  MaskPlaneDict const& planeDefs = MaskPlaneDict());
116  MaskPlaneDict const& planeDefs = MaskPlaneDict());
124  explicit Mask(lsst::geom::Extent2I const& dimensions, MaskPixelT initialValue,
125  MaskPlaneDict const& planeDefs = MaskPlaneDict());
132  explicit Mask(lsst::geom::Box2I const& bbox, MaskPlaneDict const& planeDefs = MaskPlaneDict());
140  explicit Mask(lsst::geom::Box2I const& bbox, MaskPixelT initialValue,
141  MaskPlaneDict const& planeDefs = MaskPlaneDict());
142 
163  explicit Mask(std::string const& fileName, int hdu = fits::DEFAULT_HDU,
167  bool conformMasks = false, bool allowUnsafe = false);
168 
189  explicit Mask(fits::MemFileManager& manager, int hdu = fits::DEFAULT_HDU,
193  bool conformMasks = false, bool allowUnsafe = false);
194 
212  explicit Mask(fits::Fits& fitsfile,
216  bool conformMasks = false, bool allowUnsafe = false);
217 
218  // generalised copy constructor
219  template <typename OtherPixelT>
220  Mask(Mask<OtherPixelT> const& rhs, const bool deep)
221  : image::ImageBase<MaskPixelT>(rhs, deep), _maskDict(rhs._maskDict) {}
222 
229  Mask(const Mask& src, const bool deep = false);
230  Mask(Mask&& src);
231  ~Mask() override;
240  Mask(const Mask& src, const lsst::geom::Box2I& bbox, ImageOrigin const origin = PARENT,
241  const bool deep = false);
242 
243  explicit Mask(ndarray::Array<MaskPixelT, 2, 1> const& array, bool deep = false,
245 
246  void swap(Mask& rhs);
247  // Operators
248 
249  Mask& operator=(MaskPixelT const rhs);
250  Mask& operator=(const Mask& rhs);
251  Mask& operator=(Mask&& rhs);
252 
254  Mask& operator|=(Mask const& rhs);
256  Mask& operator|=(MaskPixelT const rhs);
257 
259  Mask& operator&=(Mask const& rhs);
261  Mask& operator&=(MaskPixelT const rhs);
262 
277  return Mask(*this, bbox, origin, false);
278  }
279 
281  Mask operator[](lsst::geom::Box2I const& bbox) const { return subset(bbox); }
282 
284 
290  static MaskPixelT getPlaneBitMask(const std::vector<std::string>& names);
291 
293  Mask& operator^=(Mask const& rhs);
295  Mask& operator^=(MaskPixelT const rhs);
296 
310  typename ImageBase<MaskPixelT>::PixelConstReference operator()(int x, int y) const;
318  bool operator()(int x, int y, int plane) const;
326  typename ImageBase<MaskPixelT>::PixelReference operator()(int x, int y, CheckIndices const& check);
335  CheckIndices const& check) const;
344  bool operator()(int x, int y, int plane, CheckIndices const& check) const;
345 
353  void writeFits(std::string const& fileName,
356  std::string const& mode = "w") const;
357 
368  std::string const& mode = "w") const;
369 
378 
387  void writeFits(std::string const& filename, fits::ImageWriteOptions const& options,
388  std::string const& mode = "w",
389  std::shared_ptr<daf::base::PropertySet const> header = nullptr) const;
390 
400  std::string const& mode = "w",
401  std::shared_ptr<daf::base::PropertySet const> header = nullptr) const;
402 
410  void writeFits(fits::Fits& fitsfile, fits::ImageWriteOptions const& options,
411  std::shared_ptr<daf::base::PropertySet const> header = nullptr) const;
412 
421  static Mask readFits(std::string const& filename, int hdu = fits::DEFAULT_HDU) {
422  return Mask<MaskPixelT>(filename, hdu);
423  }
424 
433  static Mask readFits(fits::MemFileManager& manager, int hdu = fits::DEFAULT_HDU) {
434  return Mask<MaskPixelT>(manager, hdu);
435  }
436 
438  static std::string interpret(MaskPixelT value);
439  std::string getAsString(int x, int y) { return interpret((*this)(x, y)); }
440 
441  // Mask Plane ops
442 
444  void clearAllMaskPlanes();
446  void clearMaskPlane(int plane);
450  void setMaskPlaneValues(const int plane, const int x0, const int x1, const int y);
458  //
459  // Operations on the mask plane dictionary
460  //
462  static void clearMaskPlaneDict();
463  static int addMaskPlane(const std::string& name);
464  static void removeMaskPlane(const std::string& name);
474  void removeAndClearMaskPlane(const std::string& name, bool const removeFromDefault = false);
475 
481  static int getMaskPlane(const std::string& name);
487  static MaskPixelT getPlaneBitMask(const std::string& name);
488 
489  static int getNumPlanesMax() { return 8 * sizeof(MaskPixelT); }
490  static int getNumPlanesUsed();
494  MaskPlaneDict const& getMaskPlaneDict() const;
496  void printMaskPlanes() const;
497 
502  //
503  // This one isn't static, it fixes up a given Mask's planes
518  void conformMaskPlanes(const MaskPlaneDict& masterPlaneDict);
519 
520 private:
521  friend class MaskFitsReader;
522 
523  std::shared_ptr<detail::MaskDict> _maskDict; // our bitplane dictionary
524 
525  static std::shared_ptr<detail::MaskDict> _maskPlaneDict();
526  static int _setMaskPlaneDict(MaskPlaneDict const& mpd);
527  static const std::string maskPlanePrefix;
528 
537  static int addMaskPlane(std::string name, int plane);
538 
540  static int getMaskPlaneNoThrow(const std::string& name);
542  static MaskPixelT getBitMaskNoThrow(int plane);
548  static MaskPixelT getBitMask(int plane);
549 
553  void _initializePlanes(MaskPlaneDict const& planeDefs); // called by ctors
554 
555  //
556  // Make names in templatized base class visible (Meyers, Effective C++, Item 43)
557  //
560 
566  void checkMaskDictionaries(Mask const& other);
567 };
568 
569 template <typename PixelT>
570 void swap(Mask<PixelT>& a, Mask<PixelT>& b);
571 } // namespace image
572 } // namespace afw
573 } // namespace lsst
574 
575 #endif // LSST_AFW_IMAGE_MASK_H
table::Key< std::string > name
Definition: Amplifier.cc:116
AmpInfoBoxKey bbox
Definition: Amplifier.cc:117
double x
afw::table::PointKey< int > dimensions
Definition: GaussianPsf.cc:48
std::shared_ptr< RecordT > src
Definition: Match.cc:48
int y
Definition: SpanSet.cc:48
table::Key< int > b
table::Key< int > a
Basic LSST definitions.
A simple struct that combines the two arguments that must be passed to most cfitsio routines and cont...
Definition: fits.h:297
Lifetime-management for memory that goes into FITS memory files.
Definition: fits.h:121
A class used to request that array accesses be checked.
Definition: ImageBase.h:74
The base class for all image classed (Image, Mask, MaskedImage, ...)
Definition: ImageBase.h:102
typename Reference< PixelT >::type PixelReference
A Reference to a PixelT.
Definition: ImageBase.h:117
typename ConstReference< PixelT >::type PixelConstReference
A ConstReference to a PixelT.
Definition: ImageBase.h:119
A FITS reader class for Masks.
Represent a 2-dimensional array of bitmask pixels.
Definition: Mask.h:77
void writeFits(std::string const &fileName, std::shared_ptr< lsst::daf::base::PropertySet const > metadata=std::shared_ptr< lsst::daf::base::PropertySet >(), std::string const &mode="w") const
Write a mask to a regular FITS file.
void swap(Mask &rhs)
Definition: Mask.cc:126
Mask & operator=(MaskPixelT const rhs)
Definition: Mask.cc:152
void printMaskPlanes() const
print the mask plane dictionary to std::cout
Definition: Mask.cc:638
static Mask readFits(std::string const &filename, int hdu=fits::DEFAULT_HDU)
Read a Mask from a regular FITS file.
Definition: Mask.h:421
static int getMaskPlane(const std::string &name)
Return the mask plane number corresponding to a plane name.
Definition: Mask.cc:350
detail::MaskPlaneDict MaskPlaneDict
Definition: Mask.h:79
static std::string interpret(MaskPixelT value)
Interpret a mask value as a comma-separated list of mask plane names.
Definition: Mask.cc:245
static void removeMaskPlane(const std::string &name)
Definition: Mask.cc:298
void writeFits(fits::MemFileManager &manager, std::shared_ptr< lsst::daf::base::PropertySet const > metadata=std::shared_ptr< lsst::daf::base::PropertySet >(), std::string const &mode="w") const
Write a mask to a FITS RAM file.
void removeAndClearMaskPlane(const std::string &name, bool const removeFromDefault=false)
Clear all pixels of the specified mask and remove the plane from the mask plane dictionary; optionall...
Definition: Mask.cc:309
Mask(Mask< OtherPixelT > const &rhs, const bool deep)
Definition: Mask.h:220
static Mask readFits(fits::MemFileManager &manager, int hdu=fits::DEFAULT_HDU)
Read a Mask from a FITS RAM file.
Definition: Mask.h:433
static void clearMaskPlaneDict()
Reset the maskPlane dictionary.
Definition: Mask.cc:386
void setMaskPlaneValues(const int plane, const int x0, const int x1, const int y)
Set the bit specified by "planeId" for pixels (x0, y) ...
Definition: Mask.cc:560
static void addMaskPlanesToMetadata(std::shared_ptr< lsst::daf::base::PropertySet >)
Given a PropertySet, replace any existing MaskPlane assignments with the current ones.
Definition: Mask.cc:569
Mask subset(lsst::geom::Box2I const &bbox, ImageOrigin origin=PARENT) const
Return a subimage corresponding to the given box.
Definition: Mask.h:276
ImageBase< MaskPixelT >::PixelReference operator()(int x, int y)
get a reference to the specified pixel
Definition: Mask.cc:457
Mask & operator^=(Mask const &rhs)
XOR a Mask into a Mask.
Definition: Mask.cc:546
static int addMaskPlane(const std::string &name)
Definition: Mask.cc:260
Mask(std::string const &fileName, int hdu=fits::DEFAULT_HDU, std::shared_ptr< lsst::daf::base::PropertySet > metadata=std::shared_ptr< lsst::daf::base::PropertySet >(), lsst::geom::Box2I const &bbox=lsst::geom::Box2I(), ImageOrigin origin=PARENT, bool conformMasks=false, bool allowUnsafe=false)
Construct a Mask by reading a regular FITS file.
void writeFits(fits::MemFileManager &manager, fits::ImageWriteOptions const &options, std::string const &mode="w", std::shared_ptr< daf::base::PropertySet const > header=nullptr) const
Write a mask to a FITS RAM file.
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:372
Mask(fits::MemFileManager &manager, int hdu=fits::DEFAULT_HDU, std::shared_ptr< lsst::daf::base::PropertySet > metadata=std::shared_ptr< lsst::daf::base::PropertySet >(), lsst::geom::Box2I const &bbox=lsst::geom::Box2I(), ImageOrigin origin=PARENT, bool conformMasks=false, bool allowUnsafe=false)
Construct a Mask by reading a FITS image in memory.
Mask(unsigned int width, unsigned int height, MaskPlaneDict const &planeDefs=MaskPlaneDict())
Construct a Mask initialized to 0x0.
Definition: Mask.cc:62
std::string getAsString(int x, int y)
Definition: Mask.h:439
Mask(fits::Fits &fitsfile, std::shared_ptr< lsst::daf::base::PropertySet > metadata=std::shared_ptr< lsst::daf::base::PropertySet >(), lsst::geom::Box2I const &bbox=lsst::geom::Box2I(), ImageOrigin origin=PARENT, bool conformMasks=false, bool allowUnsafe=false)
Construct a Mask from an already-open FITS object.
MaskPlaneDict const & getMaskPlaneDict() const
Return the Mask's maskPlaneDict.
Definition: Mask.cc:293
void conformMaskPlanes(const MaskPlaneDict &masterPlaneDict)
Adjust this mask to conform to the standard Mask class's mask plane dictionary, adding any new mask p...
Definition: Mask.cc:401
Mask & operator|=(Mask const &rhs)
OR a Mask into a Mask.
Definition: Mask.cc:505
Mask operator[](lsst::geom::Box2I const &bbox) const
Return a subimage corresponding to the given box (interpreted as PARENT coordinates).
Definition: Mask.h:281
void writeFits(std::string const &filename, fits::ImageWriteOptions const &options, std::string const &mode="w", std::shared_ptr< daf::base::PropertySet const > header=nullptr) const
Write a mask to a regular FITS file.
static MaskPlaneDict parseMaskPlaneMetadata(std::shared_ptr< lsst::daf::base::PropertySet const > metadata)
Given a PropertySet that contains the MaskPlane assignments, setup the MaskPlanes.
Definition: Mask.cc:597
void clearAllMaskPlanes()
Clear all the pixels.
Definition: Mask.cc:391
static int getNumPlanesUsed()
Definition: Mask.cc:381
void writeFits(fits::Fits &fitsfile, std::shared_ptr< lsst::daf::base::PropertySet const > metadata=std::shared_ptr< lsst::daf::base::PropertySet const >()) const
Write a mask to an open FITS file object.
void clearMaskPlane(int plane)
Clear the specified bit in all pixels.
Definition: Mask.cc:396
static int getNumPlanesMax()
Definition: Mask.h:489
void writeFits(fits::Fits &fitsfile, fits::ImageWriteOptions const &options, std::shared_ptr< daf::base::PropertySet const > header=nullptr) const
Write a mask to an open FITS file object.
Mask & operator&=(Mask const &rhs)
AND a Mask into a Mask.
Definition: Mask.cc:525
An integer coordinate rectangle.
Definition: Box.h:55
const int DEFAULT_HDU
Specify that the default HDU should be read.
Definition: fitsDefaults.h:18
std::map< std::string, int > MaskPlaneDict
Definition: Mask.h:58
Backwards-compatibility support for depersisting the old Calib (FluxMag0/FluxMag0Err) objects.
void swap(Image< PixelT > &a, Image< PixelT > &b)
Definition: Image.cc:444
A base class for image defects.
Options for writing an image to FITS.
Definition: fits.h:219
A templated class to return this classes' type (present in Image/Mask/MaskedImage)
Definition: Mask.h:85