# -*- coding: utf-8 -*-
# Copyright 2022, SERTIT-ICube - France, https://sertit.unistra.fr/
# This file is part of eoreader project
# https://github.com/sertit/eoreader
#
# 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
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""
ICEYE products.
Take a look
`here <https://www.iceye.com/hubfs/Downloadables/ICEYE-Level-1-Product-Specs-2019.pdf>`_.
"""
import logging
import warnings
from datetime import datetime
from enum import unique
from typing import Union
import geopandas as gpd
import rasterio
from lxml import etree
from sertit import files, vectors
from sertit.misc import ListEnum
from eoreader import cache, cached_property
from eoreader.bands.bands import SarBandNames as sbn
from eoreader.exceptions import InvalidProductError, InvalidTypeError
from eoreader.products import SarProduct, SarProductType
from eoreader.utils import DATETIME_FMT, EOREADER_NAME
LOGGER = logging.getLogger(EOREADER_NAME)
# Disable georef warnings here as the SAR products are not georeferenced
warnings.filterwarnings("ignore", category=rasterio.errors.NotGeoreferencedWarning)
[docs]@unique
class IceyeProductType(ListEnum):
"""
ICEYE products types.
Take a look
`here <https://www.iceye.com/hubfs/Downloadables/ICEYE-Level-1-Product-Specs-2019.pdf>`_.
"""
GRD = "GRD"
""" Level-1 Ground Range Detected (GRD) """
SLC = "SLC"
""" Level-1 Single Look Complex (SLC) """
# Not released yet
# ORTHO = "ORTHO"
# """ ORTHO """
[docs]@unique
class IceyeSensorMode(ListEnum):
"""
ICEYE imaging mode.
Take a look
`here <https://www.iceye.com/hubfs/Downloadables/ICEYE-Level-1-Product-Specs-2019.pdf>`_.
"""
SM = "stripmap"
"""SM - Stripmap, StripmapWide, StripmapHI"""
SL = "spotlight"
"""SL - Spotlight"""
SC = "scan"
"""SC - ScanSAR"""
[docs]class IceyeProduct(SarProduct):
"""
Class for ICEYE Products
Take a look
`here <https://www.iceye.com/hubfs/Downloadables/ICEYE-Level-1-Product-Specs-2019.pdf>`_.
"""
def _set_resolution(self) -> float:
"""
Set product default resolution (in meters)
See here
<here](https://www.iceye.com/hubfs/Downloadables/ICEYE_SAR_Product_Guide_2021_V4.0.pdf>`_
for more information (table B.3).
"""
# ICEYE-SLC is not yet handled in EOReader
# For complex data, set regular ground range resolution provided by the constructor
# if self.product_type == IceyeProductType.SLC:
# if self.sensor_mode == IceyeSensorMode.SM:
# def_res = 2.5
# elif self.sensor_mode == IceyeSensorMode.SL:
# def_res = 0.5
# elif self.sensor_mode == IceyeSensorMode.SC:
# def_res = 6.0
# else:
# raise InvalidProductError(f"Unknown sensor mode: {self.sensor_mode}")
# Read metadata for default resolution
try:
root, nsmap = self.read_mtd()
# Some ICEYE product metadata has a namespace some don't
namespace = nsmap.get(None, "")
def_res = float(root.findtext(f".//{namespace}range_spacing"))
except (InvalidProductError, TypeError):
raise InvalidProductError("range_spacing not found in metadata!")
return def_res
def _pre_init(self) -> None:
"""
Function used to pre_init the products
(setting needs_extraction and so on)
"""
# Private attributes
# TODO: Allow to use SLC data ? Or SLC and GRD are always together ?
self._raw_band_regex = "*ICEYE*GRD*.tif"
self._band_folder = self.path
self._snap_path = str(next(self.path.glob("*ICEYE*GRD*.xml")).name)
# SNAP cannot process its archive
self.needs_extraction = True
# Post init done by the super class
super()._pre_init()
def _post_init(self) -> 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()
@cached_property
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"1011117-766193"
>>> prod = Reader().open(path)
>>> prod.wgs84_extent
geometry
0 POLYGON ((108.09797 15.61011, 108.48224 15.678...
Returns:
gpd.GeoDataFrame: WGS84 extent as a gpd.GeoDataFrame
"""
# Open extent KML file
try:
extent_file = next(self.path.glob("*ICEYE*QUICKLOOK*.kml"))
except IndexError as ex:
raise InvalidProductError(
f"Extent file (*.kml) not found in {self.path}"
) from ex
extent_wgs84 = vectors.read(extent_file).envelope
return gpd.GeoDataFrame(geometry=extent_wgs84.geometry, crs=extent_wgs84.crs)
def _set_product_type(self) -> None:
"""Set products type"""
# Get MTD XML file
root, nsmap = self.read_mtd()
# Some ICEYE product metadata has a namespace some don't
namespace = nsmap.get(None, "")
# Open identifier
prod_type = root.findtext(f".//{namespace}product_level")
if not prod_type:
raise InvalidProductError("mode not found in metadata!")
self.product_type = IceyeProductType.from_value(prod_type)
if self.product_type == IceyeProductType.GRD:
self.sar_prod_type = SarProductType.GDRG
elif self.product_type == IceyeProductType.SLC:
self.sar_prod_type = SarProductType.CPLX
else:
raise NotImplementedError(
f"{self.product_type.value} product type is not available for {self.name}"
)
if self.product_type != IceyeProductType.GRD:
LOGGER.warning(
"Other products type than MGD has not been tested for %s data. "
"Use it at your own risks !",
self.platform.value,
)
def _set_sensor_mode(self) -> None:
"""
Get products type from S2 products name (could check the metadata too)
"""
# Get MTD XML file
root, nsmap = self.read_mtd()
# Some ICEYE product metadata has a namespace some don't
namespace = nsmap.get(None, "")
# Open identifier
imaging_mode = root.findtext(f".//{namespace}acquisition_mode")
if not imaging_mode:
raise InvalidProductError("imagingMode not found in metadata!")
# Get sensor mode
try:
self.sensor_mode = IceyeSensorMode.from_value(imaging_mode)
except ValueError as ex:
raise InvalidTypeError(f"Invalid sensor mode for {self.name}") from ex
[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"1011117-766193"
>>> prod = Reader().open(path)
>>> prod.get_datetime(as_datetime=True)
datetime.datetime(2020, 10, 28, 22, 46, 25)
>>> prod.get_datetime(as_datetime=False)
'20201028T224625'
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, nsmap = self.read_mtd()
# Some ICEYE product metadata has a namespace some don't
namespace = nsmap.get(None, "")
# Open identifier
acq_date = root.findtext(f".//{namespace}acquisition_start_utc")
if not acq_date:
raise InvalidProductError(
"acquisition_start_utc 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(self) -> str:
"""
Set product real name from metadata
Returns:
str: True name of the product (from metadata)
"""
# Get MTD XML file
root, nsmap = self.read_mtd()
# Some ICEYE product metadata has a namespace some don't
namespace = nsmap.get(None, "")
# Open identifier
name = root.findtext(f".//{namespace}product_name")
if not name:
raise InvalidProductError("product_name not found in metadata!")
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"1001513-735093"
>>> prod = Reader().open(path)
>>> prod.read_mtd()
(<Element DeliveryNote at 0x2454ad4ee88>, {})
Returns:
(etree._Element, dict): Metadata XML root and its namespaces
"""
mtd_from_path = "ICEYE*GRD*.xml"
return self._read_mtd_xml(mtd_from_path)
def _get_raw_band_paths(self) -> dict:
"""
Return the existing path of the VV band (as they come with the archived products).
ICEYE product only contains a VV band !
Returns:
dict: Dictionary containing the path of every band existing in the raw products
"""
band_paths = {}
try:
band_paths[sbn.VV] = files.get_file_in_dir(
self._band_folder, self._raw_band_regex, exact_name=True, get_list=False
)
except FileNotFoundError:
raise InvalidProductError(
"An ICEYE product should at least contain a VV band !"
)
return band_paths
