# -*- coding: utf-8 -*-

import math, os
import pickle

from pylab import *

## USAGE: python2.6 plot_corrs.py

sys.path.extend(["..","../networks","../generators","../simulations"])

from networkConstants import * # has central_glom
from data_utils import *
from calc_corrs import *

from results_catalogue import *

fig1 = figure(facecolor='none') # 'none' is transparent
## A super axes to set common x and y axes labels
bigAxes1 = fig1.add_axes([0.1,0.1,0.8,0.8],frameon=False) # hide frame
#bigAxes1.set_xticks([0,1,2,3])
#bigAxes1.set_xticklabels(['none', 'singles',\
#    's+joints', 's+j+PGs'],fontsize=20)
bigAxes1.set_xticks([])
bigAxes1.set_yticks([])
bigAxes1.text(-0.1,0.3,'spiking probability', fontsize=24, rotation='vertical')
bigAxes1.text(-0.1,-0.11,'inhibition: none, singles, s+joints, s+j+PGs',\
    fontsize=24, rotation='horizontal')
bigAxes1.set_title('Cross-Correlogram peak',fontsize=36)

fig2 = figure(facecolor='none') # 'none' is transparent
## A super axes to set common x and y axes labels
bigAxes2 = fig2.add_axes([0.1,0.1,0.8,0.8],frameon=False) # hide frame
#bigAxes2.set_xticks([0,1,2,3])
#bigAxes2.set_xticklabels(['none', 'singles',\
#    's+joints', 's+j+PGs'],fontsize=20)
bigAxes2.set_xticks([])
bigAxes2.set_yticks([])
bigAxes2.text(-0.1,-0.11,'inhibition: none, singles, s+joints, s+j+PGs',\
    fontsize=24, rotation='horizontal')
bigAxes2.set_title('Binned Cross-Correlation peak',fontsize=36)

fig3 = figure(facecolor='none') # 'none' is transparent
## A super axes to set common x and y axes labels
bigAxes3 = fig3.add_axes([0.1,0.1,0.8,0.8],frameon=False) # hide frame
#bigAxes3.set_xticks([0,1,2,3])
#bigAxes3.set_xticklabels(['none', 'singles',\
#    's+joints', 's+j+PGs'],fontsize=20)
bigAxes3.set_xticks([])
bigAxes3.set_yticks([])
bigAxes3.text(-0.1,0.3,'time(s)', fontsize=24, rotation='vertical')
bigAxes3.text(-0.1,-0.11,'inhibition: none, singles, s+joints, s+j+PGs',\
    fontsize=24, rotation='horizontal')
bigAxes3.set_title('Cross-Correlogram shift',fontsize=36)

plotnum = 1
for runnum in range(len(filelist[0])):
    corrgramlist = []
    corrgrammaxlist = []
    corrgrampeaklist = []
    corrlist = []
    print seeds[runnum]
    for filename in array(filelist)[:,runnum]:
        filenamefull = '../results/odor_morphs/'+filename
        (air_corr,odorA_corr,odorB_corr),\
            (tcorrlist,airxcorrgram,odorAxcorrgram,odorBxcorrgram)\
            = calc_corrs(filenamefull, norm_str='overall')
        corrgramlist.append([airxcorrgram,odorAxcorrgram,odorBxcorrgram])
        maxes = [max(airxcorrgram),max(odorAxcorrgram),max(odorBxcorrgram)]
        peaks = [ where(airxcorrgram==maxes[0])[0],\
            where(odorAxcorrgram==maxes[1])[0],
            where(odorBxcorrgram==maxes[2])[0] ]
        if len(peaks[0])>1 or len(peaks[1])>1 or len(peaks[2])>1:
            print "Multiple peaks for",filenamefull,peaks
        corrgrammaxlist.append(maxes)
        ## binwidth = 1e-3 for xcorrgram, convert bin index to time shift
        peaks = ( array([mean(peaks[0]),mean(peaks[1]),mean(peaks[2])]) - 1 ) * 1e-3 - 0.5
        corrgrampeaklist.append(peaks)
        corrlist.append([air_corr,odorA_corr,odorB_corr])
    corrlist = array(corrlist)
    corrgramlist = array(corrgramlist)
    corrgrammaxlist = array(corrgrammaxlist)
    corrgrampeaklist = array(corrgrampeaklist)
    
    ## hardcoded - I know there are 4 files
    ax1 = fig1.add_subplot(2,3,plotnum)
    ax1.set_xticks([])
    ax1.plot(corrgrammaxlist[:,0],color=(0,0,0),linewidth=2)
    ax1.plot(corrgrammaxlist[:,1],color=(1,0,0),linewidth=2)
    ax1.plot(corrgrammaxlist[:,2],color=(0,1,0),linewidth=2)
    #ymax = ax1.get_ylim()[1]+0.01
    ## very important to give it after the plot functions else autoscales
    #ax1.set_ylim(-0.01,ymax)
    #ax1.set_yticks([0,ymax])
    #ax1.set_yticklabels(['0','%0.2f'%ymax],size='large')
    ## or set autoscaling off - gca().set_autoscale_on(False)
    ## - taken from http://old.nabble.com/ylim-does-not-work-td19000814.html
    ax1.set_title( str(seeds[runnum]), size='large')

    ## hardcoded - I know there are 4 files
    ax3 = fig3.add_subplot(2,3,plotnum)
    ax3.set_xticks([])
    #ax3.plot(corrgrampeaklist[:,0],color=(0,0,0),linewidth=2)
    #ax3.plot(corrgrampeaklist[:,1],color=(1,0,0),linewidth=2)
    #ax3.plot(corrgrampeaklist[:,2],color=(0,1,0),linewidth=2)
    ax3.plot(corrgramlist[3,0],color=(0,0,0),linewidth=2)
    ax3.plot(corrgramlist[3,2],color=(0,1,0),linewidth=2)
    ax3.plot(corrgramlist[3,1],color=(1,0,0),linewidth=2)
    ax3.set_title( str(seeds[runnum]), size='large')

    ## hardcoded - I know there are 4 files
    ax2 = fig2.add_subplot(2,3,plotnum)
    ax2.set_xticks([])
    ax2.plot(corrlist[:,0],color=(0,0,0),linewidth=2)
    ax2.plot(corrlist[:,1],color=(1,0,0),linewidth=2)
    ax2.plot(corrlist[:,2],color=(0,1,0),linewidth=2)
    #ymax = ax1.get_ylim()[1]+0.01
    ## very important to give it after the plot functions else autoscales
    #ax2.set_ylim(-0.01,ymax)
    #ax2.set_yticks([0,ymax])
    #ax2.set_yticklabels(['0','%0.2f'%ymax],size='large')
    ## or set autoscaling off - gca().set_autoscale_on(False)
    ## - taken from http://old.nabble.com/ylim-does-not-work-td19000814.html
    ax2.set_title( str(seeds[runnum]), size='large')

    plotnum += 1

show()