199 lines
8.7 KiB
Python
199 lines
8.7 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
水力瓶颈管道综合分析
|
||
数据源：管道属性 + 实时水力 + 节点压力 → 复合评分 → 改造建议
|
||
注：realtime links 的 setting 字段为无效值，已移除所有基于 setting 的判定。
|
||
schema 适配（2026-08 实测）：realtime links 主键为 link_id，realtime nodes 主键为 node_id，
|
||
time 为 UTC 格式，--target-time 需传 UTC 时刻（如北京时间 08:00 对应 00:00+00:00）。
|
||
"""
|
||
import json, sys, math, argparse
|
||
from collections import defaultdict
|
||
|
||
VELOCITY_THRESHOLDS = {"critical": 3.0, "severe": 2.0, "high": 1.5}
|
||
|
||
def pct(d, v):
|
||
if not d: return 0
|
||
k = (v/100)*(len(d)-1); f=math.floor(k); c=math.ceil(k)
|
||
return d[f] if f==c else d[f]*(c-k)+d[c]*(k-f)
|
||
|
||
def load_json(path):
|
||
with open(path, encoding='utf-8') as f: raw = f.read()
|
||
return json.loads(raw[raw.find('{'):])
|
||
|
||
def main():
|
||
ap = argparse.ArgumentParser()
|
||
ap.add_argument('--pipe-props', required=True)
|
||
ap.add_argument('--realtime', required=True)
|
||
ap.add_argument('--node-pressures', required=True)
|
||
ap.add_argument('--target-time', required=True)
|
||
ap.add_argument('--top', type=int, default=100)
|
||
args = ap.parse_args()
|
||
TT = args.target_time
|
||
|
||
# 1. Load
|
||
print("[1/5] Loading data...", file=sys.stderr)
|
||
props = load_json(args.pipe_props).get('data', [])
|
||
pipe_map = {p['id']: p for p in props}
|
||
|
||
rt = load_json(args.realtime).get('data', [])
|
||
rt = [r for r in rt if r.get('time') == TT]
|
||
|
||
np = load_json(args.node_pressures).get('data', [])
|
||
np = [n for n in np if n.get('time') == TT]
|
||
node_pressure = {n['node_id']: n.get('pressure', n.get('value',0)) for n in np}
|
||
print(f" Pipes: {len(props)}, Realtime: {len(rt)}, Node pressures: {len(node_pressure)}", file=sys.stderr)
|
||
|
||
# 2. Merge
|
||
print("[2/5] Merging...", file=sys.stderr)
|
||
merged = []
|
||
for r in rt:
|
||
pid = r['link_id']; prop = pipe_map.get(pid)
|
||
if prop:
|
||
merged.append({**prop, **r, '_prop_id': prop['id'], '_rt_id': r['link_id']})
|
||
print(f" Merged: {len(merged)}", file=sys.stderr)
|
||
|
||
# 3. Score
|
||
print("[3/5] Scoring...", file=sys.stderr)
|
||
hl_vals = sorted([abs(m['headloss']) for m in merged])
|
||
p80 = pct(hl_vals, 80); p90 = pct(hl_vals, 90); p95 = pct(hl_vals, 95)
|
||
|
||
scored = []
|
||
for m in merged:
|
||
vel = abs(m['velocity']); hl = abs(m['headloss'])
|
||
diam = m.get('diameter', 0)
|
||
length = m.get('length', 0); roughness = m.get('roughness', 0)
|
||
n1, n2 = m['node1'], m['node2']
|
||
pid = m['id']
|
||
|
||
vs = 3 if vel>=3 else (2 if vel>=2 else (1 if vel>=1.5 else 0))
|
||
vg = "极危" if vs==3 else ("严重" if vs==2 else ("偏高" if vs==1 else "正常"))
|
||
hs = 2 if hl>p90 else (1 if hl>p80 else 0)
|
||
hg = "严重" if hs==2 else ("中度" if hs==1 else "正常")
|
||
composite = vs + hs
|
||
is_bn = (vs>=1 and hs>=1) or (vs>=2)
|
||
|
||
# Node pressures
|
||
p1 = node_pressure.get(n1); p2 = node_pressure.get(n2)
|
||
min_p = min(p1, p2) if (p1 is not None and p2 is not None) else None
|
||
|
||
# Pipe category
|
||
if diam <= 50: dcat = "微型(≤50mm)"
|
||
elif diam <= 100: dcat = "小型(51-100mm)"
|
||
elif diam <= 200: dcat = "中型(101-200mm)"
|
||
elif diam <= 400: dcat = "大型(201-400mm)"
|
||
elif diam <= 800: dcat = "主干(401-800mm)"
|
||
else: dcat = "干管(>800mm)"
|
||
|
||
scored.append({
|
||
'id': pid, 'node1': n1, 'node2': n2,
|
||
'velocity': round(vel, 4), 'flow': round(m.get('flow',0), 4),
|
||
'headloss': round(hl, 4), 'length': round(length, 4),
|
||
'diameter': int(diam), 'roughness': int(roughness),
|
||
'vel_grade': vg, 'hl_grade': hg, 'composite_score': composite,
|
||
'is_bottleneck': is_bn,
|
||
'n1_pressure': round(p1, 2) if p1 is not None else None,
|
||
'n2_pressure': round(p2, 2) if p2 is not None else None,
|
||
'min_pressure': round(min_p, 2) if min_p is not None else None,
|
||
'diam_cat': dcat,
|
||
})
|
||
|
||
# 4. Filter & sort
|
||
print("[4/5] Filtering bottlenecks...", file=sys.stderr)
|
||
bn = [s for s in scored if s['is_bottleneck']]
|
||
bn.sort(key=lambda x: (-x['composite_score'], -x['velocity']))
|
||
|
||
critical = [b for b in bn if b['vel_grade']=='极危']
|
||
severe = [b for b in bn if b['vel_grade']=='严重']
|
||
high_vel = [b for b in bn if b['vel_grade']=='偏高']
|
||
|
||
low_p = sum(1 for b in bn if b['min_pressure'] and b['min_pressure'] < 20)
|
||
lowish_p = sum(1 for b in bn if b['min_pressure'] and 20 <= b['min_pressure'] < 25)
|
||
high_rough = sum(1 for b in bn if b['roughness'] > 130)
|
||
|
||
# Diam distribution
|
||
dd = defaultdict(int)
|
||
for b in bn:
|
||
d = b['diameter']
|
||
if d <= 50: dd['≤50mm']+=1
|
||
elif d <= 100: dd['51-100mm']+=1
|
||
elif d <= 200: dd['101-200mm']+=1
|
||
elif d <= 400: dd['201-400mm']+=1
|
||
elif d <= 800: dd['401-800mm']+=1
|
||
else: dd['>800mm']+=1
|
||
|
||
# 5. Output
|
||
print("[5/5] Generating report...", file=sys.stderr)
|
||
|
||
# Print text summary to stderr
|
||
print(f"\n{'='*70}", file=sys.stderr)
|
||
print(f" 水力瓶颈分析报告 - {TT}", file=sys.stderr)
|
||
print(f"{'='*70}", file=sys.stderr)
|
||
print(f" 总管道数: {len(scored)}", file=sys.stderr)
|
||
print(f" 瓶颈管道: {len(bn)} ({len(bn)/len(scored)*100:.1f}%)", file=sys.stderr)
|
||
print(f" 极危(>3.0m/s): {len(critical)} 条", file=sys.stderr)
|
||
print(f" 严重(2.0-3.0): {len(severe)} 条", file=sys.stderr)
|
||
print(f" 偏高(1.5-2.0): {len(high_vel)} 条", file=sys.stderr)
|
||
print(f"\n 水头损失阈值: P80={p80:.4f}m P90={p90:.4f}m P95={p95:.4f}m", file=sys.stderr)
|
||
print(f" 平均={sum(hl_vals)/len(hl_vals):.4f}m 最大={max(hl_vals):.4f}m", file=sys.stderr)
|
||
print(f"\n 低压节点(<20m): {low_p} 条, 偏低(20-25m): {lowish_p} 条", file=sys.stderr)
|
||
print(f" 高粗糙度(>130): {high_rough} 条", file=sys.stderr)
|
||
|
||
print(f"\n 瓶颈管径分布:", file=sys.stderr)
|
||
for cat in ['≤50mm','51-100mm','101-200mm','201-400mm','401-800mm','>800mm']:
|
||
print(f" {cat}: {dd.get(cat,0)} 条", file=sys.stderr)
|
||
|
||
# Text table (Top 30) to stderr
|
||
print(f"\n{'='*120}", file=sys.stderr)
|
||
print(f"{'排名':<5} {'管道ID':<10} {'流速(m/s)':<10} {'水损(m)':<10} {'管径(mm)':<9} {'管长(km)':<10} {'评分':<4} {'压力1':<8} {'压力2':<8} {'管径类别':<18}", file=sys.stderr)
|
||
print('-'*120, file=sys.stderr)
|
||
for i, b in enumerate(bn[:30]):
|
||
p1s = f"{b['n1_pressure']:.1f}" if b['n1_pressure'] is not None else "-"
|
||
p2s = f"{b['n2_pressure']:.1f}" if b['n2_pressure'] is not None else "-"
|
||
print(f"{i+1:<5} {b['id']:<10} {b['velocity']:<10.4f} {b['headloss']:<10.2f} {b['diameter']:<9} {b['length']:<10.4f} {b['composite_score']:<4} {p1s:<8} {p2s:<8} {b['diam_cat']:<18}", file=sys.stderr)
|
||
|
||
# Generate suggestions for each bottleneck
|
||
for b in bn:
|
||
sug = []
|
||
if b['velocity'] > 2.0:
|
||
sug.append(f"流速{b['velocity']:.2f}m/s过高，需扩容或分流")
|
||
elif b['velocity'] > 1.5:
|
||
sug.append(f"流速{b['velocity']:.2f}m/s偏高")
|
||
hl = b['headloss']
|
||
if hl > p95:
|
||
sug.append(f"水头损失{hl:.2f}m(>P95)严重超标")
|
||
elif hl > p90:
|
||
sug.append(f"水头损失{hl:.2f}m(>P90)")
|
||
if b['diameter'] < 100:
|
||
sug.append(f"管径{b['diameter']}mm偏小，建议扩径至≥150mm")
|
||
elif b['diameter'] < 200:
|
||
sug.append(f"管径{b['diameter']}mm，评估扩容至250-300mm")
|
||
if b['roughness'] > 130:
|
||
sug.append(f"粗糙系数{b['roughness']}偏高，建议内衬修复")
|
||
if b['min_pressure'] is not None and b['min_pressure'] < 20:
|
||
sug.append(f"端节点压力{b['min_pressure']:.1f}m(<20m)，低压区域需增压")
|
||
elif b['min_pressure'] is not None and b['min_pressure'] < 25:
|
||
sug.append(f"端节点压力{b['min_pressure']:.1f}m偏低")
|
||
if b['length'] < 0.01 and b['headloss'] > 0.5:
|
||
sug.append("短管高水损，检查是否存在模型异常或局部阻塞")
|
||
b['suggestions'] = sug
|
||
|
||
result = {
|
||
'target_time': TT,
|
||
'summary': {
|
||
'total_pipes': len(scored),
|
||
'bottleneck_count': len(bn),
|
||
'critical': len(critical), 'severe': len(severe), 'high_vel': len(high_vel),
|
||
'low_pressure_nodes': low_p, 'lowish_pressure_nodes': lowish_p,
|
||
'high_roughness_pipes': high_rough,
|
||
'headloss_p80': round(p80,4), 'headloss_p90': round(p90,4),
|
||
'headloss_p95': round(p95,4),
|
||
'diameter_distribution': dict(dd),
|
||
},
|
||
'top_bottlenecks': bn[:args.top],
|
||
}
|
||
|
||
print(json.dumps(result, ensure_ascii=False, indent=2))
|
||
|
||
if __name__ == '__main__':
|
||
main()
|