Catalog: Add data_type and ranges to processing levels (#10777)

GitOrigin-RevId: 283361d3d81fc081b2eacdb3a219f3f7f1d311f3

Author: stephencpope

descarteslabs/_dl_modules/catalog/band.py

@@ -147,11 +147,41 @@ class ProcessingStepAttribute(MappingAttribute):
         the coefficients for the processing function. Required.
     index : int
         Optional index into the named parameter (an array) for the band.
+    data_type : str or DataType
+        Optional data type for pixel values in this band.
+    data_range : tuple(float, float)
+        Optional normal range of pixel values stored in the band.
+    display_range : tuple(float, float)
+        Optional normal range of pixel values stored in the band for display purposes.
+    physical_range : tuple(float, float)
+        Optional normal range of physical values stored in the band.
+    physical_range_units : str
+        Optional unit of the physical range.
     """
 
     function = TypedAttribute(str)
     parameter = TypedAttribute(str)
     index = TypedAttribute(int)
+    data_type = EnumAttribute(DataType)
+    data_range = TupleAttribute(
+        min_length=2,
+        max_length=2,
+        coerce=True,
+        attribute_type=float,
+    )
+    display_range = TupleAttribute(
+        min_length=2,
+        max_length=2,
+        coerce=True,
+        attribute_type=float,
+    )
+    physical_range = TupleAttribute(
+        min_length=2,
+        max_length=2,
+        coerce=True,
+        attribute_type=float,
+    )
+    physical_range_unit = Attribute()
 
 
 class ProcessingLevelsAttribute(ModelAttribute, MutableMapping):

descarteslabs/_dl_modules/catalog/readme.rst

@@ -96,6 +96,8 @@ Guide <catalog_v2_guide>`.
   ~descarteslabs.catalog.DataType
   ~descarteslabs.catalog.BandType
   ~descarteslabs.catalog.Colormap
+  ~descarteslabs.catalog.ProcessingStepAttribute
+  ~descarteslabs.catalog.ProcessingLevelsAttribute
   ~descarteslabs.catalog.Resolution
   ~descarteslabs.catalog.ResolutionUnit
   ~descarteslabs.catalog.File

descarteslabs/_dl_modules/catalog/tests/test_band.py

@@ -15,6 +15,7 @@
     MicrowaveBand,
     ProcessingStepAttribute,
     DerivedParamsAttribute,
+    DataType,
 )
 from ..product import Product
 
@@ -286,6 +287,43 @@ def test_processing_levels_create(self):
             },
         )
 
+        b = SpectralBand(
+            id=band_id,
+            processing_levels={
+                "default": "toa_reflectance",
+                "toa_reflectance": [
+                    {
+                        "function": "fun",
+                        "parameter": "param",
+                        "index": 0,
+                        "data_type": "Float64",
+                        "data_range": [0, 1],
+                        "display_range": [0, 0.4],
+                        "physical_range": [0, 1],
+                        "physical_range_unit": "reflectance",
+                    }
+                ],
+            },
+        )
+        self.assertEqual(
+            b.processing_levels,
+            {
+                "default": "toa_reflectance",
+                "toa_reflectance": [
+                    ProcessingStepAttribute(
+                        function="fun",
+                        parameter="param",
+                        index=0,
+                        data_type=DataType("Float64"),
+                        data_range=(0.0, 1.0),
+                        display_range=(0, 0.4),
+                        physical_range=(0.0, 1.0),
+                        physical_range_unit="reflectance",
+                    )
+                ],
+            },
+        )
+
         with pytest.raises(AttributeValidationError):
             SpectralBand(id=band_id, processing_levels={"default": 1})