cars 1.0.0rc2__cp312-cp312-win_amd64.whl

cars/core/datasets.py

@@ -0,0 +1,140 @@
+#!/usr/bin/env python
+# coding: utf8
+#
+# Copyright (c) 2020 Centre National d'Etudes Spatiales (CNES).
+#
+# This file is part of CARS
+# (see https://github.com/CNES/cars).
+#
+# 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.
+#
+"""
+CARS Datasets module
+"""
+
+# Standard imports
+from typing import List
+
+# Third party imports
+import numpy as np
+import rasterio as rio
+import xarray as xr
+
+# CARS imports
+from cars.core import constants as cst
+from cars.core import inputs
+
+# TODO: refacto constants: define constants here as only concerning datasets
+
+
+def create_im_dataset(  # pylint: disable=too-many-positional-arguments
+    img: np.ndarray,
+    region: List[int],
+    largest_size: List[int],
+    img_path: str = None,
+    band_coords: str = None,
+    descriptions: list = None,
+    data_types: list = None,
+    msk: np.ndarray = None,
+) -> xr.Dataset:
+    """
+    Create image dataset as used in cars.
+
+    :param img: image as a numpy array
+    :param region: region as list [xmin ymin xmax ymax]
+    :param largest_size: whole image size
+    :param img_path: path to image
+    :param band_type: set to band coord names (cst.BAND_IM or BAND_CLASSIF)
+     to add band description in the dataset
+    :param msk: image mask as a numpy array (default None)
+    :return: The image dataset as used in cars
+    """
+    # Get georef and transform
+    img_crs = None
+    img_transform = None
+    if img_path is not None:
+        with rio.open(img_path) as img_srs:
+            img_crs = img_srs.profile["crs"]
+            img_transform = inputs.rasterio_get_transform(
+                img_path, convention="north"
+            )
+
+    if img_crs is None:
+        img_crs = "None"
+    if img_transform is None:
+        img_transform = "None"
+
+    # Add band dimension if needed
+    if band_coords:
+        # Reorder dimensions in color dataset in order that the first dimension
+        # is band.
+
+        dataset = xr.Dataset(
+            {
+                cst.EPI_IMAGE: (
+                    [band_coords, cst.ROW, cst.COL],
+                    img,
+                )
+            },
+            coords={
+                band_coords: descriptions,
+                cst.ROW: np.array(range(region[1], region[3])),
+                cst.COL: np.array(range(region[0], region[2])),
+            },
+        )
+        if msk is not None:
+            dataset[cst.EPI_MSK] = xr.DataArray(
+                msk.astype(np.int16), dims=[band_coords, cst.ROW, cst.COL]
+            )
+    else:
+        dataset = xr.Dataset(
+            {cst.EPI_IMAGE: ([cst.ROW, cst.COL], img[0, ...])},
+            coords={
+                cst.ROW: np.array(range(region[1], region[3])),
+                cst.COL: np.array(range(region[0], region[2])),
+            },
+        )
+        if msk is not None:
+            dataset[cst.EPI_MSK] = xr.DataArray(
+                msk[0, ...].astype(np.int16), dims=[cst.ROW, cst.COL]
+            )
+
+    dataset.attrs[cst.EPI_VALID_PIXELS] = 0
+    dataset.attrs[cst.EPI_NO_DATA_MASK] = 255
+    dataset.attrs[cst.EPI_FULL_SIZE] = largest_size
+    dataset.attrs[cst.EPI_CRS] = img_crs
+    dataset.attrs[cst.EPI_TRANSFORM] = img_transform
+    dataset.attrs["region"] = np.array(region)
+    if descriptions is not None:
+        dataset.attrs[cst.BAND_NAMES] = descriptions
+    if data_types is not None:
+        dataset.attrs[cst.IMAGE_TYPE] = data_types
+    return dataset
+
+
+def get_color_bands(dataset):
+    """
+    Get band names list from the cardataset color
+
+    :param dataset: carsdataset with the color data
+    :type dataset: CarsDataset
+    :param key: dataset color data key
+    :param key: string
+
+    :return: list of color band names
+    """
+    band_im = None
+    if cst.BAND_NAMES in dataset.attrs.keys():
+        band_im = dataset.attrs[cst.BAND_NAMES]
+
+    return band_im

cars/core/geometry/__init__.py

@@ -0,0 +1,27 @@
+# pylint: disable=missing-module-docstring
+# flake8: noqa
+#!/usr/bin/env python
+# coding: utf8
+#
+# Copyright (c) 2020 Centre National d'Etudes Spatiales (CNES).
+#
+# This file is part of CARS
+# (see https://github.com/CNES/cars).
+#
+# 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.
+#
+"""
+Init file of geometry module
+"""
+
+from . import abstract_geometry, shareloc_geometry