ras-commander 0.42.0__py3-none-any.whl → 0.43.0__py3-none-any.whl

ras_commander/HdfMesh.py

@@ -0,0 +1,299 @@
+from pathlib import Path
+import h5py
+import numpy as np
+import pandas as pd
+from geopandas import GeoDataFrame
+from shapely.geometry import Polygon, Point, LineString, MultiLineString, MultiPolygon
+from shapely.ops import polygonize  # Importing polygonize to resolve the undefined name error
+from typing import List, Tuple, Optional, Dict, Any
+import logging
+from .HdfBase import HdfBase
+from .HdfUtils import HdfUtils
+from .Decorators import standardize_input, log_call
+from .LoggingConfig import setup_logging, get_logger
+
+logger = get_logger(__name__)
+
+
+class HdfMesh:
+    """
+    A class for handling mesh-related operations on HEC-RAS HDF files.
+
+    This class provides methods to extract and analyze mesh data from HEC-RAS HDF files,
+    including mesh area names, mesh areas, cell polygons, cell points, cell faces, and
+    2D flow area attributes.
+
+    Methods in this class are designed to work with the mesh geometry data stored in
+    HEC-RAS HDF files, providing functionality to retrieve and process various aspects
+    of the 2D flow areas and their associated mesh structures.
+
+    Note: This class relies on HdfBase and HdfUtils for some underlying operations.
+    """
+
+    FLOW_AREA_2D_PATH = "Geometry/2D Flow Areas"
+
+    def __init__(self):
+        self.logger = logging.getLogger(__name__)
+
+    @staticmethod
+    @standardize_input(file_type='plan_hdf')
+    def mesh_area_names(hdf_path: Path) -> List[str]:
+        """
+        Return a list of the 2D mesh area names of the RAS geometry.
+
+        Parameters
+        ----------
+        hdf_path : Path
+            Path to the HEC-RAS geometry HDF file.
+
+        Returns
+        -------
+        List[str]
+            A list of the 2D mesh area names (str) within the RAS geometry if 2D areas exist.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf_file:
+                if HdfMesh.FLOW_AREA_2D_PATH not in hdf_file:
+                    return list()
+                return list(
+                    [
+                        HdfUtils.convert_ras_hdf_string(n)
+                        for n in hdf_file[f"{HdfMesh.FLOW_AREA_2D_PATH}/Attributes"][()]["Name"]
+                    ]
+                )
+        except Exception as e:
+            self.logger.error(f"Error reading mesh area names from {hdf_path}: {str(e)}")
+            return list()
+
+    @staticmethod
+    @standardize_input(file_type='geom_hdf')
+    def mesh_areas(hdf_path: Path) -> GeoDataFrame:
+        """
+        Return 2D flow area perimeter polygons.
+
+        Parameters
+        ----------
+        hdf_path : Path
+            Path to the HEC-RAS geometry HDF file.
+
+        Returns
+        -------
+        GeoDataFrame
+            A GeoDataFrame containing the 2D flow area perimeter polygons if 2D areas exist.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf_file:
+                mesh_area_names = HdfMesh.mesh_area_names(hdf_path)
+                if not mesh_area_names:
+                    return GeoDataFrame()
+                mesh_area_polygons = [
+                    Polygon(hdf_file[f"{HdfMesh.FLOW_AREA_2D_PATH}/{n}/Perimeter"][()])
+                    for n in mesh_area_names
+                ]
+                return GeoDataFrame(
+                    {"mesh_name": mesh_area_names, "geometry": mesh_area_polygons},
+                    geometry="geometry",
+                    crs=HdfUtils.projection(hdf_file),  # Pass the h5py.File object instead of the path
+                )
+        except Exception as e:
+            logger.error(f"Error reading mesh areas from {hdf_path}: {str(e)}")
+            return GeoDataFrame()
+
+    @staticmethod
+    @standardize_input(file_type='geom_hdf')
+    def mesh_cell_polygons(hdf_path: Path) -> GeoDataFrame:
+        """
+        Return 2D flow mesh cell polygons.
+
+        Parameters
+        ----------
+        hdf_path : Path
+            Path to the HEC-RAS geometry HDF file.
+
+        Returns
+        -------
+        GeoDataFrame
+            A GeoDataFrame containing the 2D flow mesh cell polygons.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf_file:
+                mesh_area_names = HdfMesh.mesh_area_names(hdf_path)
+                if not mesh_area_names:
+                    return GeoDataFrame()
+
+                face_gdf = HdfMesh.mesh_cell_faces(hdf_path)
+
+                cell_dict = {"mesh_name": [], "cell_id": [], "geometry": []}
+                for i, mesh_name in enumerate(mesh_area_names):
+                    cell_cnt = hdf_file[f"{HdfMesh.FLOW_AREA_2D_PATH}/Cell Info"][()][i][1]
+                    cell_ids = list(range(cell_cnt))
+                    cell_face_info = hdf_file[
+                        f"{HdfMesh.FLOW_AREA_2D_PATH}/{mesh_name}/Cells Face and Orientation Info"
+                    ][()]
+                    cell_face_values = hdf_file[
+                        f"{HdfMesh.FLOW_AREA_2D_PATH}/{mesh_name}/Cells Face and Orientation Values"
+                    ][()][:, 0]
+                    face_id_lists = list(
+                        np.vectorize(
+                            lambda cell_id: str(
+                                cell_face_values[
+                                    cell_face_info[cell_id][0] : cell_face_info[cell_id][0]
+                                    + cell_face_info[cell_id][1]
+                                ]
+                            )
+                        )(cell_ids)
+                    )
+                    mesh_faces = (
+                        face_gdf[face_gdf.mesh_name == mesh_name][["face_id", "geometry"]]
+                        .set_index("face_id")
+                        .to_numpy()
+                    )
+                    cell_dict["mesh_name"] += [mesh_name] * cell_cnt
+                    cell_dict["cell_id"] += cell_ids
+                    cell_dict["geometry"] += list(
+                        np.vectorize(
+                            lambda face_id_list: (
+                                lambda geom_col: Polygon(list(polygonize(geom_col))[0])
+                            )(
+                                np.ravel(
+                                    mesh_faces[
+                                        np.array(face_id_list.strip("[]").split()).astype(int)
+                                    ]
+                                )
+                            )
+                        )(face_id_lists)
+                    )
+                return GeoDataFrame(cell_dict, geometry="geometry", crs=HdfUtils.projection(hdf_file))
+        except Exception as e:
+            logger.error(f"Error reading mesh cell polygons from {hdf_path}: {str(e)}")
+            return GeoDataFrame()
+
+    @staticmethod
+    @standardize_input(file_type='plan_hdf')
+    def mesh_cell_points(hdf_path: Path) -> GeoDataFrame:
+        """
+        Return 2D flow mesh cell points.
+
+        Parameters
+        ----------
+        hdf_path : Path
+            Path to the HEC-RAS geometry HDF file.
+
+        Returns
+        -------
+        GeoDataFrame
+            A GeoDataFrame containing the 2D flow mesh cell points.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf_file:
+                mesh_area_names = HdfMesh.mesh_area_names(hdf_path)
+                if not mesh_area_names:
+                    return GeoDataFrame()
+                pnt_dict = {"mesh_name": [], "cell_id": [], "geometry": []}
+                for i, mesh_name in enumerate(mesh_area_names):
+                    starting_row, count = hdf_file[f"{HdfMesh.FLOW_AREA_2D_PATH}/Cell Info"][()][i]
+                    cell_pnt_coords = hdf_file[f"{HdfMesh.FLOW_AREA_2D_PATH}/Cell Points"][()][
+                        starting_row : starting_row + count
+                    ]
+                    pnt_dict["mesh_name"] += [mesh_name] * cell_pnt_coords.shape[0]
+                    pnt_dict["cell_id"] += range(count)
+                    pnt_dict["geometry"] += list(
+                        np.vectorize(lambda coords: Point(coords), signature="(n)->()")(
+                            cell_pnt_coords
+                        )
+                    )
+                return GeoDataFrame(pnt_dict, geometry="geometry", crs=HdfUtils.projection(hdf_path))
+        except Exception as e:
+            self.logger.error(f"Error reading mesh cell points from {hdf_path}: {str(e)}")
+            return GeoDataFrame()
+
+    @staticmethod
+    @standardize_input(file_type='plan_hdf')
+    def mesh_cell_faces(hdf_path: Path) -> GeoDataFrame:
+        """
+        Return 2D flow mesh cell faces.
+
+        Parameters
+        ----------
+        hdf_path : Path
+            Path to the HEC-RAS geometry HDF file.
+
+        Returns
+        -------
+        GeoDataFrame
+            A GeoDataFrame containing the 2D flow mesh cell faces.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf_file:
+                mesh_area_names = HdfMesh.mesh_area_names(hdf_path)
+                if not mesh_area_names:
+                    return GeoDataFrame()
+                face_dict = {"mesh_name": [], "face_id": [], "geometry": []}
+                for mesh_name in mesh_area_names:
+                    facepoints_index = hdf_file[
+                        f"{HdfMesh.FLOW_AREA_2D_PATH}/{mesh_name}/Faces FacePoint Indexes"
+                    ][()]
+                    facepoints_coordinates = hdf_file[
+                        f"{HdfMesh.FLOW_AREA_2D_PATH}/{mesh_name}/FacePoints Coordinate"
+                    ][()]
+                    faces_perimeter_info = hdf_file[
+                        f"{HdfMesh.FLOW_AREA_2D_PATH}/{mesh_name}/Faces Perimeter Info"
+                    ][()]
+                    faces_perimeter_values = hdf_file[
+                        f"{HdfMesh.FLOW_AREA_2D_PATH}/{mesh_name}/Faces Perimeter Values"
+                    ][()]
+                    face_id = -1
+                    for pnt_a_index, pnt_b_index in facepoints_index:
+                        face_id += 1
+                        face_dict["mesh_name"].append(mesh_name)
+                        face_dict["face_id"].append(face_id)
+                        coordinates = list()
+                        coordinates.append(facepoints_coordinates[pnt_a_index])
+                        starting_row, count = faces_perimeter_info[face_id]
+                        if count > 0:
+                            coordinates += list(
+                                faces_perimeter_values[starting_row : starting_row + count]
+                            )
+                        coordinates.append(facepoints_coordinates[pnt_b_index])
+                        face_dict["geometry"].append(LineString(coordinates))
+                return GeoDataFrame(face_dict, geometry="geometry", crs=HdfUtils.projection(hdf_path))
+        except Exception as e:
+            self.logger.error(f"Error reading mesh cell faces from {hdf_path}: {str(e)}")
+            return GeoDataFrame()
+
+    @staticmethod
+    @standardize_input(file_type='geom_hdf')
+    def get_geom_2d_flow_area_attrs(hdf_path: Path) -> Dict[str, Any]:
+        """
+        Return geometry 2D flow area attributes from a HEC-RAS HDF file.
+
+        Parameters
+        ----------
+        hdf_path : Path
+            Path to the HEC-RAS geometry HDF file.
+
+        Returns
+        -------
+        Dict[str, Any]
+            A dictionary containing the 2D flow area attributes.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf_file:
+                d2_flow_area = hdf_file.get(f"{HdfMesh.FLOW_AREA_2D_PATH}/Attributes")
+                if d2_flow_area is not None and isinstance(d2_flow_area, h5py.Dataset):
+                    result = {}
+                    for name in d2_flow_area.dtype.names:
+                        try:
+                            value = d2_flow_area[name][()]
+                            if isinstance(value, bytes):
+                                value = value.decode('utf-8')
+                            result[name] = value
+                        except Exception as e:
+                            logger.warning(f"Error converting attribute '{name}': {str(e)}")
+                    return result
+                else:
+                    logger.info("No 2D Flow Area attributes found or invalid dataset.")
+                    return {}
+        except Exception as e:
+            logger.error(f"Error reading 2D flow area attributes from {hdf_path}: {str(e)}")
+            return {}

