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
+3 -5
View File
@@ -37,7 +37,8 @@ def ensure_no_neighbors(nodes: list[Node]):
raise Cidr4MergerError(f"There are neighbors! {neighbors=}")
def find_parent(a: Node, b: Node) -> Node:
def merge_nodes(a: Node, b: Node) -> tuple[Node, int]:
def find_parent(a: Node, b: Node) -> Node:
ia, la = a
ib, lb = b
min_l = min(la, lb)
@@ -47,8 +48,7 @@ def find_parent(a: Node, b: Node) -> Node:
mask = (mask << 1) & ((1 << 32) - 1)
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):
m = lp + 1
res = 1 << (l1 - m)
@@ -58,8 +58,6 @@ def calc_dip(la: int, lb: int, lp: int) -> int:
return dip(la, lp) + dip(lb, lp)
def merge_nodes(a: Node, b: Node) -> tuple[Node, int]:
p = find_parent(a, b)
dip = calc_dip(a[1], b[1], p[1])
return p, dip