Source code for eoreader.products.sar.s1_product

# -*- coding: utf-8 -*-
# Copyright 2022, SERTIT-ICube - France,
# This file is part of eoreader project
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# See the License for the specific language governing permissions and
# limitations under the License.
""" Sentinel-1 products """
import logging
import os
import re
import tempfile
import zipfile
from datetime import datetime
from enum import unique
from typing import Union

import geopandas as gpd
from lxml import etree
from sertit import files, vectors
from sertit.misc import ListEnum

from eoreader import cache
from eoreader.exceptions import InvalidProductError
from eoreader.products import SarProduct, SarProductType
from eoreader.products.product import OrbitDirection
from eoreader.utils import DATETIME_FMT, EOREADER_NAME

LOGGER = logging.getLogger(EOREADER_NAME)

[docs]@unique class S1ProductType(ListEnum): """ S1 products types. Take a look here: """ RAW = "RAW" """Raw products (lvl 0): **not used by EOReader**""" SLC = "SLC" """Single Look Complex (SLC, lvl 1)""" GRD = "GRD" """Ground Range Detected (GRD, lvl 1, phase lost)""" OCN = "OCN" """Ocean products (lvl 2): **not used by EOReader**"""
[docs]@unique class S1SensorMode(ListEnum): """ S1 sensor mode. Take a look here: The primary conflict-free modes are IW, with VV+VH polarisation over land, and WV, with VV polarisation, over open ocean. EW mode is primarily used for wide area coastal monitoring including ship traffic, oil spill and sea-ice monitoring. SM mode is only used for small islands and on request for extraordinary events such as emergency management. """ SM = "SM" """Stripmap (SM)""" IW = "IW" """Interferometric Wide swath (IW)""" EW = "EW" """Extra-Wide swath (EW)""" WV = "WV" """Wave (WV) -> single polarisation only (HH or VV)"""
[docs]class S1Product(SarProduct): """ Class for Sentinel-1 Products You can use directly the .zip file """ def _get_resolution(self) -> float: """ Get product default resolution (in meters) See here <here](>`_ for more information """ if self.sensor_mode == S1SensorMode.SM: def_res = 9.0 # Full Resolution GRD elif self.sensor_mode == S1SensorMode.IW: def_res = 20.0 # High Resolution GRD elif self.sensor_mode == S1SensorMode.EW: def_res = 50.0 # High Resolution GRD elif self.sensor_mode == S1SensorMode.WV: def_res = 52.0 # Medium Resolution GRD else: raise InvalidProductError(f"Unknown sensor mode: {self.sensor_mode}") return def_res def _set_instrument(self) -> None: """ Set instrument Sentinel-1: """ self.instrument = "SAR C-band" def _pre_init(self, **kwargs) -> None: """ Function used to pre_init the products (setting needs_extraction and so on) """ # Private attributes self._raw_band_regex = "*-{!l}-*.tiff" # Just get the SLC-iw1 image for now self._band_folder = self.path.joinpath("measurement") self.snap_filename = "" # Its original filename is its name self._use_filename = True # Zipped and SNAP can process its archive self.needs_extraction = False # Post init done by the super class super()._pre_init(**kwargs) def _post_init(self, **kwargs) -> None: """ Function used to post_init the products (setting product-type, band names and so on) """ # Post init done by the super class super()._post_init(**kwargs)
[docs] @cache def wgs84_extent(self) -> gpd.GeoDataFrame: """ Get the WGS84 extent of the file before any reprojection. This is useful when the SAR pre-process has not been done yet. .. code-block:: python >>> from eoreader.reader import Reader >>> path = r"" >>> prod = Reader().open(path) >>> prod.wgs84_extent() Name ... geometry 0 Sentinel-1 Image Overlay ... POLYGON ((0.85336 42.24660, -2.32032 42.65493,... [1 rows x 12 columns] Returns: gpd.GeoDataFrame: WGS84 extent as a gpd.GeoDataFrame """ tmp_dir = tempfile.TemporaryDirectory() try: # Open the map-overlay file if self.is_archived: # We need to extract the file here as we need a proper file with zipfile.ZipFile(self.path, "r") as zip_ds: filenames = [f.filename for f in zip_ds.filelist] regex = re.compile(".*preview.*map-overlay.kml") preview_overlay = zip_ds.extract( list(filter(regex.match, filenames))[0], ) else: preview_overlay = self.path.joinpath("preview", "map-overlay.kml") if os.path.isfile(preview_overlay): # Open the KML file extent_wgs84 = if extent_wgs84.empty: raise InvalidProductError( f"Cannot determine the WGS84 extent of {}" ) else: raise InvalidProductError( f"Impossible to find the map-overlay.kml in {self.path}" ) except Exception as ex: raise InvalidProductError(ex) from ex finally: tmp_dir.cleanup() return extent_wgs84
def _set_product_type(self) -> None: """Set products type""" # Get MTD XML file root, _ = self.read_mtd() # Open identifier prod_type = root.findtext(".//productType") if not prod_type: raise InvalidProductError("mode not found in metadata!") self.product_type = S1ProductType.from_value(prod_type) if self.product_type == S1ProductType.GRD: self.sar_prod_type = SarProductType.GDRG elif self.product_type == S1ProductType.SLC: self.sar_prod_type = SarProductType.CPLX else: raise NotImplementedError( f"{self.product_type.value} product type is not available for {}" ) def _set_sensor_mode(self) -> None: """ Get products type from S1 products name (could check the metadata too) """ # Get MTD XML file root, _ = self.read_mtd() # Open identifier mode = root.findtext(".//mode") if not mode: raise InvalidProductError("mode not found in metadata!") # Get sensor mode self.sensor_mode = S1SensorMode.from_value(mode) # Discard invalid sensor mode if self.sensor_mode != S1SensorMode.IW: raise NotImplementedError( f"For now, only IW sensor mode is used in EOReader processes: {}" ) if not self.sensor_mode: raise InvalidProductError( f"Invalid {self.constellation.value} name: {}" )
[docs] def get_datetime(self, as_datetime: bool = False) -> Union[str, datetime]: """ Get the product's acquisition datetime, with format :code:`YYYYMMDDTHHMMSS` <-> :code:`%Y%m%dT%H%M%S` .. code-block:: python >>> from eoreader.reader import Reader >>> path = r"" >>> prod = Reader().open(path) >>> prod.get_datetime(as_datetime=True) datetime.datetime(2019, 12, 15, 6, 9, 6) >>> prod.get_datetime(as_datetime=False) '20191215T060906' Args: as_datetime (bool): Return the date as a datetime.datetime. If false, returns a string. Returns: Union[str, datetime.datetime]: Its acquisition datetime """ if self.datetime is None: # Get MTD XML file root, _ = self.read_mtd() # Open identifier acq_date = root.findtext(".//startTime") if not acq_date: raise InvalidProductError("startTime not found in metadata!") # Convert to datetime date = datetime.strptime(acq_date, "%Y-%m-%dT%H:%M:%S.%f") else: date = self.datetime if not as_datetime: date = date.strftime(DATETIME_FMT) return date
def _get_name_constellation_specific(self) -> str: """ Set product real name from metadata Returns: str: True name of the product (from metadata) """ try: if self.is_archived: pdf_file = files.get_archived_path(self.path, r".*\.pdf", as_list=False) else: pdf_file = next(self.path.glob("*.pdf")) except (FileNotFoundError, StopIteration): # The name is not in the classic metadata, but can be found in the product-preview try: mtd_from_path = "preview/product-preview.html" mtd_archived = r"preview.*product-preview\.html" root = self._read_mtd_html(mtd_from_path, mtd_archived) # Open identifier name = root.findtext(".//head/title") if not name: raise InvalidProductError("title not found in metadata!") LOGGER.warning( "Product filename is not a valid Sentinel-1 name, and the retrieved name is missing the unique ID." ) except InvalidProductError: raise InvalidProductError( "product-preview.html not found in the product, the name will be the filename (which is not a valid Sentinel-1 name)" ) else: name = files.get_filename(pdf_file) return name @cache def _read_mtd(self) -> (etree._Element, dict): """ Read metadata and outputs the metadata XML root and its namespaces as a dict .. code-block:: python >>> from eoreader.reader import Reader >>> path = r"" >>> prod = Reader().open(path) >>> prod.read_mtd() (<Element product at 0x1832895d788>, {}) Returns: (etree._Element, dict): Metadata XML root and its namespaces """ mtd_from_path = "annotation/*.xml" # When archived, other XML (in calibration folder) can be found mtd_archived = r"annotation/(?!rfi)(?!cal)(?!noise).*\.xml" return self._read_mtd_xml(mtd_from_path, mtd_archived)
[docs] def get_quicklook_path(self) -> str: """ Get quicklook path if existing. Returns: str: Quicklook path """ quicklook_path = None try: if self.is_archived: quicklook_path = files.get_archived_rio_path( self.path, file_regex=r".*preview.quick-look\.png" ) else: quicklook_path = next(self.path.glob("preview/quick-look.png")) except (StopIteration, FileNotFoundError): LOGGER.warning(f"No quicklook found in {self.condensed_name}") return quicklook_path
[docs] @cache def get_orbit_direction(self) -> OrbitDirection: """ Get cloud cover as given in the metadata .. code-block:: python >>> from eoreader.reader import Reader >>> path = r"" >>> prod = Reader().open(path) >>> prod.get_orbit_direction().value "DESCENDING" Returns: OrbitDirection: Orbit direction (ASCENDING/DESCENDING) """ # Get MTD XML file root, _ = self.read_mtd() # Get the orbit direction try: od = OrbitDirection.from_value(root.findtext(".//pass").upper()) except TypeError: raise InvalidProductError("pass not found in metadata!") return od