From 8cffdd0c6d2b2619902825178b06aefaf46d8664 Mon Sep 17 00:00:00 2001
From: Francisco Dorr <fran.dorr@gmail.com>
Date: Thu, 9 Dec 2021 20:09:07 +0000
Subject: [PATCH] Add BQA mask

---
 src/satextractor/extractor/extractor.py       |  7 ++++++-
 src/satextractor/models/constellation_info.py | 21 +++++++++++++++++++
 2 files changed, 27 insertions(+), 1 deletion(-)

diff --git a/src/satextractor/extractor/extractor.py b/src/satextractor/extractor/extractor.py
index 4cdeab5..5520e32 100644
--- a/src/satextractor/extractor/extractor.py
+++ b/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"
diff --git a/src/satextractor/models/constellation_info.py b/src/satextractor/models/constellation_info.py
index a89b923..cd1a399 100644
--- a/src/satextractor/models/constellation_info.py
+++ b/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,
+    },
 }