ras_commander/HdfPlan.py

@@ -0,0 +1,190 @@
+import h5py
+import pandas as pd
+from datetime import datetime
+from pathlib import Path
+from typing import Dict, List, Optional
+
+from .HdfBase import HdfBase
+from .HdfUtils import HdfUtils
+from .Decorators import standardize_input, log_call
+from .LoggingConfig import setup_logging, get_logger
+
+logger = get_logger(__name__)
+
+
+class HdfPlan:
+    """
+    A class for handling operations on HEC-RAS plan HDF files.
+
+    This class provides methods for extracting and analyzing data from HEC-RAS plan HDF files,
+    including simulation times, plan information, and geometry attributes.
+
+    Methods in this class use the @standardize_input decorator to handle different input types
+    (e.g., plan number, file path) and the @log_call decorator for logging method calls.
+
+    Attributes:
+        None
+
+    Methods:
+        get_simulation_start_time: Get the simulation start time.
+        get_simulation_end_time: Get the simulation end time.
+        get_unsteady_datetimes: Get a list of unsteady datetimes.
+        get_plan_info_attrs: Get plan information attributes.
+        get_plan_param_attrs: Get plan parameter attributes.
+        get_meteorology_precip_attrs: Get precipitation attributes.
+        get_geom_attrs: Get geometry attributes.
+    """
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_simulation_start_time(hdf_path: Path) -> datetime:
+        """
+        Get the simulation start time from the plan file.
+
+        Args:
+            hdf_path (Path): Path to the HEC-RAS plan HDF file.
+
+        Returns:
+            datetime: The simulation start time.
+
+        Raises:
+            ValueError: If there's an error reading the simulation start time.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf_file:
+                return HdfBase._get_simulation_start_time(hdf_file)
+        except Exception as e:
+            raise ValueError(f"Failed to get simulation start time: {str(e)}")
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_simulation_end_time(hdf_path: Path) -> datetime:
+        """
+        Get the simulation end time from the plan file.
+
+        Args:
+            hdf_path (Path): Path to the HEC-RAS plan HDF file.
+
+        Returns:
+            datetime: The simulation end time.
+
+        Raises:
+            ValueError: If there's an error reading the simulation end time.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf_file:
+                plan_info = hdf_file.get('Plan Data/Plan Information')
+                if plan_info is None:
+                    raise ValueError("Plan Information not found in HDF file")
+                time_str = plan_info.attrs.get('Simulation End Time')
+                return datetime.strptime(time_str.decode('utf-8'), "%d%b%Y %H:%M:%S")
+        except Exception as e:
+            raise ValueError(f"Failed to get simulation end time: {str(e)}")
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_unsteady_datetimes(hdf_path: Path) -> List[datetime]:
+        """
+        Get the list of unsteady datetimes from the HDF file.
+
+        Args:
+            hdf_path (Path): Path to the HEC-RAS plan HDF file.
+
+        Returns:
+            List[datetime]: A list of datetime objects representing the unsteady timestamps.
+
+        Raises:
+            ValueError: If there's an error retrieving the unsteady datetimes.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf_file:
+                return HdfBase._get_unsteady_datetimes(hdf_file)
+        except Exception as e:
+            raise ValueError(f"Failed to get unsteady datetimes: {str(e)}")
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_plan_info_attrs(hdf_path: Path) -> Dict:
+        """
+        Get plan information attributes from a HEC-RAS HDF plan file.
+
+        Args:
+            hdf_path (Path): Path to the HEC-RAS plan HDF file.
+
+        Returns:
+            Dict: A dictionary containing the plan information attributes.
+
+        Raises:
+            ValueError: If there's an error retrieving the plan information attributes.
+        """
+        try:
+            return HdfUtils.get_attrs(hdf_path, "Plan Data/Plan Information")
+        except Exception as e:
+            raise ValueError(f"Failed to get plan information attributes: {str(e)}")
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_plan_param_attrs(hdf_path: Path) -> Dict:
+        """
+        Get plan parameter attributes from a HEC-RAS HDF plan file.
+
+        Args:
+            hdf_path (Path): Path to the HEC-RAS plan HDF file.
+
+        Returns:
+            Dict: A dictionary containing the plan parameter attributes.
+
+        Raises:
+            ValueError: If there's an error retrieving the plan parameter attributes.
+        """
+        try:
+            return HdfUtils.get_attrs(hdf_path, "Plan Data/Plan Parameters")
+        except Exception as e:
+            raise ValueError(f"Failed to get plan parameter attributes: {str(e)}")
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_meteorology_precip_attrs(hdf_path: Path) -> Dict:
+        """
+        Get precipitation attributes from a HEC-RAS HDF plan file.
+
+        Args:
+            hdf_path (Path): Path to the HEC-RAS plan HDF file.
+
+        Returns:
+            Dict: A dictionary containing the precipitation attributes.
+
+        Raises:
+            ValueError: If there's an error retrieving the precipitation attributes.
+        """
+        try:
+            return HdfUtils.get_attrs(hdf_path, "Event Conditions/Meteorology/Precipitation")
+        except Exception as e:
+            raise ValueError(f"Failed to get precipitation attributes: {str(e)}")
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_geom_attrs(hdf_path: Path) -> Dict:
+        """
+        Get geometry attributes from a HEC-RAS HDF plan file.
+
+        Args:
+            hdf_path (Path): Path to the HEC-RAS plan HDF file.
+
+        Returns:
+            Dict: A dictionary containing the geometry attributes.
+
+        Raises:
+            ValueError: If there's an error retrieving the geometry attributes.
+        """
+        try:
+            return HdfUtils.get_attrs(hdf_path, "Geometry")
+        except Exception as e:
+            raise ValueError(f"Failed to get geometry attributes: {str(e)}")