1
0
forked from MTSR/mapserver
mapserver/server/db/postgres/initialblocks.go
2019-02-15 08:58:04 +01:00

152 lines
3.3 KiB
Go

package postgres
import (
"mapserver/coords"
"mapserver/db"
"mapserver/layer"
"mapserver/settings"
"math"
"github.com/sirupsen/logrus"
)
const (
SETTING_LAST_LAYER = "last_layer"
SETTING_LAST_X_BLOCK = "last_x_block"
SETTING_LAST_Y_BLOCK = "last_y_block"
)
// x -> 0 ... 256
//zoom/mapblock-width
//13 1
//12 2
//11 4
//10 8
//9 16
//Zoom 9:
//10 mapblocks height * 16 * 16 == 2560
func (this *PostgresAccessor) FindNextInitialBlocks(s settings.Settings, layers []*layer.Layer, limit int) (*db.InitialBlocksResult, error) {
lastlayer := s.GetInt(SETTING_LAST_LAYER, 0)
lastxblock := s.GetInt(SETTING_LAST_X_BLOCK, -129)
lastyblock := s.GetInt(SETTING_LAST_Y_BLOCK, -128)
if lastxblock >= 128 {
lastxblock = -128
lastyblock++
} else {
lastxblock++
}
if lastyblock > 128 {
//done
//TODO: next layer
result := &db.InitialBlocksResult{}
result.HasMore = false
return result, nil
}
tc := coords.NewTileCoords(lastxblock, lastyblock, 9, lastlayer)
currentlayer := layer.FindLayerById(layers, lastlayer)
fromY := int(currentlayer.From / 16)
toY := int(currentlayer.To / 16)
tcr := coords.GetMapBlockRangeFromTile(tc, 0)
tcr.Pos1.Y = fromY
tcr.Pos2.Y = toY
fields := logrus.Fields{
"layerId": lastlayer,
"pos1": tcr.Pos1,
"pos2": tcr.Pos2,
"tile": tc,
}
log.WithFields(fields).Debug("Initial-Query")
minX := int(math.Min(float64(tcr.Pos1.X), float64(tcr.Pos2.X)))
maxX := int(math.Max(float64(tcr.Pos1.X), float64(tcr.Pos2.X)))
minY := int(math.Min(float64(tcr.Pos1.Y), float64(tcr.Pos2.Y)))
maxY := int(math.Max(float64(tcr.Pos1.Y), float64(tcr.Pos2.Y)))
minZ := int(math.Min(float64(tcr.Pos1.Z), float64(tcr.Pos2.Z)))
maxZ := int(math.Max(float64(tcr.Pos1.Z), float64(tcr.Pos2.Z)))
if lastxblock <= -128 {
//first x entry, check z stride
stridecount := this.intQuery(`
select count(*) from blocks
where posz >= $1 and posz <= $2
and posy >= $3 and posy <= $4
and mtime = 0`,
minZ, maxZ,
minY, maxY,
)
if stridecount == 0 {
fields = logrus.Fields{
"minX": minX,
"maxX": maxX,
"minY": minY,
"maxY": maxY,
}
log.WithFields(fields).Debug("Skipping stride")
s.SetInt(SETTING_LAST_LAYER, lastlayer)
s.SetInt(SETTING_LAST_X_BLOCK, -129)
s.SetInt(SETTING_LAST_Y_BLOCK, lastyblock+1)
result := &db.InitialBlocksResult{}
result.Progress = float64(((lastyblock+128) * 256) + (lastxblock+128)) / float64(256*256)
result.HasMore = true
return result, nil
}
}
rows, err := this.db.Query(getBlocksByInitialTileQuery,
minX, minY, minZ, maxX, maxY, maxZ,
)
if err != nil {
return nil, err
}
defer rows.Close()
blocks := make([]*db.Block, 0)
for {
for rows.Next() {
var posx, posy, posz int
var data []byte
var mtime int64
err = rows.Scan(&posx, &posy, &posz, &data, &mtime)
if err != nil {
return nil, err
}
mb := convertRows(posx, posy, posz, data, mtime)
blocks = append(blocks, mb)
}
if !rows.NextResultSet() {
break
}
}
s.SetInt(SETTING_LAST_LAYER, lastlayer)
s.SetInt(SETTING_LAST_X_BLOCK, lastxblock)
s.SetInt(SETTING_LAST_Y_BLOCK, lastyblock)
result := &db.InitialBlocksResult{}
result.Progress = float64(((lastyblock+128) * 256) + (lastxblock+128)) / float64(256*256)
result.List = blocks
result.HasMore = true
return result, nil
}