Variants demo

main.go

// Demonstrate game variants in muniverse.

package main

import (
	"fmt"
	"image/png"
	"log"
	"os"
	"time"

	"github.com/unixpickle/muniverse"
)

func main() {
	names := []string{
		"GoldMinerTom-v0",
		"GoldMinerTom-v1",
		"BurninRubber-v0",
		"BurninRubber-v1",
		"BurninRubber-v2",
		"BurninRubber-v3",
	}
	for _, name := range names {
		log.Println("Doing", name, "...")
		spec := muniverse.SpecForName(name)
		if spec == nil {
			panic("no spec")
		}
		env, err := muniverse.NewEnv(spec)
		if err != nil {
			panic(err)
		}
		if err := env.Reset(); err != nil {
			panic(err)
		}
		rew, done, err := env.Step(time.Second * 2)
		if err != nil {
			panic(err)
		}
		fmt.Println("rew", rew, "done", done)
		obs, err := env.Observe()
		if err != nil {
			panic(err)
		}
		img, err := obs.Image()
		if err != nil {
			panic(err)
		}
		f, _ := os.Create(name + ".png")
		png.Encode(f, img)
		f.Close()
	}
}