plottraffic.py 文件源码

def updatePlot(self):

        # Lists for positions of cars
        x=[]
        y=[]
    color = []
    carSymbol = []
        for car in Plotter.instance()._trafficManager.cars:
            x.append(car.getPosition())
            y.append((car.getLane() * self._laneWidth) - (self._laneWidth/2.))
        color.append(pg.mkBrush(car.getColor()))          
        if (car.getType() == 'b'):
        carSymbol.append('d')
        elif (car.getType() == 's'):
                carSymbol.append('o')
        elif (car.getType() == 'a'):
        carSymbol.append('+')   
        else:
        carSymbol.append('t')  

        self._pw.plot(x, y, clear=True, pen=None, symbol=carSymbol, symbolSize=20, symbolBrush = color)
        self._pw.addItem(self._backgroundImage)
        self._backgroundImage.setZValue(-100)  # make sure image is behind other data
        self._backgroundImage.setRect(pg.QtCore.QRectF(0, 0, self._roadLength, self._roadWidth))
        pg.QtGui.QApplication.processEvents()