package main
|
|
import (
|
"analysis/demo"
|
"analysis/valib/ipc"
|
srv "analysis/work/service"
|
"bytes"
|
"context"
|
"encoding/gob"
|
"encoding/json"
|
"flag"
|
"fmt"
|
"strconv"
|
"time"
|
)
|
|
var (
|
picFolder string
|
|
ipcURL string
|
streamURL string
|
|
ipcServer bool
|
|
procType string
|
procCount int
|
)
|
|
func init() {
|
flag.StringVar(&streamURL, "i", "rtsp://192.168.1.203:8554/16.mkv", "input url")
|
flag.StringVar(&picFolder, "f", ".", "test pic folder")
|
|
flag.StringVar(&ipcURL, "ipc", "ipc:///tmp/piipc", "ipc label")
|
|
flag.BoolVar(&ipcServer, "server", false, "run ipc as server")
|
|
flag.IntVar(&procCount, "c", 1, "proc run count")
|
}
|
|
const (
|
sender = "snd"
|
reciever = "rcv"
|
)
|
|
// 每一个任务标签
|
type TaskLabel struct {
|
Taskid string
|
Sdkids []string
|
Index int
|
}
|
|
//每一条加工后的数据流
|
type SdkMessage struct {
|
Cid string
|
Tasklab TaskLabel
|
Data []byte
|
}
|
|
type ChSdkMsg struct {
|
sdkMsg SdkMessage
|
ch chan<- srv.ImageInfo
|
}
|
|
var mapSdkMsg = make(map[string]ChSdkMsg)
|
|
func recvSdkMsgInfoFromIPC(ctx context.Context, url string) {
|
ipc := ipc.NewClient(ctx, url)
|
fmt.Println("ipc address: ", url)
|
|
for {
|
msg := ipc.Recv()
|
if msg != nil {
|
|
var m SdkMessage
|
if err := json.Unmarshal(msg, &m); err == nil {
|
|
if v, ok := mapSdkMsg[m.Cid]; !ok {
|
imageChan := make(chan srv.ImageInfo)
|
mapSdkMsg[m.Cid] = ChSdkMsg{m, imageChan}
|
y := demo.NewYolo()
|
go y.ShowYoloWithName(imageChan, m.Cid)
|
} else {
|
|
fmt.Println("recv ipc : ", url, " cameraid : ", v.sdkMsg.Cid)
|
|
var buf bytes.Buffer
|
buf.Write(v.sdkMsg.Data)
|
dec := gob.NewDecoder(&buf)
|
var i srv.ImageInfo
|
if err := dec.Decode(&i); err != nil {
|
fmt.Println("gob decode CameraImage error", err)
|
continue
|
}
|
|
v.ch <- i
|
}
|
|
} else {
|
fmt.Println(err)
|
}
|
|
}
|
}
|
}
|
|
func main() {
|
flag.Parse()
|
|
fmt.Println("start test, pic folder: ", picFolder)
|
|
tcp := "tcp://192.168.1.124:"
|
port := 9000
|
ctx, cancel := context.WithCancel(context.Background())
|
for i := 0; i < procCount; i++ {
|
go recvSdkMsgInfoFromIPC(ctx, tcp+strconv.Itoa(port))
|
port++
|
}
|
for {
|
time.Sleep(time.Duration(2) * time.Second)
|
}
|
cancel()
|
}
|
|
func oneTest(ctx context.Context) {
|
imageChan := make(chan srv.ImageInfo)
|
d := srv.NewReciever(ctx, ipcURL, imageChan)
|
|
if ipcServer {
|
go d.RunAsServer()
|
} else {
|
go d.RunAsClient()
|
}
|
|
y := demo.NewYolo()
|
y.ShowYolo(imageChan)
|
|
}
|
