Exemplo n.º 1
0
def plot_laser(ax,arena,ldf,tdf,hdf,geom):
    madplot.plot_laser_trajectory(ax, ldf, arena)
    ax.add_patch(arena.get_patch(color='k', alpha=0.1))
Exemplo n.º 2
0
def plot_data(arena, path, data):
    target_name = data['target_name']
    pool_df = data['pooldf']
    ldf = data['ldf']
    hdf = data['hdf']
    target_ranges = data['target_ranges']

    fig = plt.figure("Image Processing Time (%s)" % target_name)
    ax = fig.add_subplot(1, 1, 1)
    ax.hist(hdf['h_processing_time'].values, bins=100)
    ax.set_xlabel('processing time (s)')
    ax.set_title("Distribution of time taken for %s detection" % target_name)
    fig.savefig('imgproc_%s.png' % target_name)

    fig = plt.figure("TTM Tracking", figsize=(8, 8))
    ax = fig.add_subplot(1, 1, 1)
    ax.add_patch(arena.get_patch(color='k', alpha=0.1))
    madplot.plot_laser_trajectory(ax, ldf, arena)
    ax.set_title("The effect of TTM tracking on laser position.\n"\
                 "Values required to hit the fly %s" % target_name)
    ax.legend()
    fig.savefig('ttmeffect_%s.png' % target_name)

    all_v = []
    all_e = []

    #YAY finally plot
    for k in target_ranges:
        vmin, vmax = k

        fvels = "%s-%s px" % (vmin, vmax)

        figf = plt.figure("tFly %s/s" % fvels)
        axf = figf.add_subplot(1, 1, 1)
        axf_w = axf.twinx()
        figt = plt.figure("tTTM %s/s" % fvels)
        axt = figt.add_subplot(1, 1, 1)
        axt_w = axt.twinx()

        for trg in target_ranges[k]:
            #widefiled
            err = trg.wf_err
            axf_w.plot(err.dt_thisfly, wide_conv(err.err), 'r', alpha=0.3)
            axt_w.plot(err.dt_ttm, wide_conv(err.err), 'r', alpha=0.3)

            #ttm
            err = trg.ttm_err
            axf.plot(err.dt_thisfly, ttm_conv(err.err), 'k', alpha=0.2)
            axt.plot(err.dt_ttm, ttm_conv(err.err), 'k', alpha=0.2)

            all_v.append(trg.v)
            all_e.append(np.mean(ttm_conv(err.err)))

        for ax in (axf, axt):
            ax.set_xlim([0, 0.25])
            ax.set_xlabel('time (s)')
            ax.set_ylabel('TTM error (%s)' % ttm_unit)
            ax.set_ylim([0, 1.5])

        for ax in (axf_w, axt_w):
            ax.set_ylabel('deviation from WF COM (%s)' % wide_unit)
            ax.set_ylim([0, 4])

        for axttm, axwf in [(axt, axt_w), (axf, axf_w)]:
            axttm.set_zorder(axwf.get_zorder() + 1)  # put ax in front of ax2
            axttm.patch.set_visible(False)

        axf.set_title("Spacial accuracy of %s targeting (fly speed %s/s)\n"\
                      "(time since targeting this fly)" % (target_name,fvels))
        axt.set_title("Spacial accuracy of %s targeting (fly speed %s/s)\n"\
                      "(time since switching to TTM targeting)" % (target_name,fvels))

        figf.savefig(('tFly%s%s.png' % (fvels, target_name)).replace(' ', ''))
        figt.savefig(('tTTM%s%s.png' % (fvels, target_name)).replace(' ', ''))

    fig = plt.figure("Speed %s" % target_name)
    ax = fig.add_subplot(1, 1, 1)
    ax.scatter(wide_conv(all_v), all_e)
    ax.set_ylim([0, 1.5])
    ax.set_xlim([0, 2.5])
    ax.set_title("Spacial accuracy of %s targeting" % target_name)
    ax.set_xlabel('fly speed (%s/s)' % wide_unit)
    ax.set_ylabel('error (%s)' % ttm_unit)
    fig.savefig('flyv_errpx_%s.png' % target_name)
