Add apCorrMap interface to Exposure

Fowards everything to ExposureInfo, but we have such an interface for several other such components.
lsst · Jul 25, 2023 · 8b7d701 · 8b7d701
1 parent baf8900
commit 8b7d701
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/include/lsst/afw/image/Exposure.h b/include/lsst/afw/image/Exposure.h
@@ -231,6 +231,8 @@ class Exposure {
 
     std::shared_ptr<geom::SkyWcs const> getWcs() const { return _info->getWcs(); }
 
+    std::shared_ptr<ApCorrMap const> getApCorrMap() const { return _info->getApCorrMap(); }
+
     /// Return the Exposure's Detector information
     std::shared_ptr<lsst::afw::cameraGeom::Detector const> getDetector() const {
         return _info->getDetector();
@@ -313,6 +315,9 @@ class Exposure {
     /// Does this Exposure have a Wcs?
     bool hasWcs() const { return _info->hasWcs(); }
 
+    /// Set the Exposure's aperture correction map.
+    void setApCorrMap(std::shared_ptr<ApCorrMap const> apCorrMap) { _info->setApCorrMap(apCorrMap); }
+
     /// Get the ExposureInfo that aggregates all the non-image components.  Never null.
     std::shared_ptr<ExposureInfo> getInfo() { return _info; }
 

diff --git a/python/lsst/afw/image/_exposure/_exposure.cc b/python/lsst/afw/image/_exposure/_exposure.cc
@@ -30,6 +30,7 @@
 #include "lsst/afw/image/FilterLabel.h"
 #include "lsst/afw/image/Exposure.h"
 #include "lsst/afw/detection/Psf.h"
+#include "lsst/afw/image/ApCorrMap.h"
 
 namespace py = pybind11;
 using namespace pybind11::literals;
@@ -135,6 +136,9 @@ PyExposure<PixelT> declareExposure(lsst::utils::python::WrapperCollection &wrapp
                 cls.def_property_readonly("visitInfo",
                                           [](ExposureT &self) { return self.getInfo()->getVisitInfo(); });
 
+                cls.def("setApCorrMap", &ExposureT::setApCorrMap, "apCorrMap"_a);
+                cls.def_property_readonly("apCorrMap", [](ExposureT &self) { return self.getApCorrMap(); });
+
                 cls.def("subset", &ExposureT::subset, "bbox"_a, "origin"_a = PARENT);
 
                 cls.def("writeFits", (void(ExposureT::*)(std::string const &) const) & ExposureT::writeFits);