def select_laps_to_render(videos, lap_comparison_mode=False, select_sessions=False): if select_sessions: return select_sessions_to_render(videos) laps = {} for video in videos: for lap in video.matched_laps: print lap['lap'] key = float(lap['lap']) laps[key] = lap lap["render"] = False keys = laps.keys() keys.sort() title = 'Select laps to render' if lap_comparison_mode: title = "Select at most 2 laps to render in side-by-side mode" picker = Picker(title=title, options=keys) picker.start() opts = picker.getSelected() if lap_comparison_mode: opts = opts[:2] for lap in opts: laps[lap]['render'] = True
def select_sessions_to_render(videos): s_videos = {} for video in videos: key = str(video) s_videos[key] = video for lap in video.matched_laps: lap["render"] = False keys = s_videos.keys() keys.sort() title = 'Select sessions to render' picker = Picker(title=title, options=keys) picker.window_width = 150 picker.window_height = 30 picker.start() opts = picker.getSelected() for videoname in opts: video = s_videos[videoname] for lap in video.matched_laps: lap['render'] = True