Method updateScene

limsim/simModel/egoTracking/movingScene.py:210–269

(self, dataBase: str, timeStep: int)

        self.junctions: set = None

    def updateScene(self, dataBase: str, timeStep: int):
        ex, ey = self.ego.x, self.ego.y
        currGeox = int(ex // 100)
        currGeoy = int(ey // 100)

        sceGeohashIDs = (
            (currGeox-1, currGeoy-1),
            (currGeox, currGeoy-1),
            (currGeox+1, currGeoy-1),
            (currGeox-1, currGeoy),
            (currGeox, currGeoy),
            (currGeox+1, currGeoy),
            (currGeox-1, currGeoy+1),
            (currGeox, currGeoy+1),
            (currGeox+1, currGeoy+1),
        )

        NowEdges: set = set()
        NowJuncs: set = set()

        for sgh in sceGeohashIDs:
            try:
                geohash = self.netInfo.geoHashes[sgh]
            except KeyError:
                continue
            NowEdges = NowEdges | geohash.edges
            NowJuncs = NowJuncs | geohash.junctions

        self.edges = NowEdges
        self.junctions = NowJuncs

        NowTLs = {}
        conn = sqlite3.connect(dataBase)
        cur = conn.cursor()
        cur.execute(
            '''SELECT * FROM trafficLightStates WHERE frame=%i;''' % timeStep)
        tlsINFO = cur.fetchall()
        if tlsINFO:
            for tls in tlsINFO:
                frame, tlid, currPhase, nextPhase, switchTime = tls
                NowTLs[tlid] = (currPhase, nextPhase, switchTime)

        cur.close()
        conn.close()

        if NowTLs:
            for jid in NowJuncs:
                junc = self.netInfo.getJunction(jid)
                if junc:
                    for jlid in junc.JunctionLanes:
                        jl = self.netInfo.getJunctionLane(jlid)
                        tlid = jl.tlLogic
                        if tlid:
                            try:
                                currPhase, nextPhase, switchTime = NowTLs[tlid]
                            except KeyError:
                                continue
                            jl.currTlState = currPhase[jl.tlsIndex]