package util

import (
	"basic.com/pubsub/protomsg.git"
	"bufio"
	"fmt"
	"gocv.io/x/gocv"
	"image"
	"image/color"
	"os"
	"ruleprocess/cache"
	"basic.com/valib/logger.git"
	"ruleprocess/ruleserver"
	"ruleprocess/structure"
	"strings"
)

func CvRTSP() {
	url := `rtsp://admin:a1234567@192.168.1.188:554/h264/ch1/main/av_stream`

	webcam, _ := gocv.OpenVideoCapture(url)
	window := gocv.NewWindow("Hello")
	img := gocv.NewMat()
	for {
		webcam.Read(&img)
		window.IMShow(img)
		window.WaitKey(1)
	}
}

func cvFaceDetect() {
	// set to use a video capture device 0
	deviceID := 0
	url := `rtsp://admin:a1234567@192.168.1.188:554/h264/ch1/main/av_stream`

	// open webcam
	webcam, err := gocv.OpenVideoCapture(url)
	if err != nil {
		fmt.Println(err)
		return
	}
	defer webcam.Close()

	// open display window
	window := gocv.NewWindow("Face Detect")
	defer window.Close()

	// prepare image matrix
	img := gocv.NewMat()
	defer img.Close()

	// color for the rect when faces detected
	blue := color.RGBA{0, 0, 255, 0}

	// load classifier to recognize faces
	classifier := gocv.NewCascadeClassifier()
	defer classifier.Close()

	if !classifier.Load("data/haarcascade_frontalface_default.xml") {
		fmt.Println("Error reading cascade file: data/haarcascade_frontalface_default.xml")
		return
	}

	fmt.Printf("start reading camera device: %v\n", deviceID)
	for {
		if ok := webcam.Read(&img); !ok {
			fmt.Printf("cannot read device %v\n", deviceID)
			return
		}
		if img.Empty() {
			continue
		}

		// detect faces
		rects := classifier.DetectMultiScale(img)
		fmt.Printf("found %d faces\n", len(rects))

		// draw a rectangle around each face on the original image
		for _, r := range rects {
			gocv.Rectangle(&img, r, blue, 3)
		}

		// show the image in the window, and wait 1 millisecond
		window.IMShow(img)
		window.WaitKey(1)
	}
}

var w = 400

// CVDraw demo
func CVDraw() {
	windowA := gocv.NewWindow("basic drawing: atom")
	windowR := gocv.NewWindow("basic drawing: rook")
	defer windowA.Close()
	defer windowR.Close()

	atom := gocv.NewMatWithSize(w, w, gocv.MatTypeCV8UC3)
	defer atom.Close()

	rook := gocv.NewMatWithSize(w, w, gocv.MatTypeCV8UC3)
	defer rook.Close()

	black := color.RGBA{0, 0, 0, 0}
	blue := color.RGBA{0, 0, 255, 0}
	red := color.RGBA{255, 0, 0, 0}
	white := color.RGBA{255, 255, 255, 0}
	yellow := color.RGBA{255, 255, 0, 0}

	// draw the atom
	gocv.Ellipse(&atom, image.Pt(w/2., w/2.), image.Pt(w/4.0, w/16.0), 90., 0, 360, blue, 2)
	gocv.Ellipse(&atom, image.Pt(w/2., w/2.), image.Pt(w/4.0, w/16.0), 0., 0, 360, blue, 2)
	gocv.Ellipse(&atom, image.Pt(w/2., w/2.), image.Pt(w/4.0, w/16.0), 45., 0, 360, blue, 2)
	gocv.Ellipse(&atom, image.Pt(w/2., w/2.), image.Pt(w/4.0, w/16.0), -45., 0, 360, blue, 2)
	gocv.Circle(&atom, image.Pt(w/2., w/2.), w/32., red, -1)

	// draw the rook
	points := [][]image.Point{
		{
			image.Pt(w/4., 7*w/8.),
			image.Pt(3*w/4., 7*w/8.),
			image.Pt(3*w/4., 13*w/16.),
			image.Pt(11*w/16., 13*w/16.),
			image.Pt(19*w/32., 3*w/8.),
			image.Pt(3*w/4., 3*w/8.),
			image.Pt(3*w/4., w/8.),
			image.Pt(26*w/40., w/8.),
			image.Pt(26*w/40., w/4.),
			image.Pt(22*w/40., w/4.),
			image.Pt(22*w/40., w/8.),
			image.Pt(18*w/40., w/8.),
			image.Pt(18*w/40., w/4.),
			image.Pt(14*w/40., w/4.),
			image.Pt(14*w/40., w/8.),
			image.Pt(w/4., w/8.),
			image.Pt(w/4., 3*w/8.),
			image.Pt(13*w/32., 3*w/8.),
			image.Pt(5*w/16., 13*w/16.),
			image.Pt(w/4., 13*w/16.),
		},
	}
	gocv.FillPoly(&rook, points, white)
	gocv.Rectangle(&rook, image.Rect(0, 7*w/8.0, w, w), yellow, -1)
	gocv.Line(&rook, image.Pt(0, 15*w/16), image.Pt(w, 15*w/16), black, 2)
	gocv.Line(&rook, image.Pt(w/4, 7*w/8), image.Pt(w/4, w), black, 2)
	gocv.Line(&rook, image.Pt(w/2, 7*w/8), image.Pt(w/2, w), black, 2)
	gocv.Line(&rook, image.Pt(3*w/4, 7*w/8), image.Pt(3*w/4, w), black, 2)

	for {
		windowA.IMShow(atom)
		windowR.IMShow(rook)

		if windowA.WaitKey(10) >= 0 || windowR.WaitKey(10) >= 0 {
			break
		}
	}
}

