Mercurial Hosting > traffic-intelligence

diff python/moving.py @ 501:c81cbd6953fb

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
update to classify speed to remove data at both ends
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Tue, 20 May 2014 15:18:55 -0400
parents 6464e4f0cc26
children a40c75f04903
line wrap: on
line diff
--- a/python/moving.py	Fri May 09 14:09:14 2014 -0400
+++ b/python/moving.py	Tue May 20 15:18:55 2014 -0400
@@ -805,11 +805,15 @@
         at constant speed'''
         return predictPositionNoLimit(nTimeSteps, self.getPositionAtInstant(instant), self.getVelocityAtInstant(instant), externalAcceleration)
 
-    def classifyUserTypeSpeed(self, threshold, statisticsFunc = median):
+    def classifyUserTypeSpeed(self, threshold, statisticsFunc = median, ignoreNInstantsAtEnds = 0):
         '''Classifies slow and fast road users
         slow: non-motorized -> pedestrians
         fast: motorized -> cars'''
-        if statisticsFunc(self.velocities.norm()) >= threshold:
+        if ignoreNInstantsAtEnds > 0:
+            speeds = self.velocities.norm()[ignoreNInstantsAtEnds:-ignoreNInstantsAtEnds]
+        else:
+            speeds = self.velocities.norm()
+        if statisticsFunc(speeds) >= threshold:
             self.setUserType(userType2Num['car'])
         else:
             self.setUserType(userType2Num['pedestrian'])