diff --git a/simulation.py b/simulation.py index d11c9aa..9b8e8b7 100644 --- a/simulation.py +++ b/simulation.py @@ -1,7 +1,7 @@ import numpy as np from tjnetwork import * from api.s36_wda_cal import * -# from get_real_status import * +from get_real_status import * from datetime import datetime,timedelta from math import modf import json @@ -17,135 +17,6 @@ import psycopg import logging import globals -# 数据接口 -# url_path = 'http://10.101.15.16:9000/loong' # 内网 -# url_path = 'http://183.64.62.100:9057/loong' # 外网 -# url_real = url_path + '/api/mpoints/realValue' -# url_hist = url_path + '/api/curves/data' - -# # 实时数据的设备编号 -# DN_900_ID='2498' -# DN_500_ID='3854' -# DN_1000_ID='3853' -# H_RESSURE='2510' -# L_PRESURE='2514' -# H_TANK='4780' -# L_TANK='4854' -# -# # inp文件数据信息 -PATTERN_TIME_STEP = 15.0 - -# regions -# regions = ['hp', 'lp'] -# regions_demand_patterns = {'hp': ['DN900', 'DN500'], 'lp': ['DN1000']} # 出厂水量近似表示用水量 -# # regions_patterns = {'hp': ['ChuanYiJiXiao', 'BeiQuanHuaYuan', 'ZhuangYuanFuDi', 'JingNingJiaYuan', -# # '308', 'JiaYinYuan', 'XinChengGuoJi', 'YiJingBeiChen', 'ZhongYangXinDu', -# # 'XinHaiJiaYuan', 'DongFengJie', 'DingYaXinYu', 'ZiYunTai', 'XieMaGuangChang', -# # 'YongJinFu', 'BianDianZhan', 'BeiNanDaDao', 'TianShengLiJie', 'XueYuanXiaoQu', -# # 'YunHuaLu', 'GaoJiaQiao', 'LuZuoFuLuXiaDuan', 'TianRunCheng', 'CaoJiaBa', -# # 'PuLingChang', 'QiLongXiaoQu', 'TuanXiao', -# # 'TuanShanBaoZhongShiHua', 'XieMa', 'BeiWenQuanJiuHaoErQi', 'LaiYinHuSiQi', -# # 'DN500', 'DN900'], -# # 'lp': ['PanXiMingDu', 'WanKeJinYuHuaFuGaoCeng', 'KeJiXiao', -# # 'LuGouQiao', 'LongJiangHuaYuan', 'LaoQiZhongDui', 'ShiYanCun', 'TianQiDaSha', -# # 'TianShengPaiChuSuo', 'TianShengShangPin', 'JiaoTang', 'RenMinHuaYuan', -# # 'TaiJiBinJiangYiQi', 'TianQiHuaYuan', 'TaiJiBinJiangErQi', '122Zhong', -# # 'WanKeJinYuHuaFuYangFang', 'ChengBeiCaiShiKou', 'WenXingShe', 'YueLiangTianBBGJCZ', -# # 'YueLiangTian', 'YueLiangTian200', 'ChengTaoChang', 'HuoCheZhan', 'LiangKu', 'QunXingLu', -# # 'JiuYuanErTongYiYuan', 'TangDouHua', 'TaiJiBinJiangErQi(SanJi)', -# # 'ZhangDouHua', 'JinYunXiaoQuDN400', -# # 'DN1000']} -# -# # nodes -# monitor_single_patterns = ['ChuanYiJiXiao', 'BeiQuanHuaYuan', 'ZhuangYuanFuDi', 'JingNingJiaYuan', -# '308', 'JiaYinYuan', 'XinChengGuoJi', 'YiJingBeiChen', 'ZhongYangXinDu', -# 'XinHaiJiaYuan', 'DongFengJie', 'DingYaXinYu', 'ZiYunTai', 'XieMaGuangChang', -# 'YongJinFu', 'PanXiMingDu', 'WanKeJinYuHuaFuGaoCeng', 'KeJiXiao', -# 'LuGouQiao', 'LongJiangHuaYuan', 'LaoQiZhongDui', 'ShiYanCun', 'TianQiDaSha', -# 'TianShengPaiChuSuo', 'TianShengShangPin', 'JiaoTang', 'RenMinHuaYuan', -# 'TaiJiBinJiangYiQi', 'TianQiHuaYuan', 'TaiJiBinJiangErQi', '122Zhong', -# 'WanKeJinYuHuaFuYangFang'] -# monitor_single_patterns_id = {'ChuanYiJiXiao': '7338', 'BeiQuanHuaYuan': '7315', 'ZhuangYuanFuDi': '7316', -# 'JingNingJiaYuan': '7528', '308': '8272', 'JiaYinYuan': '7304', -# 'XinChengGuoJi': '7325', 'YiJingBeiChen': '7328', 'ZhongYangXinDu': '7329', -# 'XinHaiJiaYuan': '9138', 'DongFengJie': '7302', 'DingYaXinYu': '7331', -# 'ZiYunTai': '7420,9059', 'XieMaGuangChang': '7326', 'YongJinFu': '9059', -# 'PanXiMingDu': '7320', 'WanKeJinYuHuaFuGaoCeng': '7419', -# 'KeJiXiao': '7305', 'LuGouQiao': '7306', 'LongJiangHuaYuan': '7318', -# 'LaoQiZhongDui': '9075', 'ShiYanCun': '7309', 'TianQiDaSha': '7323', -# 'TianShengPaiChuSuo': '7335', 'TianShengShangPin': '7324', 'JiaoTang': '7332', -# 'RenMinHuaYuan': '7322', 'TaiJiBinJiangYiQi': '7333', 'TianQiHuaYuan': '8235', -# 'TaiJiBinJiangErQi': '7334', '122Zhong': '7314', 'WanKeJinYuHuaFuYangFang': '7418'} -# -# monitor_unity_patterns = ['BianDianZhan', 'BeiNanDaDao', 'TianShengLiJie', 'XueYuanXiaoQu', -# 'YunHuaLu', 'GaoJiaQiao', 'LuZuoFuLuXiaDuan', 'TianRunCheng', -# 'CaoJiaBa', 'PuLingChang', 'QiLongXiaoQu', 'TuanXiao', -# 'ChengBeiCaiShiKou', 'WenXingShe', 'YueLiangTianBBGJCZ', -# 'YueLiangTian', 'YueLiangTian200', -# 'ChengTaoChang', 'HuoCheZhan', 'LiangKu', 'QunXingLu', -# 'TuanShanBaoZhongShiHua', 'XieMa', 'BeiWenQuanJiuHaoErQi', 'LaiYinHuSiQi', -# 'JiuYuanErTongYiYuan', 'TangDouHua', 'TaiJiBinJiangErQi(SanJi)', -# 'ZhangDouHua', '', -# 'DN500', 'DN900', 'DN1000'] -# monitor_unity_patterns_id = {'BianDianZhan': '7339', 'BeiNanDaDao': '7319', 'TianShengLiJie': '8242', -# 'XueYuanXiaoQu': '7327', 'YunHuaLu': '7312', 'GaoJiaQiao': '7340', -# 'LuZuoFuLuXiaDuan': '7343', 'TianRunCheng': '7310', 'CaoJiaBa': '7300', -# 'PuLingChang': '7307', 'QiLongXiaoQu': '7321', 'TuanXiao': '8963', -# 'ChengBeiCaiShiKou': '7330', 'WenXingShe': '7311', -# 'YueLiangTianBBGJCZ': '7313', 'YueLiangTian': '7313', 'YueLiangTian200': '7313', -# 'ChengTaoChang': '7301', 'HuoCheZhan': '7303', -# 'LiangKu': '7296', 'QunXingLu': '7308', -# 'DN500': '3854', 'DN900': '2498', 'DN1000': '3853'} -# monitor_patterns = monitor_single_patterns + monitor_unity_patterns -# monitor_patterns_id = {**monitor_single_patterns_id, **monitor_unity_patterns_id} - -# flow -# hp_flow_pattern_id = {'DN900': '2498', 'DN500': '3854'} -# lp_flow_pattern_id = {'DN1000': '3853'} -# -# # pumps -# pump_pattern_ids = ['1#', '2#', '3#', '4#', '5#', '6#', '7#'] -# pumps = ['PU00000', 'PU00001', 'PU00002', 'PU00003', 'PU00004', 'PU00005', 'PU00006'] -# variable_frequency_pumps = ['PU00004', 'PU00005', 'PU00006'] -# fixed_pumps_id = {'PU00000': '2747', 'PU00001': '2776', 'PU00002': '2730', 'PU00003': '2787'} -# variable_pumps_id = {'PU00004': '2500', 'PU00005': '2502', 'PU00006': '2504'} -# -# # reservoirs -# # reservoirs = ['ZBBDJSCP000002', 'R00003'] -# # reservoirs_id = {'ZBBDJSCP000002': '2497', 'R00003': '2571'} -# # tanks -# tanks = ['ZBBDTJSC000002', 'ZBBDTJSC000001'] -# tanks_id = {'ZBBDTJSC000002': '4780', 'ZBBDTJSC000001': '9774'} -# -# # 用于更改数据的SCADA设的ID -# change_data_device_ids = ['2498', '3854', '3853', '2497', '2571', '4780', '9774', -# '2747', '2776', '2730', '2787', '2500', '2502', '2504'] - -# # 实时数据类：element_id和api_query_id对应 -# reservoirs_id = {} -# tanks_id = {} -# fixed_pumps_id ={} -# variable_pumps_id = {} -# pressure_id = {} -# demand_id = {} -# quality_id = {} -# -# # 实时数据类：pattern_id和api_query_id对应 -# source_outflow_pattern_id = {} -# realtime_pipe_flow_pattern_id = {} -# pipe_flow_region_patterns = {} # 根据realtime的pipe_flow，对non_realtime的demand进行分区 -# -# # 分区查询 -# source_outflow_region = {} # 以绑定的管段作为value -# source_outflow_region_id = {} # 以api_query_id作为value -# source_outflow_region_patterns = {} # 以associated_pattern作为value -# # 非实时数据的pattern -# non_realtime_region_patterns = {} # 基于source_outflow_region进行区分 -# -# realtime_region_pipe_flow_and_demand_id = {} # 基于source_outflow_region搜索该分区中的实时pipe_flow和demand的api_query_id，后续用region的流量 - 实时流量计的流量 -# realtime_region_pipe_flow_and_demand_patterns = {} # 基于source_outflow_region搜索该分区中的实时pipe_flow和demand的associated_pattern，后续用region的流量 - 实时流量计的流量 - - logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s') @@ -497,6 +368,46 @@ def query_pipe_flow_region_patterns(name: str, column_prefix: str = 'associated_ return globals.pipe_flow_region_patterns + +# 2025/02/15 +def query_SCADA_ID_corresponding_info(name: str, SCADA_ID: str) -> dict: + """ + 在地图上拾取SCADA元素后，获取SCADA_ID，在pg数据库中查询该SCADA设备对应的associated_element_id和api_query_id + :param name: pg数据库的名称 + :param SCADA_ID: SCADA设备的ID + :return: 包含associated_element_id和api_query_id的字典 + """ + conn_string = f"dbname={name} host=127.0.0.1" + try: + # 使用 psycopg.connect 创建连接 + with psycopg.connect(conn_string) as conn: + with conn.cursor() as cur: + # 执行查询 + query = """ + SELECT associated_element_id, API_query_id + FROM scada_info + WHERE id = %s + """ + cur.execute(query, (SCADA_ID,)) # 执行查询并传递参数 + + # 获取查询结果 + result = cur.fetchone() + + if result: + # 将结果转换为字典 + associated_info = { + 'associated_element_id': result[0], + 'API_query_id': result[1] + } + return associated_info + else: + # 如果没有找到记录 + return None + except Exception as e: + print(f"An error occurred: {e}") + return None + + # 2025/01/11 def get_source_outflow_region_id(name: str, source_outflow_region: dict, column_prefix: str = 'associated_source_outflow_id') -> dict: @@ -626,7 +537,7 @@ def get_pattern_index(cur_datetime: str) -> int: dt = datetime.strptime(cur_datetime, str_format) hr = dt.hour mnt = dt.minute - i = int((hr * 60 + mnt) / PATTERN_TIME_STEP) + i = int((hr * 60 + mnt) / globals.PATTERN_TIME_STEP) return i @@ -637,7 +548,7 @@ def get_pattern_index_str(current_time: str) -> str: :return: str， 以“HH:MM:00”格式返回 """ i = get_pattern_index(current_time) - [minN, hrN] = modf(i * PATTERN_TIME_STEP / 60) + [minN, hrN] = modf(i * globals.PATTERN_TIME_STEP / 60) minN_str = str(int(minN * 60)) minN_str = minN_str.zfill(2) hrN_str = str(int(hrN)) @@ -686,18 +597,23 @@ def get_history_pattern_info(project_name, pattern_name): def run_simulation(name: str, simulation_type: str, modify_pattern_start_time: str, modify_total_duration: int = 0, modify_reservoir_head_pattern: dict[str, list] = None, modify_tank_initial_level: dict[str, float] = None, modify_junction_base_demand: dict[str, float] = None, modify_junction_damand_pattern: dict[str, list] = None, - modify_pump_pattern: dict[str, list] = None): + modify_fixed_pump_pattern: dict[str, list] = None, modify_variable_pump_pattern: dict[str, list] = None, + modify_valve_opening: dict[str, float] = None, scheme_Type: str = None, scheme_Name: str = None) -> None: """ 传入需要修改的参数，改变数据库中对应位置的值，然后计算，返回结果 :param name: 模型名称，数据库中对应的名字 :param simulation_type: 模拟的类型，realtime为实时模拟，修改原数据库；extended为多步长模拟，需要复制数据库 :param modify_pattern_start_time: 模拟开始时间,格式为'2024-11-25T09:00:00+08:00' - :param modify_total_duration: 模拟总历时 + :param modify_total_duration: 模拟总历时，秒 :param modify_reservoir_head_pattern: dict中包含多个水库模式，str为水库head_pattern的id，list为修改后的head_pattern :param modify_tank_initial_level: dict中包含多个水塔，str为水塔的id，float为修改后的initial_level :param modify_junction_base_demand: dict中包含多个节点，str为节点的id，float为修改后的base_demand :param modify_junction_damand_pattern: dict中包含多个节点模式，str为节点demand_pattern的id，list为修改后的demand_pattern - :param modify_pump_pattern: dict中包含多个水泵模式，str为水泵pattern的id，list为修改后的pattern + :param modify_fixed_pump_pattern: dict中包含多个水泵模式，str为工频水泵的id，list为修改后的pattern + :param modify_variable_pump_pattern: dict中包含多个水泵模式，str为变频水泵的id，list为修改后的pattern + :param modify_valve_opening: dict中包含多个阀门开启度，str为阀门的id，float为修改后的阀门开启度 + :param scheme_Type: 模拟方案类型 + :param scheme_Name：模拟方案名称 :return: """ # 记录开始时间 @@ -709,19 +625,27 @@ def run_simulation(name: str, simulation_type: str, modify_pattern_start_time: s if is_project_open(name): close_project(name) # 判断是实时模拟还是多步长模拟 - if simulation_type.upper() == 'REALTIME': # 实时模拟(修改原数据库) - name_c = name - elif simulation_type.upper() == 'EXTENDED': # 扩展模拟(复制数据库) - name_c = '_'.join([name, 'c']) - if have_project(name_c): - if is_project_open(name_c): - close_project(name_c) - delete_project(name_c) - copy_project(name, name_c) # 备份项目 - else: - raise Exception('Incorrect simulation type, choose in (realtime, extended)') + # if simulation_type.upper() == 'REALTIME': # 实时模拟(修改原数据库) + # name_c = name + # elif simulation_type.upper() == 'EXTENDED': # 扩展模拟(复制数据库) + # name_c = '_'.join([name, 'c']) + # if have_project(name_c): + # if is_project_open(name_c): + # close_project(name_c) + # delete_project(name_c) + # copy_project(name, name_c) # 备份项目 + # else: + # raise Exception('Incorrect simulation type, choose in (realtime, extended)') + name_c = name # 打开数据库 open_project(name_c) + dic_time = get_time(name_c) + # 获取水力模拟步长，如’0:15:00‘ + globals.hydraulic_timestep = dic_time['HYDRAULIC TIMESTEP'] + # 将时间字符串转换为 timedelta 对象 + time_obj = datetime.strptime(globals.hydraulic_timestep, '%H:%M:%S') + # 转换为分钟浮点数 + globals.PATTERN_TIME_STEP = float(time_obj.hour * 60 + time_obj.minute + time_obj.second / 60) # 对输入的时间参数进行处理 pattern_start_time = convert_time_format(modify_pattern_start_time) @@ -746,10 +670,13 @@ def run_simulation(name: str, simulation_type: str, modify_pattern_start_time: s dic_time = get_time(name_c) dic_time['PATTERN START'] = str_pattern_start dic_time['DURATION'] = from_seconds_to_clock(modify_total_duration) + if simulation_type.upper() == 'REALTIME': + dic_time['DURATION'] = 0 cs = ChangeSet() cs.operations.append(dic_time) set_time(name_c, cs) + # 根据SCADA实时数据进行修改，如果没有对应的SCADA数据，如未来的时间点，则不改变pg数据库的数据 if globals.reservoirs_id: # reservoirs_id = {'ZBBDJSCP000002': '2497', 'R00003': '2571'} # 1.获取reservoir的SCADA数据,形式如{'2497': '3.1231', '2571': '2.7387'} @@ -928,19 +855,24 @@ def run_simulation(name: str, simulation_type: str, modify_pattern_start_time: s region_realtime_region_pipe_flow_and_demand_data_dict = influxdb_api.query_SCADA_data_by_device_ID_and_time( query_ids_list=temp_realtime_region_pipe_flow_and_demand_id, query_time=modify_pattern_start_time) - region_total_source_outflow = sum(float(value) for value in region_source_outflow_data_dict.values()) - history_region_total_source_outflow = 0 - for source_outflow_pattern_name in temp_source_outflow_region_patterns: - temp_history_source_outflow_list = get_history_pattern_info(name_c, source_outflow_pattern_name) - history_region_total_source_outflow += temp_history_source_outflow_list[modify_index] + # 2025/02/12 确保 region_source_outflow_data_dict 和 + # region_realtime_region_pipe_flow_and_demand_data_dict中的每个值都不是 None 且不为 0 + region_source_outflow_valid_values = [float(value) for value in region_source_outflow_data_dict.values() if value not in [None, 0]] + valid_values = [float(value) for value in region_realtime_region_pipe_flow_and_demand_data_dict.values() if value not in [None, 0]] + # 如果都非空，则执行 sum 操作 + if region_source_outflow_valid_values and valid_values: + region_total_source_outflow = sum(region_source_outflow_valid_values) + history_region_total_source_outflow = 0 + for source_outflow_pattern_name in temp_source_outflow_region_patterns: + temp_history_source_outflow_list = get_history_pattern_info(name_c, source_outflow_pattern_name) + history_region_total_source_outflow += temp_history_source_outflow_list[modify_index] - region_total_realtime_region_pipe_flow_and_demand = sum(float(value) for value in region_realtime_region_pipe_flow_and_demand_data_dict.values()) - history_region_total_realtime_region_pipe_flow_and_demand = 0 - for pipe_flow_and_demand_pattern_name in temp_realtime_region_pipe_flow_and_demand_patterns: - temp_history_pipe_flow_and_demand_list = get_history_pattern_info(name_c, pipe_flow_and_demand_pattern_name) - history_region_total_realtime_region_pipe_flow_and_demand += temp_history_pipe_flow_and_demand_list[modify_index] + region_total_realtime_region_pipe_flow_and_demand = sum(valid_values) + history_region_total_realtime_region_pipe_flow_and_demand = 0 + for pipe_flow_and_demand_pattern_name in temp_realtime_region_pipe_flow_and_demand_patterns: + temp_history_pipe_flow_and_demand_list = get_history_pattern_info(name_c, pipe_flow_and_demand_pattern_name) + history_region_total_realtime_region_pipe_flow_and_demand += temp_history_pipe_flow_and_demand_list[modify_index] - if (region_total_source_outflow - region_total_realtime_region_pipe_flow_and_demand): temp_multiply_factor = (region_total_source_outflow - region_total_realtime_region_pipe_flow_and_demand) / (history_region_total_source_outflow - history_region_total_realtime_region_pipe_flow_and_demand) for non_realtime_region_pattern_name in temp_non_realtime_region_patterns: pattern = get_pattern(name_c, non_realtime_region_pattern_name) @@ -949,6 +881,98 @@ def run_simulation(name: str, simulation_type: str, modify_pattern_start_time: s cs.append(pattern) set_pattern(name_c, cs) + # 根据输入的参数进行数据修改，后面修改的可以覆盖前面的，用于EXTENDED类的方案模拟 + # 修改清水池(reservoir)液位的pattern + if modify_reservoir_head_pattern: + for reservoir_name in modify_reservoir_head_pattern.keys(): + # 这句代码的作用是判断modify_reservoir_head_pattern[reservoir_name][0]是否不是NaN。 + # 如果modify_reservoir_head_pattern[reservoir_name][0]不是NaN，则条件成立，代码块会执行 + if not np.isnan(modify_reservoir_head_pattern[reservoir_name][0]): + # 给 list 中的所有元素加上 RESERVOIR_BASIC_HEIGHT + modified_values = [value + globals.RESERVOIR_BASIC_HEIGHT for value in + modify_reservoir_head_pattern[reservoir_name]] + reservoir_pattern = get_pattern(name_c, get_reservoir(name_c, reservoir_name)['pattern']) + reservoir_pattern['factors'][modify_index + :modify_index + len(modified_values)] = modified_values + cs = ChangeSet() + cs.append(reservoir_pattern) + set_pattern(name_c, cs) + + # 修改调节池(tank)初始液位 + if modify_tank_initial_level: + for tank_name in modify_tank_initial_level.keys(): + if (not np.isnan(modify_tank_initial_level[tank_name])) and (modify_tank_initial_level[tank_name] != 0): + tank = get_tank(name_c, tank_name) + tank['init_level'] = modify_tank_initial_level[tank_name] + cs = ChangeSet() + cs.append(tank) + set_tank(name_c, cs) + + # 修改节点（junction）基础水量（demand） + if modify_junction_base_demand: + for junction_name in modify_junction_base_demand.keys(): + if not np.isnan(modify_junction_base_demand[junction_name]): + junction = get_demand(name_c, junction_name) + junction['demand'] = modify_junction_base_demand[junction_name] + cs = ChangeSet() + cs.append(junction) + set_demand(name_c, cs) + + # 修改节点（junction）的水量模式（pattern） + if modify_junction_damand_pattern: + for pattern_name in modify_junction_damand_pattern.keys(): + if not np.isnan(modify_junction_damand_pattern[pattern_name][0]): + junction_pattern = get_pattern(name_c, pattern_name) + junction_pattern['factors'][modify_index + :modify_index + len(modify_junction_damand_pattern[pattern_name])] \ + = modify_junction_damand_pattern[pattern_name] + cs = ChangeSet() + cs.append(junction_pattern) + set_pattern(name_c, cs) + + # 修改工频水泵（fixed_pump）的pattern + if modify_fixed_pump_pattern: + for pump_name in modify_fixed_pump_pattern.keys(): + if not np.isnan(modify_fixed_pump_pattern[pump_name][0]): + pump_pattern = get_pattern(name_c, get_pump(name_c, pattern_name)['pattern']) + pump_pattern['factors'][modify_index + :modify_index + len(modify_fixed_pump_pattern)] \ + = modify_fixed_pump_pattern[pump_name] + cs = ChangeSet() + cs.append(pump_pattern) + set_pattern(name_c, cs) + + # 修改变频水泵（variable_pump）的pattern + if modify_variable_pump_pattern: + for pump_name in modify_variable_pump_pattern.keys(): + if not np.isnan(modify_variable_pump_pattern[pump_name][0]): + # 给 list 中的所有元素除以 50Hz + modified_values = [value / 50 for value in modify_variable_pump_pattern[pump_name]] + pump_pattern = get_pattern(name_c, get_pump(name_c, pattern_name)['pattern']) + pump_pattern['factors'][modify_index + :modify_index + len(modified_values)] = modified_values + cs = ChangeSet() + cs.append(pump_pattern) + set_pattern(name_c, cs) + + # 修改阀门（valve）的状态setting和status + if modify_valve_opening: + for valve_name in modify_valve_opening.keys(): + if not np.isnan(modify_valve_opening[valve_name]): + valve_status = get_status(name_c, valve_name) + if modify_valve_opening[valve_name] == 0: + valve_status['status'] = 'CLOSED' + valve_status['setting'] = 0 + if modify_valve_opening[valve_name] < 1: + valve_status['status'] = 'OPEN' + valve_status['setting'] = 0.1036 * pow(modify_valve_opening[valve_name], -3.105) + if modify_valve_opening[valve_name] == 1: + valve_status['status'] = 'OPEN' + valve_status['setting'] = 0 + cs = ChangeSet() + cs.append(valve_status) + set_status(name_c, cs) + # 根据高压出厂流量，更改高压用水模式 # hp_flow_SCADA_data_dict = influxdb_api.query_SCADA_data_by_device_ID_and_time( # query_ids_list=list(hp_flow_pattern_id.values()), query_time=modify_pattern_start_time) @@ -996,12 +1020,8 @@ def run_simulation(name: str, simulation_type: str, modify_pattern_start_time: s # cs = ChangeSet() # cs.append(pattern) # set_pattern(name_c, cs) - - - # 运行并返回结果 - print(f'Before run_project') + # 运行并返回结果 result = run_project(name_c) - # print(f'Simulation result : ' + result) time_cost_end = time.perf_counter() print('{} -- Hydraulic simulation finished, cost time: {:.2f} s.'.format( @@ -1010,21 +1030,19 @@ def run_simulation(name: str, simulation_type: str, modify_pattern_start_time: s close_project(name_c) - time.sleep(5) # wait 5 seconds - - tmp_file = './temp/simulation.result.out' - shutil.copy(f'./temp/{name_c}.db.opt', tmp_file) - output = Output(tmp_file) + output = Output("./temp/{}.db.out".format(name_c)) node_result = output.node_results() link_result = output.link_results() + num_periods_result = output.times()['num_periods'] - - # print(link_result[:3]) - influxdb_api.store_realtime_simulation_result_to_influxdb(node_result, link_result, modify_pattern_start_time) - - - - + # print(num_periods_result) + # print(node_result) + # 存储 + if simulation_type.upper() == 'REALTIME': + influxdb_api.store_realtime_simulation_result_to_influxdb(node_result, link_result, modify_pattern_start_time) + elif simulation_type.upper() == 'EXTENDED': + influxdb_api.store_scheme_simulation_result_to_influxdb(node_result, link_result, modify_pattern_start_time, + num_periods_result, scheme_Type, scheme_Name) if __name__ == "__main__": @@ -1032,7 +1050,6 @@ if __name__ == "__main__": query_corresponding_element_id_and_query_id("bb") query_corresponding_pattern_id_and_query_id('bb') region_result = query_non_realtime_region('bb') - print(region_result) globals.source_outflow_region_id = get_source_outflow_region_id('bb', region_result) globals.realtime_region_pipe_flow_and_demand_id = query_realtime_region_pipe_flow_and_demand_id('bb', region_result) @@ -1059,7 +1076,16 @@ if __name__ == "__main__": # print("Realtime Region Pipe Flow And Demand ID:", globals.realtime_region_pipe_flow_and_demand_id) # print("Realtime Region Pipe Flow And Demand Patterns:", globals.realtime_region_pipe_flow_and_demand_patterns) - run_simulation(name='bb', simulation_type="realtime", modify_pattern_start_time='2025-02-08T10:30:00+08:00') + # 模拟示例1 + # run_simulation(name='bb', simulation_type="realtime", modify_pattern_start_time='2025-02-14T10:30:00+08:00') + # 模拟示例2 + # run_simulation(name='bb', simulation_type="extended", modify_pattern_start_time='2025-02-14T10:30:00+08:00', modify_total_duration=900, + # scheme_Type="burst_Analysis", scheme_Name="scheme1") + + # 查询示例1：query_SCADA_ID_corresponding_info + # result = query_SCADA_ID_corresponding_info(name='bb', SCADA_ID='P10755') + # print(result) +