sen2p 0.0.1__py3-none-any.whl

sen2p/__init__.py

@@ -0,0 +1 @@
+from .download import download

sen2p/download.py

@@ -0,0 +1,157 @@
+import os
+from typing import List, Optional, Dict, Any, Union
+from tqdm import tqdm
+import xarray as xr
+import rioxarray as rxr
+import geopandas as gpd
+from shapely.geometry import mapping, Point
+from pystac_client import Client
+import planetary_computer as pc
+from rasterio.enums import Resampling
+
+def download(
+    start_date: str,
+    end_date: str,
+    location: Union[List[float], Dict[str, Any], str],
+    bands: Optional[List[str]] = None,
+    output_dir: Optional[str] = None,
+    show_progress: bool = True,
+    merge_bands: bool = False,
+    merged_filename: Optional[str] = None,
+    overwrite: bool = False,
+    cell_size: Optional[float] = None,
+    max_items: int = 10,
+    collection: str = "sentinel-2-l2a"
+) -> List[Dict[str, Any]]:
+    '''
+    Download Sentinel-2 data from Planetary Computer using search parameters.
+
+    Args:
+        start_date (str): Start date (e.g., "2023-06-01").
+        end_date (str): End date (e.g., "2023-06-30").
+        location (list, dict, or str): [lon, lat], GeoJSON geometry, or path to shapefile.
+        bands (list, optional): List of bands to download (e.g., ["B02", "B03"]).
+        output_dir (str, optional): Where to save output files.
+        show_progress (bool): Show download progress.
+        merge_bands (bool): Merge into single file.
+        merged_filename (str, optional): Filename for merged file.
+        overwrite (bool): Overwrite existing files.
+        cell_size (float, optional): Target resolution.
+        max_items (int): How many items to fetch.
+        collection (str): STAC collection name.
+
+    Returns:
+        List of result dicts, one per STAC item.
+    '''
+
+    catalog = Client.open("https://planetarycomputer.microsoft.com/api/stac/v1", modifier=pc.sign_inplace)
+
+    if isinstance(location, list):
+        geometry = mapping(Point(location[0], location[1]))
+    elif isinstance(location, dict):
+        geometry = location
+    elif isinstance(location, str) and location.endswith(".shp"):
+        gdf = gpd.read_file(location)
+        geometry = mapping(gdf.unary_union)
+    else:
+        raise ValueError("Location must be [lon, lat], GeoJSON geometry, or shapefile path")
+
+    search = catalog.search(
+        collections=[collection],
+        intersects=geometry,
+        datetime=f"{start_date}/{end_date}",
+        max_items=max_items,
+    )
+
+    items = list(search.get_items())
+    if not items:
+        raise ValueError("No Sentinel-2 items found for the given location and time range.")
+
+    all_results = []
+
+    for item in items:
+        item_id = item.id
+        available_assets = list(item.assets.keys())
+
+        if bands is None:
+            bands_to_download = [asset for asset in available_assets if asset.startswith("B")]
+        else:
+            missing_bands = [band for band in bands if band not in available_assets]
+            if missing_bands:
+                print(f"Skipping {item_id}: Missing bands {missing_bands}")
+                continue
+            bands_to_download = bands
+
+        item_output_dir = os.path.join(output_dir, item_id) if output_dir else None
+        if item_output_dir and not os.path.exists(item_output_dir):
+            os.makedirs(item_output_dir)
+
+        result = {}
+        band_data_arrays = []
+        resampled_arrays = []
+        band_names = []
+        progress_iter = tqdm(bands_to_download, desc=f"{item_id}") if show_progress else bands_to_download
+
+        for band in progress_iter:
+            band_url = item.assets[band].href
+            file_path = os.path.join(item_output_dir, f"{item.id}_{band}.tif") if item_output_dir else None
+
+            if file_path and os.path.exists(file_path) and not overwrite:
+                if show_progress:
+                    progress_iter.write(f"{file_path} exists, skipping.")
+                if merge_bands:
+                    band_data = rxr.open_rasterio(file_path)
+                    band_data_arrays.append((band, band_data))
+                    band_names.append(band)
+                result[band] = file_path
+                continue
+
+            if show_progress:
+                progress_iter.set_description(f"{item_id}: {band}")
+
+            band_data = rxr.open_rasterio(band_url)
+            if merge_bands:
+                band_data_arrays.append((band, band_data))
+                band_names.append(band)
+
+            if item_output_dir:
+                band_data.rio.to_raster(file_path)
+                result[band] = file_path
+            else:
+                result[band] = band_data
+
+        if merge_bands and item_output_dir:
+            merged_path = os.path.join(item_output_dir, merged_filename or f"{item.id}_merged.tif")
+            if os.path.exists(merged_path) and not overwrite:
+                if show_progress:
+                    print(f"Merged file {merged_path} exists, skipping.")
+                result["merged"] = merged_path
+            else:
+                if cell_size is None and band_data_arrays:
+                    cell_size = abs(band_data_arrays[0][1].rio.transform()[0])
+                elif cell_size is None:
+                    cell_size = 10
+
+                for band_name, data_array in band_data_arrays:
+                    current_res = abs(data_array.rio.transform()[0])
+                    if abs(current_res - cell_size) > 0.01:
+                        resampling_method = Resampling.bilinear if current_res < cell_size else Resampling.nearest
+                        resampled = data_array.rio.reproject(
+                            data_array.rio.crs,
+                            resolution=(cell_size, cell_size),
+                            resampling=resampling_method,
+                        )
+                        resampled_arrays.append(resampled)
+                    else:
+                        resampled_arrays.append(data_array)
+
+                merged_data = xr.concat(resampled_arrays, dim="band")
+                merged_data.attrs["description"] = f"Merged bands: {', '.join(band_names)}"
+                merged_data.rio.to_raster(merged_path, tags={"BAND_NAMES": ",".join(band_names)}, descriptions=band_names)
+                result["merged"] = merged_path
+                if show_progress:
+                    print(f"Merged file written: {merged_path}")
+
+        all_results.append(result)
+
+    return all_results