Exemplo n.º 3
0
def plot_laser(ax, arena, ldf, tdf, hdf, geom):
    madplot.plot_laser_trajectory(ax, ldf, arena)
    ax.add_patch(arena.get_patch(color='k', alpha=0.1))
Exemplo n.º 4
0
def plot_data(arena, path, data):
    target_name = data['target_name']
    pool_df = data['pooldf']
    ldf = data['ldf']
    hdf = data['hdf']
    target_ranges = data['target_ranges']

    fig = plt.figure("Image Processing Time (%s)" %  target_name)
    ax = fig.add_subplot(1,1,1)
    ax.hist(hdf['h_processing_time'].values, bins=100)
    ax.set_xlabel('processing time (s)')
    ax.set_title("Distribution of time taken for %s detection" % target_name)
    fig.savefig('imgproc_%s.png' % target_name)

    fig = plt.figure("TTM Tracking", figsize=(8,8))
    ax = fig.add_subplot(1,1,1)
    ax.add_patch(arena.get_patch(color='k', alpha=0.1))
    madplot.plot_laser_trajectory(ax, ldf, arena)
    ax.set_title("The effect of TTM tracking on laser position.\n"\
                 "Values required to hit the fly %s" % target_name)
    ax.legend()
    fig.savefig('ttmeffect_%s.png' % target_name)

    all_v = []
    all_e = []

    #YAY finally plot
    for k in target_ranges:
        vmin,vmax = k

        fvels = "%s-%s px" % (vmin,vmax)

        figf = plt.figure("tFly %s/s" % fvels)
        axf = figf.add_subplot(1,1,1)
        axf_w = axf.twinx()
        figt = plt.figure("tTTM %s/s" % fvels)
        axt = figt.add_subplot(1,1,1)
        axt_w = axt.twinx()

        for trg in target_ranges[k]:
            #widefiled
            err = trg.wf_err
            axf_w.plot(err.dt_thisfly,
                       wide_conv(err.err),
                       'r', alpha=0.3)
            axt_w.plot(err.dt_ttm,
                       wide_conv(err.err),
                       'r', alpha=0.3)

            #ttm
            err = trg.ttm_err
            axf.plot(err.dt_thisfly,
                     ttm_conv(err.err),
                     'k', alpha=0.2)
            axt.plot(err.dt_ttm,
                     ttm_conv(err.err),
                     'k', alpha=0.2)

            all_v.append(trg.v)
            all_e.append(np.mean(ttm_conv(err.err)))

        for ax in (axf,axt):
            ax.set_xlim([0, 0.25])
            ax.set_xlabel('time (s)')
            ax.set_ylabel('TTM error (%s)' % ttm_unit)
            ax.set_ylim([0, 1.5])

        for ax in (axf_w,axt_w):
            ax.set_ylabel('deviation from WF COM (%s)' % wide_unit)
            ax.set_ylim([0, 4])

        for axttm,axwf in [(axt,axt_w),(axf,axf_w)]:
            axttm.set_zorder(axwf.get_zorder()+1) # put ax in front of ax2
            axttm.patch.set_visible(False)

        axf.set_title("Spacial accuracy of %s targeting (fly speed %s/s)\n"\
                      "(time since targeting this fly)" % (target_name,fvels))
        axt.set_title("Spacial accuracy of %s targeting (fly speed %s/s)\n"\
                      "(time since switching to TTM targeting)" % (target_name,fvels))

        figf.savefig(('tFly%s%s.png' % (fvels,target_name)).replace(' ',''))
        figt.savefig(('tTTM%s%s.png' % (fvels,target_name)).replace(' ',''))

    fig = plt.figure("Speed %s" % target_name)
    ax = fig.add_subplot(1,1,1)
    ax.scatter(wide_conv(all_v),all_e)
    ax.set_ylim([0, 1.5])
    ax.set_xlim([0, 2.5])
    ax.set_title("Spacial accuracy of %s targeting" % target_name)
    ax.set_xlabel('fly speed (%s/s)' % wide_unit)
    ax.set_ylabel('error (%s)' % ttm_unit)
    fig.savefig('flyv_errpx_%s.png' % target_name)