Source code for src.Record.frame

[docs]class Frame:
    def __init__(self):
        '''

        Frame include all the vehicle state information
        entities::List{Vehicle} , Scene
        n::Int
        '''
        self.entities = []
        self.n = 0

    def __getitem__(self, item):
        return self.entities[item]

[docs]    def findfirst(self, id: int):
        '''

        :param id: the vehicle id
        :return: the first vehicle's index who has the corresponding id
        '''
        for entity_index in range(self.n):
            entity = self.entities[entity_index]
            if entity.id == id:
                return entity_index

        return None

[docs]    def deleteat(self, entity_index: int):
        '''

        :param entity_index: the index of the vehicle that we want to delete
        :return: no return
        '''
        for i in range(entity_index, self.n - 1):
            self.entities[i] = self.entities[i + 1]
        self.n -= 1

[docs]    def delete_by_id(self, id: int):
        '''

        :param id: the id of the vehicle that you want to delete
        :return: no return
        '''
        entity_index = self.findfirst(id)
        if entity_index is not None:
            self.deleteat(entity_index)

[docs]    def init(self, n: int):
        '''

        :param n: the number of vehicle
        :return: no return
        '''
        for i in range(n):
            self.entities.append(None)

[docs]    def push(self, entity):
        '''

        :param entity: the entity to push back
        :return: no return
        '''
        self.entities[self.n] = entity
        self.n += 1

[docs]    def empty(self):
        '''

        :return: no return, empty the whole list
        '''
        self.n = 0

    def __setitem__(self, key, value):
        self.entities[key] = value


[docs]def copyto(dest: Frame, src: Frame):
    '''

    :param dest: the frame we are copying to
    :param src: the frame we are copying from
    :return: destination frame
    '''
    for i in range(src.n):
        dest.entities[i] = src.entities[i]
    dest.n = src.n
    return dest