diff trafficintelligence/moving.py @ 1037:6a6c37eb3a74

added function to load prototype assignments
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Tue, 03 Jul 2018 23:58:16 -0400
parents 933588568bec
children fc7c0f38e8a6
line wrap: on
line diff
--- a/trafficintelligence/moving.py	Tue Jul 03 16:47:37 2018 -0400
+++ b/trafficintelligence/moving.py	Tue Jul 03 23:58:16 2018 -0400
@@ -1784,6 +1784,10 @@
         self.movingObject = o
     def __str__(self):
         return '{} {} {}'.format(self.filename, self.num, self.trajectoryType)
+    def __eq__(self, p2):
+        return self.filename == p2.filename and self.num == p2.num and self.trajectoryType == p2.trajectoryType
+    def __hash__(self):
+        return hash((self.filename, self.num, self.trajectoryType))
     
 ##################
 # Annotations