epyt-flow 0.4.0__tar.gz → 0.5.0__tar.gz

{epyt_flow-0.4.0 → epyt_flow-0.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: epyt-flow
-Version: 0.4.0
+Version: 0.5.0
 Summary: EPyT-Flow -- EPANET Python Toolkit - Flow
 Author-email: André Artelt <aartelt@techfak.uni-bielefeld.de>, "Marios S. Kyriakou" <kiriakou.marios@ucy.ac.cy>, "Stelios G. Vrachimis" <vrachimis.stelios@ucy.ac.cy>
 License: MIT License

epyt_flow-0.5.0/epyt_flow/VERSION

@@ -0,0 +1 @@
+0.5.0

{epyt_flow-0.4.0 → epyt_flow-0.5.0}/epyt_flow/simulation/scada/scada_data.py

@@ -1163,6 +1163,9 @@ class ScadaData(Serializable):
             :func:`~epyt_flow.simulation.scada.scada_data.ScadaData.get_data` --
             calling this function is the only way of accessing the energy usage of each pump.
 
+            The odering in the returned NumPy array corresponds to the ordering
+            of the pumps in EPANET.
+
         Returns
         -------
         `numpy.ndarray`
@@ -1180,6 +1183,9 @@ class ScadaData(Serializable):
             :func:`~epyt_flow.simulation.scada.scada_data.ScadaData.get_data` --
             calling this function is the only way of accessing the pumps' efficiency.
 
+            The odering in the returned NumPy array corresponds to the ordering
+            of the pumps in EPANET.
+
         Returns
         -------
         `numpy.ndarray`

{epyt_flow-0.4.0 → epyt_flow-0.5.0}/epyt_flow/simulation/scenario_visualizer.py

@@ -12,7 +12,7 @@ class ScenarioVisualizer():
 
     Parameters
     ----------
-    scenario : :class:`epyt_flow.simulation.scenario_simulator.ScenarioSimulator`
+    scenario : :class:`~epyt_flow.simulation.scenario_simulator.ScenarioSimulator`
         Scenario to be visualized.
     """
     def __init__(self, scenario: ScenarioSimulator):

{epyt_flow-0.4.0 → epyt_flow-0.5.0}/epyt_flow/simulation/sensor_config.py

@@ -1647,6 +1647,42 @@ class SensorConfig(JsonSerializable):
         """
         return deepcopy(self.__sensors_id_to_idx)
 
+    def get_as_dict(self) -> dict:
+        """
+        Gets the sensor configuration as a dictionary.
+
+        Returns
+        -------
+        `dict`
+            Dictionary of set sensors -- the keys are the sensor types.
+        """
+        r = {}
+
+        if self.__pressure_sensors != []:
+            r["pressure"] = self.__pressure_sensors
+        if self.__flow_sensors != []:
+            r["flow"] = self.__flow_sensors
+        if self.__demand_sensors != []:
+            r["demand"] = self.__demand_sensors
+        if self.__tank_volume_sensors != []:
+            r["tank_volume"] = self.__tank_volume_sensors
+        if self.__valve_state_sensors != []:
+            r["valve_state"] = self.__valve_state_sensors
+        if self.__pump_state_sensors != []:
+            r["pump_state"] = self.__pump_state_sensors
+        if self.__quality_node_sensors != []:
+            r["node_quality"] = self.__quality_node_sensors
+        if self.__quality_link_sensors != []:
+            r["link_quality"] = self.__quality_link_sensors
+        if self.__bulk_species_node_sensors != {}:
+            r["node_bulk_species"] = self.__bulk_species_node_sensors
+        if self.__bulk_species_link_sensors != {}:
+            r["link_bulk_species"] = self.__bulk_species_link_sensors
+        if self.__surface_species_sensors != {}:
+            r["surface_species"] = self.__surface_species_sensors
+
+        return r
+
     def get_attributes(self) -> dict:
         attr = {"nodes": self.__nodes, "links": self.__links,
                 "valves": self.__valves, "pumps": self.__pumps,
@@ -1719,7 +1755,7 @@ class SensorConfig(JsonSerializable):
             f"pump_id_to_idx: {self.__pump_id_to_idx} tank_id_to_idx: {self.__tank_id_to_idx} " +\
             f"valve_id_to_idx: {self.__valve_id_to_idx} " +\
             f"bulkspecies_id_to_idx: {self.__bulkspecies_id_to_idx} " +\
-            f"surfacespecies_id_to_idx: {self.__surfacespecies_id_to_idx}" +\
+            f"surfacespecies_id_to_idx: {self.__surfacespecies_id_to_idx} " +\
             f"pressure_sensors: {self.__pressure_sensors} flow_sensors: {self.__flow_sensors} " +\
             f"demand_sensors: {self.__demand_sensors} " +\
             f"quality_node_sensors: {self.__quality_node_sensors} " +\

{epyt_flow-0.4.0 → epyt_flow-0.5.0}/epyt_flow/topology.py

@@ -108,7 +108,7 @@ class NetworkTopology(nx.Graph, JsonSerializable):
 
     def convert_units(self, units: int) -> Any:
         """
-        Converts this instance to a :class:`epyt_flow.topology.NetworkTopology` instance
+        Converts this instance to a :class:`~epyt_flow.topology.NetworkTopology` instance
         where everything is measured in given measurement units category
         (US Customary or SI Metric).
 
@@ -124,7 +124,7 @@ class NetworkTopology(nx.Graph, JsonSerializable):
 
         Returns
         -------
-        :class:`epyt_flow.topology.NetworkTopology`
+        :class:`~epyt_flow.topology.NetworkTopology`
             Network topology with the new measurements units.
         """
         if self.__units is None:
@@ -199,6 +199,76 @@ class NetworkTopology(nx.Graph, JsonSerializable):
         """
         return [(link_id, end_points) for link_id, end_points, _ in self.__links]
 
+    def get_all_junctions(self) -> list[str]:
+        """
+        Gets all junctions -- i.e. nodes that are not tanks or reservoirs.
+
+        Returns
+        -------
+        `list[str]`
+            List of all junctions.
+        """
+        r = []
+
+        for node_id in self.get_all_nodes():
+            if self.get_node_info(node_id)["type"] == "JUNCTION":
+                r.append(node_id)
+
+        return r
+
+    def get_all_tanks(self) -> list[str]:
+        """
+        Gets all tanks -- i.e. nodes that are not junctions or reservoirs.
+
+        Returns
+        -------
+        `list[str]`
+            List of all tanks.
+        """
+        r = []
+
+        for node_id in self.get_all_nodes():
+            if self.get_node_info(node_id)["type"] == "TANK":
+                r.append(node_id)
+
+        return r
+
+    def get_all_reservoirs(self) -> list[str]:
+        """
+        Gets all reservoirs -- i.e. nodes that are not junctions or tanks.
+
+        Returns
+        -------
+        `list[str]`
+            List of all reservoirs.
+        """
+        r = []
+
+        for node_id in self.get_all_nodes():
+            if self.get_node_info(node_id)["type"] == "RESERVOIR":
+                r.append(node_id)
+
+        return r
+
+    def get_all_pipes(self) -> list[tuple[str, tuple[str, str]]]:
+        """
+        Gets all pipes -- i.e. links that not valves or pumps.
+
+        Returns
+        -------
+        `list[tuple[str, tuple[str, str]]]`
+            List of all pipes -- (link ID, (left node ID, right node ID)).
+        """
+        r = []
+
+        for link_id, link_nodes in self.get_all_links():
+            link_info = self.get_link_info(link_id)
+
+            if link_info["type"] == "PIPE":
+                r.append((link_id, link_nodes))
+
+        return r
+
     def get_all_pumps(self) -> list[str]:
         """
         Gets the IDs of all pumps.

{epyt_flow-0.4.0 → epyt_flow-0.5.0}/epyt_flow/utils.py

@@ -153,8 +153,10 @@ def plot_timeseries_data(data: np.ndarray, labels: list[str] = None, x_axis_labe
 
 
 def plot_timeseries_prediction(y: np.ndarray, y_pred: np.ndarray,
-                               confidence_interval: np.ndarray = None, show: bool = True,
-                               ax: matplotlib.axes.Axes = None) -> matplotlib.axes.Axes:
+                               confidence_interval: np.ndarray = None,
+                               x_axis_label: str = None, y_axis_label: str = None,
+                               show: bool = True, ax: matplotlib.axes.Axes = None
+                               ) -> matplotlib.axes.Axes:
     """
     Plots the prediction (e.g. forecast) of *single* time series together with the
     ground truth time series. In addition, confidence intervals can be plotted as well.
@@ -169,6 +171,14 @@ def plot_timeseries_prediction(y: np.ndarray, y_pred: np.ndarray,
         Confidence interval (upper and lower value) for each prediction in `y_pred`.
         If not None, the confidence interval is plotted as well.
 
+        The default is None.
+    x_axis_label : `str`, optional
+        X axis label.
+
+        The default is None.
+    y_axis_label : `str`, optional
+        Y axis label.
+
         The default is None.
     show : `bool`, optional
         If True, the plot/figure is shown in a window.
@@ -198,6 +208,14 @@ def plot_timeseries_prediction(y: np.ndarray, y_pred: np.ndarray,
         raise ValueError("'y_pred' must be a 1d array")
     if len(y.shape) != 1:
         raise ValueError("'y' must be a 1d array")
+    if x_axis_label is not None:
+        if not isinstance(x_axis_label, str):
+            raise TypeError("'x_axis_label' must be an instance of 'str' " +
+                            f"but not of '{type(x_axis_label)}'")
+    if y_axis_label is not None:
+        if not isinstance(y_axis_label, str):
+            raise TypeError("'y_axis_label' must be an instance of 'str' " +
+                            f"but not of '{type(y_axis_label)}'")
     if not isinstance(show, bool):
         raise TypeError(f"'show' must be an instance of 'bool' but not of '{type(show)}'")
     if ax is not None:
@@ -218,6 +236,11 @@ def plot_timeseries_prediction(y: np.ndarray, y_pred: np.ndarray,
     ax.plot(y_pred, ".-", label="Prediction")
     ax.legend()
 
+    if x_axis_label is not None:
+        ax.set_xlabel(x_axis_label)
+    if y_axis_label is not None:
+        ax.set_ylabel(y_axis_label)
+
     if show is True and fig is not None:
         plt.show()
 

{epyt_flow-0.4.0 → epyt_flow-0.5.0}/epyt_flow.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: epyt-flow
-Version: 0.4.0
+Version: 0.5.0
 Summary: EPyT-Flow -- EPANET Python Toolkit - Flow
 Author-email: André Artelt <aartelt@techfak.uni-bielefeld.de>, "Marios S. Kyriakou" <kiriakou.marios@ucy.ac.cy>, "Stelios G. Vrachimis" <vrachimis.stelios@ucy.ac.cy>
 License: MIT License

epyt_flow-0.5.0/epyt_flow.egg-info/top_level.txt

@@ -0,0 +1,12 @@
+Hanoi
+MacOS
+Net1
+Tut
+build
+dist
+docs
+epyt_flow
+examples
+paper
+quality-example-data
+tests

epyt_flow-0.4.0/epyt_flow/VERSION

@@ -1 +0,0 @@
-0.4.0

epyt_flow-0.4.0/epyt_flow.egg-info/top_level.txt

@@ -1,6 +0,0 @@
-dist
-docs
-epyt_flow
-examples
-paper
-tests