Example #1
0
        '--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()