/usr/share/sumo/tools/traci/poi.py is in sumo-tools 0.15.0~dfsg-2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 | # -*- coding: utf-8 -*-
"""
@file poi.py
@author Michael Behrisch
@author Lena Kalleske
@date 2008-10-09
@version $Id: poi.py 11671 2012-01-07 20:14:30Z behrisch $
Python implementation of the TraCI interface.
SUMO, Simulation of Urban MObility; see http://sumo.sourceforge.net/
Copyright (C) 2008-2012 DLR (http://www.dlr.de/) and contributors
All rights reserved
"""
import struct, traci
import traci.constants as tc
_RETURN_VALUE_FUNC = {tc.ID_LIST: traci.Storage.readStringList,
tc.VAR_TYPE: traci.Storage.readString,
tc.VAR_POSITION: lambda(result): result.read("!dd"),
tc.VAR_COLOR: lambda(result): result.read("!BBBB")}
subscriptionResults = {}
def _getUniversal(varID, poiID):
result = traci._sendReadOneStringCmd(tc.CMD_GET_POI_VARIABLE, varID, poiID)
return _RETURN_VALUE_FUNC[varID](result)
def getIDList():
"""getIDList() -> list(string)
Returns a list of all pois in the network.
"""
return _getUniversal(tc.ID_LIST, "")
def getType(poiID):
"""getType(string) -> string
.
"""
return _getUniversal(tc.VAR_TYPE, poiID)
def getPosition(poiID):
"""getPosition(string) -> (double, double)
.
"""
return _getUniversal(tc.VAR_POSITION, poiID)
def getColor(poiID):
"""getColor(string) -> (integer, integer, integer, integer)
.
"""
return _getUniversal(tc.VAR_COLOR, poiID)
def subscribe(poiID, varIDs=(tc.VAR_POSITION,), begin=0, end=2**31-1):
"""subscribe(string, list(integer), double, double) -> None
Subscribe to one or more poi values for the given interval.
A call to this method clears all previous subscription results.
"""
_resetSubscriptionResults()
traci._subscribe(tc.CMD_SUBSCRIBE_POI_VARIABLE, begin, end, poiID, varIDs)
def _resetSubscriptionResults():
subscriptionResults.clear()
def _addSubscriptionResult(poiID, varID, data):
if poiID not in subscriptionResults:
subscriptionResults[poiID] = {}
subscriptionResults[poiID][varID] = _RETURN_VALUE_FUNC[varID](data)
def getSubscriptionResults(poiID=None):
"""getSubscriptionResults(string) -> dict(integer: <value_type>)
Returns the subscription results for the last time step and the given poi.
If no poi id is given, all subscription results are returned in a dict.
If the poi id is unknown or the subscription did for any reason return no data,
'None' is returned.
It is not possible to retrieve older subscription results than the ones
from the last time step.
"""
if poiID == None:
return subscriptionResults
return subscriptionResults.get(poiID, None)
def setType(poiID, poiType):
traci._beginMessage(tc.CMD_SET_POI_VARIABLE, tc.VAR_TYPE, poiID, 1+4+len(poiType))
traci._message.string += struct.pack("!Bi", tc.TYPE_STRING, len(poiType)) + poiType
traci._sendExact()
def setPosition(poiID, x, y):
traci._beginMessage(tc.CMD_SET_POI_VARIABLE, tc.VAR_POSITION, poiID, 1+8+8)
traci._message.string += struct.pack("!Bdd", tc.POSITION_2D, x, y)
traci._sendExact()
def setColor(poiID, color):
traci._beginMessage(tc.CMD_SET_POI_VARIABLE, tc.VAR_COLOR, poiID, 1+1+1+1+1)
traci._message.string += struct.pack("!BBBBB", tc.TYPE_COLOR, int(color[0]), int(color[1]), int(color[2]), int(color[3]))
traci._sendExact()
def add(poiID, x, y, color, poiType="", layer=0):
traci._beginMessage(tc.CMD_SET_POI_VARIABLE, tc.ADD, poiID, 1+4 + 1+4+len(poiType) + 1+1+1+1+1 + 1+4 + 1+8+8)
traci._message.string += struct.pack("!Bi", tc.TYPE_COMPOUND, 4)
traci._message.string += struct.pack("!Bi", tc.TYPE_STRING, len(poiType)) + poiType
traci._message.string += struct.pack("!BBBBB", tc.TYPE_COLOR, int(color[0]), int(color[1]), int(color[2]), int(color[3]))
traci._message.string += struct.pack("!Bi", tc.TYPE_INTEGER, layer)
traci._message.string += struct.pack("!Bdd", tc.POSITION_2D, x, y)
traci._sendExact()
def remove(poiID, layer=0):
traci._beginMessage(tc.CMD_SET_POI_VARIABLE, tc.REMOVE, poiID, 1+4)
traci._message.string += struct.pack("!Bi", tc.TYPE_INTEGER, layer)
traci._sendExact()
|