mirrors/PMTiles
1
0
Fork 0
mirror of https://github.com/protomaps/PMTiles.git synced 2026-02-04 10:51:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

python reader can read v2 archives (non-fixed leaf level) [#36]

Browse Source
This commit is contained in:
Brandon Liu
2022-03-17 15:50:17 +08:00
parent 3784132161
commit c9b0ddc558
2 changed files with 18 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
python/pmtiles/reader.py
View File

@@ -53,19 +53,24 @@ class Reader:
def root_entries(self):
return int.from_bytes(self.mmap[8:10],byteorder='little')
@property
def leaf_level(self):
return next(iter(self.leaves))[0]
def get(self,z,x,y):
val = self.root_dir.get((z,x,y))
if val:
return self.mmap[val[0]:val[0]+val[1]]
else:
z7_tile_diff = z - 7
z7_tile = (7,x // (1 << z7_tile_diff),y // (1 << z7_tile_diff))
val = self.leaves.get(z7_tile)
if val:
directory, _ = self.load_directory(val[0],val[1]//17)
val = directory.get((z,x,y))
if len(self.leaves) > 0:
level_diff = z - self.leaf_level
leaf = (self.leaf_level,x // (1 << level_diff),y // (1 << level_diff))
val = self.leaves.get(leaf)
if val:
return self.mmap[val[0]:val[0]+val[1]]
directory, _ = self.load_directory(val[0],val[1]//17)
val = directory.get((z,x,y))
if val:
return self.mmap[val[0]:val[0]+val[1]]
def tiles(self):
for k,v in self.root_dir.items():