From b8c8df1fb66923dbc889188e45ac535060f44e74 Mon Sep 17 00:00:00 2001
From: Florian <12199342+fwfichtner@users.noreply.github.com>
Date: Mon, 19 Sep 2022 17:01:22 +0200
Subject: [PATCH] Change schema Sentinel-2 (#168)

* change schema

* bump version
---
 CHANGELOG.rst          |  6 ++++++
 tests/test_raster.py   | 34 +++++++++++++++++++++++++++++++---
 ukis_pysat/__init__.py |  2 +-
 ukis_pysat/raster.py   | 26 +++++++++++++-------------
 4 files changed, 51 insertions(+), 17 deletions(-)

diff --git a/CHANGELOG.rst b/CHANGELOG.rst
index 0d6943f..11b3fea 100644
--- a/CHANGELOG.rst
+++ b/CHANGELOG.rst
@@ -1,6 +1,12 @@
 Changelog
 =========
 
+[1.4.2] (2022-09-19)
+---------------------
+Changed
+^^^^^^^
+- Adapt band schema of Sentinel-2 #168
+
 [1.4.1] (2022-09-19)
 ---------------------
 Changed
diff --git a/tests/test_raster.py b/tests/test_raster.py
index 3a9840f..23b9cd2 100644
--- a/tests/test_raster.py
+++ b/tests/test_raster.py
@@ -268,14 +268,42 @@ def test_dn2toa(self):
                 "dn_file": target_dir.joinpath("S2B_MSIL1C_20200406T101559_N0209_R065_T32UPC_20200406T130159.tif"),
                 "toa_file": target_dir.joinpath("S2B_MSIL1C_20200406T101559_N0209_R065_T32UPC_20200406T130159_toa.tif"),
                 "mtd_file": target_dir.joinpath("S2B_MSIL1C_20200406T101559_N0209_R065_T32UPC_20200406T130159_MTD.xml"),
-                "wavelengths": ["B1", "B2", "B3", "B4", "B5", "B6", "B7", "B8", "B8A", "B9", "B10", "B11", "B12"],
+                "wavelengths": [
+                    "Aerosol",
+                    "Blue",
+                    "Green",
+                    "Red",
+                    "Rededge1",
+                    "Rededge2",
+                    "Rededge3",
+                    "NIR",
+                    "Rededge4",
+                    "Watervapor",
+                    "Cirrus",
+                    "Swir1",
+                    "Swir2",
+                ],
             },
             {  # post product upgrades, see https://github.com/dlr-eoc/ukis-pysat/issues/165
                 "platform": Platform.Sentinel2,
                 "dn_file": target_dir.joinpath("S2B_MSIL1C_20220615T101559_N0400_R065_T32UPC_20220615T122549.tif"),
                 "toa_file": target_dir.joinpath("S2B_MSIL1C_20220615T101559_N0400_R065_T32UPC_20220615T122549_toa.tif"),
                 "mtd_file": target_dir.joinpath("S2B_MSIL1C_20220615T101559_N0400_R065_T32UPC_20220615T122549_MTD.xml"),
-                "wavelengths": ["B1", "B2", "B3", "B4", "B5", "B6", "B7", "B8", "B8A", "B9", "B10", "B11", "B12"],
+                "wavelengths": [
+                    "Aerosol",
+                    "Blue",
+                    "Green",
+                    "Red",
+                    "Rededge1",
+                    "Rededge2",
+                    "Rededge3",
+                    "NIR",
+                    "Rededge4",
+                    "Watervapor",
+                    "Cirrus",
+                    "Swir1",
+                    "Swir2",
+                ],
             },
         ]
 
@@ -329,7 +357,7 @@ def test__lookup_bands(self):
             ["8", "10", "11"],
             self.img._lookup_bands(Platform.Landsat8, ["PAN", "Tirs1", "Tirs2"]),
         )
-        self.assertEqual(["1", "2", "3"], self.img._lookup_bands(Platform.Sentinel2, ["B2", "B3", "B4"]))
+        self.assertEqual(["1", "2", "3"], self.img._lookup_bands(Platform.Sentinel2, ["Blue", "Green", "Red"]))
 
     def test_get_tiles(self):
         for idx, each in enumerate(self.img.get_tiles(5, 5, 1)):
diff --git a/ukis_pysat/__init__.py b/ukis_pysat/__init__.py
index 05a4070..c902d56 100644
--- a/ukis_pysat/__init__.py
+++ b/ukis_pysat/__init__.py
@@ -1 +1 @@
-__version__ = "1.4.1"
+__version__ = "1.4.2"
diff --git a/ukis_pysat/raster.py b/ukis_pysat/raster.py
index 0e93306..1bc6b51 100644
--- a/ukis_pysat/raster.py
+++ b/ukis_pysat/raster.py
@@ -407,19 +407,19 @@ def _lookup_bands(platform, wavelengths):
                 "tirs2": "11",
             },
             Platform.Sentinel2: {
-                "b1": "0",
-                "b2": "1",
-                "b3": "2",
-                "b4": "3",
-                "b5": "4",
-                "b6": "5",
-                "b7": "6",
-                "b8": "7",
-                "b8a": "8",
-                "b9": "9",
-                "b10": "10",
-                "b11": "11",
-                "b12": "12",
+                "aerosol": "0",
+                "blue": "1",
+                "green": "2",
+                "red": "3",
+                "rededge1": "4",
+                "rededge2": "5",
+                "rededge3": "6",
+                "nir": "7",
+                "rededge4": "8",
+                "watervapor": "9",
+                "cirrus": "10",
+                "swir1": "11",
+                "swir2": "12",
             },
         }