import math

# 沿程压力损失计算
def darcy_loss(length, diameter, velocity, roughness_coefficient):
    # 计算雷诺数
    reynolds_number = velocity * diameter / 0.0000114  # 假设运动粘度为 0.0000114 m²/s
    # 计算沿程阻力系数（这里为简化计算，假设一个固定值）
    if reynolds_number < 2000:
        friction_factor = 64 / reynolds_number
    else:
        friction_factor = 0.04
    return friction_factor * length / diameter * velocity ** 2 / (2 * 9.81)

# 局部压力损失计算
def local_loss(local_resistance_coefficient, velocity):
    return local_resistance_coefficient * velocity ** 2 / (2 * 9.81)

# 测试
pipe_length = 100  # 管道长度，单位：m
pipe_diameter = 0.5  # 管道内径，单位：m
flow_velocity = 2  # 流速，单位：m/s
roughness = 0.01  # 粗糙度系数
local_resistance_coeff = 0.5  # 局部阻力系数

total_head_loss = darcy_loss(pipe_length, pipe_diameter, flow_velocity, roughness) + local_loss(
    local_resistance_coeff, flow_velocity
)
print(f"总压力损失为：{total_head_loss}米水柱")