Mercurial > hg > nsaunier > traffic-intelligence

comparison python/events.py @ 887:e2452abba0e7

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
added option to compute PET in safety analysis script, and save in database
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 22 Mar 2017 10:44:24 -0400
parents 4749b71aa7fb
children b67a784beb69
comparison
equal deleted inserted replaced
886:d2eb8c93f7de 887:e2452abba0e7
163 163
164 def plotOnWorldImage(self, nPixelsPerUnitDistance, options = '', withOrigin = False, timeStep = 1, **kwargs): 164 def plotOnWorldImage(self, nPixelsPerUnitDistance, options = '', withOrigin = False, timeStep = 1, **kwargs):
165 self.roadUser1.plotOnWorldImage(nPixelsPerUnitDistance, options, withOrigin, timeStep, **kwargs) 165 self.roadUser1.plotOnWorldImage(nPixelsPerUnitDistance, options, withOrigin, timeStep, **kwargs)
166 self.roadUser2.plotOnWorldImage(nPixelsPerUnitDistance, options, withOrigin, timeStep, **kwargs) 166 self.roadUser2.plotOnWorldImage(nPixelsPerUnitDistance, options, withOrigin, timeStep, **kwargs)
167 167
168 def play(self, videoFilename, homography = None, undistort = False, intrinsicCameraMatrix = None, distortionCoefficients = None, undistortedImageMultiplication = 1.): 168 def play(self, videoFilename, homography = None, undistort = False, intrinsicCameraMatrix = None, distortionCoefficients = None, undistortedImageMultiplication = 1., allUserInstants = False):
169 if self.roadUser1 is not None and self.roadUser2 is not None: 169 if self.roadUser1 is not None and self.roadUser2 is not None:
170 cvutils.displayTrajectories(videoFilename, [self.roadUser1, self.roadUser2], homography = homography, firstFrameNum = self.getFirstInstant(), lastFrameNumArg = self.getLastInstant(), undistort = undistort, intrinsicCameraMatrix = intrinsicCameraMatrix, distortionCoefficients = distortionCoefficients, undistortedImageMultiplication = undistortedImageMultiplication) 170 if allUserInstants:
171 firstFrameNum = min(self.roadUser1.getFirstInstant(), self.roadUser2.getFirstInstant())
172 lastFrameNum = max(self.roadUser1.getLastInstant(), self.roadUser2.getLastInstant())
173 else:
174 firstFrameNum = self.getFirstInstant()
175 lastFrameNum = self.getLastInstant()
176 cvutils.displayTrajectories(videoFilename, [self.roadUser1, self.roadUser2], homography = homography, firstFrameNum = firstFrameNum, lastFrameNumArg = lastFrameNum, undistort = undistort, intrinsicCameraMatrix = intrinsicCameraMatrix, distortionCoefficients = distortionCoefficients, undistortedImageMultiplication = undistortedImageMultiplication)
171 else: 177 else:
172 print('Please set the interaction road user attributes roadUser1 and roadUser1 through the method setRoadUsers') 178 print('Please set the interaction road user attributes roadUser1 and roadUser1 through the method setRoadUsers')
173 179
174 def computeIndicators(self): 180 def computeIndicators(self):
175 '''Computes the collision course cosine only if the cosine is positive''' 181 '''Computes the collision course cosine only if the cosine is positive'''
235 pPETs[i] = prediction.SafetyPoint.computeExpectedIndicator(cz) 241 pPETs[i] = prediction.SafetyPoint.computeExpectedIndicator(cz)
236 self.addIndicator(indicators.SeverityIndicator(Interaction.indicatorNames[9], pPETs, mostSevereIsMax=False)) 242 self.addIndicator(indicators.SeverityIndicator(Interaction.indicatorNames[9], pPETs, mostSevereIsMax=False))
237 # TODO add probability of collision, and probability of successful evasive action 243 # TODO add probability of collision, and probability of successful evasive action
238 244
239 def computePET(self, collisionDistanceThreshold): 245 def computePET(self, collisionDistanceThreshold):
240 # TODO add crossing zone 246 pet, t1, t2= moving.MovingObject.computePET(self.roadUser1, self.roadUser2, collisionDistanceThreshold)
241 pet = moving.MovingObject.computePET(self.roadUser1, self.roadUser2, collisionDistanceThreshold) 247 if pet is not None:
242 if pet is not None: # TODO get crossing zone and time 248 self.addIndicator(indicators.SeverityIndicator(Interaction.indicatorNames[10], {min(t1, t2): pet}, mostSevereIsMax = False))
243 self.addIndicator(indicators.SeverityIndicator(Interaction.indicatorNames[10], {0: pet}, mostSevereIsMax = False))
244 249
245 def setCollision(self, collision): 250 def setCollision(self, collision):
246 '''indicates if it is a collision: argument should be boolean''' 251 '''indicates if it is a collision: argument should be boolean'''
247 self.collision = collision 252 self.collision = collision
248 253