mirror of
https://bitbucket.org/s_l_teichmann/mtsatellite
synced 2024-11-08 19:20:25 +01:00
122 lines
3.3 KiB
Go
122 lines
3.3 KiB
Go
// Copyright 2014 by Sascha L. Teichmann
|
|
// Use of this source code is governed by the MIT license
|
|
// that can be found in the LICENSE file.
|
|
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"image"
|
|
"image/color"
|
|
"log"
|
|
|
|
"bitbucket.org/s_l_teichmann/mtsatellite/common"
|
|
)
|
|
|
|
func main() {
|
|
var (
|
|
port int
|
|
host string
|
|
x, y, z int
|
|
width, height, depth int
|
|
colorsfile string
|
|
outfile string
|
|
shaded bool
|
|
)
|
|
|
|
flag.IntVar(&port, "port", 6379, "port to of mtredisalize server")
|
|
flag.IntVar(&port, "p", 6379, "port to of mtredisalize server (shorthand)")
|
|
flag.StringVar(&host, "host", "localhost", "host to mtredisalize server")
|
|
flag.IntVar(&x, "x", 0, "x of query cuboid")
|
|
flag.IntVar(&y, "y", -75, "y of query cuboid")
|
|
flag.IntVar(&z, "z", 0, "z of query cuboid")
|
|
flag.IntVar(&width, "width", 16, "width of query cuboid")
|
|
flag.IntVar(&height, "height", 16, "height of query cuboid")
|
|
flag.IntVar(&depth, "depth", 150, "depth of query cuboid")
|
|
flag.IntVar(&width, "w", 16, "width of query cuboid (shorthand)")
|
|
flag.IntVar(&height, "h", 16, "height of query cuboid (shorthand)")
|
|
flag.IntVar(&depth, "d", 150, "depth of query cuboid (shorthand)")
|
|
flag.StringVar(&colorsfile, "colors", "colors.txt", "definition of colors")
|
|
flag.StringVar(&outfile, "output", "out.png", "image file of result")
|
|
flag.StringVar(&outfile, "o", "out.png", "image file of result (shorthand)")
|
|
flag.BoolVar(&shaded, "shaded", true, "draw relief")
|
|
|
|
flag.Parse()
|
|
|
|
var err error
|
|
var colors *common.Colors
|
|
|
|
if colors, err = common.ParseColors(colorsfile); err != nil {
|
|
log.Fatalf("Cannot open color file: %s", err)
|
|
}
|
|
|
|
address := fmt.Sprintf("%s:%d", host, port)
|
|
|
|
var client *common.RedisClient
|
|
|
|
if client, err = common.NewRedisClient("tcp", address); err != nil {
|
|
log.Fatalf("Cannot connect to '%s': %s", address, err)
|
|
}
|
|
defer client.Close()
|
|
|
|
if shaded {
|
|
width += 2
|
|
height += 2
|
|
x--
|
|
z--
|
|
}
|
|
|
|
q1x, q1y, q1z := int16(x), int16(y), int16(z)
|
|
q2x, q2y, q2z := q1x+int16(width)-1, q1y+int16(depth)-1, q1z+int16(height)-1
|
|
|
|
renderer := common.NewRenderer(width, height)
|
|
renderer.SetPos(q1x, q1z)
|
|
yOrder := common.NewYOrder(renderer, 512)
|
|
|
|
numBlocks := 0
|
|
drawBlock := func(block *common.Block) {
|
|
if err := yOrder.RenderBlock(block, colors); err != nil {
|
|
log.Printf("WARN: rendering block failed: %s", err)
|
|
}
|
|
numBlocks++
|
|
}
|
|
|
|
c1 := common.Coord{X: q1x, Z: q1z}
|
|
c2 := common.Coord{X: q2x, Z: q2z}
|
|
for c2.Y = q2y; c2.Y > q1y; c2.Y -= 8 {
|
|
c1.Y = c2.Y - 7
|
|
if c1.Y < q1y {
|
|
c1.Y = q1y
|
|
}
|
|
cuboid := common.Cuboid{P1: common.MinCoord(c1, c2), P2: common.MaxCoord(c1, c2)}
|
|
if err = client.QueryCuboid(cuboid, drawBlock); err != nil {
|
|
log.Fatalf("query failed: %s", err)
|
|
}
|
|
if err = yOrder.Drain(colors); err != nil {
|
|
log.Printf("WARN: rendering block failed: %s", err)
|
|
}
|
|
if renderer.IsFilled() {
|
|
break
|
|
}
|
|
}
|
|
|
|
var image image.Image
|
|
|
|
if shaded {
|
|
image = renderer.CreateShadedImage(
|
|
16, 16, (width-2)*16, (height-2)*16,
|
|
colors.Colors, color.RGBA{R: 0xff, G: 0xff, B: 0xff, A: 0xff})
|
|
} else {
|
|
image = renderer.CreateImage(
|
|
colors.Colors, color.RGBA{R: 0xff, G: 0xff, B: 0xff, A: 0xff})
|
|
}
|
|
|
|
if err = common.SaveAsPNG(outfile, image); err != nil {
|
|
log.Fatalf("writing image failed: %s", err)
|
|
}
|
|
|
|
log.Printf("num blocks: %d\n", numBlocks)
|
|
log.Printf("rejected blocks: %d\n", renderer.Rejected)
|
|
}
|