package service

import (
	"basic.com/pubsub/protomsg.git"
	"basic.com/valib/deliver.git"
	"encoding/base64"
	"fmt"
	"github.com/gogo/protobuf/proto"
	"github.com/pierrec/lz4"
	"github.com/satori/go.uuid"
	"gocv.io/x/gocv"
	"time"
)

const (
	Ipc_Push_Ext = "_2.ipc"
	Ipc_Pull_Ext = "_1.ipc"
	Ipc_Url_Pre = "ipc:///tmp///"
	Virtual_FaceTaskId = "92496BDF-2BFA-98F2-62E8-96DD9866ABD2"
	Virtual_FaceSdkId = "virtual-faceextract-sdk-pull"
	Url_Service_PUSH = Ipc_Url_Pre + Virtual_FaceSdkId + Ipc_Push_Ext
	Url_Service_PULL = Ipc_Url_Pre + Virtual_FaceSdkId + Ipc_Pull_Ext
)
var imgPushChan chan []byte
var client_push deliver.Deliver
var client_pull deliver.Deliver

func TestPushImgMsg() {
	InitService()

	i := readImgFile()

	fmt.Printf("width:%d,height:%d,data.length:%d,timestamp:%s,id:%d\n",i.Width,i.Height,len(i.Data),i.Timestamp,i.Id)
	if b, err := proto.Marshal(&i);err !=nil{
		fmt.Println("protoImage marshal err")
		return
	} else {
		bc := make([]byte,len(b))
		ht := make([]int, 64<<10)
		n,err := lz4.CompressBlock(b, bc, ht)
		if err !=nil {
			fmt.Println(err)
		}
		if n >= len(b){
			fmt.Println("image is not compressible")
		}
		bc = bc[:n]
		for {
			PushImgMsg(bc)
			fmt.Println("pushed img")
			time.Sleep(5*time.Second)
		}

	}
}

func readImgFile() protomsg.Image{
	var i protomsg.Image
	timeUnix := time.Now().Unix()
	formatTimeStr := time.Unix(timeUnix, 0).Format("2006-01-02 15:04:05")
	filePath := "/home/user/workspace/timg.jpg"

	picMat := gocv.IMRead(filePath, gocv.IMReadColor)

	defer picMat.Close()

	if picMat.Empty() {
		fmt.Println("file not exist")
		return i
	}
	height := int32(picMat.Rows())
	width := int32(picMat.Cols())
	data := picMat.ToBytes()
	//wrMat,_ := gocv.NewMatFromBytes(picMat.Rows(),picMat.Cols(),gocv.MatTypeCV8UC3,data)
	//
	//gocv.IMWrite("xxx.jpg", wrMat)

	i = protomsg.Image{
		Width: width,
		Height: height,
		Timestamp: formatTimeStr,
		Data: data,
		Id: timeUnix,
	}
	i.Cid = uuid.NewV4().String()//数据唯一id
	fmt.Println("gocv read img completed")
	return i
}

func PushImgMsg(is []byte){
	imgPushChan <- is
}

var resultMap map[string]protomsg.SdkMessage


func InitService(){
	fmt.Println("service init!")
	imgPushChan = make(chan []byte)
	resultMap = make(map[string]protomsg.SdkMessage,0)
	client_push = deliver.NewClient(deliver.PushPull, Url_Service_PUSH)
	client_pull = deliver.NewClient(deliver.PushPull, Url_Service_PULL)
	defer func() {
		client_push.Close()
		client_pull.Close()
	}()
	go thSend()

	go thRecv()
}

func thSend(){
	for {
		select {
		case d := <- imgPushChan:
			fmt.Println("imgPushChan in")
			err := client_push.Send(d)
			if err !=nil {
				fmt.Println("img Send err:",err)
			}
		default:
			//fmt.Println("no img in")
		}
	}
}

func thRecv(){
	for {
		resultBytes, err := client_pull.Recv()
		if err !=nil{
			fmt.Println("pull err:",err)
			continue
		}
		rMsg := protomsg.SdkMessage{}
		if err := proto.Unmarshal(resultBytes, &rMsg);err ==nil{
			fmt.Println("received MSG:",rMsg.Cid)
			perId := rMsg.Cid //数据id
			if rMsg.Tasklab !=nil && rMsg.Tasklab.Taskid == Virtual_FaceTaskId {
				sdkInfos := rMsg.Tasklab.Sdkinfos
				fmt.Println("Len(sdkInfos)=",len(sdkInfos))
					for _,swt :=range sdkInfos{
						fmt.Println("sdkName:",swt.SdkName)
						if swt.Sdktype =="FaceDetect"{
							fmt.Println("人脸检测结果")
							var pfp protomsg.ParamFacePos
							err := proto.Unmarshal(swt.Sdkdata, &pfp)
							if err !=nil {
								fmt.Println("faceDetect result unmarshal err:",err)
							} else {
								for _,face :=range pfp.Faces{
									fmt.Println("FacePos:",face.Pos)
									fmt.Println("ThftResult:",face.Result)
								}
							}
						}
						if swt.Sdktype == "FaceExtract"{
							fmt.Println("sdkData.len:",len(swt.Sdkdata))
							var pff protomsg.ParamFaceFeature
							if err := proto.Unmarshal(swt.Sdkdata, &pff);err !=nil{
								//fmt.Println("ParamFaceFeature unmarshal err:",err)
							} else {
								fmt.Println("目标数：",len(pff.ExtComp))
								for _,fea :=range pff.ExtComp{
									base64Fea := base64.StdEncoding.EncodeToString(fea.Feats)
									fmt.Println("perId:",perId)
									fmt.Println("faceFeature:",base64Fea)
								}
							}
							break
						}
					}

			}
			//resultMap[rMsg.Cid] = rMsg
		} else {
			fmt.Println("recv msg Err:",err)
		}

	}
}