implemented wap parsing and info extraction

2025-08-18 12:29:17 +02:00
parent ca5fe83e32
commit 2af5bd222e
2 changed files with 282 additions and 3 deletions
--- a/main.go
+++ b/main.go
@@ -1,6 +1,92 @@
 package main

-// init
-func main() {
+import (
+	"bspviz/internal/wad"
+	"flag"
+	"fmt"
+	"log"
+	"os"
+	"path/filepath"
+	"strings"
+)

+func main() {
+	// Flags
+	wadPath := flag.String("wad", "", "Pfad zur WAD (required)")
+	mapMarker := flag.String("map", "MAP01", "Map-Marker (z.B. MAP01, E1M1, MYMAP)")
+	listOnly := flag.Bool("list", false, "Nur Directory auflisten und beenden")
+	info := flag.Bool("info", false, "Roh-Infos zur Map (Counts von VERTEXES/LINEDEFS)")
+	extract := flag.String("extract", "", "Kommagetrennte Lump-Namen aus der Map extrahieren (z.B. VERTEXES,LINEDEFS)")
+	outdir := flag.String("out", ".", "Zielordner für -extract")
+	flag.Parse()
+
+	if *wadPath == "" {
+		fmt.Fprintf(os.Stderr, "Usage:\n")
+		fmt.Fprintf(os.Stderr, "  go run ./cmd/bspviz -wad MYMAP.wad -list\n")
+		fmt.Fprintf(os.Stderr, "  go run ./cmd/bspviz -wad MYMAP.wad -map MYMAP -info\n")
+		fmt.Fprintf(os.Stderr, "  go run ./cmd/bspviz -wad MYMAP.wad -map MYMAP -extract VERTEXES,LINEDEFS -out dumps/\n")
+		os.Exit(2)
+	}
+
+	w, err := wad.Open(*wadPath)
+	if err != nil {
+		log.Fatalf("open wad: %v", err)
+	}
+	defer w.Close()
+
+	if *listOnly {
+		fmt.Printf("WAD: %s\n", *wadPath)
+		for i, d := range w.Dir() {
+			fmt.Printf("%3d: %-8s size=%-7d pos=%-8d\n", i, d.Name(), d.Size, d.FilePos)
+		}
+		return
+	}
+
+	start, end, err := w.FindMap(*mapMarker)
+	if err != nil {
+		log.Fatalf("find map: %v", err)
+	}
+	fmt.Printf("Map %s: Directory [%d, %d)\n", strings.ToUpper(*mapMarker), start, end)
+
+	if *info {
+		lumps, err := w.LoadMapLumps(*mapMarker, "VERTEXES", "LINEDEFS")
+		if err != nil {
+			log.Fatalf("load map lumps: %v", err)
+		}
+		vb := lumps["VERTEXES"]
+		lb := lumps["LINEDEFS"]
+
+		verts := len(vb) / 4
+		lines := len(lb) / 14
+		fmt.Printf("VERTEXES: bytes=%d  count=%d\n", len(vb), verts)
+		fmt.Printf("LINEDEFS: bytes=%d  count=%d\n", len(lb), lines)
+
+		if len(vb)%4 != 0 {
+			fmt.Println("WARN: VERTEXES size ist kein Vielfaches von 4 → Format prüfen")
+		}
+		if len(lb)%14 != 0 {
+			fmt.Println("WARN: LINEDEFS size ist kein Vielfaches von 14 → Format prüfen")
+		}
+	}
+
+	if *extract != "" {
+		want := strings.Split(*extract, ",")
+		for i := range want {
+			want[i] = strings.ToUpper(strings.TrimSpace(want[i]))
+		}
+		lumps, err := w.LoadMapLumps(*mapMarker, want...)
+		if err != nil {
+			log.Fatalf("extract: %v", err)
+		}
+		if err := os.MkdirAll(*outdir, 0o755); err != nil {
+			log.Fatalf("mkdir %s: %v", *outdir, err)
+		}
+		for name, data := range lumps {
+			dst := filepath.Join(*outdir, fmt.Sprintf("%s.lmp", name))
+			if err := os.WriteFile(dst, data, 0o644); err != nil {
+				log.Fatalf("write %s: %v", dst, err)
+			}
+			fmt.Printf("wrote %s (%d bytes)\n", dst, len(data))
+		}
+	}
 }