Merge pull request #32 from FrontierDevelopmentLab/feature/bqa-masks

Add BQA mask

Author: frandorr

src/satextractor/extractor/extractor.py

@@ -115,13 +115,18 @@ def download_and_extract_tiles_window(
             with rasterio.open(f) as ds:
                 window = get_window_union(task.tiles, ds)
 
+                if task.band == "BQA":
+                    resampling = Resampling.nearest
+                else:
+                    resampling = Resampling.bilinear
+
                 rst_arr = ds.read(
                     1,
                     window=window,
                     out_shape=out_shp,
                     fill_value=0,
                     boundless=True,
-                    resampling=Resampling.bilinear,
+                    resampling=resampling,
                 )
 
         out_f = f"{task.task_id}_{ii}.tif"

src/satextractor/models/constellation_info.py

@@ -115,6 +115,13 @@
         ),
         "gsd": 100.0,
     },
+    "BQA": {
+        "band": Band.create(
+            name="BQA",
+            common_name="bqa",
+        ),
+        "gsd": 30.0,
+    },
 }
 
 LANDSAT7_BAND_INFO = {
@@ -199,6 +206,13 @@
         ),
         "gsd": 15.0,
     },
+    "BQA": {
+        "band": Band.create(
+            name="BQA",
+            common_name="bqa",
+        ),
+        "gsd": 30.0,
+    },
 }
 
 LANDSAT5_BAND_INFO = {
@@ -265,6 +279,13 @@
         ),
         "gsd": 30.0,
     },
+    "BQA": {
+        "band": Band.create(
+            name="BQA",
+            common_name="bqa",
+        ),
+        "gsd": 60.0,
+    },
 }