package main

import (
	"flag"
	"fmt"
	"time"

	"track/gohumantrack"
	"track/goconv"
	"basic.com/valib/goffmpeg.git"
)

var (
	url1 string
	url2 string
)

func init() {
	flag.StringVar(&url1, "cam1", "", "url")
	flag.StringVar(&url2, "cam2", "", "url")
}

func run(cam1, cam2 *goffmpeg.GoFFMPEG, tracker *gohumantrack.HumanTracker) bool {
	data1, ow1, oh1, _ := cam1.GetYUV()
	data2, ow2, oh2, _ := cam2.GetYUV()
	if ow1 > 0 && oh1 > 0 && ow2 > 0 && oh2 > 0 {
		bgr1 := goconv.YUV2BGR(data1, ow1, oh1)
		bgr2 := goconv.YUV2BGR(data2, ow2, oh2)
		if bgr1 == nil || bgr2 == nil{
			return false
		}
		var images []gohumantrack.ImageHumanTracker
		img := gohumantrack.ImageHumanTracker{
			Data:    bgr1,
			Width:   ow1,
			Height:  oh1,
			Channel: 3,
		}
		images = append(images, img)
		img = gohumantrack.ImageHumanTracker{
			Data:    bgr2,
			Width:   ow2,
			Height:  oh2,
			Channel: 3,
		}
		images = append(images, img)
		res, err := tracker.Process(images)
		if err == nil {
			for _, v := range res {
				fmt.Printf("result size: %d\n", v.FgNum)
				for i := 0; i < int(v.FgNum); i++ {
					r := v.Fginfo[i]
					if r.Confidence > 0 {
						fmt.Printf(" Index %d Rect: %dx%dx%dx%d", i, r.Left, r.Top, r.Right, r.Bottom)
						fmt.Printf(" Confidence: %f", r.Confidence*100)
						fmt.Printf(" Center: %dx%d", r.X, r.Y)
						fmt.Printf(" ID: %d\n", r.ID)
					}
				}

			}
		} else {
			fmt.Println("process error: ", err)
		}
		return true
	} 
	return false
}

func main() {
	flag.Parse()
	fmt.Println("url1: ", url1, " url2: ", url2)
	if len(url1) == 0 || len(url2) == 0 {
		fmt.Println("url null")
	}

	tracker := gohumantrack.NewHumanTracker(0, 2, 0)
	fmt.Println("version: ", tracker.GetVersion())

	goffmpeg.InitFFmpeg("./runtime/libcffmpeg.so")

	cam1 := goffmpeg.New(false, false)
	cam1.Run(url1)
	cam1.BuildDecoder()
	cam1.CloseStream()

	cam2 := goffmpeg.New(false, false)
	cam2.Run(url2)
	cam2.BuildDecoder()
	cam2.CloseStream()

	for {
		if !run(cam1, cam2, tracker){
			time.Sleep(38 * time.Millisecond)
		}

	}
	tracker.Free()

	goffmpeg.FreeFFmpeg()
}