import math
from pickle import dumps as pdumps, loads as ploads
import lz4.frame as gzip
from py3dtiles.points.utils import split_aabb
from py3dtiles.points.node import Node
[docs]
class NodeCatalog:
"""NodeCatalog is a store of Node objects.py3dtiles
Using a NodeCatalog allows to only store a children names
in nodes, instead of storing a full recursive structure.
"""
def __init__(self, nodes, name, octree_metadata):
super(NodeCatalog, self).__init__()
self.nodes = {}
self.root_aabb = octree_metadata.aabb
self.root_spacing = octree_metadata.spacing
self.node_bytes = {}
self._load_from_store(name, nodes)
[docs]
def get_node(self, name):
"""Returns the node mathing the given name"""
if name not in self.nodes:
spacing = self.root_spacing / math.pow(2, len(name))
aabb = self.root_aabb
for i in name:
aabb = split_aabb(aabb, int(i))
node = Node(name, aabb, spacing)
self.nodes[name] = node
else:
node = self.nodes[name]
return node
[docs]
def dump(self, name, max_depth):
"""Serialize the stored nodes to a bytes list"""
node = self.nodes[name]
if node.dirty:
self.node_bytes[name] = node.save_to_bytes()
if node.children is not None and max_depth > 0:
for n in node.children:
self.dump(n, max_depth - 1)
return pdumps(self.node_bytes)
def _load_from_store(self, name, data):
if len(data) > 0:
out = ploads(gzip.decompress(data))
for n in out:
spacing = self.root_spacing / math.pow(2, len(n))
aabb = self.root_aabb
for i in n:
aabb = split_aabb(aabb, int(i))
node = Node(n, aabb, spacing)
node.load_from_bytes(out[n])
self.node_bytes[n] = out[n]
self.nodes[n] = node
else:
spacing = self.root_spacing / math.pow(2, len(name))
aabb = self.root_aabb
for i in name:
aabb = split_aabb(aabb, int(i))
node = Node(name, aabb, spacing)
self.nodes[name] = node
return self.nodes[name]