ras-commander 0.44.0__py3-none-any.whl → 0.45.0__py3-none-any.whl

ras_commander/HdfPipe.py

@@ -0,0 +1,262 @@
+import h5py
+import numpy as np
+import pandas as pd
+import geopandas as gpd
+import xarray as xr
+from pathlib import Path
+from shapely.geometry import LineString, Point
+from typing import List, Dict, Any, Optional, Union
+from .HdfBase import HdfBase
+from .HdfUtils import HdfUtils
+from .Decorators import standardize_input, log_call
+from .LoggingConfig import get_logger
+
+logger = get_logger(__name__)
+
+class HdfPipe:
+    """
+    A class for handling pipe network related data from HEC-RAS HDF files.
+    """
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_pipe_conduits(hdf_path: Path) -> gpd.GeoDataFrame:
+        """
+        Extract pipe conduit data from the HDF file.
+
+        Args:
+            hdf_path (Path): Path to the HDF file.
+
+        Returns:
+            gpd.GeoDataFrame: GeoDataFrame containing pipe conduit data.
+
+        Raises:
+            KeyError: If the required datasets are not found in the HDF file.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf:
+                # Extract pipe conduit data
+                polyline_info = hdf['/Geometry/Pipe Conduits/Polyline Info'][()]
+                polyline_points = hdf['/Geometry/Pipe Conduits/Polyline Points'][()]
+                terrain_profiles_info = hdf['/Geometry/Pipe Conduits/Terrain Profiles Info'][()]
+
+                # Create geometries
+                geometries = []
+                for start, count, _, _ in polyline_info:
+                    points = polyline_points[start:start+count]
+                    geometries.append(LineString(points))
+
+                # Create GeoDataFrame
+                gdf = gpd.GeoDataFrame(geometry=geometries)
+                gdf['conduit_id'] = range(len(gdf))
+                gdf['terrain_profile_start'] = terrain_profiles_info[:, 0]
+                gdf['terrain_profile_count'] = terrain_profiles_info[:, 1]
+
+                # Set CRS if available
+                crs = HdfUtils.projection(hdf_path)
+                if crs:
+                    gdf.set_crs(crs, inplace=True)
+
+                return gdf
+
+        except KeyError as e:
+            logger.error(f"Required dataset not found in HDF file: {e}")
+            raise
+        except Exception as e:
+            logger.error(f"Error extracting pipe conduit data: {e}")
+            raise
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_pipe_nodes(hdf_path: Path) -> gpd.GeoDataFrame:
+        """
+        Extract pipe node data from the HDF file.
+
+        Args:
+            hdf_path (Path): Path to the HDF file.
+
+        Returns:
+            gpd.GeoDataFrame: GeoDataFrame containing pipe node data.
+
+        Raises:
+            KeyError: If the required datasets are not found in the HDF file.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf:
+                # Extract pipe node data
+                points = hdf['/Geometry/Pipe Nodes/Points'][()]
+                attributes = hdf['/Geometry/Pipe Nodes/Attributes'][()]
+
+                # Create geometries
+                geometries = [Point(x, y) for x, y in points]
+
+                # Create GeoDataFrame
+                gdf = gpd.GeoDataFrame(geometry=geometries)
+                gdf['node_id'] = range(len(gdf))
+
+                # Add attributes
+                for name in attributes.dtype.names:
+                    gdf[name] = attributes[name]
+
+                # Set CRS if available
+                crs = HdfUtils.projection(hdf_path)
+                if crs:
+                    gdf.set_crs(crs, inplace=True)
+
+                return gdf
+
+        except KeyError as e:
+            logger.error(f"Required dataset not found in HDF file: {e}")
+            raise
+        except Exception as e:
+            logger.error(f"Error extracting pipe node data: {e}")
+            raise
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_pipe_network_timeseries(hdf_path: Path, variable: str) -> xr.DataArray:
+        """
+        Extract timeseries data for a specific variable in the pipe network.
+
+        Args:
+            hdf_path (Path): Path to the HDF file.
+            variable (str): Variable to extract (e.g., "Cell Courant", "Cell Water Surface").
+
+        Returns:
+            xr.DataArray: DataArray containing the timeseries data.
+
+        Raises:
+            KeyError: If the required datasets are not found in the HDF file.
+            ValueError: If an invalid variable is specified.
+        """
+        valid_variables = [
+            "Cell Courant", "Cell Water Surface", "Face Flow", "Face Velocity",
+            "Face Water Surface", "Pipes/Pipe Flow DS", "Pipes/Pipe Flow US",
+            "Pipes/Vel DS", "Pipes/Vel US", "Nodes/Depth", "Nodes/Drop Inlet Flow",
+            "Nodes/Water Surface"
+        ]
+
+        if variable not in valid_variables:
+            raise ValueError(f"Invalid variable. Must be one of: {', '.join(valid_variables)}")
+
+        try:
+            with h5py.File(hdf_path, 'r') as hdf:
+                # Extract timeseries data
+                data_path = f"/Results/Unsteady/Output/Output Blocks/DSS Hydrograph Output/Unsteady Time Series/Pipe Networks/Davis/{variable}"
+                data = hdf[data_path][()]
+
+                # Extract time information
+                time = HdfBase._get_unsteady_datetimes(hdf)
+
+                # Create DataArray
+                da = xr.DataArray(
+                    data=data,
+                    dims=['time', 'location'],
+                    coords={'time': time, 'location': range(data.shape[1])},
+                    name=variable
+                )
+
+                # Add attributes
+                da.attrs['units'] = hdf[data_path].attrs.get('Units', b'').decode('utf-8')
+                da.attrs['variable'] = variable
+
+                return da
+
+        except KeyError as e:
+            logger.error(f"Required dataset not found in HDF file: {e}")
+            raise
+        except Exception as e:
+            logger.error(f"Error extracting pipe network timeseries data: {e}")
+            raise
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_pipe_network_summary(hdf_path: Path) -> pd.DataFrame:
+        """
+        Extract summary data for pipe networks from the HDF file.
+
+        Args:
+            hdf_path (Path): Path to the HDF file.
+
+        Returns:
+            pd.DataFrame: DataFrame containing pipe network summary data.
+
+        Raises:
+            KeyError: If the required datasets are not found in the HDF file.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf:
+                # Extract summary data
+                summary_path = "/Results/Unsteady/Summary/Pipe Network"
+                if summary_path not in hdf:
+                    logger.warning("Pipe Network summary data not found in HDF file")
+                    return pd.DataFrame()
+
+                summary_data = hdf[summary_path][()]
+                
+                # Create DataFrame
+                df = pd.DataFrame(summary_data)
+
+                # Convert column names
+                df.columns = [col.decode('utf-8') for col in df.columns]
+
+                return df
+
+        except KeyError as e:
+            logger.error(f"Required dataset not found in HDF file: {e}")
+            raise
+        except Exception as e:
+            logger.error(f"Error extracting pipe network summary data: {e}")
+            raise
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_pipe_profile(hdf_path: Path, conduit_id: int) -> pd.DataFrame:
+        """
+        Extract the profile data for a specific pipe conduit.
+
+        Args:
+            hdf_path (Path): Path to the HDF file.
+            conduit_id (int): ID of the conduit to extract profile for.
+
+        Returns:
+            pd.DataFrame: DataFrame containing the pipe profile data.
+
+        Raises:
+            KeyError: If the required datasets are not found in the HDF file.
+            IndexError: If the specified conduit_id is out of range.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf:
+                # Get conduit info
+                terrain_profiles_info = hdf['/Geometry/Pipe Conduits/Terrain Profiles Info'][()]
+                
+                if conduit_id >= len(terrain_profiles_info):
+                    raise IndexError(f"conduit_id {conduit_id} is out of range")
+
+                start, count = terrain_profiles_info[conduit_id]
+
+                # Extract profile data
+                profile_values = hdf['/Geometry/Pipe Conduits/Terrain Profiles Values'][start:start+count]
+
+                # Create DataFrame
+                df = pd.DataFrame(profile_values, columns=['Station', 'Elevation'])
+
+                return df
+
+        except KeyError as e:
+            logger.error(f"Required dataset not found in HDF file: {e}")
+            raise
+        except IndexError as e:
+            logger.error(f"Invalid conduit_id: {e}")
+            raise
+        except Exception as e:
+            logger.error(f"Error extracting pipe profile data: {e}")
+            raise
+        
+        

ras_commander/HdfPump.py

@@ -0,0 +1,255 @@
+import h5py
+import numpy as np
+import pandas as pd
+import geopandas as gpd
+import xarray as xr
+from pathlib import Path
+from shapely.geometry import Point
+from typing import List, Dict, Any, Optional, Union
+from .HdfUtils import HdfUtils
+from .Decorators import standardize_input, log_call
+from .LoggingConfig import get_logger
+
+logger = get_logger(__name__)
+
+class HdfPump:
+    """
+    A class for handling pump station related data from HEC-RAS HDF files.
+    """
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_pump_stations(hdf_path: Path) -> gpd.GeoDataFrame:
+        """
+        Extract pump station data from the HDF file.
+
+        Args:
+            hdf_path (Path): Path to the HDF file.
+
+        Returns:
+            gpd.GeoDataFrame: GeoDataFrame containing pump station data.
+
+        Raises:
+            KeyError: If the required datasets are not found in the HDF file.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf:
+                # Extract pump station data
+                attributes = hdf['/Geometry/Pump Stations/Attributes'][()]
+                points = hdf['/Geometry/Pump Stations/Points'][()]
+
+                # Create geometries
+                geometries = [Point(x, y) for x, y in points]
+
+                # Create GeoDataFrame
+                gdf = gpd.GeoDataFrame(geometry=geometries)
+                gdf['station_id'] = range(len(gdf))
+
+                # Add attributes
+                for name in attributes.dtype.names:
+                    gdf[name] = attributes[name]
+
+                # Set CRS if available
+                crs = HdfUtils.projection(hdf_path)
+                if crs:
+                    gdf.set_crs(crs, inplace=True)
+
+                return gdf
+
+        except KeyError as e:
+            logger.error(f"Required dataset not found in HDF file: {e}")
+            raise
+        except Exception as e:
+            logger.error(f"Error extracting pump station data: {e}")
+            raise
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_pump_groups(hdf_path: Path) -> pd.DataFrame:
+        """
+        Extract pump group data from the HDF file.
+
+        Args:
+            hdf_path (Path): Path to the HDF file.
+
+        Returns:
+            pd.DataFrame: DataFrame containing pump group data.
+
+        Raises:
+            KeyError: If the required datasets are not found in the HDF file.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf:
+                # Extract pump group data
+                attributes = hdf['/Geometry/Pump Stations/Pump Groups/Attributes'][()]
+                efficiency_curves_info = hdf['/Geometry/Pump Stations/Pump Groups/Efficiency Curves Info'][()]
+                efficiency_curves_values = hdf['/Geometry/Pump Stations/Pump Groups/Efficiency Curves Values'][()]
+
+                # Create DataFrame
+                df = pd.DataFrame(attributes)
+
+                # Add efficiency curve data
+                df['efficiency_curve_start'] = efficiency_curves_info[:, 0]
+                df['efficiency_curve_count'] = efficiency_curves_info[:, 1]
+
+                # Process efficiency curves
+                def get_efficiency_curve(start, count):
+                    return efficiency_curves_values[start:start+count].tolist()
+
+                df['efficiency_curve'] = df.apply(lambda row: get_efficiency_curve(row['efficiency_curve_start'], row['efficiency_curve_count']), axis=1)
+
+                return df
+
+        except KeyError as e:
+            logger.error(f"Required dataset not found in HDF file: {e}")
+            raise
+        except Exception as e:
+            logger.error(f"Error extracting pump group data: {e}")
+            raise
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_pump_station_timeseries(hdf_path: Path, pump_station: str) -> xr.DataArray:
+        """
+        Extract timeseries data for a specific pump station.
+
+        Args:
+            hdf_path (Path): Path to the HDF file.
+            pump_station (str): Name of the pump station.
+
+        Returns:
+            xr.DataArray: DataArray containing the timeseries data.
+
+        Raises:
+            KeyError: If the required datasets are not found in the HDF file.
+            ValueError: If the specified pump station is not found.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf:
+                # Check if the pump station exists
+                pumping_stations_path = "/Results/Unsteady/Output/Output Blocks/DSS Hydrograph Output/Unsteady Time Series/Pumping Stations"
+                if pump_station not in hdf[pumping_stations_path]:
+                    raise ValueError(f"Pump station '{pump_station}' not found in HDF file")
+
+                # Extract timeseries data
+                data_path = f"{pumping_stations_path}/{pump_station}/Structure Variables"
+                data = hdf[data_path][()]
+
+                # Extract time information
+                time = HdfBase._get_unsteady_datetimes(hdf)
+
+                # Create DataArray
+                da = xr.DataArray(
+                    data=data,
+                    dims=['time', 'variable'],
+                    coords={'time': time, 'variable': ['Flow', 'Stage HW', 'Stage TW', 'Pump Station', 'Pumps on']},
+                    name=pump_station
+                )
+
+                # Add attributes
+                da.attrs['units'] = hdf[data_path].attrs.get('Variable_Unit', b'').decode('utf-8')
+                da.attrs['pump_station'] = pump_station
+
+                return da
+
+        except KeyError as e:
+            logger.error(f"Required dataset not found in HDF file: {e}")
+            raise
+        except ValueError as e:
+            logger.error(str(e))
+            raise
+        except Exception as e:
+            logger.error(f"Error extracting pump station timeseries data: {e}")
+            raise
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_pump_station_summary(hdf_path: Path) -> pd.DataFrame:
+        """
+        Extract summary data for pump stations from the HDF file.
+
+        Args:
+            hdf_path (Path): Path to the HDF file.
+
+        Returns:
+            pd.DataFrame: DataFrame containing pump station summary data.
+
+        Raises:
+            KeyError: If the required datasets are not found in the HDF file.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf:
+                # Extract summary data
+                summary_path = "/Results/Unsteady/Summary/Pump Station"
+                if summary_path not in hdf:
+                    logger.warning("Pump Station summary data not found in HDF file")
+                    return pd.DataFrame()
+
+                summary_data = hdf[summary_path][()]
+                
+                # Create DataFrame
+                df = pd.DataFrame(summary_data)
+
+                # Convert column names
+                df.columns = [col.decode('utf-8') for col in df.columns]
+
+                return df
+
+        except KeyError as e:
+            logger.error(f"Required dataset not found in HDF file: {e}")
+            raise
+        except Exception as e:
+            logger.error(f"Error extracting pump station summary data: {e}")
+            raise
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_pump_operation_data(hdf_path: Path, pump_station: str) -> pd.DataFrame:
+        """
+        Extract pump operation data for a specific pump station.
+
+        Args:
+            hdf_path (Path): Path to the HDF file.
+            pump_station (str): Name of the pump station.
+
+        Returns:
+            pd.DataFrame: DataFrame containing pump operation data.
+
+        Raises:
+            KeyError: If the required datasets are not found in the HDF file.
+            ValueError: If the specified pump station is not found.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf:
+                # Check if the pump station exists
+                pump_stations_path = "/Results/Unsteady/Output/Output Blocks/DSS Profile Output/Unsteady Time Series/Pumping Stations"
+                if pump_station not in hdf[pump_stations_path]:
+                    raise ValueError(f"Pump station '{pump_station}' not found in HDF file")
+
+                # Extract pump operation data
+                data_path = f"{pump_stations_path}/{pump_station}/Structure Variables"
+                data = hdf[data_path][()]
+
+                # Extract time information
+                time = HdfBase._get_unsteady_datetimes(hdf)
+
+                # Create DataFrame
+                df = pd.DataFrame(data, columns=['Flow', 'Stage HW', 'Stage TW', 'Pump Station', 'Pumps on'])
+                df['Time'] = time
+
+                return df
+
+        except KeyError as e:
+            logger.error(f"Required dataset not found in HDF file: {e}")
+            raise
+        except ValueError as e:
+            logger.error(str(e))
+            raise
+        except Exception as e:
+            logger.error(f"Error extracting pump operation data: {e}")
+            raise

ras_commander/HdfResultsPlan.py

@@ -396,3 +396,6 @@ class HdfResultsPlan:
         except Exception as e:
             logger.error(f"Error in reference_summary_output: {str(e)}")
             return pd.DataFrame()  # Return an empty DataFrame on general error
+
+
+

ras_commander/HdfResultsXsec.py

@@ -8,20 +8,21 @@ released under MIT license and Copyright (c) 2024 fema-ffrd
 The file has been forked and modified for use in RAS Commander.
 """
 
