python wrappers for the Stormwater Management Model (SWMM5)
Getting started
-
Poject Website: www.pyswmm.org
We provide support on Stack Overflow or GitHub Discussions to answer support questions related to PySWMM.
Cite our Paper
McDonnell, Bryant E., Ratliff, Katherine M., Tryby, Michael E., Wu, Jennifer Jia Xin, & Mullapudi, Abhiram. (2020). PySWMM: The Python Interface to Stormwater Management Model (SWMM). Journal of Open Source Software, 5(52), 2292, https://doi.org/10.21105/joss.02292
YouTube Training Vidoes
Setting a manhole inflow during a running simulation!
Overview
PySWMM is a Python language software package for the creation, manipulation, and study of the structure, dynamics, and function of complex networks.
With PySWMM you can load and manipulate USEPA Stormwater Management Models. With the development of PySWMM, control algorithms can now be developed exclusively in Python which allows the use of functions and objects as well as storing and tracking hydraulic trends for control actions.
As of version v1.1.0, PySWMM includes new features to process metadata and timeseries stored in SWMM binary output file.
Who uses PySWMM?
PySWMM is used by engineers, modelers, and researchers who want to streamline stormwater modeling optimization, controls, and post-processing results.
Goals
PySWMM is intended to provide
- tools for the study of the structure and dynamics within USEPA SWMM5,
- a standard programming interface and graph implementation that is suitable for many applications,
- a rapid development environment for collaborative, multidisciplinary projects,
- an interface to USEPA SWMM5,
- development and implementation of control logic outside of native EPA-SWMM Controls,
- methods for users to establish their own node inflows,
- a coding interface to binary output files,
- new modeling possibilities for the SWMM5 Community.
Install
Get the latest version of PySWMM from PyPI See the Quick Guide!
$ pip install pyswmm
As of version 1.3.1, pyswmm can be installed with specific versions of the SWMM engine ranging from 5.1.14 to 5.2.3 using pip extras:
$ pip install pyswmm[swmm5.2.1]
SWMM and Python Compatibility Table
pyswmm version | swmm-toolkit compatibility (SWMM engine) |
python compatibility |
---|---|---|
1.0.0 - 1.1.1 | 0.8.2 (SWMM 5.1.13) | 3.6 - 3.9 |
1.2.0 - 1.3.1 | 0.9.1 - 0.14.0 (SWMM 5.1.14 - 5.2.3) Note: 0.11.0 (SWMM 5.2.0) only supported on windows |
3.7 - 3.11 |
Usage
A quick example that steps through a simulation:
Examples:
See the Latte Example
from pyswmm import Simulation, Nodes, Links
with Simulation(r'Example1.inp') as sim:
Node21 = Nodes(sim)["21"]
print("Invert Elevation: {}". format(Node21.invert_elevation))
Link15 = Links(sim)['15']
print("Outlet Node ID: {}".format(Link15.outlet_node))
# Launch a simulation!
for ind, step in enumerate(sim):
if ind % 100 == 0:
print(sim.current_time,",",round(sim.percent_complete*100),"%",\
Node21.depth, Link15.flow)
Opening a SWMM binary output file and accessing model metadata and timeseries.
from swmm.toolkit.shared_enum import SubcatchAttribute, NodeAttribute, LinkAttribute
from pyswmm import Output
with Output('model.out') as out:
print(len(out.subcatchments))
print(len(out.nodes))
print(len(out.links))
print(out.version)
sub_ts = out.subcatch_series('S1', SubcatchAttribute.RUNOFF_RATE)
node_ts = out.node_series('J1', NodeAttribute.INVERT_DEPTH)
link_ts = out.link_series('C2', LinkAttribute.FLOW_RATE)
Bugs
Our issue tracker is at https://github.com/OpenWaterAnalytics/pyswmm/issues. Please report any bugs that you find. Or, even better, fork the repository on GitHub and create a pull request. All changes are welcome, big or small, and we will help you make the pull request if you are new to git (just ask on the issue).
Contributing
Please check out our Wiki https://github.com/OpenWaterAnalytics/pyswmm/wiki for more information on contributing, including an Author Contribution Checklist.
License
Distributed with a BSD2 license; see LICENSE.txt:
Copyright (C) 2014-2023 PySWMM Developers
Community-Owned See AUTHORS and CITATION.cff
Acknowledgements
- Assela Pathirana