Commit cdab1972 authored by Kirill Smelkov's avatar Kirill Smelkov

.

parent 8c550297
...@@ -240,6 +240,8 @@ def Restructure(ztree, newStructure): ...@@ -240,6 +240,8 @@ def Restructure(ztree, newStructure):
# Bj ∈ RNv is mapped into Ai ∈ RZv such that that sum_j D(A_i, Bj) is minimal. # Bj ∈ RNv is mapped into Ai ∈ RZv such that that sum_j D(A_i, Bj) is minimal.
# it is used to associate nodes in tnew to ztree nodes. # it is used to associate nodes in tnew to ztree nodes.
def assign(RZv, RNv): def assign(RZv, RNv):
print()
print('assign %s <- %s' % (RZv, RNv))
for rzn in RZv: for rzn in RZv:
assert isinstance(rzn, _NodeInRange) assert isinstance(rzn, _NodeInRange)
assert isinstance(rzn.node, (ztreeType, zbucketType)) assert isinstance(rzn.node, (ztreeType, zbucketType))
...@@ -268,8 +270,10 @@ def Restructure(ztree, newStructure): ...@@ -268,8 +270,10 @@ def Restructure(ztree, newStructure):
# assignments more efficiently taking that Av and Bv are key↑ and the # assignments more efficiently taking that Av and Bv are key↑ and the
# property of D function so that if B2 > B1 (all keys in B2 > all keys # property of D function so that if B2 > B1 (all keys in B2 > all keys
# in B1) and A < B1.hi, then D(B2, A) > D(B1, A). # in B1) and A < B1.hi, then D(B2, A) > D(B1, A).
print()
print(C) print(C)
jv, iv = scipy.optimize.linear_sum_assignment(C) jv, iv = scipy.optimize.linear_sum_assignment(C)
print(jv, iv)
for (j,i) in zip(jv, iv): for (j,i) in zip(jv, iv):
RNv[j].node.Z = RZv[i].node RNv[j].node.Z = RZv[i].node
...@@ -278,6 +282,7 @@ def Restructure(ztree, newStructure): ...@@ -278,6 +282,7 @@ def Restructure(ztree, newStructure):
# to newly created Z nodes. # to newly created Z nodes.
for j2 in range(len(RNv)): for j2 in range(len(RNv)):
if j2 not in jv: if j2 not in jv:
print('\n\n\nXXX create new @%d\n\n\n' % j2)
n = RNv[j2].node n = RNv[j2].node
assert n.Z is None assert n.Z is None
assert isinstance(n, (Tree,Bucket)) assert isinstance(n, (Tree,Bucket))
...@@ -349,8 +354,10 @@ def Restructure(ztree, newStructure): ...@@ -349,8 +354,10 @@ def Restructure(ztree, newStructure):
assert len(rbucketv) > 0 assert len(rbucketv) > 0
for i in range(len(rbucketv)-1): for i in range(len(rbucketv)-1):
assert rbucketv[i].range.khi <= rbucketv[i+1].klo assert rbucketv[i].range.khi <= rbucketv[i+1].klo
rbucketv[i].node.next_bucket = rbucketv[i+1].node rbucketv[i] .node.next_bucket = rbucketv[i+1].node
rbucketv[i+1].node.prev_bucket = rbucketv[i] .node
rbucketv[-1].node.next_bucket = None rbucketv[-1].node.next_bucket = None
rbucketv[0] .node.prev_bucket = None
# associate every bucket to zbucket # associate every bucket to zbucket
assign(zrbucketv, rbucketv) assign(zrbucketv, rbucketv)
...@@ -361,18 +368,24 @@ def Restructure(ztree, newStructure): ...@@ -361,18 +368,24 @@ def Restructure(ztree, newStructure):
for rnodev in rlevelv: for rnodev in rlevelv:
for rn in rnodev: for rn in rnodev:
node = rn.node node = rn.node
assert isinstance(node, Tree) # buckets were filtered out to rbucketv assert isinstance(node, Tree), node # buckets were filtered out to rbucketv
znode = node.Z znode = node.Z
# https://github.com/zopefoundation/BTrees/blob/4.5.0-1-gc8bf24e/BTrees/BTreeTemplate.c#L1087 # https://github.com/zopefoundation/BTrees/blob/4.5.0-1-gc8bf24e/BTrees/BTreeTemplate.c#L1087
zstate = unset = object()
if len(node.keyv) == 0: if len(node.keyv) == 0:
child = node.children[0] child = node.children[0]
if isinstance(child, Bucket): if isinstance(child, Bucket):
if len(child.keyv) == 0: # empty bucket noone links to
panic("TODO: empty tree") if len(child.keyv) == 0 and child.prev_bucket is None:
if child.Z._p_oid is None: zstate = None # -> empty tree
panic("TODO: tree with bucket without oid")
# tree with single bucket noone links to
elif child.Z._p_oid is None and child.prev_bucket is None:
zstate = ((child.Z.__getstate__(),),) # tree with bucket without oid
if zstate is unset:
# normal tree node
zstate = () zstate = ()
assert len(node.children) == len(node.keyv) + 1 assert len(node.children) == len(node.keyv) + 1
zstate += node.children[0] zstate += node.children[0]
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment