multi model, custom output support

snowboy-2022/main.go

@@ -12,6 +12,8 @@ import (
 	"flag"
 	"fmt"
 	"log"
+	"path"
+	"strings"
 	"time"
 
 	"github.com/brentnd/go-snowboy"
@@ -76,12 +78,12 @@ func (s *Sound) Read(p []byte) (int, error) {
 
 func main() {
 	resources := flag.String("r", "", "path to the .res file")
-	model := flag.String("m", "", "path to the .?mdl file")
+	models := flag.String("ms", "", "comma delimited path to the .?mdl file/output")
 	sensitivity := flag.Float64("s", 0.45, "0..1")
 	quiet := flag.Bool("q", false, "emit '1' on detect else silent")
 	flag.Parse()
 
-	if *resources == "" || *model == "" {
+	if *resources == "" || *models == "" {
 		panic("all flags must be set")
 	}
 
@@ -95,13 +97,14 @@ func main() {
 	defer d.Close()
 
 	// set the handlers
-	d.HandleFunc(snowboy.NewHotword(*model, float32(*sensitivity)), func(string) {
+	for _, modelStr := range strings.Split(*models, ",") {
+		d.HandleFunc(snowboy.NewHotword(path.Dir(modelStr), float32(*sensitivity)), func(string) {
 			if !*quiet {
 				log.Println("GOTCHA!")
-		} else {
-			fmt.Printf("1")
 			}
+			fmt.Println(path.Base(modelStr))
 		})
+	}
 
 	d.HandleSilenceFunc(1*time.Second, func(string) {
 		if !*quiet {