'--overlays', type=bool, required=False, default=False, help='turn on overlays to plot wing angles and distance bars') args = parser.parse_args() zoom_fmf = args.zoom wide_fmf = args.wide savedir = args.savedir overlays_value = args.overlays #os.system(ffmpeg -f image2 -r 15 -i _tmp%05d.png -vcodec mpeg4 -y (_VIDEO_DIR + '/flymad_annotated.mp4')) vid = data_movies.FlyPanel(zoom_fmf, savedir, utilities.parse_fmftime(zoom_fmf)[0], overlays_value) image_height = vid.get_frame(0)[0].shape[0] image_width = vid.get_frame(0)[0].shape[1] for frame_number in range(vid.get_n_frames()): if os.path.exists(savedir + '/temp_png/_tmp%05d.png' % (frame_number)): continue fig = plt.figure(figsize=(image_width / 100, image_height * 1.2 / 100), dpi=200.399, frameon=False) make_panel(vid, 0, 0, frame_number)
'--bottomlabel', type=str, required=False, default="Experiment", help= 'Label for bottom data. example: P1>CsCHRIMSON & $\mathregular{SHIBIRE^{TS}}$' ) args = parser.parse_args() top_fmf = args.topfmf bottom_fmf = args.bottomfmf savedir = args.savedir #os.system(ffmpeg -f image2 -r 15 -i _tmp%05d.png -vcodec mpeg4 -y (_VIDEO_DIR + '/flymad_annotated.mp4')) ctrlfly = data_movies.FlyPanel(top_fmf, savedir, utilities.parse_fmftime(top_fmf)[0]) expfly = data_movies.FlyPanel(bottom_fmf, savedir, utilities.parse_fmftime(bottom_fmf)[0]) image_height = ctrlfly.get_frame(0)[0].shape[0] + expfly.get_frame( 0)[0].shape[0] image_width = expfly.get_frame(0)[0].shape[1] * 3 for frame_number in range(9450): #ctrlfly.get_n_frames()): if os.path.exists(savedir + '/temp_png/_tmp%05d.png' % (frame_number)): continue fig = plt.figure(figsize=(image_width / 100, image_height / 100), dpi=200.399)
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter) parser.add_argument('--fmf', type=str, required=True, help='path to fmf directory') parser.add_argument('--savedir', type=str, required=True, help='path to save directory') parser.add_argument('--label', type=str, required=False, default="Control", help='label for top data') args = parser.parse_args() fmf = args.fmf savedir = args.savedir #os.system(ffmpeg -f image2 -r 15 -i _tmp%05d.png -vcodec mpeg4 -y (_VIDEO_DIR + '/flymad_annotated.mp4')) fly = data_movies.FlyPanel(fmf, savedir, utilities.parse_fmftime(fmf)[0]) image_height = (fly.get_frame(0)[0].shape[0] )*2 image_width = fly.get_frame(0)[0].shape[1] *2: if not os.path.exists(savedir + '/temp_png): os.makedirs(savedir + '/temp_png) for frame_number in range(fly.get_n_frames()): if os.path.exists(savedir + '/temp_png/_tmp%05d.png'%(frame_number)): continue fig = plt.figure(figsize=(image_width/100, image_height/100), dpi=200.399 ) make_panel(fly, 0, 0, frame_number, args.label) #plt.show()