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)
|
}
|
|
}
|
}
|
