| 12
 3
 4
 5
 6
 7
 8
 9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 
 |  
import pickle
from matplotlib import pyplot as plt
import Tkinter, tkFileDialog
 
root = Tkinter.Tk()
dirname = tkFileDialog.askdirectory(parent=root,initialdir="/",title='Please select a directory')
dirname_out=dirname + '/results/'
 
def plotting_trials(trial_number):
    outfile_trial_name=dirname_out + 'trial_name.p'
    outfile_trial_result=dirname_out + 'trial_result.p'
    outfile_trial_labels=dirname_out + 'trial_labels.p'
    outfile_trial_labels_time=dirname_out + 'trial_labels_time.p'
    outfile_trial_x=dirname_out + 'trial_x.p'
    outfile_trial_y=dirname_out + 'trial_y.p'
    outfile_trial_time=dirname_out + 'trial_time.p'
    outfile_trial_saccofinterest=dirname_out + 'trial_sacc_of_interest.p'
    outfile_trial_latency=dirname_out + 'trial_latency.p'
    #loading files
    trial_name = pickle.load(open(outfile_trial_name, 'rb'))
    trial_result=pickle.load(open(outfile_trial_result, 'rb'))
    trial_time=pickle.load(open(outfile_trial_time, 'rb'))
    trial_x=pickle.load(open(outfile_trial_x, 'rb'))
    trial_labels=pickle.load(open(outfile_trial_labels, 'rb'))
    trial_labels_time=pickle.load(open(outfile_trial_labels_time, 'rb'))
    trial_y=pickle.load(open(outfile_trial_y, 'rb'))
    saccofinterest=pickle.load(open(outfile_trial_saccofinterest, 'rb'))
    latency=pickle.load(open(outfile_trial_latency, 'rb'))
    #end loading files
 
    plt.figure(1)
    plt.subplot(211)
    plt.plot(trial_time[trial_number],trial_x[trial_number])
    plt.xlabel('Time [ms]')
    plt.ylabel('Eye Position [V]')
    for t in range(len(trial_labels_time[trial_number])-3): 
        plt.vlines(trial_labels_time[trial_number][t], -5, 5, colors='k')
        if (trial_labels[trial_number][t]=='Fixation'):
            plt.text(trial_labels_time[trial_number][t],4,trial_labels[trial_number][t],color='g')
        if (trial_labels[trial_number][t]=='TrialStart_'):
            plt.text(trial_labels_time[trial_number][t],3,trial_labels[trial_number][t],color='g')
        if (trial_labels[trial_number][t]=='Target'):
            plt.text(trial_labels_time[trial_number][t],3,trial_labels[trial_number][t],color='g') 
        if (trial_labels[trial_number][t]=='Target2'):
            plt.text(trial_labels_time[trial_number][t],2,trial_labels[trial_number][t],color='g') 
        if (trial_labels[trial_number][t]=='EndTrial'):
            plt.text(trial_labels_time[trial_number][t],4,trial_labels[trial_number][t],color='g') 
    plt.vlines(saccofinterest[trial_number], -5, 5, colors='r')
    titre='trial number: ' + str(trial_number)+ ', ' + trial_name[trial_number] + ', ' + trial_result[trial_number] + ' latency: ' + str(latency[trial_number]) + ' ms'
    plt.title(titre)
 
    plt.subplot(212)
    plt.plot(trial_time[trial_number],trial_y[trial_number])
 
    plt.vlines(saccofinterest[trial_number], -5, 5, colors='r')
    for t in range(len(trial_labels_time[trial_number])-3): 
        plt.vlines(trial_labels_time[trial_number][t], -5, 5, colors='k')
        if (trial_labels[trial_number][t]=='Fixation'):
            plt.text(trial_labels_time[trial_number][t],4,trial_labels[trial_number][t],color='g')
        if (trial_labels[trial_number][t]=='TrialStart_'):
            plt.text(trial_labels_time[trial_number][t],3,trial_labels[trial_number][t],color='g')
        if (trial_labels[trial_number][t]=='Target'):
            plt.text(trial_labels_time[trial_number][t],3,trial_labels[trial_number][t],color='g') 
        if (trial_labels[trial_number][t]=='Target2'):
            plt.text(trial_labels_time[trial_number][t],2,trial_labels[trial_number][t],color='g') 
        if (trial_labels[trial_number][t]=='EndTrial'):
            plt.text(trial_labels_time[trial_number][t],4,trial_labels[trial_number][t],color='g')      
    plt.xlabel('Time [ms]')
    plt.ylabel('Eye Position [V]')
    plt.show()
 
 
plotting_trials(11) | 
Partager