Evo: turning keeping all history in memory optional #11

README.md

@@ -1,10 +1,10 @@
 # node-moving-things-tracker
 
-node-moving-things-tracker is a javascript implementation of the _"tracker by detections"_ for realtime multiple object tracking (MOT) in node.js / browsers. 
+node-moving-things-tracker is a javascript implementation of a _"tracker by detections"_ for realtime multiple object tracking (MOT) in node.js / browsers. 
 
 Commissioned by moovel lab for [Beat the Traffic X](http://beatthetraffic.moovellab.com/) and the [Open Data Cam](http://opendatacam.moovellab.com/) project.
 
-## confidencelem
+## Problem
 
 How to track persistently multiple moving things from frame-by-frame object detections inputs? How to assign an unique identifier to frame-by-frame object detection results?
 

main.js

@@ -52,6 +52,11 @@ var detections = {}
 // Store tracker output
 var tracker = {}
 
+// If MODE is BEATTHETRAFFIC keep all tracker history in memory
+if(MODE_BEATTHETRAFFIC) {
+  Tracker.enableKeepInMemory();
+}
+
 // Parse detections input
 fs.readFile(`${pathRawDetectionsInput}`, function(err, f){
     var lines = f.toString().split('\n');

package.json

@@ -1,6 +1,6 @@
 {
   "name": "node-moving-things-tracker",
-  "version": "0.5.4",
+  "version": "0.6.0",
   "description": "Tracker by detections in javascript for node.js / browsers",
   "url": "https://github.com/tdurand/node-moving-things-tracker",
   "main": "main.js",

tracker.js

@@ -21,6 +21,9 @@ var mapOfItemsTracked = new Map();
 // Useful to ouput the file of all items tracked
 var mapOfAllItemsTracked = new Map();
 
+// By default, we do not keep all the history in memory
+var keepAllHistoryInMemory = false;
+
 // Implementation detail, we store the distance in a KDTREE, we want to be able to exclude values from 
 // the kdtree search by assigning them KDTREESEARCH_LIMIT + 1
 var KDTREESEARCH_LIMIT = 10000;
@@ -202,7 +205,9 @@ exports.updateTrackedItemsWithNewFrame = function(detectionsOfThisFrame, frameNb
         if(itemTracked.isDead()) {
           mapOfItemsTracked.delete(itemTracked.id);
           treeItemsTracked.remove(itemTracked);
-          mapOfAllItemsTracked.set(itemTracked.id, itemTracked);
+          if(keepAllHistoryInMemory) {
+            mapOfAllItemsTracked.set(itemTracked.id, itemTracked);
+          }
         }
       }
     });
@@ -215,6 +220,14 @@ exports.reset = function() {
   mapOfAllItemsTracked = new Map();
 }
 
+exports.enableKeepInMemory = function() {
+  keepAllHistoryInMemory = true;
+}
+
+exports.disableKeepInMemory = function() {
+  keepAllHistoryInMemory = false;
+}
+
 exports.getJSONOfTrackedItems = function() {
   return Array.from(mapOfItemsTracked.values()).map(function(itemTracked) {
     return itemTracked.toJSON();
@@ -227,10 +240,12 @@ exports.getJSONDebugOfTrackedItems = function() {
   });
 };
 
+// Work only if keepInMemory is enabled
 exports.getAllTrackedItems = function() {
   return mapOfAllItemsTracked;
 };
 
+// Work only if keepInMemory is enabled
 exports.getJSONOfAllTrackedItems = function() {
   return Array.from(mapOfAllItemsTracked.values()).map(function(itemTracked) {
     return itemTracked.toJSONGenericInfo();