geoai-py 0.5.4__py2.py3-none-any.whl → 0.5.6__py2.py3-none-any.whl

geoai/__init__.py

@@ -2,7 +2,7 @@
 
 __author__ = """Qiusheng Wu"""
 __email__ = "giswqs@gmail.com"
-__version__ = "0.5.4"
+__version__ = "0.5.6"
 
 
 import os

geoai/download.py

@@ -905,7 +905,7 @@ def pc_stac_download(
     items,
     output_dir=".",
     assets=None,
-    max_workers=4,
+    max_workers=1,
     skip_existing=True,
 ):
     """
@@ -922,7 +922,7 @@ def pc_stac_download(
         assets (list, optional): List of asset keys to download. If None,
             downloads all available assets. Defaults to None.
         max_workers (int, optional): Maximum number of concurrent download threads.
-            Defaults to 4.
+            Defaults to 1.
         skip_existing (bool, optional): Skip download if the file already exists.
             Defaults to True.
         sign_urls (bool, optional): Whether to sign URLs for authenticated access.
@@ -1135,7 +1135,37 @@ def view_pc_item(
     map_args=None,
     **kwargs,
 ):
+    """
+    Visualize a STAC item from the Planetary Computer on an interactive map.
+
+    This function allows users to display a STAC item on a map using either the
+    `folium` or `ipyleaflet` backend. It supports adding basemaps, configuring
+    map options, and customizing the visualization of the STAC item.
+
+    Args:
+        url (str, optional): URL of the STAC item to visualize. Defaults to None.
+        collection (str, optional): The collection ID of the STAC item. Defaults to None.
+        item (str or pystac.Item, optional): The STAC item or its ID. Defaults to None.
+        assets (list, optional): List of asset keys to visualize. Defaults to None.
+        bands (list, optional): List of specific bands to visualize. Defaults to None.
+        titiler_endpoint (str, optional): URL of the Titiler endpoint for rendering. Defaults to None.
+        name (str, optional): Name of the layer to display on the map. Defaults to "STAC Item".
+        attribution (str, optional): Attribution text for the layer. Defaults to "Planetary Computer".
+        opacity (float, optional): Opacity of the layer (0.0 to 1.0). Defaults to 1.0.
+        shown (bool, optional): Whether the layer is visible by default. Defaults to True.
+        fit_bounds (bool, optional): Whether to fit the map bounds to the layer. Defaults to True.
+        layer_index (int, optional): Index of the layer in the map's layer stack. Defaults to None.
+        backend (str, optional): Map backend to use ('folium' or 'ipyleaflet'). Defaults to "folium".
+        basemap (str, optional): Name of the basemap to add to the map. Defaults to None.
+        map_args (dict, optional): Additional arguments for configuring the map. Defaults to None.
+        **kwargs: Additional keyword arguments passed to the `add_stac_layer` method.
 
