package service

import (
	"basic.com/pubsub/protomsg.git"
	"basic.com/valib/godraw.git"
	"basic.com/valib/logger.git"
	"bytes"
	"github.com/gogo/protobuf/proto"
	"image"
	"time"
)

type FaceSdkService struct {
	File []byte
	Id int64
	Result []*protomsg.ResultFaceDetect
	DeadTime time.Duration
}

const (
	faceExtractWebCID = "virtual-face-extract-web-camera-id"
)

func GetFaceFeaFromSdk(fileBytes []byte,deadTime time.Duration) ([]*protomsg.ResultFaceDetect,error,*protomsg.Image){
	t1 := time.Now()
	s := newFaceSdkService(fileBytes, deadTime)
	i, err := s.readFromUploadImg()
	logger.Debug("ReadFromUploadImg用时:", time.Since(t1))
	if err !=nil{
		logger.Debug("readFromUploadImg err:",err)
		return nil,err,i
	}
	imgBytes, err := proto.Marshal(i)
	if err !=nil {
		logger.Debug("i marshal err:",err)
		return nil,err,i
	}
	detectResults, err := DoFDetectWithRpc(&protomsg.FaceDetectRequest{ReqParam: imgBytes}, deadTime)
	logger.Debug("提取face共用时：", time.Since(t1))
	if err !=nil {
		logger.Debug("DoFDetectWithRpc err:",err)
		return nil,err,i
	} else {
		return detectResults,nil,i
	}
}

func newFaceSdkService(fileBytes []byte, deadTime time.Duration) *FaceSdkService{
	return &FaceSdkService{
		File:fileBytes,
		Id:time.Now().UnixNano(),
		DeadTime:deadTime,
	}
}

func (s *FaceSdkService) readFromUploadImg() (*protomsg.Image,error){
	bt := bytes.NewBuffer(s.File)
	img, _, err := image.Decode(bt)
	if err != nil {
		return nil, err
	}
	bgr := godraw.Image2BGR(img)

	timeUnix := time.Now().Unix()
	formatTimeStr := time.Unix(timeUnix, 0).Format("2006-01-02 15:04:05")

	return &protomsg.Image{
		Width: int32(img.Bounds().Dx()),
		Height: int32(img.Bounds().Dy()),
		Timestamp: formatTimeStr,
		Data: bgr,
		Id: s.Id,
		Cid: faceExtractWebCID,
	},nil
}