Skip to content

Commit f092b95

Browse files
authored
Update epamodule.py
Following the release of epanet 2.2 version and new API functions available, two functions were added to epamodule.py : 1) ENsettitle 2) ENsetflowunits
1 parent 3d1d360 commit f092b95

1 file changed

Lines changed: 28 additions & 0 deletions

File tree

epanet-module/epamodule.py

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -98,6 +98,34 @@ def ENgetnodeid(index):
9898
if ierr!=0: raise ENtoolkitError(ierr)
9999
return label.value
100100

101+
def ENsettitle(title):
102+
"""Set inp file title
103+
104+
Arguments:
105+
title: inp title"""
106+
ierr = _lib.ENsettitle(ctypes.c_char_p(title.encode()),'','')
107+
if ierr!=0: raise ENtoolkitError(ierr)
108+
109+
110+
def ENsetflowunits(units_code):
111+
"""Set flow units
112+
113+
Arguments:
114+
units_code: int code for unit type:
115+
EN_CFS 0 cubic feet per second
116+
EN_GPM 1 gallons per minute
117+
EN_MGD 2 million gallons per day
118+
EN_IMGD 3 Imperial mgd
119+
EN_AFD 4 acre-feet per day
120+
EN_LPS 5 liters per second
121+
EN_LPM 6 liters per minute
122+
EN_MLD 7 million liters per day
123+
EN_CMH 8 cubic meters per hour
124+
EN_CMD 9 cubic meters per day
125+
"""
126+
ierr = _lib.ENsetflowunits(ctypes.c_int(units_code))
127+
if ierr != 0: raise ENtoolkitError(ierr)
128+
101129

102130
def ENgetnodetype(index):
103131
"""Retrieves the node-type code for a specific node.

0 commit comments

Comments
 (0)