diff scripts/safety-analysis.py @ 948:584b9405e494

added safety analysis parameters for motion patterns
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Fri, 21 Jul 2017 12:11:55 -0400
parents e5970606066f
children d6c1c05d11f5
line wrap: on
line diff
--- a/scripts/safety-analysis.py	Fri Jul 21 11:31:42 2017 -0400
+++ b/scripts/safety-analysis.py	Fri Jul 21 12:11:55 2017 -0400
@@ -56,7 +56,7 @@
         prototypes = storage.loadPrototypesFromSqlite(args.prototypeDatabaseFilename)
     for p in prototypes:
         p.getMovingObject().computeCumulativeDistances()
-    predictionParameters = prediction.PrototypePredictionParameters(prototypes, params.nPredictedTrajectories, 2., 0.4, 'cityblock', 10, params.constantSpeedPrototypePrediction, params.useFeaturesForPrediction)
+    predictionParameters = prediction.PrototypePredictionParameters(prototypes, params.nPredictedTrajectories, params.maxLcssDistance, params.minLcssSimilarity, params.lcssMetric, params.minFeatureTime, params.constantSpeedPrototypePrediction, params.useFeaturesForPrediction)
 # else:
 # no else required, since parameters is required as argument
 
@@ -68,16 +68,13 @@
 #                                                                                  params.useFeaturesForPrediction)
 
 objects = storage.loadTrajectoriesFromSqlite(params.databaseFilename, 'object', args.nObjects, withFeatures = (params.useFeaturesForPrediction or predictionMethod == 'ps' or predictionMethod == 'mp'))
-# if params.useFeaturesForPrediction:
-#     features = storage.loadTrajectoriesFromSqlite(params.databaseFilename,'feature') # needed if normal adaptation
-#     for obj in objects:
-#         obj.setFeatures(features)
 
 interactions = events.createInteractions(objects)
 for inter in interactions:
+    print('processing interaction {}'.format(inter.getNum())
     inter.computeIndicators()
     if not args.noMotionPrediction:
-        inter.computeCrossingsCollisions(predictionParameters, params.collisionDistance, params.predictionTimeHorizon, params.crossingZones, nProcesses = args.nProcesses, debug = True)
+        inter.computeCrossingsCollisions(predictionParameters, params.collisionDistance, params.predictionTimeHorizon, params.crossingZones, nProcesses = args.nProcesses)
 
 if args.computePET:
     for inter in interactions: