pflag

cmd/wgpu-gen/main.go

@@ -4,23 +4,81 @@
 package main
 
 import (
+	"fmt"
 	"log"
 	"os"
 	"os/signal"
 	"syscall"
 
 	"github.com/qvest-digital/go-mxl/mxl"
+	"github.com/spf13/pflag"
 
 	flowdef "mxl-pattern-generator/internal/flow-def"
 	"mxl-pattern-generator/internal/generator"
 )
 
+type appArgs struct {
+	showHelp bool
+
+	domain           string
+	videoFlowDefFile string
+	audioFlowDefFile string
+	pattern          uint8
+	listPatterns     bool
+	textOverlay      string
+
+	videoWidth  uint16
+	videoHeight uint16
+	videoFPS    float32
+
+	audioChannels     uint8
+	audioSamplingFreq uint32
+}
+
+func printUsage() {
+	fmt.Println("CLI USAGE FUNC")
+	fmt.Fprintln(os.Stderr, "Usage: ")
+}
+
 func main() {
+	var args appArgs
+	flagSet := pflag.NewFlagSet("args", pflag.ContinueOnError)
+	flagSet.SortFlags = false
+	flagSet.Usage = func() { printUsage() }
+	// common flags
+	flagSet.BoolVarP(&args.showHelp, "help", "h", false, "Show help message and exit")
+	// MXL flags
+	flagSet.StringVarP(&args.domain, "domain", "d", "", "MXL domain")
+	flagSet.StringVarP(&args.videoFlowDefFile, "video", "v", "", "Video flow definition JSON file path")
+	flagSet.StringVarP(&args.audioFlowDefFile, "audio", "a", "", "Audio flow definition JSON file path [TODO]")
+	// Video pattern flags
+	flagSet.Uint8VarP(&args.pattern, "pattern", "p", 0, "Video pattern type [TODO]")
+	flagSet.BoolVar(&args.listPatterns, "list-patterns", false, "List video available video patterns and exit [TODO]")
+	flagSet.StringVarP(&args.textOverlay, "overlay", "o", "", "Text overlay above video pattern [TODO]")
+	flagSet.Uint16VarP(&args.videoWidth, "width", "w", 1920, "Video pattern width")
+	// flagSet.Uint16VarP(&args.videoHeight, "height", "h", 1920, "Video pattern height")
+	args.videoFPS = *flagSet.Float32P("fps", "f", 1920, "Video pattern FPS")
+	// Audio pattern flags
+	flagSet.Uint8VarP(&args.audioChannels, "channel", "c", 2, "Amount of audio channels. Each channel: num * 1kHz [TODO]")
+	flagSet.Uint32VarP(&args.audioSamplingFreq, "freq", "f", 0, "Sampling frequency of test audio feed [TODO]")
+
 	var width, height uint = 1920, 1080
 	var fpsNum, fpsDen uint = 25, 1
 	flowUUID := "8f1d2a4b-6c3e-4f5a-9b2c-1d7e8a3f0b5d"
 	domain := "/dev/shm/mxl"
 
+	// create domain if not exist
+	if _, err := os.Stat(domain); err != nil {
+		err := os.MkdirAll(domain, 0775)
+		if err != nil {
+			log.Fatalf("Could not create domain dir: %v", err)
+		}
+	}
+	// TODO: move to parse args
+	if _, err := mxl.IsTmpFs(domain); err != nil {
+		log.Fatalf("Domain is not tmpfs dir: %v", err)
+	}
+
 	log.Printf("Go Pattern Gen (wgpu)")
 	log.Printf("Video: %dx%dp%d/%d", width, height, fpsNum, fpsDen)
 	log.Printf("UUID: %s", flowUUID)