move internal functions to the function where they are used

This commit is contained in:
Pavel Patsey
2025-01-26 10:48:08 +03:00
parent e0ecafa9d0
commit a4991518ba
+1 -3
View File
@@ -37,6 +37,7 @@ def ensure_no_neighbors(nodes: list[Node]):
raise Cidr4MergerError(f"There are neighbors! {neighbors=}") raise Cidr4MergerError(f"There are neighbors! {neighbors=}")
def merge_nodes(a: Node, b: Node) -> tuple[Node, int]:
def find_parent(a: Node, b: Node) -> Node: def find_parent(a: Node, b: Node) -> Node:
ia, la = a ia, la = a
ib, lb = b ib, lb = b
@@ -47,7 +48,6 @@ def find_parent(a: Node, b: Node) -> Node:
mask = (mask << 1) & ((1 << 32) - 1) mask = (mask << 1) & ((1 << 32) - 1)
return ia & mask, min_l return ia & mask, min_l
def calc_dip(la: int, lb: int, lp: int) -> int: def calc_dip(la: int, lb: int, lp: int) -> int:
def dip(l1, lp): def dip(l1, lp):
m = lp + 1 m = lp + 1
@@ -58,8 +58,6 @@ def calc_dip(la: int, lb: int, lp: int) -> int:
return dip(la, lp) + dip(lb, lp) return dip(la, lp) + dip(lb, lp)
def merge_nodes(a: Node, b: Node) -> tuple[Node, int]:
p = find_parent(a, b) p = find_parent(a, b)
dip = calc_dip(a[1], b[1], p[1]) dip = calc_dip(a[1], b[1], p[1])
return p, dip return p, dip