wntr从零开始学习 day4

前言

本文将单独使用wntr构建一个网络模型，并且学习有关图像输出的功能，力求解决调参方面的问题。首先我们列出wntr中的所有单位，以便输入数据时的转换

一、构建模型

本次练习使用同济大学汉化手册的第二章例子，也是我们day1介绍的最基础，经典的一个示例管网。使用wntr将模型完成并且输出保存：

使用wntr穿件该模型并且完成模拟，步骤如下：

1.创建空白模型

#coding = utf-8
import wntr
import matplotlib.pyplot as plt

#创建一个空白模型
wn = wntr.network.WaterNetworkModel()

2.设置选项options

#按照例子，我们把总历时设置为72小时，模式时间步长设置为6小时
wn.options.time.duration = 72*3600
wn.options.time.pattern_timestep=6*3600

3.添加曲线和模式

#创建一个模式，对应浏览器的模式类，输入多重乘子
p1 = wntr.network.elements.Pattern('p1',[0.5,1.3,1.0,1.2],time_options=wn.options.time)
wn.add_pattern(name='pat1',pattern=p1)
#创建泵站的曲线，以便之后加入泵站中
wntr.network.elements.Curve('pump_curve1',curve_type='HEAD',points=[(0.25,35)],original_units='SI',current_units='SI')
wn.add_curve('pump_curve1',curve_type='HEAD',xy_tuples_list=[(0.25,35)])

4.画出图形的框架并添加模式和曲线

由于我们没有可交互的图形界面，所以通过大致坐标来对管网的各个结点和管段进行建模，坐标(0,0)在左下角，所以结点的其他位置可以大概确定，base_demand的单位是m^3/s,而我们的例子中的需水量单位是LPS（升每秒），通过换算要乘0.001进行转化。demand_pattern需要在指定之前进行创建和添加。

#创建连接点,demand_pattern需要在前面创建和添加到管网
wn.add_junction('n2',coordinates=(1,3),elevation=13.6,demand_pattern='pat1',base_demand=0)
wn.add_junction('n3',coordinates=(2,2),elevation=18.8,demand_pattern='pat1',base_demand=0.0146)
wn.add_junction('n4',coordinates=(2,0),elevation=18.3,demand_pattern='pat1',base_demand=0.0351)
wn.add_junction('n5',coordinates=(5,2),elevation=19.1,demand_pattern='pat1',base_demand=0.0512)
wn.add_junction('n6',coordinates=(5,0),elevation=17.3,demand_pattern='pat1',base_demand=0.0823)
wn.add_junction('n7',coordinates=(8,2),elevation=22.0,demand_pattern='pat1',base_demand=0.0408)
#创建水源
wn.add_reservoir('r1',coordinates=(0,4),base_head=13.6)
#创建高位水箱 注意单位
wn.add_tank('t1',init_level=0.3,min_level=0,max_level=4.5,diameter=12,elevation=32.2,coordinates=(10,4))
#创建管段，注意单位，默认海曾威廉公式的系数是100，直径diameter的单位是m，也要进行换算
wn.add_pipe('1','n2','n3',320,0.4)
wn.add_pipe('2','n3','n5',650,0.3)
wn.add_pipe('3','n3','n4',330,0.3)
wn.add_pipe('4','n4','n6',590,0.3)
wn.add_pipe('5','n5','n6',350,0.2)
wn.add_pipe('6','n5','n7',550,0.2)
wn.add_pipe('7','n7','t1',270,0.3)
wn.add_pipe('8','n6','n7',660,0.2)
#添加泵站管段
wn.add_pump('9',start_node_name='r1',end_node_name='n2',pump_type='HEAD',pump_parameter='pump_curve1')

5.模拟运行并且输出为.inp文件

#输出文件
wn.write_inpfile('filename.inp', version=2.2)
#进行模拟
sim = wntr.sim.EpanetSimulator(wn)
results = sim.run_sim()

报错，sad，输出到epanet2.2查看，原来是添加的需水量模式叫p1,但是我们定义的pattren叫pat1

结果是在wntr中水源的参数少设置了一个

wn.add_reservoir('r1',coordinates=(0,4),base_head=13.6,head_pattern='p1')

