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

fix root dir writing, refactor js getZxy

Browse Source
This commit is contained in:
Brandon Liu
2021-02-18 16:21:30 +08:00
parent ab8f1e889d
commit f309d81aa2
4 changed files with 79 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

110
js/pmtiles.js
View File

@@ -42,7 +42,7 @@
class PMTiles {
constructor(url,ready) {
this.url = url
this.apex = fetch(this.url,{method:'HEAD',headers:{'Range':'bytes=0-511999'}}).then(resp => {
this.rootdir = fetch(this.url,{method:'HEAD',headers:{'Range':'bytes=0-511999'}}).then(resp => {
if (resp.status == 206) { // this does not work on Azure, it returns 200 instead of 206
console.log("Check succeeded: server supports byte ranges")
return fetch(this.url,{headers:{'Range':'bytes=0-511999'}}).then(resp => {
@@ -70,19 +70,20 @@
this.outstanding_requests.get(tilestr).push(resolve)
} else {
this.outstanding_requests.set(tilestr,[])
this.apex.then(apex_map => {
if (apex_map.has(tilestr)) {
var val = apex_map.get(tilestr)
this.rootdir.then(rootdir => {
if (rootdir.has(tilestr)) {
var val = rootdir.get(tilestr)
if (val[2] == 1) { // it is a directory
fetch(this.url, {headers:{'range':'bytes=' + val[0] + '-' + (val[0] + val[1]-1)}}).then(resp => {
return resp.arraybuffer()
return resp.arrayBuffer()
}).then(buf => {
var map = bytestomap(buf,val[1]/17)
var map = bytesToMap(new DataView(buf),val[1]/17)
this.leaves.set(tilestr,map)
resolve(map)
this.outstanding_requests.get(tilestr).foreach(f => {
this.outstanding_requests.get(tilestr).forEach(f => {
f(map)
})
// TODO fix duplication of multiple leaves in same directory
console.log("leaves: ", this.leaves.size)
})
}
@@ -92,52 +93,71 @@
})
}
transformRequest = (u,t,tile,done) => {
if (t == 'Tile' && done) {
var tid = tile.tileID.canonical
var strid = tid.z + '_' + tid.x + '_' + tid.y
this.apex.then(map => {
if (map.has(strid) && map.get(strid)[2] == 0) {
var val = map.get(strid)
done({url: this.url, headers:{'Range':'bytes=' + val[0] + '-' + (val[0]+val[1]-1)}})
} else {
if (tid.z >= 7) {
var z7_tile_diff = (tid.z - 7)
var z7_tile = [7,Math.trunc(tid.x / (1 << z7_tile_diff)), Math.trunc(tid.y / (1 << z7_tile_diff))]
var z7_tile_str = z7_tile[0] + "_" + z7_tile[1] + "_" + z7_tile[2]
this.getLeaf(z7_tile_str).then(map => {
if (map.has(strid)) {
var val = map.get(strid)
done({url: this.url, headers:{'Range':'bytes=' + val[0] + '-' + (val[0]+val[1]-1)}})
}
})
}
}
})
}
return {url: u}
getZxy = (z,x,y) => {
var strid = z + '_' + x + '_' + y
return this.rootdir.then(rootdir => {
if (rootdir.has(strid) && rootdir.get(strid)[2] == 0) {
return rootdir.get(strid)
} else {
if (z >= 7) {
var z7_tile_diff = (z - 7)
var z7_tile = [7,Math.trunc(x / (1 << z7_tile_diff)), Math.trunc(y / (1 << z7_tile_diff))]
var z7_tile_str = z7_tile[0] + "_" + z7_tile[1] + "_" + z7_tile[2]
return this.getLeaf(z7_tile_str).then(leafdir => {
if (leafdir.has(strid)) {
return leafdir.get(strid)
}
return null
})
}
}
return null
})
}
// transformRequest = (u,t,tile,done) => {
// if (t == 'Tile' && done) {
// var tid = tile.tileID.canonical
// var strid = tid.z + '_' + tid.x + '_' + tid.y
// this.rootdir.then(rootdir => {
// if (rootdir.has(strid) && rootdir.get(strid)[2] == 0) {
// var val = rootdir.get(strid)
// done({url: this.url, headers:{'Range':'bytes=' + val[0] + '-' + (val[0]+val[1]-1)}})
// } else {
// if (tid.z >= 7) {
// var z7_tile_diff = (tid.z - 7)
// var z7_tile = [7,Math.trunc(tid.x / (1 << z7_tile_diff)), Math.trunc(tid.y / (1 << z7_tile_diff))]
// var z7_tile_str = z7_tile[0] + "_" + z7_tile[1] + "_" + z7_tile[2]
// this.getLeaf(z7_tile_str).then(leafdir => {
// if (leafdir.has(strid)) {
// var val = leafdir.get(strid)
// done({url: this.url, headers:{'Range':'bytes=' + val[0] + '-' + (val[0]+val[1]-1)}})
// }
// })
// }
// }
// })
// }
// return {url: u}
// }
leafletLayer = options => {
const self = this
var cls = L.GridLayer.extend({
createTile: function(coords, done){
createTile: function(coord, done){
var tile = document.createElement('img')
var error
self.apex.then(map => {
var strid = coords.z + '_' + coords.x + '_' + coords.y
if (map.has(strid)) {
var val = map.get(strid)
fetch(self.url,{headers:{'Range':'bytes=' + val[0] + '-' + (val[0]+val[1]-1)}}).then(resp => {
return resp.arrayBuffer()
}).then(buf => {
var blob = new Blob( [buf], { type: "image/png" } )
var imageUrl = window.URL.createObjectURL(blob)
tile.src = imageUrl
done(error,tile)
})
}
self.getZxy(coord.z,coord.x,coord.y).then(result => {
if (result === null) return
fetch(self.url,{headers:{'Range':'bytes=' + result[0] + '-' + (result[0]+result[1]-1)}}).then(resp => {
return resp.arrayBuffer()
}).then(buf => {
var blob = new Blob( [buf], { type: "image/png" } )
var imageUrl = window.URL.createObjectURL(blob)
tile.src = imageUrl
done(error,tile)
})
})
return tile
},

16
python/bin/pmtiles-show
View File

@@ -8,8 +8,14 @@ if len(sys.argv) <= 1:
exit(1)
with read(sys.argv[1]) as reader:
print('spec version: ',reader.version)
print('metadata:')
for k, v in reader.metadata.items():
print(k,'=',v)
print('root entries:', reader.root_entries)
if len(sys.argv) == 2:
print('spec version: ',reader.version)
print('metadata:')
for k, v in reader.metadata.items():
print(k,'=',v)
print('root entries:', reader.root_entries)
else:
z = int(sys.argv[2])
x = int(sys.argv[3])
y = int(sys.argv[4])
print(reader.get(z,x,y))

1
python/pmtiles/reader.py
View File

@@ -47,6 +47,7 @@ class Reader:
def root_entries(self):
return int.from_bytes(self.mmap[8:10],byteorder='little')
# TODO support recursive directories
def get(self,z,x,y):
val = self.root_dir.get((z,x,y))
if val:

4
python/pmtiles/writer.py
View File

@@ -97,8 +97,8 @@ class Writer:
if len(leafdir_tiles):
self.write_leafdir(leafdir_tiles)
root = [group for group in itertools.groupby(self.tiles,key=by_parent) if group[0][0] == 0][0]
root_tiles = list(root[1])
root = [(group[0],list(group[1])) for group in itertools.groupby(self.tiles,key=by_parent) if group[0][0] == 0][0]
root_tiles = root[1]
self.f.seek(0)
self.write_header(metadata,len(root_tiles) + len(self.leaves))
for entry in root_tiles: