| | |
| | | |
| | | type channel struct { |
| | | index int |
| | | msg chan protomsg.SdkMessage |
| | | msgs *sdkhelper.LockList |
| | | } |
| | | |
| | | ////////////////////////////////////////////////////////////////// |
| | | const ( |
| | | cacheFrameNum = 3 |
| | | trackChnTimeout = time.Duration(10) |
| | | ) |
| | | |
| | |
| | | |
| | | c := &channel{ |
| | | index: index, |
| | | msg: make(chan protomsg.SdkMessage, cacheFrameNum), |
| | | msgs: sdkhelper.NewLockList(6), |
| | | } |
| | | |
| | | f.mapChan[id] = c |
| | |
| | | |
| | | for _, msg := range msgs { |
| | | if c, ok := f.channelExist(msg.Cid); ok { |
| | | f.fnLogger("######push image") |
| | | c.msg <- msg |
| | | f.fnLogger("######push over image") |
| | | f.fnLogger("######push image: ", c.index) |
| | | c.msgs.Push(msg) |
| | | f.fnLogger("######push over image: ", c.index) |
| | | } else { |
| | | |
| | | nc := f.channelNew(msg.Cid) |
| | |
| | | ret := f.handle.TrackerResize(imgW, imgH, nc.index) |
| | | f.fnLogger("ResizeFaceTracker: cid: ", msg.Cid, " chan: ", nc.index, " wXh: ", imgW, "x", imgH, " result:", ret) |
| | | go f.detectTrackOneChn(nc, out) |
| | | nc.msg <- msg |
| | | nc.msgs.Push(msg) |
| | | } |
| | | } |
| | | f.fnLogger("######run over batch images") |
| | |
| | | |
| | | for { |
| | | select { |
| | | case rMsg := <-c.msg: |
| | | |
| | | case <-time.After(trackChnTimeout * time.Second): |
| | | f.fnLogger("Timeout to get image, curCid:", curCid) |
| | | if curCid != "" { |
| | | f.channelRelease(curCid) |
| | | } |
| | | return |
| | | default: |
| | | |
| | | elems := c.msgs.Drain() |
| | | var msgs []protomsg.SdkMessage |
| | | for _, v := range elems { |
| | | msgs = append(msgs, v.(protomsg.SdkMessage)) |
| | | } |
| | | for _, rMsg := range msgs { |
| | | |
| | | if !sdkhelper.ValidRemoteMessage(rMsg, f.typ, f.fnLogger) { |
| | | sdkhelper.EjectResult(nil, rMsg, out) |
| | |
| | | // conv to bgr24 and resize |
| | | imgW, imgH := int(i.Width), int(i.Height) |
| | | |
| | | f.fnLogger("######face sdk run") |
| | | f.fnLogger("######face sdk run: ", dtchn) |
| | | count, data, _ := f.handle.Run(i.Data, imgW, imgH, 3, dtchn) |
| | | f.fnLogger("######face over sdk run") |
| | | f.fnLogger("######face over sdk run: ", dtchn) |
| | | |
| | | f.fnLogger("######face eject data") |
| | | f.fnLogger("######face eject data: ", dtchn) |
| | | sdkhelper.EjectResult(data, rMsg, out) |
| | | f.fnLogger("######face over eject data") |
| | | f.fnLogger("######face over eject data: ", dtchn) |
| | | |
| | | var id, name string |
| | | if rMsg.Tasklab != nil { |
| | | id, name = rMsg.Tasklab.Taskid, rMsg.Tasklab.Taskname |
| | | } |
| | | f.fnLogger("CAMERAID: ", rMsg.Cid, " TASKID: ", id, " TASKNAME: ", name, " DETECT ", f.typ, " COUNT: ", count) |
| | | } |
| | | |
| | | sc++ |
| | | if sc == 25 { |
| | |
| | | sc = 0 |
| | | tm = time.Now() |
| | | } |
| | | case <-time.After(trackChnTimeout * time.Second): |
| | | f.fnLogger("Timeout to get image, curCid:", curCid) |
| | | if curCid != "" { |
| | | f.channelRelease(curCid) |
| | | } |
| | | return |
| | | |
| | | } |
| | | } |
| | | } |