forked from MTSR/mapserver
98 lines
1.7 KiB
Go
98 lines
1.7 KiB
Go
package tilerenderer
|
|
|
|
import (
|
|
"bytes"
|
|
"github.com/sirupsen/logrus"
|
|
"io/ioutil"
|
|
"mapserver/colormapping"
|
|
"mapserver/coords"
|
|
"mapserver/db/sqlite"
|
|
"mapserver/layer"
|
|
"mapserver/mapblockaccessor"
|
|
"mapserver/mapblockrenderer"
|
|
"mapserver/testutils"
|
|
"mapserver/tiledb"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestTileRender(t *testing.T) {
|
|
logrus.SetLevel(logrus.DebugLevel)
|
|
|
|
tmpfile, err := ioutil.TempFile("", "TestTileRender.*.sqlite")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
defer os.Remove(tmpfile.Name())
|
|
testutils.CreateTestDatabase(tmpfile.Name())
|
|
|
|
a, err := sqlite.New(tmpfile.Name())
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
err = a.Migrate()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
cache := mapblockaccessor.NewMapBlockAccessor(a, 500*time.Millisecond, 1000*time.Millisecond, 1000)
|
|
c := colormapping.NewColorMapping()
|
|
_, err = c.LoadVFSColors(false, "/colors.txt")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
r := mapblockrenderer.NewMapBlockRenderer(cache, c)
|
|
|
|
tiletmpdir, err := ioutil.TempDir("", "TestTileRenderTiles.*.sqlite")
|
|
defer os.RemoveAll(tiletmpdir)
|
|
|
|
tdb, _ := tiledb.New(tiletmpdir)
|
|
|
|
layers := []*layer.Layer{
|
|
&layer.Layer{
|
|
Id: 0,
|
|
Name: "Base",
|
|
From: -16,
|
|
To: 160,
|
|
},
|
|
}
|
|
|
|
tr := NewTileRenderer(r, tdb, a, layers)
|
|
|
|
if tr == nil {
|
|
panic("no renderer")
|
|
}
|
|
|
|
coord := coords.NewTileCoords(0, 0, 12, 0)
|
|
data, err := tr.Render(coord)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if data == nil {
|
|
panic("no data")
|
|
}
|
|
|
|
f, _ := os.Create("../test-output/0_0_12.png")
|
|
bytes.NewReader(data).WriteTo(f)
|
|
|
|
coord1 := coord.GetZoomedOutTile()
|
|
|
|
data, err = tr.Render(coord1)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if data == nil {
|
|
panic("no data")
|
|
}
|
|
|
|
f, _ = os.Create("../test-output/0_0_13.png")
|
|
bytes.NewReader(data).WriteTo(f)
|
|
|
|
}
|