+    Returns:
+        leafmap.Map: An interactive map with the STAC item visualized.
+
+    Raises:
+        ValueError: If an unsupported backend is specified.
+    """
     if backend == "folium":
         import leafmap.foliumap as leafmap
 
@@ -1181,3 +1211,103 @@ def view_pc_item(
         **kwargs,
     )
     return m
+
+
+def view_pc_items(
+    urls=None,
+    collection=None,
+    items=None,
+    assets=None,
+    bands=None,
+    titiler_endpoint=None,
+    attribution="Planetary Computer",
+    opacity=1.0,
+    shown=True,
+    fit_bounds=True,
+    layer_index=None,
+    backend="folium",
+    basemap=None,
+    map_args=None,
+    **kwargs,
+):
+    """
+    Visualize multiple STAC items from the Planetary Computer on an interactive map.
+
+    This function allows users to display multiple STAC items on a map using either
+    the `folium` or `ipyleaflet` backend. It supports adding basemaps, configuring
+    map options, and customizing the visualization of the STAC items.
+
+    Args:
+        urls (list, optional): List of URLs of the STAC items to visualize. Defaults to None.
+        collection (str, optional): The collection ID of the STAC items. Defaults to None.
+        items (list or str, optional): List of STAC items or their IDs. Defaults to None.
+        assets (list, optional): List of asset keys to visualize. Defaults to None.
+        bands (list, optional): List of specific bands to visualize. Defaults to None.
+        titiler_endpoint (str, optional): URL of the Titiler endpoint for rendering. Defaults to None.
+        attribution (str, optional): Attribution text for the layers. Defaults to "Planetary Computer".
+        opacity (float, optional): Opacity of the layers (0.0 to 1.0). Defaults to 1.0.
+        shown (bool, optional): Whether the layers are visible by default. Defaults to True.
+        fit_bounds (bool, optional): Whether to fit the map bounds to the layers. Defaults to True.
+        layer_index (int, optional): Index of the layers in the map's layer stack. Defaults to None.
+        backend (str, optional): Map backend to use ('folium' or 'ipyleaflet'). Defaults to "folium".
+        basemap (str, optional): Name of the basemap to add to the map. Defaults to None.
+        map_args (dict, optional): Additional arguments for configuring the map. Defaults to None.
+        **kwargs: Additional keyword arguments passed to the `add_stac_layer` method.
+
+    Returns:
+        leafmap.Map: An interactive map with the STAC items visualized.
+
+    Raises:
+        ValueError: If an unsupported backend is specified.
+    """
+    if backend == "folium":
+        import leafmap.foliumap as leafmap
+
+    elif backend == "ipyleaflet":
+        import leafmap.leafmap as leafmap
+
+    else:
+        raise ValueError(
+            f"Unsupported backend: {backend}. Supported backends are 'folium' and 'ipyleaflet'."
+        )
+
+    if map_args is None:
+        map_args = {}
+
+    if "draw_control" not in map_args:
+        map_args["draw_control"] = False
+
+    if urls is not None:
+        items = [pystac.Item.from_file(url) for url in urls]
+
+    if isinstance(items, str):
+        items = [pystac.Item.from_file(items)]
+    m = leafmap.Map(**map_args)
+
+    if basemap is not None:
+        m.add_basemap(basemap)
+    if isinstance(items, list):
+
+        for item in items:
+
+            if isinstance(item, pystac.Item):
+                collection = item.collection_id
+                if assets is None:
+                    assets = [list(item.assets.keys())[0]]
+                item = item.id
+
+                m.add_stac_layer(
+                    collection=collection,
+                    item=item,
+                    assets=assets,
+                    bands=bands,
+                    titiler_endpoint=titiler_endpoint,
+                    name=item,
+                    attribution=attribution,
+                    opacity=opacity,
+                    shown=shown,
+                    fit_bounds=fit_bounds,
+                    layer_index=layer_index,
+                    **kwargs,
+                )
+    return m

geoai/geoai.py

@@ -11,12 +11,15 @@ from .download import (
     download_naip,
     download_overture_buildings,
     download_pc_stac_item,
+    get_overture_data,
+    extract_building_stats,
     pc_collection_list,
     pc_item_asset_list,
     pc_stac_search,
     pc_stac_download,
     read_pc_item_asset,
     view_pc_item,
+    view_pc_items,
 )
 from .classify import train_classifier, classify_image, classify_images
 from .extract import *

geoai/utils.py

@@ -2112,7 +2112,7 @@ def raster_to_vector(
         return gdf
 
 
-def batch_raster_to_vector(
+def raster_to_vector_batch(
     input_dir,
     output_dir,
     pattern="*.tif",
@@ -6235,7 +6235,7 @@ def download_model_from_hf(model_path, repo_id=None):
         # Define the repository ID and model filename
         if repo_id is None:
             print(
-                "Repo is is not specified, using default Hugging Face repo_id: giswqs/geoai"
+                "Repo is not specified, using default Hugging Face repo_id: giswqs/geoai"
             )
             repo_id = "giswqs/geoai"
 

{geoai_py-0.5.4.dist-info → geoai_py-0.5.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: geoai-py
-Version: 0.5.4
+Version: 0.5.6
 Summary: A Python package for using Artificial Intelligence (AI) with geospatial data
 Author-email: Qiusheng Wu <giswqs@gmail.com>
 License: MIT License

geoai_py-0.5.6.dist-info/RECORD

@@ -0,0 +1,16 @@
+geoai/__init__.py,sha256=aEive0DNN76ztdahgu3zV4CwuIRiQQ6E8CCpD4qI3eE,3765
+geoai/classify.py,sha256=_e-193QzAx3pIxUflPIsIs1qZevQx5ADu7i3bOL1G70,35055
+geoai/download.py,sha256=EQpcrcqMsYhDpd7bpjf4hGS5xL2oO-jsjngLgjjP3cE,46599
+geoai/extract.py,sha256=GocJufMmrwEWxNBL1J91EXXHL8AKcO8m_lmtUF5AKPw,119102
+geoai/geoai.py,sha256=jB5LvapKG0yg1jiVceyU1GSRZj-RSSwWuAmogvsfRpU,9340
+geoai/hf.py,sha256=mLKGxEAS5eHkxZLwuLpYc1o7e3-7QIXdBv-QUY-RkFk,17072
+geoai/segment.py,sha256=g3YW17ftr--CKq6VB32TJEPY8owGQ7uQ0sg_tUT2ooE,13681
+geoai/segmentation.py,sha256=AtPzCvguHAEeuyXafa4bzMFATvltEYcah1B8ZMfkM_s,11373
+geoai/train.py,sha256=mQXat2yuddT-2rME4xnX_m3SkY23E_-zdxLnBIKxw8o,44091
+geoai/utils.py,sha256=sPEVSGDHmgbTs078pES_K0HGmYMwNWw7XNCF-DwydWg,244383
+geoai_py-0.5.6.dist-info/licenses/LICENSE,sha256=vN2L5U7cZ6ZkOHFmc8WiGlsogWsZc5dllMeNxnKVOZg,1070
+geoai_py-0.5.6.dist-info/METADATA,sha256=oH93RwL6Y_PB29uc9q0kioYo986fZe_XEX2vtP2ErtA,6623
+geoai_py-0.5.6.dist-info/WHEEL,sha256=AeO2BvogYWm3eGaHCvhzmUYt8ia7KfURiHzO_1atlys,109
+geoai_py-0.5.6.dist-info/entry_points.txt,sha256=uGp3Az3HURIsRHP9v-ys0hIbUuBBNUfXv6VbYHIXeg4,41
+geoai_py-0.5.6.dist-info/top_level.txt,sha256=1YkCUWu-ii-0qIex7kbwAvfei-gos9ycyDyUCJPNWHY,6
+geoai_py-0.5.6.dist-info/RECORD,,

{geoai_py-0.5.4.dist-info → geoai_py-0.5.6.dist-info}/WHEEL

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

geoai_py-0.5.4.dist-info/RECORD

@@ -1,16 +0,0 @@
-geoai/__init__.py,sha256=_SUkF87skch6o0kWnBmcY5ki_cjE1uSsIVM_ByCU6eA,3765
-geoai/classify.py,sha256=_e-193QzAx3pIxUflPIsIs1qZevQx5ADu7i3bOL1G70,35055
-geoai/download.py,sha256=lJ1GsJOZsKc2i6_dQyPV-XXIXmlADOpmSBo-wha4DEU,40892
-geoai/extract.py,sha256=GocJufMmrwEWxNBL1J91EXXHL8AKcO8m_lmtUF5AKPw,119102
-geoai/geoai.py,sha256=CeVLiK_ly8SG6HBXb8ZrHv_WbPQS3v68NYfXskr1vMM,9270
-geoai/hf.py,sha256=mLKGxEAS5eHkxZLwuLpYc1o7e3-7QIXdBv-QUY-RkFk,17072
-geoai/segment.py,sha256=g3YW17ftr--CKq6VB32TJEPY8owGQ7uQ0sg_tUT2ooE,13681
-geoai/segmentation.py,sha256=AtPzCvguHAEeuyXafa4bzMFATvltEYcah1B8ZMfkM_s,11373
-geoai/train.py,sha256=mQXat2yuddT-2rME4xnX_m3SkY23E_-zdxLnBIKxw8o,44091
-geoai/utils.py,sha256=075-CoiYIYs7ZgtWOnyw5pBaQIeWuBW49JwTYfOBHDI,244386
-geoai_py-0.5.4.dist-info/licenses/LICENSE,sha256=vN2L5U7cZ6ZkOHFmc8WiGlsogWsZc5dllMeNxnKVOZg,1070
-geoai_py-0.5.4.dist-info/METADATA,sha256=aCOkYuGkhV6Iuzv5o5RvJNhAMApwANK9aJqmgPWG6y4,6623
-geoai_py-0.5.4.dist-info/WHEEL,sha256=MAQBAzGbXNI3bUmkDsiV_duv8i-gcdnLzw7cfUFwqhU,109
-geoai_py-0.5.4.dist-info/entry_points.txt,sha256=uGp3Az3HURIsRHP9v-ys0hIbUuBBNUfXv6VbYHIXeg4,41
-geoai_py-0.5.4.dist-info/top_level.txt,sha256=1YkCUWu-ii-0qIex7kbwAvfei-gos9ycyDyUCJPNWHY,6
-geoai_py-0.5.4.dist-info/RECORD,,