Function process_points

app.py:355–397

(traj_list, num_frames=81)

def process_points(traj_list, num_frames=81):


    if len(traj_list) < 2:     # First point
        return [traj_list[0]] * num_frames

    elif len(traj_list) >= num_frames:
        raise gr.Info("The number of trajectory points is more than limits, we will do cropping!")
        skip = len(traj_list) // num_frames
        return traj_list[::skip][: num_frames - 1] + traj_list[-1:]

    else:

        insert_num = num_frames - len(traj_list)
        insert_num_dict = {}
        interval = len(traj_list) - 1
        n = insert_num // interval
        m = insert_num % interval

        for i in range(interval):
            insert_num_dict[i] = n

        for i in range(m):
            insert_num_dict[i] += 1

        res = []
        for i in range(interval):
            insert_points = []
            x0, y0 = traj_list[i]
            x1, y1 = traj_list[i + 1]

            delta_x = x1 - x0
            delta_y = y1 - y0
            for j in range(insert_num_dict[i]):
                x = x0 + (j + 1) / (insert_num_dict[i] + 1) * delta_x
                y = y0 + (j + 1) / (insert_num_dict[i] + 1) * delta_y
                insert_points.append([int(x), int(y)])

            res += traj_list[i : i + 1] + insert_points
        res += traj_list[-1:]

        # return
        return res