Merge pull request #1 from GalPerelman/GalPerelman-patch-1

Update epamodule.py

Author: GalPerelman

epanet-module/epamodule.py

@@ -98,6 +98,34 @@ def ENgetnodeid(index):
     if ierr!=0: raise ENtoolkitError(ierr)
     return label.value
 
+def ENsettitle(title):
+	"""Set inp file title
+	
+	Arguments:
+	title: inp title"""
+	ierr = _lib.ENsettitle(ctypes.c_char_p(title.encode()),'','')
+	if ierr!=0: raise ENtoolkitError(ierr)
+
+
+def ENsetflowunits(units_code):
+	"""Set flow units
+	
+	Arguments:
+	units_code:	int code for unit type: 
+		EN_CFS	0	cubic feet per second
+		EN_GPM	1	gallons per minute
+		EN_MGD	2	million gallons per day
+		EN_IMGD	3	Imperial mgd
+		EN_AFD	4	acre-feet per day
+		EN_LPS	5	liters per second
+		EN_LPM	6	liters per minute
+		EN_MLD	7	million liters per day
+		EN_CMH	8	cubic meters per hour
+		EN_CMD	9	cubic meters per day
+	"""
+    	ierr = _lib.ENsetflowunits(ctypes.c_int(units_code))
+    	if ierr != 0: raise ENtoolkitError(ierr)
+
 
 def ENgetnodetype(index):
     """Retrieves the node-type code for a specific node.