From adb478c329c669f8709593d16dddffc0505e71ee Mon Sep 17 00:00:00 2001
From: xuxiuxi <554325746@qq.com>
Date: 星期三, 10 七月 2019 16:04:42 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'

---
 service/FaceSdkService.go |  129 ++++++++++++++++++++++--------------------
 1 files changed, 68 insertions(+), 61 deletions(-)

diff --git a/service/FaceSdkService.go b/service/FaceSdkService.go
index 8275437..a3f0387 100644
--- a/service/FaceSdkService.go
+++ b/service/FaceSdkService.go
@@ -3,7 +3,6 @@
 import (
 	"basic.com/pubsub/protomsg.git"
 	"basic.com/valib/deliver.git"
-	"fmt"
 	"github.com/gogo/protobuf/proto"
 	"github.com/pierrec/lz4"
 	"github.com/pkg/errors"
@@ -11,12 +10,13 @@
 	"gocv.io/x/gocv"
 	"sync"
 	"time"
+	"webserver/extend/logger"
 )
 
 type FaceSdkService struct {
 	File []byte
 	Id string
-	Result []*protomsg.ResultFaceExtCom
+	Result []*protomsg.ResultFaceDetect
 }
 
 const (
@@ -28,6 +28,27 @@
 	Url_Service_PUSH   = Ipc_Url_Pre + Virtual_FaceSdkId + Ipc_Push_Ext
 	Url_Service_PULL   = Ipc_Url_Pre + Virtual_FaceSdkId + Ipc_Pull_Ext
 )
