US-EPA-SWMM python interface
Project description
© Institute of Urban Water Management and Landscape Water Engineering, Graz University of Technology and Markus Pichler
💡 Getting started
With this package you can read INP-files, manipulate them and write new ones. You can run swmm within the python api. And you can read the RPT- and OUT-files as a pandas DataFrame for further analysis.
This package is based on the command line SWMM syntax. (see Appendix D in the SWMM User Manual 5.2)
Introduction
The swmm-api package is a powerful tool for modellers and researchers who use the Storm Water Management Model (SWMM). This software enables the manipulation and analysis of SWMM models, both in terms of the input data and the simulation results. The package is written in Python, making it an attractive option for those who use this language for data management and advanced analysis.
One of the key features of swmm-api is its ability to read and write SWMM import-files (.inp), allowing the user to manipulate the model structure and input data. The package also has the capability to run the SWMM model within the Python environment, providing users with quick access to simulation results. Furthermore, swmm-api can read both the report (.rpt) and binary output-files (.out), presenting the results as a Pandas DataFrame for easy analysis. The ability to read binary hotstart-files (.hst) is also included, which enables the acceleration of simulation time by using initial values stored in the file.
The swmm-api package is designed to be flexible and user-friendly, with an object-oriented structure that is lightweight and fast. The package is based on the SWMM command line syntax, making it easy to use for those familiar with this model. Additionally, swmm-api has the ability to interact with GIS data, making it a valuable tool for modellers working with spatial data.
Install the package
pip install swmm-api
... to install the package with all dependencies for macros use:
pip install swmm-api[macros]
... to install the package with all dependencies for GIS I/O use (for Linux or for Windows using python >= 3.10):
pip install swmm-api[gis]
... and to install the package with all dependencies for macros and GIS I/O use (for Linux or for Windows using python >= 3.10):
pip install swmm-api[full]
To add the GIS functionality on Windows, I recommend using python version >= 3.10 or with Mamba
(or miniconda or Anaconda)
and run mamba install geopandas
to install the GIS dependencies (see GeoPandas).
📖 Documentation
Link to the documentation of the api and some example jupyter notebook.
Here are example files for other use-cases.
Read, manipulate and write the INP-File
Read the INP-File
from swmm_api import read_inp_file, SwmmInput
inp = read_inp_file('inputfile.inp') # type: swmm_api.SwmmInput
# or
inp = SwmmInput.read_file('inputfile.inp')
# or
inp = SwmmInput('inputfile.inp')
Getting information
from swmm_api.input_file.section_labels import TIMESERIES
# getting a specific section of the inp-file
sec_timeseries = inp[TIMESERIES] # type: swmm_api.input_file.helpers.InpSection
# or
sec_timeseries = inp.TIMESERIES # type: swmm_api.input_file.helpers.InpSection
# getting a specific timeseries as pandas.Series
ts = inp[TIMESERIES]['regenseries'].pandas # type: pandas.Series
Manipulate the INP-File
from swmm_api.input_file.section_labels import JUNCTIONS
# setting the elevation of a specific node to a new value
inp[JUNCTIONS]['J01'].elevation = 210
# or
inp.JUNCTIONS['J01'].elevation = 210
# or
inp.JUNCTIONS['J01']['elevation'] = 210
Write the manipulated INP-File
inp.write_file('new_inputfile.inp')
see examples/inp_file_reader.ipynb
see examples/inp_file_structure.ipynb
see examples/inp_file_macros.ipynb
Run SWMM
Run SWMM with a specified executable.
from swmm_api import swmm5_run
swmm5_run('new_inputfile.inp', swmm_lib_path=r'C:\path\to\runswmm.exe')
Or run SWMM with pyswmm. This would be platform independent as pyswmm is compiled for all platforms. Additionally, you gain the advantage of a progress bar.
from swmm_api import swmm5_run
swmm5_run('new_inputfile.inp', progress_size=100)
swmm5 C:\path\to\new_inputfile.inp: 77%|███████▋ | 77/100 [00:03<00:01, 22.36it/s, 2007-02-16 08:46:27]
Read the OUT-File
from swmm_api import read_out_file
out = read_out_file('new_inputfile.out') # type: swmm_api.SwmmOut
df = out.to_frame() # type: pandas.DataFrame
see examples/out_file_reader.ipynb
Read the RPT-File
from swmm_api import read_rpt_file
rpt = read_rpt_file('new_inputfile.rpt') # type: swmm_api.SwmmReport
node_flooding_summary = rpt.node_flooding_summary # type: pandas.DataFrame
see examples/rpt_file_reader.ipynb
🗺️ GIS interactions
geopandas
and its requirements (Shapely
, pyproj
, Rtree
) must be installed! (Use python version >3.10 or conda (Anaconda|miniconda) on Windows)
from swmm_api import SwmmInput
from swmm_api.input_file.macros.gis import write_geo_package, gpkg_to_swmm
inp = SwmmInput.read_file('inputfile.inp')
coords = inp.COORDINATES.geo_series # type: geoandas.GeoSeries with points for all nodes
vertices = inp.VERTICES.geo_series # type: geoandas.GeoSeries with lines for all links
polygons = inp.POLYGONS.geo_series # type: geoandas.GeoSeries with polygons for all subcatchments
# create geopackage of all objects in inp file
write_geo_package(inp, gpkg_fn='geopackage.gpkg', driver='GPKG', label_sep='.', crs="EPSG:32633")
# read above written geopackage and convert it to inp-data
inp_new = gpkg_to_swmm('geopackage.gpkg', label_sep='.')
inp_new.write_file('new_inputfile.inp')
⚠️ Be Aware!
As python is case-sensitive this API is also case-sensitive, but SWMM is case-insensitive. This is important for the naming of the objects. For example, you could create a junction 'a' and 'A' with this API, but SWMM would only consider one and ignore the other.
AND
This package uses
utf-8
as default encoding for the file I/O (reading and writing inp, rpt and out files.) Every function to read a file has the option to define a custom encoding (for example Windows uses this as default for germanencoding='iso-8859-1'
).
But one can set a default encoding for the package using:
from swmm_api import CONFIG
CONFIG['encoding'] = 'iso-8859-1'
This documentation will be continuously extended and enhanced. If you have any question, don't hesitate to write the author and email or create an issue on GitLab or GitHub.
MORE INFORMATION COMING SOON
Cite as
Pichler, Markus. (2022). swmm_api: API for reading, manipulating and running SWMM-Projects with python (0.3). Zenodo. https://doi.org/10.5281/zenodo.7054804
Publications mentioning swmm-api
- Baumann, H., Ravn, N. H., & Schaum, A. (2022). Efficient hydrodynamic modelling of urban stormwater systems for real-time applications. Modelling, 3(4), 464–480. https://doi.org/10.3390/modelling3040030
- Farina, A., Di Nardo, A., Gargano, R., & Greco, R. (2022). Assessing the environmental impact of combined sewer overflows through a parametric study. EWaS5, 8. https://doi.org/10.3390/environsciproc2022021008
- Schilling, J., & Tränckner, J. (2022). Generate_swmm_inp: An open-source qgis plugin to import and export model input files for swmm. Water, 14(14), 2262. https://doi.org/10.3390/w14142262
- Zhang, Z., Tian, W., & Liao, Z. (2023). Towards coordinated and robust real-time control: A decentralized approach for combined sewer overflow and urban flooding reduction based on multi-agent reinforcement learning. Water Research, 229, 119498. https://doi.org/10.1016/j.watres.2022.119498
- van der Werf, J. A., Kapelan Z., Langeveld, J. G. (2023). Predictive heuristic control: inferring risks from heterogeneous nowcast accuracy. Water Sci Technol 2023; 87 (4): 1009–1028. https://doi.org/10.2166/wst.2023.027
Alternative packages
R-language:
python:
- swmmio / docs / pypi / GitHub / simular to this package but more high-level approach (= slower for specific tasks)
- GisToSWMM5 / GitHub / converting gis data to swmm model (also possible with swmm_api:
swmm_api.input_file.macro_snippets.gis_standard_import
andswmm_api.input_file.macro_snippets.gis_export
) - swmmtoolbox / GitHub / Thanks to Tim Cera for this package! I used his package to understand the .out-files but completely rewrote the reading process in this package.
- swmmnetwork / GitHub / create graph network from swmm model (see
swmm_api.input_file.macros.inp_to_graph
) - SWMMOutputAPI / GitHub / read the output file (see
swmm_api.output_file.out
) / (OpenWaterAnalytics) - swmm-pandas / pypi / equal functionalities to this package, but not feature complete
- swmmout / pypi / docs / simular to
swmmtoolbox
andSWMMOutputAPI
- swmmtonetcdf / pypi / GitHub
- hymo / GitHub Input and Report Reader (Lucas Nguyen)
- shmm / GitHub Input Reader (Lucas Nguyen)
- swmmreport / GitHub Report Reader (Lucas Nguyen)
- swmmdoodler / GitHub
Other SWMM-related packages
python:
- pyswmm / pypi / GitHub / RTC, etc. / based on swmm-toolkit (OpenWaterAnalytics)
- swmm-toolkit / pypi / GitHub / by Michael Tryby (OpenWaterAnalytics)
- SWMM5 / pypi / GitHub / simular approach to swmm-toolkit (by Assela Pathirana)
- SWMM-xsections-shape-generator / pypi / tool to generate custom shapes (by me)
- SWMM_EA / pypi / usage of genetic algorithms with SWMM (by Assela Pathirana)
- OSTRICH-SWMM / GitHub / OSTRICH optimization software toolkit with SWMM
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for swmm_api-0.4.23-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | b96160bcb84810a41641d7a215b9a670c962e15897c613ce28bb3d2f13f0a56d |
|
MD5 | 99f08b5b745ae73be5b5b1be0af606c6 |
|
BLAKE2b-256 | 42784b901bbd8863f02df5c46fb0a235da425c3c0498ee359e2ffef6a9b364ea |