From a25aa1cd44030cb75cae87acde17affbecbf6f41 Mon Sep 17 00:00:00 2001
From: michaelpivato <michael.pivato@powerhealthsolutions.com>
Date: Wed, 18 Mar 2020 12:43:31 +1030
Subject: [PATCH] Add option to set the colour for points based on their group.

---
 tracking/animate_alg.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tracking/animate_alg.py b/tracking/animate_alg.py
index 4fd0abd..879b12c 100644
--- a/tracking/animate_alg.py
+++ b/tracking/animate_alg.py
@@ -25,6 +25,8 @@ def update_line(num, iterator, line, prev_groups):
     line.set_offsets(offsets)
     intens = np.array([meas[0] for meas in scan])
     line.set_array(intens)
+    # Set the colour matrix: Just set the colours to 2 * np.pi * group number (for every group number)
+    # line.set_color()
     return line,
 
 class Bunch: