Class Grid

join/Pexeso/Hierarchical_Grid.py:5–31

import pdb

class Grid:

    def __init__(self, id, level, mlevel, o, l):
        self.id = id
        self.level = level
        self.mlevel = mlevel
        self.o = o #中点座标
        self.l = l #网格边长
        self.child = {} # id - grid
        self.vector = []
        self.vec_ids = []
        self.emb = []

    def get_size(self):
        return len(self.vector)

    def is_leaf(self):
        return self.mlevel == self.level

    def get_leaf(self):
        # pdb.set_trace()
        if self.is_leaf():
            return [self]
        leaves = []
        for ch in self.child.values():
            leaves.extend(ch.get_leaf())
        return leaves

class HierarchicalGrid:
    # tree = {}