Support setting null ValidPolygon on ExposureRecord

This works around DM-10289.
lsst · May 16, 2017 · 1849a83 · 1849a83
1 parent 3deb52d
commit 1849a83
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/python/lsst/afw/table/exposure/exposure.cc b/python/lsst/afw/table/exposure/exposure.cc
@@ -80,7 +80,18 @@ PyExposureRecord declareExposureRecord(py::module &mod) {
     cls.def("getApCorrMap", &ExposureRecord::getApCorrMap);
     cls.def("setApCorrMap", &ExposureRecord::setApCorrMap, "appCorrMap"_a);
     cls.def("getValidPolygon", &ExposureRecord::getValidPolygon);
-    cls.def("setValidPolygon", &ExposureRecord::setValidPolygon, "polygon"_a);
+
+    // Workaround for DM-10289.
+    cls.def("setValidPolygon",
+            [](ExposureRecord &self, py::object polygon) {
+                if (polygon == py::none()) {
+                    self.setValidPolygon(nullptr);
+                } else {
+                    self.setValidPolygon(py::cast<std::shared_ptr<afw::geom::polygon::Polygon>>(polygon));
+                }
+            },
+            "polygon"_a);
+
     cls.def("getVisitInfo", &ExposureRecord::getVisitInfo);
     cls.def("setVisitInfo", &ExposureRecord::setVisitInfo, "visitInfo"_a);
     return cls;