+
+func GetFaceFeaFromSdk(fileBytes []byte) ([]*protomsg.ResultFaceDetect,error,*protomsg.Image){
+	s := NewFaceSdkService(fileBytes)
+	i, err := s.ReadFromUploadImg()
+	if err !=nil{
+		logger.Debug("readFromUploadImg err:",err)
+		return nil,err,i
+	}
+	bc, err := ImgCompress(i)
+	if err !=nil {
+		logger.Debug("ImgCompress err:",err)
+		return nil,err,i
+	}
+	s.PushImgMsg(bc)
+	s.GetFaceFea()
+	if s.Result == nil{
+		return nil,errors.New("no fea"),i
+	} else {
+		return s.Result,nil,i
+	}
+}
 
 func NewFaceSdkService(fileBytes []byte) FaceSdkService{
 	return FaceSdkService{
@@ -45,11 +66,11 @@
 
 	i := readTestImgFile()
 
-	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)
+	logger.Debug("width:%d,height:%d,data.length:%d,timestamp:%s,id:%d\n",i.Width,i.Height,len(i.Data),i.Timestamp,i.Id)
 
 	bc, err := ImgCompress(&i)
 	if err !=nil {
-		fmt.Println("image is not compressible")
+		logger.Debug("image is not compressible")
 	} else {
 		var s FaceSdkService
 		s.PushImgMsg(bc)
@@ -60,18 +81,18 @@
 	//defer (*(s.File)).Close()
 	//imgB, err := ioutil.ReadAll(*(s.File))
 	//if err !=nil{
-	//	fmt.Println("File.Read err:",err)
+	//	logger.Debug("File.Read err:",err)
 	//	return nil,err
 	//}
 
 	picMat, err := gocv.IMDecode(s.File, gocv.IMReadColor)
 	if err !=nil {
-		fmt.Println("gocv.IMDecode err:",err)
+		logger.Debug("gocv.IMDecode err:",err)
 		return nil,err
 	}
 	defer picMat.Close()
 	if picMat.Empty() {
-		fmt.Println("file not exist")
+		logger.Debug("file not exist")
 		return nil,errors.New("picMat is empty")
 	}
 	height := int32(picMat.Rows())
@@ -92,18 +113,18 @@
 
 func ImgCompress(i *protomsg.Image) ([]byte,error){
 	if b, err := proto.Marshal(i); err != nil {
-		fmt.Println("protoImage marshal err")
+		logger.Debug("protoImage marshal err")
 		return nil,err
 	} else {
 		bc := make([]byte, len(b))
 		ht := make([]int, 64<<10)
 		n, err := lz4.CompressBlock(b, bc, ht)
 		if err != nil {
-			fmt.Println(err)
+			logger.Debug(err)
 			return nil,err
 		}
 		if n >= len(b) {
-			fmt.Println("image is not compressible")
+			logger.Debug("image is not compressible")
 			return nil,errors.New("compressed len is 0")
 		}
 		bc = bc[:n]
@@ -124,8 +145,8 @@
 			case <-ticker.C:
 				return
 			default:
-				if feas,ok := resultMap.Get(s.Id);ok {
-					s.Result = feas
+				if faces,ok := resultMap.Get(s.Id);ok {
+					s.Result = faces
 					return
 				}
 			}
@@ -145,7 +166,7 @@
 	defer picMat.Close()
 
 	if picMat.Empty() {
-		fmt.Println("file not exist")
+		logger.Debug("file not exist")
 		return i
 	}
 	height := int32(picMat.Rows())
@@ -163,7 +184,7 @@
 		Id:        timeUnix,
 	}
 	i.Cid = uuid.NewV4().String() //鏁版嵁鍞竴id
-	fmt.Println("gocv read img completed")
+	logger.Debug("gocv read img completed")
 	return i
 }
 
@@ -172,35 +193,35 @@
 }
 
 type FeaResult struct {
-	FeaM map[string][]*protomsg.ResultFaceExtCom
+	FaceM map[string][]*protomsg.ResultFaceDetect
 	Lock sync.Mutex
 }
 
-func (f *FeaResult) Write(id string,faceResult []*protomsg.ResultFaceExtCom){
+func (f *FeaResult) Write(id string,faceDetectResult []*protomsg.ResultFaceDetect){
 	f.Lock.Lock()
 	defer f.Lock.Unlock()
-	f.FeaM[id] = faceResult
+	f.FaceM[id] = faceDetectResult
 }
 
-func (f *FeaResult) Get(id string) ([]*protomsg.ResultFaceExtCom,bool){
+func (f *FeaResult) Get(id string) ([]*protomsg.ResultFaceDetect,bool){
 	f.Lock.Lock()
 	defer f.Lock.Unlock()
-	coms,ok := f.FeaM[id]
+	coms,ok := f.FaceM[id]
 	return coms,ok
 }
 
 func (f *FeaResult) Delete(id string){
 	f.Lock.Lock()
 	defer f.Lock.Unlock()
-	delete(f.FeaM,id)
+	delete(f.FaceM,id)
 }
 
 var resultMap =  FeaResult{}
 
 func InitService() {
-	fmt.Println("service init!")
+	logger.Debug("service init!")
 	imgPushChan = make(chan []byte)
-	resultMap.FeaM = make(map[string][]*protomsg.ResultFaceExtCom,0)
+	resultMap.FaceM = make(map[string][]*protomsg.ResultFaceDetect,0)
 	client_push = deliver.NewClient(deliver.PushPull, Url_Service_PUSH)
 	client_pull = deliver.NewClient(deliver.PushPull, Url_Service_PULL)
 	defer func() {
@@ -216,13 +237,13 @@
 	for {
 		select {
 		case d := <-imgPushChan:
-			fmt.Println("imgPushChan in")
+			logger.Debug("imgPushChan in")
 			err := client_push.Send(d)
 			if err != nil {
-				fmt.Println("img Send err:", err)
+				logger.Debug("img Send err:", err)
 			}
 		default:
-			//fmt.Println("no img in")
+			//logger.Debug("no img in")
 		}
 	}
 }
@@ -231,60 +252,46 @@
 	for {
 		resultBytes, err := client_pull.Recv()
 		if err != nil {
-			//fmt.Println("pull err:", err)
+			//logger.Debug("pull err:", err)
 			continue
 		}
 		rMsg := protomsg.SdkMessage{}
 		if err := proto.Unmarshal(resultBytes, &rMsg); err == nil {
-			fmt.Println("received MSG:", rMsg.Cid)
+			logger.Debug("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))
+				logger.Debug("Len(sdkInfos)=",len(sdkInfos))
 				for _,swt :=range sdkInfos{
 					//浜鸿劯妫�娴嬬殑缁撴灉
-					//if swt.Sdktype =="FaceDetect"{
-					//	fmt.Println("浜鸿劯妫�娴嬬粨鏋滈暱搴︼細",len(swt.Sdkdata))
-					//	if len(swt.Sdkdata)>1{
-					//		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("浜鸿劯鎻愬彇缁撴灉闀垮害:",len(swt.Sdkdata))
-						if len(swt.Sdkdata) > 1{
-							var pff protomsg.ParamFaceFeature
-							if err := proto.Unmarshal(swt.Sdkdata, &pff);err !=nil{
-								//fmt.Println("ParamFaceFeature unmarshal err:",err)
+					if swt.Sdktype =="FaceDetect"{
+						logger.Debug("浜鸿劯妫�娴嬬粨鏋滈暱搴︼細",len(swt.Sdkdata))
+						if len(swt.Sdkdata)>1{
+							var pfp protomsg.ParamFacePos
+							err := proto.Unmarshal(swt.Sdkdata, &pfp)
+							if err !=nil {
+								logger.Debug("faceDetect result unmarshal err:",err)
 							} else {
-								fmt.Println("鐩爣鏁帮細",len(pff.ExtComp))
-								if len(pff.ExtComp) >0 {
-									resultMap.Write(perId,pff.ExtComp)
-
-									//for _,fea :=range pff.ExtComp{
-									//	base64Fea := base64.StdEncoding.EncodeToString(fea.Feats)
-									//	fmt.Println("perId:",perId)
-									//	fmt.Println("faceFeature:",base64Fea)
-									//}
+								logger.Debug("妫�娴嬩汉鑴告暟:",len(pfp.Faces))
+								if len(pfp.Faces)>0{
+									resultMap.Write(perId,pfp.Faces)
 								}
+								//for _,face :=range pfp.Faces{
+									//logger.Debug("FacePos:",face.Pos)
+									//logger.Debug("ThftResult:",face.Result)
+									//	base64Fea := base64.StdEncoding.EncodeToString(face.Feats)
+									//	logger.Debug("perId:",perId)
+									//	logger.Debug("faceFeature:",base64Fea)
+								//}
 							}
-							break
 						}
+						break
 					}
 				}
 			}
 		} else {
-			fmt.Println("recv msg Err:", err)
+			logger.Debug("recv msg Err:", err)
 		}
 
 	}

--
Gitblit v1.8.0