Mercurial Hosting > traffic-intelligence
view trafficintelligence/processing.py @ 1039:5621e4ad2428
removing prefix option to loadtrajectories from SQLite
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Wed, 04 Jul 2018 12:24:37 -0400 |
parents | c6cf75a2ed08 |
children | 75a6ad604cc5 |
line wrap: on
line source
#! /usr/bin/env python '''Algorithms to process trajectories and moving objects''' import numpy as np from trafficintelligence import moving def extractSpeeds(objects, zone): speeds = {} objectsNotInZone = [] import matplotlib.nxutils as nx for o in objects: inPolygon = nx.points_inside_poly(o.getPositions().asArray().T, zone.T) if inPolygon.any(): objspeeds = [o.getVelocityAt(i).norm2() for i in range(int(o.length()-1)) if inPolygon[i]] speeds[o.num] = np.mean(objspeeds) # km/h else: objectsNotInZone.append(o) return speeds, objectsNotInZone