'''
Created on Jul 9, 2014

@author: roj-idl71
'''
import os
import datetime
import numpy

from schainpy.model.graphics.jroplot_base import Plot, plt


class ScopePlot(Plot):

    '''
       Plot for Scope
    '''  

    CODE = 'scope'
    plot_name = 'Scope'
    plot_type = 'scatter'
    
    def setup(self):

        self.xaxis = 'Range (Km)'
        self.ncols = 1
        self.nrows = 1
        self.nplots = 1
        self.ylabel = 'Intensity [dB]'
        self.titles = ['Scope']
        self.colorbar = False
        self.width = 6
        self.height = 4

    def plot_iq(self, x, y, channelIndexList, thisDatetime, wintitle):
        
        yreal = y[channelIndexList,:].real
        yimag = y[channelIndexList,:].imag
        title = wintitle + " Scope: %s" %(thisDatetime.strftime("%d-%b-%Y"))
        self.xlabel = "Range (Km)"
        self.ylabel = "Intensity - IQ"
                
        self.y = yreal
        self.x = x
        self.xmin = min(x)
        self.xmax = max(x)
        

        self.titles[0] = title        

        for i,ax in enumerate(self.axes):
            title = "Channel %d" %(i)
            if ax.firsttime:
                ax.plt_r = ax.plot(x, yreal[i,:], color='b')[0]
                ax.plt_i = ax.plot(x, yimag[i,:], color='r')[0]
            else:
                ax.plt_r.set_data(x, yreal[i,:])
                ax.plt_i.set_data(x, yimag[i,:])
        
    def plot_power(self, x, y, channelIndexList, thisDatetime, wintitle):
        y = y[channelIndexList,:] * numpy.conjugate(y[channelIndexList,:])
        yreal = y.real
        self.y = yreal
        title = wintitle + " Scope: %s" %(thisDatetime.strftime("%d-%b-%Y"))
        self.xlabel = "Range (Km)"
        self.ylabel = "Intensity"
        self.xmin = min(x)
        self.xmax = max(x)
        
        
        self.titles[0] = title

        for i,ax in enumerate(self.axes):
            title = "Channel %d" %(i)
            
            ychannel = yreal[i,:]
            
            if ax.firsttime:    
                ax.plt_r = ax.plot(x, ychannel)[0]
            else:
                #pass
                ax.plt_r.set_data(x, ychannel)
        

    def plot(self):
        
        if self.channels:
            channels = self.channels
        else:
            channels = self.data.channels

        thisDatetime = datetime.datetime.utcfromtimestamp(self.data.times[-1])
        
        scope = self.data['scope']
        
        if self.data.flagDataAsBlock:
            
            for i in range(self.data.nProfiles):

                wintitle1 = " [Profile = %d] " %i

                if self.type == "power":
                    self.plot_power(self.data.heights, 
                                    scope[:,i,:],                                
                                    channels,
                                    thisDatetime, 
                                    wintitle1
                                    )

                if self.type == "iq":
                    self.plot_iq(self.data.heights, 
                                 scope[:,i,:],
                                 channels,
                                 thisDatetime, 
                                 wintitle1 
                                )
        else:
            wintitle = " [Profile = %d] " %self.data.profileIndex
            
            if self.type == "power":
                 self.plot_power(self.data.heights, 
                            scope,
                            channels,
                            thisDatetime,
                            wintitle
                            )
            
            if self.type == "iq":
                self.plot_iq(self.data.heights, 
                            scope,
                            channels,
                            thisDatetime,
                            wintitle
                            )