sen2p-0.0.1.dist-info/METADATA

@@ -0,0 +1,78 @@
+Metadata-Version: 2.4
+Name: sen2p
+Version: 0.0.1
+Summary: Download Sentinel-2 data from Microsoft Planetary Computer
+Home-page: https://github.com/tnmthai/sen2p
+Author: Thai Tran
+Author-email: ThaiTran@outlook.co.nz
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Intended Audience :: Science/Research
+Classifier: Topic :: Scientific/Engineering :: GIS
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: planetary-computer
+Requires-Dist: pystac-client
+Requires-Dist: rioxarray
+Requires-Dist: xarray
+Requires-Dist: tqdm
+Requires-Dist: geopandas
+Requires-Dist: shapely
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# Sentinel Downloader
+
+This Python library allows you to download Sentinel-2 data from Microsoft Planetary Computer.
+
+## Installation
+
+```bash
+pip install -e .
+```
+
+## Usage
+
+```python
+from sen2p import download
+
+location = [172.1, -43.5]
+
+results = download(
+    start_date="2023-06-01",
+    end_date="2023-06-10",
+    location=location,
+    bands=["B02", "B03", "B04"],
+    output_dir="test_outputs",
+    merge_bands=True,
+    max_items=3  # download up to 3 images
+)
+
+for r in results:
+    print("Downloaded:", r)
+```
+```python
+from sen2p import download
+
+# Path to your shapefile
+shapefile_path = "Site.shp"  # Update with your actual shapefile path
+
+# Call the function
+results = download(
+    start_date="2023-06-01",
+    end_date="2023-06-30",
+    location=shapefile_path,
+    bands=["B02", "B03", "B04"],
+    output_dir="test_output",
+    merge_bands=True,
+    max_items=5  # Download up to 5 matching images
+)
+```

sen2p-0.0.1.dist-info/RECORD

@@ -0,0 +1,6 @@
+sen2p/__init__.py,sha256=uHbP27YsU5dG2fGH97zyc_kzdTGKCPcAY9m2j0i7Wnw,31
+sen2p/download.py,sha256=WjTuhc_EyKeRk_jpJwxBYupLZIqp4LKp-AEShUcy4is,6308
+sen2p-0.0.1.dist-info/METADATA,sha256=43qaqMDVhtHfQ1bpNuRATwVfId0CS8CcpsD8WP6wX40,1825
+sen2p-0.0.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sen2p-0.0.1.dist-info/top_level.txt,sha256=iI3IMU2lB-xjusho-zqzlRXWCF4uBRNhI8lnqixO3bo,6
+sen2p-0.0.1.dist-info/RECORD,,

sen2p-0.0.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

sen2p-0.0.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+sen2p