metadata: add and use generic map translation and scaling. (#37)

Author: Michael Bang

examples/de_cache.jpg

@@ -18,6 +18,7 @@ import (
 	common "github.com/markus-wa/demoinfocs-golang/common"
 	events "github.com/markus-wa/demoinfocs-golang/events"
 	ex "github.com/markus-wa/demoinfocs-golang/examples"
+	"github.com/markus-wa/demoinfocs-golang/metadata"
 	st "github.com/markus-wa/demoinfocs-golang/sendtables"
 )
 
@@ -35,6 +36,7 @@ var (
 	colorFlash       color.Color = color.RGBA{0x00, 0x00, 0xff, 0xff} // Blue, because of the color on the nade
 	colorSmoke       color.Color = color.RGBA{0xbe, 0xbe, 0xbe, 0xff} // Light gray
 	colorDecoy       color.Color = color.RGBA{0x96, 0x4b, 0x00, 0xff} // Brown, because it's shit :)
+	curMap           metadata.Map
 )
 
 type inferno []r3.Vector
@@ -55,9 +57,11 @@ func main() {
 
 	p := dem.NewParser(f)
 
-	_, err = p.ParseHeader()
+	hd, err := p.ParseHeader()
 	checkError(err)
 
+	curMap = metadata.MapNameToMap[hd.MapName]
+
 	nadeTrajectories := make(map[int64]*nadePath) // Trajectories of all destroyed nades
 
 	p.RegisterEventHandler(func(e events.NadeProjectileDestroyedEvent) {
@@ -122,18 +126,18 @@ func main() {
 	err = p.ParseToEnd()
 	checkError(err)
 
-	// Draw image
-
-	// Create output canvas
-	dest := image.NewRGBA(image.Rect(0, 0, 1024, 1024))
-
 	// Use cache map overview as base
-	fCache, err := os.Open("../de_cache.jpg")
+	fMap, err := os.Open(fmt.Sprintf("../../metadata/maps/%s.jpg", hd.MapName))
 	checkError(err)
 
-	imgCache, _, err := image.Decode(fCache)
+	imgMap, _, err := image.Decode(fMap)
 	checkError(err)
-	draw.Draw(dest, dest.Bounds(), imgCache, image.Point{0, 0}, draw.Src)
+
+	// Create output canvas
+	dest := image.NewRGBA(imgMap.Bounds())
+
+	// Draw image
+	draw.Draw(dest, dest.Bounds(), imgMap, image.Point{0, 0}, draw.Src)
 
 	// Initialize the graphic context
 	gc := draw2dimg.NewGraphicContext(dest)
@@ -171,11 +175,15 @@ func drawInfernos(gc *draw2dimg.GraphicContext, hulls []*s2.Loop) {
 
 func buildInfernoPath(gc *draw2dimg.GraphicContext, hull *s2.Loop) {
 	vertices := hull.Vertices()
-	gc.MoveTo(translateX(vertices[0].X), translateY(vertices[0].Y))
+	x, y, _ := curMap.TranslateScale(vertices[0].X, vertices[0].Y, 0)
+	gc.MoveTo(x, y)
+
 	for _, fire := range vertices[1:] {
-		gc.LineTo(translateX(fire.X), translateY(fire.Y))
+		x, y, _ := curMap.TranslateScale(fire.X, fire.Y, 0)
+		gc.LineTo(x, y)
 	}
-	gc.LineTo(translateX(vertices[0].X), translateY(vertices[0].Y))
+
+	gc.LineTo(x, y)
 }
 
 func drawTrajectories(gc *draw2dimg.GraphicContext, trajectories []*nadePath) {
@@ -209,38 +217,18 @@ func drawTrajectories(gc *draw2dimg.GraphicContext, trajectories []*nadePath) {
 		}
 
 		// Draw path
-		gc.MoveTo(translateX(np.path[0].X), translateY(np.path[0].Y)) // Move to a position to start the new path
+		x, y, _ := curMap.TranslateScale(np.path[0].X, np.path[0].Y, 0)
+		gc.MoveTo(x, y) // Move to a position to start the new path
 
 		for _, pos := range np.path[1:] {
-			gc.LineTo(translateX(pos.X), translateY(pos.Y))
+			x, y, _ := curMap.TranslateScale(pos.X, pos.Y, 0)
+			gc.LineTo(x, y)
 		}
 
 		gc.FillStroke()
 	}
 }
 
-// Rough translations for x & y coordinates from de_cache to 1024x1024 px.
-// This could be done nicer by only having to provide the mapping between two source & target coordinates and the max size.
-// Then we could calculate the correct stretch & offset automatically.
-
-func translateX(x float64) float64 {
-	const (
-		stretch = 0.18
-		offset  = 414
-	)
-
-	return x*stretch + offset
-}
-
-func translateY(y float64) float64 {
-	const (
-		stretch = -0.18
-		offset  = 630
-	)
-
-	return y*stretch + offset
-}
-
 func checkError(err error) {
 	if err != nil {
 		log.Fatal(err)

examples/nade-trajectories/nade_trajectories.go

@@ -0,0 +1,109 @@
+package metadata
+
+// Map represents a CS:GO map. It contains information required to translate
+// in-game world coordinates to coordinates relative to (0, 0)
+type Map struct {
+	Name string
+
+	PosX float64
+	PosY float64
+
+	Scale  float64
+	Rotate float64
+	Zoom   float64
+}
+
+// Translate translates in-game world-relative coordinates to (0, 0) relative coordinates
+func (m Map) Translate(x, y, z float64) (float64, float64, float64) {
+	return x - m.PosX, m.PosY - y, z
+}
+
+// TranslateScale translates and scales in-game world-relative coordinates to (0, 0) relative coordinates
+func (m Map) TranslateScale(x, y, z float64) (float64, float64, float64) {
+	x, y, z = m.Translate(x, y, z)
+	return x / m.Scale, y / m.Scale, z
+}
+
+var (
+	MapDeCache = Map{
+		Name:  "de_cache",
+		PosX:  -2000,
+		PosY:  3250,
+		Scale: 5.5,
+	}
+
+	MapDeCanals = Map{
+		Name:  "de_canals",
+		PosX:  -2496,
+		PosY:  1792,
+		Scale: 4,
+	}
+
+	MapDeCbble = Map{
+		Name:  "de_cbble",
+		PosX:  -3840,
+		PosY:  3072,
+		Scale: 6,
+	}
+
+	MapDeDust2 = Map{
+		Name:   "de_dust2",
+		PosX:   -2400,
+		PosY:   3383,
+		Scale:  4.4,
+		Rotate: 1,
+		Zoom:   1.1,
+	}
+
+	MapDeInferno = Map{
+		Name:  "de_inferno",
+		PosX:  -2087,
+		PosY:  3870,
+		Scale: 4.9,
+	}
+
+	MapDeMirage = Map{
+		Name:   "de_mirage",
+		PosX:   -3230,
+		PosY:   1713,
+		Scale:  5,
+		Rotate: 0,
+		Zoom:   0,
+	}
+
+	MapDeNuke = Map{
+		Name:  "de_nuke",
+		PosX:  -3453,
+		PosY:  2887,
+		Scale: 7,
+	}
+
+	MapDeOverpass = Map{
+		Name:   "de_overpass",
+		PosX:   -4831,
+		PosY:   1781,
+		Scale:  5.2,
+		Rotate: 0,
+		Zoom:   0,
+	}
+
+	MapDeTrain = Map{
+		Name:  "de_train",
+		PosX:  -2477,
+		PosY:  2392,
+		Scale: 4.7,
+	}
+)
+
+// MapNameToMap translates a map name to a Map.
+var MapNameToMap = map[string]Map{
+	"de_cache":    MapDeCache,
+	"de_canals":   MapDeCanals,
+	"de_cbble":    MapDeCbble,
+	"de_dust2":    MapDeDust2,
+	"de_inferno":  MapDeInferno,
+	"de_mirage":   MapDeMirage,
+	"de_nuke":     MapDeNuke,
+	"de_overpass": MapDeOverpass,
+	"de_train":    MapDeTrain,
+}

metadata/maps.go

@@ -0,0 +1,23 @@
+Map files courtesy of @zoidbergwill/csgo-overviews, granted under the MIT license:
+
+> The MIT License (MIT)
+>
+> Copyright (c) 2015 William Martin Stewart
+>
+> Permission is hereby granted, free of charge, to any person obtaining a copy
+> of this software and associated documentation files (the "Software"), to deal
+> in the Software without restriction, including without limitation the rights
+> to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+> copies of the Software, and to permit persons to whom the Software is
+> furnished to do so, subject to the following conditions:
+>
+> The above copyright notice and this permission notice shall be included in all
+> copies or substantial portions of the Software.
+>
+> THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+> IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+> FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+> AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+> LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+> OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+> SOFTWARE.