From 616d017a0dd0cb92cdc1aa1d41c39fc539d8f7f7 Mon Sep 17 00:00:00 2001
From: sipp11 <sipp11@zzyzx.co>
Date: Mon, 28 Oct 2019 18:37:52 +0900
Subject: [PATCH] ID assignment delayed a bit

---
 src/main.py | 53 +++++++++++++++++++++++++++++------------------------
 1 file changed, 29 insertions(+), 24 deletions(-)

diff --git a/src/main.py b/src/main.py
index 8c58efe..f85787a 100644
--- a/src/main.py
+++ b/src/main.py
@@ -274,6 +274,33 @@ while True:
                     trackers[idx]["status"] = "dont-care"
                     untracking.append(trackers[idx])
 
+        # TODO: assign id here! for anything that is -1 id
+        gid = None
+        if tk['id'] == -1 and 5 < h_count and h_count < 15:
+            _area_id = f"id_{tk['origin']['id']}"
+            if _area_id in W4A:
+                # check candidates which has "matched" opportunity too
+                _po = W4A[_area_id]["objects"]
+                _po = [_ for _ in _po if _frame_count > _[2]]  # opportunity
+                _po = sorted(_po, key=lambda kk: kk[2])  # first one first
+                if _po:
+                    trackers[idx]['id'] = gid = _po[0][0]["id"]
+                    # remove this id out of next W4A
+                    W4A[_area_id]["objects"] = [
+                        _ for _ in W4A[_area_id]["objects"] if _[0]["id"] != gid
+                    ]
+                    # print(f" --- {len(_po)} candicate: picked id={gid}")
+                # else:
+                #     print(f" --- no candidate")
+                print(f"         {gid} RE-ENTERS")
+        elif tk['id'] == -1 and h_count >= 15:
+            # assign an id
+            trackers[idx]['id'] = tracker_counter
+            tracker_counter += 1
+            print(f"         {trackers[idx]['id']} ENTERS (& ID assigned)")
+
+
+
         for lvng_cnd in OBJ_LEAVING_COND:
             _origin = tk["origin"]["id"]
             cond = dict(lvng_cnd)
@@ -340,7 +367,7 @@ while True:
         cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)
         cv2.putText(
             frame,
-            f"{tk['id']} - {tk['type']}",
+            f"{tk['id'] if tk['id'] > 0 else '?'} - {tk['type']}",
             (x, y - 5),
             cv2.FONT_HERSHEY_SIMPLEX,
             0.5,
@@ -404,29 +431,12 @@ while True:
                     break
 
             if not is_same:
-                gid = None
-                _area_id = f"id_{found_at['id']}"
-                if _area_id in W4A:
-                    # check candidates which has "matched" opportunity too
-                    _po = W4A[_area_id]["objects"]
-                    _po = [_ for _ in _po if _frame_count > _[2]]  # opportunity
-                    _po = sorted(_po, key=lambda kk: kk[2])  # first one first
-                    if _po:
-                        gid = _po[0][0]["id"]
-                        # remove this id out of next W4A
-                        W4A[_area_id]["objects"] = [
-                            _ for _ in W4A[_area_id]["objects"] if _[0]["id"] != gid
-                        ]
-                        # print(f" --- {len(_po)} candicate: picked id={gid}")
-                    # else:
-                    #     print(f" --- no candidate")
-
                 # create tracker and add to multi-object tracker
                 _tracker = OPENCV_OBJECT_TRACKERS["csrt"]()
                 bbox = (x, y, w, h)
                 cv_trackers.add(_tracker, frame, bbox)
                 t = {
-                    "id": tracker_counter if gid is None else gid,
+                    "id": -1,  # NOTE: assign later; tracker_counter if gid is None else gid,
                     "type": _class,
                     "status": "",
                     "curr_position": bbox,
@@ -437,11 +447,6 @@ while True:
                     "last_position": bbox,
                     "still": 0,
                 }
-                if gid is None:
-                    print(f"         {t['id']} ENTERS")
-                    tracker_counter += 1
-                else:
-                    print(f"         {t['id']} RE-ENTERS")
                 trackers.append(t)
                 print(f"         total #{len(trackers)}")
                 pp.pprint(t)