+from pathlib import Path
+from typing import Union, Optional, List
+
 import h5py
 import numpy as np
 import pandas as pd
-from pathlib import Path
-from typing import Union, Optional, List
+import xarray as xr
+
 from .HdfBase import HdfBase
 from .HdfUtils import HdfUtils
 from .Decorators import standardize_input, log_call
-from .LoggingConfig import setup_logging, get_logger
-import xarray as xr
+from .LoggingConfig import get_logger
 
 logger = get_logger(__name__)
 
-
 class HdfResultsXsec:
     """
     A class for handling cross-section results from HEC-RAS HDF files.
@@ -235,3 +236,208 @@ class HdfResultsXsec:
         """
         return HdfResultsXsec.steady_profile_xs_output(hdf_path, "Velocity Total")
 
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_pipe_network_summary(hdf_path: Path) -> pd.DataFrame:
+        """
+        Extract summary data for pipe networks from the HDF file.
+
+        Args:
+            hdf_path (Path): Path to the HDF file.
+
+        Returns:
+            pd.DataFrame: DataFrame containing pipe network summary data.
+
+        Raises:
+            KeyError: If the required datasets are not found in the HDF file.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf:
+                # Extract summary data
+                summary_path = "/Results/Unsteady/Summary/Pipe Network"
+                if summary_path not in hdf:
+                    logger.warning("Pipe Network summary data not found in HDF file")
+                    return pd.DataFrame()
+
+                summary_data = hdf[summary_path][()]
+                
+                # Create DataFrame
+                df = pd.DataFrame(summary_data)
+
+                # Convert column names
+                df.columns = [col.decode('utf-8') if isinstance(col, bytes) else col for col in df.columns]
+
+                # Convert byte string values to regular strings
+                for col in df.columns:
+                    if df[col].dtype == object:
+                        df[col] = df[col].apply(lambda x: x.decode('utf-8') if isinstance(x, bytes) else x)
+
+                return df
+
+        except KeyError as e:
+            logger.error(f"Required dataset not found in HDF file: {e}")
+            raise
+        except Exception as e:
+            logger.error(f"Error extracting pipe network summary data: {e}")
+            raise
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_pump_station_summary(hdf_path: Path) -> pd.DataFrame:
+        """
+        Extract summary data for pump stations from the HDF file.
+
+        Args:
+            hdf_path (Path): Path to the HDF file.
+
+        Returns:
+            pd.DataFrame: DataFrame containing pump station summary data.
+
+        Raises:
+            KeyError: If the required datasets are not found in the HDF file.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf:
+                # Extract summary data
+                summary_path = "/Results/Unsteady/Summary/Pump Station"
+                if summary_path not in hdf:
+                    logger.warning("Pump Station summary data not found in HDF file")
+                    return pd.DataFrame()
+
+                summary_data = hdf[summary_path][()]
+                
+                # Create DataFrame
+                df = pd.DataFrame(summary_data)
+
+                # Convert column names
+                df.columns = [col.decode('utf-8') if isinstance(col, bytes) else col for col in df.columns]
+
+                # Convert byte string values to regular strings
+                for col in df.columns:
+                    if df[col].dtype == object:
+                        df[col] = df[col].apply(lambda x: x.decode('utf-8') if isinstance(x, bytes) else x)
+
+                return df
+
+        except KeyError as e:
+            logger.error(f"Required dataset not found in HDF file: {e}")
+            raise
+        except Exception as e:
+            logger.error(f"Error extracting pump station summary data: {e}")
+            raise
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_pipe_network_profile_output(hdf_path: Path) -> pd.DataFrame:
+        """
+        Extract pipe network profile output data from the HDF file.
+
+        Args:
+            hdf_path (Path): Path to the HDF file.
+
+        Returns:
+            pd.DataFrame: DataFrame containing pipe network profile output data.
+
+        Raises:
+            KeyError: If the required datasets are not found in the HDF file.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf:
+                # Extract profile output data
+                profile_path = "/Results/Unsteady/Output/Output Blocks/DSS Profile Output/Unsteady Time Series/Pipe Networks"
+                if profile_path not in hdf:
+                    logger.warning("Pipe Network profile output data not found in HDF file")
+                    return pd.DataFrame()
+
+                # Initialize an empty list to store data from all pipe networks
+                all_data = []
+
+                # Iterate through all pipe networks
+                for network in hdf[profile_path].keys():
+                    network_path = f"{profile_path}/{network}"
+                    
+                    # Extract data for each variable
+                    for var in hdf[network_path].keys():
+                        data = hdf[f"{network_path}/{var}"][()]
+                        
+                        # Create a DataFrame for this variable
+                        df = pd.DataFrame(data)
+                        df['Network'] = network
+                        df['Variable'] = var
+                        
+                        all_data.append(df)
+
+                # Concatenate all DataFrames
+                result_df = pd.concat(all_data, ignore_index=True)
+
+                # Add time information
+                time = HdfBase._get_unsteady_datetimes(hdf)
+                result_df['Time'] = [time[i] for i in result_df.index]
+
+                return result_df
+
+        except KeyError as e:
+            logger.error(f"Required dataset not found in HDF file: {e}")
+            raise
+        except Exception as e:
+            logger.error(f"Error extracting pipe network profile output data: {e}")
+            raise
+
+    @staticmethod
+    @log_call
+    @standardize_input(file_type='plan_hdf')
+    def get_pump_station_profile_output(hdf_path: Path) -> pd.DataFrame:
+        """
+        Extract pump station profile output data from the HDF file.
+
+        Args:
+            hdf_path (Path): Path to the HDF file.
+
+        Returns:
+            pd.DataFrame: DataFrame containing pump station profile output data.
+
+        Raises:
+            KeyError: If the required datasets are not found in the HDF file.
+        """
+        try:
+            with h5py.File(hdf_path, 'r') as hdf:
+                # Extract profile output data
+                profile_path = "/Results/Unsteady/Output/Output Blocks/DSS Profile Output/Unsteady Time Series/Pumping Stations"
+                if profile_path not in hdf:
+                    logger.warning("Pump Station profile output data not found in HDF file")
+                    return pd.DataFrame()
+
+                # Initialize an empty list to store data from all pump stations
+                all_data = []
+
+                # Iterate through all pump stations
+                for station in hdf[profile_path].keys():
+                    station_path = f"{profile_path}/{station}/Structure Variables"
+                    
+                    data = hdf[station_path][()]
+                    
+                    # Create a DataFrame for this pump station
+                    df = pd.DataFrame(data, columns=['Flow', 'Stage HW', 'Stage TW', 'Pump Station', 'Pumps on'])
+                    df['Station'] = station
+                    
+                    all_data.append(df)
+
+                # Concatenate all DataFrames
+                result_df = pd.concat(all_data, ignore_index=True)
+
+                # Add time information
+                time = HdfBase._get_unsteady_datetimes(hdf)
+                result_df['Time'] = [time[i] for i in result_df.index]
+
+                return result_df
+
+        except KeyError as e:
+            logger.error(f"Required dataset not found in HDF file: {e}")
+            raise
+        except Exception as e:
+            logger.error(f"Error extracting pump station profile output data: {e}")
+            raise

ras_commander/__init__.py

@@ -32,6 +32,8 @@ from .HdfResultsXsec import HdfResultsXsec
 from .HdfStruc import HdfStruc
 from .HdfUtils import HdfUtils
 from .HdfXsec import HdfXsec
+from .HdfPump import HdfPump
+from .HdfPipe import HdfPipe
 
 # Define __all__ to specify what should be imported when using "from ras_commander import *"
 __all__ = [
@@ -45,6 +47,8 @@ __all__ = [
     "HdfStruc",
     "HdfUtils",
     "HdfXsec",
+    "HdfPump",
+    "HdfPipe",
     "standardize_input",
     "ras",
     "init_ras_project",

{ras_commander-0.44.0.dist-info → ras_commander-0.45.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ras-commander
-Version: 0.44.0
+Version: 0.45.0
 Summary: A Python library for automating HEC-RAS operations
 Home-page: https://github.com/billk-FM/ras-commander
 Author: William M. Katzenmeyer

{ras_commander-0.44.0.dist-info → ras_commander-0.45.0.dist-info}/RECORD

@@ -2,10 +2,12 @@ ras_commander/Decorators.py,sha256=i5AEQbe7JeI8Y3O_dQ5OO4Ab0KO5SiZTiysFBGxqTRU,4
 ras_commander/HdfBase.py,sha256=HV5ccV9QH2lz4ZRYqK2d7_S833cTSUcostzxxSPb4O4,7129
 ras_commander/HdfBndry.py,sha256=LWaDMHeo0V_VOpx7D9q7W_0WvWsD6NQThiSLYPY8ApE,20761
 ras_commander/HdfMesh.py,sha256=JKWn6S-FcdFs3fZQMQ8mkIEIFFxUdVjWb5r7jAh3ujw,12668
+ras_commander/HdfPipe.py,sha256=agGhMNeANi92eS62CEoaPDimS0I0yGoAoMv1c8YAIk4,9432
 ras_commander/HdfPlan.py,sha256=nt1Q03cPNX_yVX0mOUqeWGPicqtusUnuBSTcBw8n0Vg,6967
+ras_commander/HdfPump.py,sha256=8mBfFg0epaq74txDyUfBPkWqzvRQCb8OojEgDGCridM,9416
 ras_commander/HdfResultsMesh.py,sha256=Ke6HSpdKBd-TUa0QIuMaxhyj--lfuQGlMDlDuk2qrd8,28064
-ras_commander/HdfResultsPlan.py,sha256=zkosLBpyXIkoWrqAXh8sRxcmKvca2F-urQUGCnT8PmQ,17293
-ras_commander/HdfResultsXsec.py,sha256=sQokzqL5Fpnr7bsHppoygL6hrr07dTXiLvaecNOp64s,8426
+ras_commander/HdfResultsPlan.py,sha256=-4tTMLiSkuOs3pKn7sA_1RoGmBxaG8jMUJ9qvVlc2Q0,17299
+ras_commander/HdfResultsXsec.py,sha256=4BHIX-VTfkh1YwIUJDZpnnz_-wGctOJO2XVCTsiUXDs,16393
 ras_commander/HdfStruc.py,sha256=NNTSwUQmLMLn8SPyrLIKXaUaqKavTos_TDDB6Mja7-4,5595
 ras_commander/HdfUtils.py,sha256=dUU9QkNO-FhaCjD_TEOv_FNNrZI3TUOQ6QjSiXrJGSc,18403
 ras_commander/HdfXsec.py,sha256=ccJ-GAVKwS2Z-k6CCWAYvw974xCBTTF-5Hh1CnikVG4,11562
@@ -18,9 +20,9 @@ ras_commander/RasPlan.py,sha256=_0EjxQnamd9FhzzKlYNqJ3chv2bvxfk0DXEwfHntXFo,4028
 ras_commander/RasPrj.py,sha256=ePLcPNcKVMgijvJKkJIWwQslQi2D6zq2UTW-8XCFmto,36159
 ras_commander/RasUnsteady.py,sha256=oBVC9QFQMrSx8LY2Cb1CjUyCQEUoIEnEuVJsATYiVYs,4649
 ras_commander/RasUtils.py,sha256=NBMxTHWHoTH2MJzqJ0y1_00fgKSS1GnNuEikwZ3Pqzs,34153
-ras_commander/__init__.py,sha256=3MuobRXVEQNvoGwf9AlbvuJg8meni3eWw_JZMFOuMt4,1630
-ras_commander-0.44.0.dist-info/LICENSE,sha256=_pbd6qHnlsz1iQ-ozDW_49r86BZT6CRwO2iBtw0iN6M,457
-ras_commander-0.44.0.dist-info/METADATA,sha256=zbNmO6OwnKbX-B3bIgdrPYnLA2aPFZ1PqfBstFzEkCA,15935
-ras_commander-0.44.0.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-ras_commander-0.44.0.dist-info/top_level.txt,sha256=i76S7eKLFC8doKcXDl3aiOr9RwT06G8adI6YuKbQDaA,14
-ras_commander-0.44.0.dist-info/RECORD,,
+ras_commander/__init__.py,sha256=8sg6T5I_uAqGe2pInrMnBrhS8NZbfnF7Pp7uwzQOqIo,1722
+ras_commander-0.45.0.dist-info/LICENSE,sha256=_pbd6qHnlsz1iQ-ozDW_49r86BZT6CRwO2iBtw0iN6M,457
+ras_commander-0.45.0.dist-info/METADATA,sha256=Y7H1Fq4aG4C2ujRl6VkjxBLOaNZUzX1X7gNd5UJDU7s,15935
+ras_commander-0.45.0.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+ras_commander-0.45.0.dist-info/top_level.txt,sha256=i76S7eKLFC8doKcXDl3aiOr9RwT06G8adI6YuKbQDaA,14
+ras_commander-0.45.0.dist-info/RECORD,,