Merge pull request #11 from supathdhitalGEO/main

Fine Tune the applications and code formatting

Author: supathdhitalGEO

pyproject.toml

@@ -28,9 +28,8 @@ dependencies = [
   "nbformat>=5.10.4,<6.0.0",
   "pyproj>=3.7.0,<4.0.0",
   "notebook>=7.3.2,<8.0.0",
-  "boto3>=1.36.16,<2.0.0",
+  "boto3>=1.40.0",
   "geemap",
-  "uv",
   "seaborn"
 ]
 

src/fimeval/ContingencyMap/evaluationFIM.py

@@ -277,8 +277,8 @@ def resize_image(
                                 out_transform1,
                             )
                             merged = out_image1 + out_image2_resized
-                            merged[merged==7] = 5 
-                            
+                            merged[merged == 7] = 5
+
             # Get Evaluation Metrics
             (
                 unique_values,

src/fimeval/FIMs/__init__.py

@@ -0,0 +1,3 @@
+from .owphandfim import get_HANDFIM
+
+__all__ = ["get_HANDFIM"]

src/fimeval/FIMs/owphandfim.py

@@ -0,0 +1,86 @@
+import os
+import rasterio
+import shutil
+from pathlib import Path
+from rasterio.mask import mask
+import fimserve as fm
+
+from .utilis import *
+
+
+# GET LOW FIDELITY USING FIMSERVE
+def get_HANDFIM(
+    huc_id,
+    event_date=None,
+    data="retrospective",
+    forecast_range=None,
+    forecast_date=None,
+    sort_by=None,
+):
+    original_cwd = os.getcwd()
+    try:
+        createCWD("owp_fim")
+        fm.DownloadHUC8(huc_id)
+
+        # For retrospective event
+        if data == "retrospective":
+            if not event_date:
+                raise ValueError("event_date is required for retrospective analysis.")
+            huc_event_dict = initialize_huc_event(huc_id, event_date)
+            fm.getNWMretrospectivedata(huc_event_dict=huc_event_dict)
+
+        # For forecasting event
+        elif data == "forecast":
+            if not forecast_range:
+                raise ValueError(
+                    "forecast_range ('short_range', 'medium_range', or 'long_range') is required for forecast."
+                )
+
+            if forecast_range in ["medium_range", "long_range"]:
+                if not sort_by:
+                    sort_by = "maximum"
+                fm.getNWMForecasteddata(
+                    huc_id=huc_id,
+                    forecast_range=forecast_range,
+                    forecast_date=forecast_date,
+                    sort_by=sort_by,
+                )
+            else:
+                fm.getNWMForecasteddata(
+                    huc_id=huc_id,
+                    forecast_range=forecast_range,
+                    forecast_date=forecast_date,
+                )
+        else:
+            raise ValueError("data_type must be either 'retrospective' or 'forecast'.")
+
+        # Run the FIM
+        fm.runOWPHANDFIM(huc_id)
+
+    finally:
+        os.chdir(original_cwd)
+
+
+# Raster to binary
+def raster2binary(input_raster_path, geometry, final_raster_path):
+    # Mask the raster with the geometry
+    with rasterio.open(input_raster_path) as src:
+        out_image, out_transform = mask(src, geometry, crop=True, filled=True, nodata=0)
+        out_meta = src.meta.copy()
+        out_meta.update(
+            {
+                "driver": "GTiff",
+                "height": out_image.shape[1],
+                "width": out_image.shape[2],
+                "transform": out_transform,
+                "crs": src.crs,
+                "nodata": 0,
+            }
+        )
+
+    # Convert to binary (HAND logic: flooded if value > 0)
+    binary_image = (out_image > 0).astype("uint8")
+
+    # Save the binary raster
+    with rasterio.open(final_raster_path, "w", **out_meta) as dst:
+        dst.write(binary_image)

src/fimeval/FIMs/utilis.py

@@ -0,0 +1,28 @@
+import os
+import rasterio
+
+
+# INITIALIZE IN HUC EVENT DICT
+def initialize_huc_event(huc_id, event_times):
+    if isinstance(event_times, str):
+        event_times = [event_times]
+    return {huc_id: event_times}
+
+
+# Create a folder and make it a working directory
+def createCWD(folder_name):
+    os.makedirs(folder_name, exist_ok=True)
+    os.chdir(folder_name)
+    new_path = os.getcwd()
+    return new_path
+
+
+def compress_tif_lzw(tif_path):
+    # Read original file
+    with rasterio.open(tif_path) as src:
+        profile = src.profile.copy()
+        data = src.read()
+    profile.update(compress="lzw")
+
+    with rasterio.open(tif_path, "w", **profile) as dst:
+        dst.write(data)

src/fimeval/__init__.py

@@ -10,11 +10,15 @@
 # Evaluation with Building foorprint module
 from .BuildingFootprint.evaluationwithBF import EvaluationWithBuildingFootprint
 
+# FIM evaluation
+from .FIMs.owphandfim import get_HANDFIM
+
 __all__ = [
     "EvaluateFIM",
     "PrintContingencyMap",
     "PlotEvaluationMetrics",
     "get_PWB",
     "EvaluationWithBuildingFootprint",
     "compress_tif_lzw",
+    "get_HANDFIM",
 ]