func DrawPolygonOnImageForYolo(cameraId string, img protomsg.Image, results []structure.Result,url string) (maps map[string]interface{}, err0 error) {

	rook, _ := gocv.NewMatFromBytes(int(img.Height), int(img.Width), gocv.MatTypeCV8UC3, img.Data)
	//rook := gocv.IMRead("/home/user/workspace/ruleprocess/util/105.jpg",gocv.IMReadColor)
	defer rook.Close()

	yellow := color.RGBA{255, 255, 0, 0}
	red := color.RGBA{255, 0, 0, 0}
	scale := float64(img.Width) / 960 // 画图比例
	logger.Info("width:",img.Width,"--画图比例：",scale)
	// 分割区域id集合并根据id查询区域然后画框
	for _,result := range results  {
		polygonIds := strings.Split(result.AlarmPolygon,",")
		logger.Info("-----------------------看看报警区域id：",polygonIds)
		for i := 0; i < len(polygonIds); i++ {
			polygon := getPolygonById(polygonIds[i],cameraId)
			if polygon.Polygon != "[]" && polygon.Polygon != ""{
				logger.Debug("所画区域：",polygon.Polygon)
				DrawAPolygon(&rook,polygon.Polygon,yellow,scale)
			}
		}
	}
	// 把目标框出来
	for _,result := range results  {
		for _,rect := range result.Location {
			gocv.Rectangle(&rook, image.Rect(int(rect.X), int(rect.Y), int(rect.X+rect.Width), int(rect.Y+rect.Height)), red, 1)
		}
	}
	//return nil,nil
	maps,err0 = UploadFromMat(url,rook)
	return
}

func DrawPolygonOnImageForFace(cameraId string, img protomsg.Image, results []structure.FaceResult,url string) (maps map[string]interface{}, err0 error) {

	rook, _ := gocv.NewMatFromBytes(int(img.Height), int(img.Width), gocv.MatTypeCV8UC3, img.Data)
	//rook := gocv.IMRead("/home/user/workspace/ruleprocess/util/105.jpg",gocv.IMReadColor)
	defer rook.Close()

	yellow := color.RGBA{255, 255, 0, 0}
	scale := float64(img.Width) / 960 // 画图比例
	// 分割区域id集合并根据id查询区域然后画框
	for _,result := range results  {
		polygonIds := strings.Split(result.AlarmPolygon,",")
		logger.Info("-----------------------看看报警区域id：",polygonIds)
		for i := 0; i < len(polygonIds); i++ {
			polygon := getPolygonById(polygonIds[i],cameraId)
			logger.Info("----查到的报警框：",polygon)
			if polygon.Polygon != "[]" && polygon.Polygon != ""{
				logger.Debug("所画区域：",polygon.Polygon)
				DrawAPolygon(&rook,polygon.Polygon,yellow,scale)
			}
		}
	}
	//return nil,nil
	maps,err0 = UploadFromMat(url,rook)
	return
}

// 把图片转成二进制流
func RetrieveROM(filename string) ([]byte, error) {
	file, err := os.Open(filename)

	if err != nil {
		return nil, err
	}
	defer file.Close()

	stats, statsErr := file.Stat()
	if statsErr != nil {
		return nil, statsErr
	}

	var size int64 = stats.Size()
	bytes := make([]byte, size)

	bufr := bufio.NewReader(file)
	_, err = bufr.Read(bytes)

	return bytes, err
}
// 根据id去缓存里查询多边形
func getPolygonById(polygonId string, cameraId string) (protomsg.CameraPolygon){
	// 查到摄像机所有的区域并画框
	var cameraPolygons []protomsg.CameraPolygon
	cameraPolygons = cache.GetPolygonsByCameraId(cameraId)
	logger.Info("根据id查到的报警框：",cameraPolygons)
	for _, polygon := range cameraPolygons {
		logger.Info("查到的区域id：",polygon.Id,"--要匹配的数据id：",polygonId,"---是否相等：",strings.Contains(polygonId,polygon.Id))
		if strings.Contains(polygonId,polygon.Id) {
			logger.Info("进来即相等")
			return polygon
		}
	}
	return protomsg.CameraPolygon{}
}
// 在图上画一个框
func DrawAPolygon(rook *gocv.Mat,polygonString string, color color.RGBA,scale float64) {
	points := ruleserver.Json2points(polygonString)
	for index := 0; index < len(points); index++ {
		if index == len(points)-1 { // 闭合图形
			gocv.Line(rook, image.Pt(int(points[index].X * scale), int(points[index].Y * scale)), image.Pt(int(points[0].X * scale), int(points[0].Y * scale)), color, 2)
		} else {
			gocv.Line(rook, image.Pt(int(points[index].X * scale), int(points[index].Y * scale)), image.Pt(int(points[index+1].X * scale), int(points[index+1].Y * scale)), color, 2)
		}
	}
}