减少爆管定位代码中引入的不确定性

2026-03-09 11:29:57 +08:00
parent d55e23bc44
commit f9111ab9c1
5 changed files with 145 additions and 31 deletions
@@ -20,6 +20,17 @@ def _to_metis_edge_weight(edge_weight):
    return max(1, int(round(weight)))


+def _dedupe_preserve_order(items):
+    seen = set()
+    output = []
+    for item in items:
+        if item in seen:
+            continue
+        seen.add(item)
+        output.append(item)
+    return output
+
+
 def pick_center_pipe(node_x, node_y, candidate_pipe, pipe_start_node, pipe_end_node):
    candidate_pipe_list = list(candidate_pipe)
    start_nodes = pipe_start_node[candidate_pipe_list]
@@ -34,14 +45,52 @@ def pick_center_pipe(node_x, node_y, candidate_pipe, pipe_start_node, pipe_end_n
    return candidate_pipe_list[center_idx]


+def pick_max_diameter_pipe(candidate_pipe, pipe_diameter):
+    candidate_pipe_list = list(candidate_pipe)
+    diameters = pd.to_numeric(
+        pipe_diameter.reindex(candidate_pipe_list), errors="coerce"
+    ).dropna()
+    if len(diameters) != len(candidate_pipe_list):
+        missing = sorted(set(candidate_pipe_list) - set(diameters.index))
+        preview = ", ".join(map(str, missing[:10]))
+        raise ValueError(f"Missing or invalid diameter for pipes: {preview}")
+
+    max_diameter = float(diameters.max())
+    max_diameter_pipes = sorted(
+        [pipe for pipe, diameter in diameters.items() if float(diameter) == max_diameter],
+        key=str,
+    )
+    return max_diameter_pipes[0]
+
+
+def pick_dual_center_pipes(
+    node_x, node_y, candidate_pipe, pipe_start_node, pipe_end_node, pipe_diameter
+):
+    geometric_center = pick_center_pipe(
+        node_x, node_y, candidate_pipe, pipe_start_node, pipe_end_node
+    )
+    diameter_center = pick_max_diameter_pipe(candidate_pipe, pipe_diameter)
+    return _dedupe_preserve_order([geometric_center, diameter_center])
+
+
 def find_new_center_pipe(
-    node_x, node_y, candidate_pipe, pipe_start_node, pipe_end_node, record_center
+    node_x,
+    node_y,
+    candidate_pipe,
+    pipe_start_node,
+    pipe_end_node,
+    pipe_diameter,
+    record_center,
 ):
    new_candidate_pipe = sorted(set(candidate_pipe) - set(record_center))
    if new_candidate_pipe == []:
        new_candidate_pipe = candidate_pipe
    center_t = pick_center_pipe(
-        node_x, node_y, new_candidate_pipe, pipe_start_node, pipe_end_node
+        node_x,
+        node_y,
+        new_candidate_pipe,
+        pipe_start_node,
+        pipe_end_node,
    )
    return center_t

@@ -66,6 +115,7 @@ def metis_grouping_pipe_weight(
    node_pipe_dic,
    all_node_series,
    couple_node_length,
+    pipe_diameter,
 ):
    all_node_iter_series_new = all_node_series[all_node_iter]
    all_node_iter_series_new = all_node_iter_series_new.sort_values(ascending=True)
@@ -82,10 +132,12 @@ def metis_grouping_pipe_weight(
    correspond_dic = {}
    count_node = 0
    w = []
-    for node_name, neighbors in G1.adjacency():
+    for node_name in all_node_iter_new:
+        neighbors = G1[node_name]
        w_temp = []
        n_t = [node_dict[node_name]]
-        for neighbor_name, edge_data in neighbors.items():
+        for neighbor_name in sorted(neighbors.keys()):
+            edge_data = neighbors[neighbor_name]
            edge_key = f"{node_name},{neighbor_name}"
            reverse_edge_key = f"{neighbor_name},{node_name}"
            if edge_key in couple_node_length:
@@ -122,14 +174,14 @@ def metis_grouping_pipe_weight(
        w_f = w_f + w_new[i]

    # (edgecuts, parts) = pymetis.part_graph(nparts=group_num, adjacency=adjacency_list_new)
-    # metis_options = pymetis.Options()
-    # metis_options.seed = 42
+    metis_options = pymetis.Options()
+    metis_options.seed = 42
    (edgecuts, parts) = pymetis.part_graph(
        nparts=group_num,
        adjncy=final_adjacency_list,
        xadj=xadj,
        eweights=w_f,
-        # options=metis_options,
+        options=metis_options,
    )
    # (edgecuts, parts) = pymetis.part_graph(nparts=group_num, adjacency=adjacency_list_new)
    candidate_group_list = [[] * 1 for i in range(group_num)]
@@ -145,6 +197,7 @@ def metis_grouping_pipe_weight(

    new_center = []
    new_group = []
+    new_center_candidates = []
    new_all_node = []
    candidate_pipe_set = set(candidate_pipe_input)
    all_grouped_pipe = []
@@ -158,7 +211,7 @@ def metis_grouping_pipe_weight(
            # 求交集
            nodeset = G_sub.nodes()
            pipeset_set = set(cal_area_node_linked_pipe(nodeset, node_pipe_dic))
-            candidate_pipe = list(pipeset_set.intersection(candidate_pipe_set))
+            candidate_pipe = sorted(pipeset_set.intersection(candidate_pipe_set))

            # 判断集合是否保留
            if len(candidate_pipe) > 0:
@@ -170,21 +223,30 @@ def metis_grouping_pipe_weight(
                    pipe_start_node_all,
                    pipe_end_node_all,
                )
+                center_candidates_t = pick_dual_center_pipes(
+                    node_x,
+                    node_y,
+                    candidate_pipe,
+                    pipe_start_node_all,
+                    pipe_end_node_all,
+                    pipe_diameter,
+                )

                # 更新
                new_center.append(center_t)
+                new_center_candidates.append(center_candidates_t)
                new_group.append(candidate_pipe)
                new_all_node.append(nodeset)
                all_grouped_pipe = all_grouped_pipe + candidate_pipe
        else:
-            for c in sub_graphs:
+            for c in sorted(sub_graphs, key=lambda c: min(c)):
                G_temp = G0.subgraph(c)
                nodeset = G_temp.nodes()
                pipeset = cal_area_node_linked_pipe(nodeset, node_pipe_dic)
                pipeset_set = set(pipeset)

                # 求交集
-                candidate_pipe = list(pipeset_set.intersection(candidate_pipe_set))
+                candidate_pipe = sorted(pipeset_set.intersection(candidate_pipe_set))
                # print(len(candidate_node))
                # 判断集合是否保留
                if len(candidate_pipe) > 0:
@@ -196,8 +258,17 @@ def metis_grouping_pipe_weight(
                        pipe_start_node_all,
                        pipe_end_node_all,
                    )
+                    center_candidates_t = pick_dual_center_pipes(
+                        node_x,
+                        node_y,
+                        candidate_pipe,
+                        pipe_start_node_all,
+                        pipe_end_node_all,
+                        pipe_diameter,
+                    )
                    # 更新
                    new_center.append(center_t)
+                    new_center_candidates.append(center_candidates_t)
                    new_group.append(candidate_pipe)
                    new_all_node.append(nodeset)
                    all_grouped_pipe = all_grouped_pipe + candidate_pipe
@@ -217,8 +288,12 @@ def metis_grouping_pipe_weight(
                    each_group,
                    pipe_start_node_all,
                    pipe_end_node_all,
+                    pipe_diameter,
                    record_center,
                )
+                new_center_candidates[c_g] = _dedupe_preserve_order(
+                    [new_center[c_g]] + list(new_center_candidates[c_g])
+                )
            record_center.append(new_center[c_g])
        c_g += 1

@@ -227,7 +302,7 @@ def metis_grouping_pipe_weight(
    #     node_x, node_y,
    #     pipe_start_node_all, pipe_end_node_all
    # )
-    return new_center, new_group, new_all_node
+    return new_center, new_group, new_all_node, new_center_candidates


 def visualize_metis_partition(