add functions getNodeReservoirHeadPatternIndex and setNodeReservoirHeadPatternIndex and update

The functions below are already available to the user, but new functions have been added specifically for setting and retrieving reservoir head pattern.
setNodeDemandPatternIndex
getNodeDemandPatternIndex

Author: Mariosmsk

README.md

@@ -283,6 +283,7 @@ If you want to contribute, please check out our [Code of Conduct](https://github
 | getNodeReservoirCount             |Retrieves the number of Reservoirs
 | getNodeReservoirIndex             |Retrieves the indices of reservoirs
 | getNodeReservoirNameID            |Retrieves the reservoir ID label
+| getNodeReservoirHeadPatternIndex  |Retrieves the indices of all reservoir head patterns
 | getNodeResultIndex                |Retrieves the order in which a node's results were saved to an output file
 | getNodeSourcePatternIndex         |Retrieves the value of all node source pattern index
 | getNodeSourceQuality              |Retrieves the value of all node source quality
@@ -448,6 +449,7 @@ If you want to contribute, please check out our [Code of Conduct](https://github
 | setNodeComment                    |Sets the comment string assigned to the node object
 | setNodeCoordinates                |Sets node coordinates
 | setNodeDemandPatternIndex         |Sets the values of demand time pattern indices
+| setNodeReservoirHeadPatternIndex  |Sets the pattern index for a reservoir node head
 | setNodeElevations                 |Sets the values of elevation for nodes
 | setNodeEmitterCoeff               |Sets the values of emitter coefficient for nodes
 | setNodeInitialQuality             |Sets the values of initial quality for nodes

epyt/epanet.py

@@ -4380,6 +4380,24 @@ def getNodeDemandPatternIndex(self):
             value[i + 1] = list(val[i])
         return value
 
+    def getNodeReservoirHeadPatternIndex(self):
+        """ Retrieves the value of all reservoir head pattern index.
+
+        Example:
+            d = epanet('net2-cl2.inp')
+            res_index = d.addNodeReservoir("res-1")
+            pidx = d.addPattern("pat-1", [1, 3])
+            d.setNodeReservoirHeadPatternIndex(res_index, pidx)
+            print(d.getNodeDemandPatternIndex())
+            print(d.getNodeReservoirHeadPatternIndex())
+
+        """
+        value = []
+        for i in self.getNodeReservoirIndex():
+            pattern_index = self.api.ENgetnodevalue(i, self.ToolkitConstants.EN_PATTERN)
+            value.append(pattern_index)
+        return value
+
     def getNodeDemandPatternNameID(self):
         """ Retrieves the value of all node base demands pattern name ID.
 
@@ -8281,6 +8299,22 @@ def setNodeDemandPatternIndex(self, value, *argv):
         """
         self.__setNodeDemandPattern('ENsetdemandpattern', self.ToolkitConstants.EN_PATTERN, value, *argv)
 
+    def setNodeReservoirHeadPatternIndex(self, value, *argv):
+        """ Sets the pattern index for a reservoir node head
+        This is a duplicate function—identical in behavior to setNodeDemandPatternIndex
+
+        Example 1:
+            d = epanet('net2-cl2.inp')
+            res_index = d.addNodeReservoir("res-1")
+            pidx = d.addPattern("pat-1")
+            d.setNodeReservoirHeadPatternIndex(res_index, pidx)
+            d.setPattern(pidx, 1)
+         """
+        if value in self.getNodeReservoirIndex():
+            self.__setNodeDemandPattern('ENsetdemandpattern', self.ToolkitConstants.EN_PATTERN, value, *argv)
+        else:
+            warnings.warn("Invalid reservoir index. For non-reservoir nodes, please use the setNodeDemandPatternIndex function.")
+
     def setNodeElevations(self, value, *argv):
         """ Sets the values of elevation for nodes.