add mask channel to consideration for number of channels

EmilDohne · Dec 18, 2024 · eea9d14 · eea9d14
1 parent 8032cf7
commit eea9d14
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/python/src/DeclareImageLayer.h b/python/src/DeclareImageLayer.h
@@ -664,7 +664,7 @@ void declareImageLayer(py::module& m, const std::string& extension) {
 
     imageLayer.def_property_readonly("num_channels", [](Class& self)
         {
-            return self.m_ImageData.size();
+            return self.m_ImageData.size() + self.m_LayerMask.has_value();
 		});
 
     imageLayer.def_property_readonly("channels", [](Class& self)
@@ -674,6 +674,10 @@ void declareImageLayer(py::module& m, const std::string& extension) {
             {
                 indices.push_back(key.index);
             }
+            if (self.m_LayerMask)
+            {
+                indices.push_back(-2);
+            }
             return indices;
         });
 }