mtsatellite/cmd/mttilemapper/main.go

156 lines
4.5 KiB
Go
Raw Normal View History

// Copyright 2014, 2015 by Sascha L. Teichmann
2014-09-09 15:22:29 +02:00
// Use of this source code is governed by the MIT license
// that can be found in the LICENSE file.
2014-09-07 21:46:55 +02:00
package main
import (
"flag"
"fmt"
2014-09-11 17:35:03 +02:00
"image"
2014-09-07 21:46:55 +02:00
"log"
2014-10-28 11:51:45 +01:00
"os"
"runtime/pprof"
2014-09-07 21:46:55 +02:00
"bitbucket.org/s_l_teichmann/mtsatellite/common"
2014-09-07 21:46:55 +02:00
)
func main() {
var (
port int
host string
x, y, z int
width, height, depth int
2014-09-09 15:01:14 +02:00
colorsfile string
bgColor string
2014-09-10 18:58:12 +02:00
outfile string
2014-09-11 17:35:03 +02:00
shaded bool
transparent bool
2014-10-28 11:51:45 +01:00
cpuProfile string
transparentDim float64
version bool
2014-09-07 21:46:55 +02:00
)
defaultBgColor := common.ColorToHex(common.BackgroundColor)
2014-09-07 21:46:55 +02:00
flag.IntVar(&port, "port", 6379, "port to of mtredisalize server")
2014-09-10 18:58:12 +02:00
flag.IntVar(&port, "p", 6379, "port to of mtredisalize server (shorthand)")
2014-09-07 21:46:55 +02:00
flag.StringVar(&host, "host", "localhost", "host to mtredisalize server")
flag.IntVar(&x, "x", 0, "x of query cuboid")
2014-09-07 21:46:55 +02:00
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")
2014-09-07 21:46:55 +02:00
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)")
2014-09-07 21:46:55 +02:00
flag.IntVar(&depth, "d", 150, "depth of query cuboid (shorthand)")
2014-09-09 15:01:14 +02:00
flag.StringVar(&colorsfile, "colors", "colors.txt", "definition of colors")
flag.StringVar(&bgColor, "background", defaultBgColor, "background color")
flag.StringVar(&bgColor, "bg", defaultBgColor, "background color (shorthand)")
2014-09-10 18:58:12 +02:00
flag.StringVar(&outfile, "output", "out.png", "image file of result")
flag.StringVar(&outfile, "o", "out.png", "image file of result (shorthand)")
2014-09-11 17:35:03 +02:00
flag.BoolVar(&shaded, "shaded", true, "draw relief")
flag.BoolVar(&transparent, "transparent", false, "render transparent blocks")
flag.Float64Var(
&transparentDim, "transparent-dim", common.DefaultTransparentDim*100,
"Extra dimming of transparent nodes every depth meter in percent (0-100).")
2014-10-28 11:51:45 +01:00
flag.StringVar(&cpuProfile, "cpuprofile", "", "write cpu profile to file")
flag.BoolVar(&version, "version", false, "Print version and exit.")
2014-09-07 21:46:55 +02:00
flag.Parse()
if version {
common.PrintVersionAndExit()
}
2016-04-23 21:09:23 +02:00
bg := common.ParseColorDefault(bgColor, common.BackgroundColor)
2014-10-28 11:51:45 +01:00
if cpuProfile != "" {
f, err := os.Create(cpuProfile)
if err != nil {
log.Fatal(err)
}
pprof.StartCPUProfile(f)
defer pprof.StopCPUProfile()
}
2014-09-14 00:02:04 +02:00
var colors *common.Colors
2016-04-23 21:09:23 +02:00
var err error
2014-09-14 00:02:04 +02:00
if colors, err = common.ParseColors(colorsfile); err != nil {
2014-09-09 15:01:14 +02:00
log.Fatalf("Cannot open color file: %s", err)
}
colors.TransparentDim = common.Clamp32f(
float32(transparentDim/100.0), 0.0, 100.0)
2014-09-07 21:46:55 +02:00
address := fmt.Sprintf("%s:%d", host, port)
var client *common.RedisClient
2014-09-07 21:46:55 +02:00
if client, err = common.NewRedisClient("tcp", address); err != nil {
2014-09-07 21:46:55 +02:00
log.Fatalf("Cannot connect to '%s': %s", address, err)
}
defer client.Close()
if shaded {
width += 2
height += 2
x--
z--
}
2014-09-07 21:46:55 +02:00
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, transparent)
renderer.SetPos(q1x, q1z)
yOrder := common.NewYOrder(renderer, 512)
numBlocks := 0
2014-09-10 18:58:12 +02:00
drawBlock := func(block *common.Block) {
if err := yOrder.RenderBlock(block, colors); err != nil {
log.Printf("WARN: rendering block failed: %s\n", err)
}
numBlocks++
2014-09-07 21:46:55 +02:00
}
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)}
2017-02-26 21:22:14 +01:00
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\n", err)
}
if renderer.IsFilled() {
break
}
2014-09-07 21:46:55 +02:00
}
2014-09-11 17:35:03 +02:00
var image image.Image
if shaded {
image = renderer.CreateShadedImage(
16, 16, (width-2)*16, (height-2)*16,
colors, bg)
2014-09-11 17:35:03 +02:00
} else {
image = renderer.CreateImage(colors.Colors, bg)
2014-09-11 17:35:03 +02:00
}
2014-09-14 00:31:28 +02:00
if err = common.SaveAsPNG(outfile, image); err != nil {
2014-09-10 18:58:12 +02:00
log.Fatalf("writing image failed: %s", err)
}
log.Printf("num blocks: %d\n", numBlocks)
log.Printf("rejected blocks: %d\n", renderer.RejectedBlocks)
log.Printf("transparent blocks: %d\n", renderer.TransparentBlocks)
log.Printf("solid blocks: %d\n", renderer.SolidBlocks)
2014-09-07 21:46:55 +02:00
}