最后附上该模型的完全代码，基本调试了一天，包括看文档和官方文献，实属不易：

#coding = utf-8
import wntr
import matplotlib.pyplot as plt

#创建空白模型
wn = wntr.network.WaterNetworkModel()

#创建模式，并且添加到模型中
wn.options.time.duration = 48*3600
wn.options.time.pattern_timestep=6*3600
print(wn.options.hydraulic.demand_model)#查看使用demand-driven analysis (DDA) 还是 pressure driven analysis (PDA)

#添加需水量模式，注意模式的命名和导入的命名
pat1 = wntr.network.elements.Pattern('p1',[0.7,1.2,1.1,1.3],time_options=wn.options.time)
wn.add_pattern(name='p1',pattern=pat1)

#创建曲线，并添加到管网中
wntr.network.elements.Curve('pump_curve1',curve_type='HEAD',points=[(0.250,35)],original_units='SI',current_units='SI')
wn.add_curve('pump_curve1',curve_type='HEAD',xy_tuples_list=[(0.250,35)])

#创建模型
wn.add_junction('n2',coordinates=(1,3),elevation=13.6,demand_pattern='p1',demand_category=None,base_demand=0)
wn.add_junction('n3',coordinates=(2,2),elevation=18.8,demand_pattern='p1',demand_category=None,base_demand=0.0146)
wn.add_junction('n4',coordinates=(2,0),elevation=18.3,demand_pattern='p1',demand_category=None,base_demand=0.0351)
wn.add_junction('n5',coordinates=(5,2),elevation=19.1,demand_pattern='p1',demand_category=None,base_demand=0.0512)
wn.add_junction('n6',coordinates=(5,0),elevation=17.3,demand_pattern='p1',demand_category=None,base_demand=0.0823)
wn.add_junction('n7',coordinates=(8,2),elevation=22.0,demand_pattern='p1',demand_category=None,base_demand=0.0408)

wn.add_reservoir('r1',coordinates=(0,4),base_head=13.6,head_pattern='p1')

wn.add_tank('t1',init_level=0.3,min_level=0,max_level=4.5,diameter=12,elevation=32.2,coordinates=(10,4))

wn.add_pipe('1','n2','n3',320,0.4)
wn.add_pipe('2','n3','n5',650,0.3)
wn.add_pipe('3','n3','n4',330,0.3)
wn.add_pipe('4','n4','n6',590,0.3)
wn.add_pipe('5','n5','n6',350,0.2)
wn.add_pipe('6','n5','n7',550,0.2)
wn.add_pipe('7','n7','t1',270,0.3)
wn.add_pipe('8','n6','n7',660,0.2)

wn.add_pump('9',start_node_name='r1',end_node_name='n2',pump_type='HEAD',pump_parameter='pump_curve1')

二、结果进行分析

1.图像检查

首先我们看看模型的泵站曲线有没有搞错

pump = wn.get_link('9')
ax = wntr.graphics.plot_pump_curve(pump)
plt.show()

应该泵站曲线没有问题，单位正确

其次画出管网的示意图，看看和我们的例子区别：

坐标基本正确，管段连接和编号完全一致。

2.结果分析

首先我们查看模拟是采用什么需求分析（模拟没传其他参数的情况下，默认是采用需求驱动分析demand-driven (DD/DDA)）

print(wn.options.hydraulic.demand_model)

结果输出DDA，是需求驱动分析，我们可以通过改变参数来换其他模型查看结果

>>> wn.options.hydraulic.demand_model = 'DD'
>>> wn.options.hydraulic.demand_model = 'PDD'

同时也可以换不同的求解器看看结果是什么样的

sim = wntr.sim.WNTRSimulator(wn)
results = sim.run_sim()

---

本次就采用需求驱动模型，使用epanet2.2求解器，下次再研究wntr求解器：

画出节点n5的压力时间序列

pressure_at_noden5 = results.node['pressure'].loc[:,'n5']
ax = pressure_at_noden5.plot()
text = ax.set_xlabel("Time (s)")
text = ax.set_ylabel("Pressure (m)")
plt.show()

终于成功了！！！！！

几个参数调一天，希望我的文章对你有所帮助，并且如果有需要交流wntr和epanet的使用方法的朋友可以留言。