实现与SWMM数据模型快速交互pyswmm快速入门。

 一、SWMM模型执行模拟

1、简单的执行模型，不进行任何参数调整，与SWMM模型交互。

from pyswmm import Simulation

sim = Simulation('./test.inp')
sim.execute()

2、设置模型模拟时间步长函数为step_advance，可以输出模型模拟时间步长。

from pyswmm import Simulation

with Simulation('test.inp') as sim:
    sim.step_advance(300)
    for step in sim:
            print(sim.current_time)
            # or here! sim.step_advance(newvalue)

2015-11-01 14:05:00
2015-11-01 14:10:00
2015-11-01 14:15:00
2015-11-01 14:20:00

二、模型节点交互

初始化节点数据，函数为Nodes。实例化节点数据。

from pyswmm import Simulation, Nodes

with Simulation('./test.inp') as sim:
    node_object = Nodes(sim)

    #J1 node instantiation
    J1 = node_object["J1"]
    print(J1.invert_elevation)
    print(J1.is_junction())

    #Step through a simulation
    for step in sim:
            print(J1.total_inflow)

三、模型管线交互

初始化管线数据，函数为Links。

from pyswmm import Simulation, Links

with Simulation('./test.inp') as sim:
    link_object = Links(sim)

    #C1:C2 link instantiation
    c1c2 = link_object["C1:C2"]
    print(c1c2.flow_limit)
    print(c1c2.is_conduit())

    #Step through a simulation
    for step in sim:
            print(c1c2.flow)
            if c1c2.flow > 10.0:
                    c1c2.target_setting = 0.5

四、子汇水区模型交互

与子集水区交互，函数为Subcatchments。

from pyswmm import Simulation, Subcatchments

with Simulation('./test.inp') as sim:
    subcatch_object = Subcatchments(sim)

    #SC1 subcatchment instantiation
    SC1 = subcatch_object["S1"]
    print(SC1.area)

    #Step through a simulation
    for step in sim:
            print(SC1.runoff)

五、pyswmm结合SWMM实现更复杂的模型调控

比较两个模型节点高程差

from pyswmm import Simulation, Links, Nodes

def TestDepth(node, node2):
    if node > node2:
        return True
    else:
        return False

with Simulation('./testmodel.inp') as sim:
    link_object = Links(sim)

    #C1:C2 link instantiation
    c1c2 = link_object["C1:C2"]

    node_object = Nodes(sim)
    #J1 node instantiation
    J1 = node_object["J1"]
    #J2 node instantiation
    J2 = node_object["J2"]

    #Step through a simulation
    for step in sim:
        if TestDepth(J1.depth, J2.depth):
            c1c2.target_setting = 0.5

pyswmm的操作有较高的优先级，pyswmm 更新的任何模型调控都会输出到 *.rpt 文件中。

六、设置节点流入流量

设置节点的流入流量。例如径流或季节性的流量。流量单位为模型中设置单位。

from pyswmm import Simulation, Nodes

with Simulation('/testmodel.inp') as sim:
    j1 = Nodes(sim)["J1"]
    for step in sim:
        j1.generated_inflow(9)

七、SWMM 输出二进制文件转换

初始化pyswmm.output。就可以使用预定义的方法访问SWMM输出的二进制文件中的数据。

1、下面的示例打开SWMM输出二进制文件，输出识子集、节点和链接的数量，以及用于生成二进制文件的SWMM引擎。

from pyswmm import Output

with Output('tests/data/model_full_features.out') as out:
    print(len(out.subcatchments))
    print(len(out.nodes))
    print(len(out.links))
    print(out.version)

2、根据SWMM输出的二进制文件输出节点J1的整个深度时间序列。

from pyswmm import Output, SubcatchSeries, NodeSeries, LinkSeries, SystemSeries

with Output('model.out') as out:
    print(len(out.subcatchments))
    print(len(out.nodes))
    print(len(out.links))
    print(out.version)

    sub_ts = SubcatchSeries(out)['S1'].runoff_rate
    node_ts = NodeSeries(out)['J1'].invert_depth
    link_ts = LinkSeries(out)['C2'].flow_rate
    sys_ts = SystemSeries(out).rainfall

八、低影响开发（海绵城市）

1、低影响开发模型调参

所有LID模型参数可在模型仿真前设置。模型仿真时只能设置部分LID模型参数。

from pyswmm import Simulation, LidControls

with Simulation('/testmodel.inp') as sim:
    rain_barrel = LidControls(sim)["rain_barrel"]
    print(rain_barrel.drain.coefficient)
    rain_barrel.drain.coefficient = 0.60
    print(rain_barrel.drain.coefficient)

2、LID设施模块

导入LidGroups函数后，可以在概化模块上定义的LID单元进行交互。可以使用LidGroups遍历LID单元列表。

from pyswmm import Simulation, LidGroups

with Simulation('/testmodel.inp') as sim:
    lid_on_sub = LidGroups(sim)["subcatch_id"]
    for lid in lid_on_sub:
        print(lid)
    print(lid_on_sub[0])
    for step in sim:
        print(lid_on_sub.old_drain_flow)

3、LID单元

from pyswmm import Simulation, LidGroups

with Simulation('/testmodel.inp') as sim:
    lid_on_sub = LidGroups(sim)["subcatch_id"]
    for lid in lid_on_sub:
        print(lid)
    print(lid_on_sub[0])
    for step in sim:
        print(lid_on_sub.water_balance.inflow)
        print(lid_on_sub.water_balance.evaporation)

LID模块参数：

[LID_CONTROLS]
;;Name           Type_Layer Parameters
;;-------------- ---------- ----------
LID_C1  IT
LID_C1  SURFACE    0.0    0.0    0.03   1.0    5
LID_C1  STORAGE    800    0.3    3.6    0      NO
LID_C1  DRAIN      0.5    1.41   400    6      0       0

pyswmm库更多调用方法参见：pyswmm

更多水务建模教程、海绵城市及给排水专业规范等相关知识内容请关注公众号“给排水科学与技术”。