'''
Created on Feb 7, 2012

@author $Author$
@version $Id$
'''

import numpy
import os
import sys
import plplot
import datetime

path = os.path.split(os.getcwd())[0]
sys.path.append(path)

from Graphics.BaseGraph import *
from Model.Spectra import Spectra

class Spectrum:
    colorplotObj = None
    
    def __init__(self, spectraObj, index):
        self.__isPlotConfig = False
        self.__isPlotIni = False
        self.__xrange = None
        self.__yrange = None
        self.nsubplots = 0
        self.indexPlot = index
        self.spectraObj = spectraObj
    
    def setup(self,indexPlot, nsubplots, winTitle='', colormap="br_green", showColorbar=False, showPowerProfile=False, XAxisAsTime=False):
        """
        Crea un objeto colorPlot con las opciones seleccinoadas
        """
        
        self.nsubplots = nsubplots
        self.colorplotObj = PcolorPlot(indexPlot,
                                        nsubplots,
                                        winTitle,
                                        colormap,
                                        showColorbar,
                                        showPowerProfile,
                                        XAxisAsTime)
        
    def createObjects(self,xmin,xmax,ymin,ymax,zmin,zmax,titleList,xlabelList,ylabelList):
        """
        Configura cada subplot con los valores maximos y minimos incluyendo los subtitulos
        """
        
        for index in range(self.nsubplots):
            title = titleList[index]
            xlabel = xlabelList[index]
            ylabel = ylabelList[index]
            subplot = index
            self.colorplotObj.createObjects(subplot+1,xmin,xmax,ymin,ymax,zmin,zmax,title,xlabel,ylabel)
    
    def initPlot(self):
        """
        Configura cada subplot con los valores maximos y minimos incluyendo los subtitulos
        """
        
        
        for index in range(self.nsubplots):
            subplot = index
            self.colorplotObj.iniPlot(subplot+1)

    
    def plotData(self,
                 xmin=None,
                 xmax=None,
                 ymin=None,
                 ymax=None,
                 zmin=None,
                 zmax=None,
                 titleList=None,
                 xlabelList=None,
                 ylabelList=None,
                 winTitle='',
                 colormap = "br_green",
                 showColorbar = True,
                 showPowerProfile = True,
                 XAxisAsTime = False,
                 save = False,
                 channelList=[]):
        
        if channelList == []:
            channelList = numpy.arange(self.spectraObj.nChannels)
        
        
        nsubplots = len(channelList)
        nX = self.spectraObj.nFFTPoints
        nY = self.spectraObj.nHeights
        
        if self.spectraObj.noise == None:
            noise = numpy.ones(nsubplots)
        else:
            noise = 10.*numpy.log10(self.spectraObj.noise[channelList])
        
        datadB = 10.*numpy.log10(self.spectraObj.data_spc[channelList,:,:])    
        noisedB = 10.*numpy.log10(noise)
        
        x = numpy.arange(nX)        
        y = self.spectraObj.heightList
        
        indexPlot = self.indexPlot
        
        if not(self.__isPlotConfig):
            self.setup(indexPlot,
                       nsubplots,
                       winTitle,
                       colormap,
                       showColorbar,
                       showPowerProfile,
                       XAxisAsTime)
            
            
            self.__isPlotConfig = True
            
        if not(self.__isPlotIni):
            if titleList == None:
                titleList = []
                for i in range(nsubplots):
                    titleList.append("Channel: %d - Noise: %.2f" %(i, noise[i]))

            if xlabelList == None:
                xlabelList = []
                for i in range(nsubplots):
                    xlabelList.append("")

            if ylabelList == None:
                ylabelList = []
                for i in range(nsubplots):
                    ylabelList.append("Range (Km)")

            if xmin == None: xmin = x[0]
            if xmax == None: xmax = x[-1] 
            if ymin == None: ymin = y[0]
            if ymax == None: ymax = y[-1]                
            if zmin == None: zmin = datadB.min()
            if zmax == None: zmax = datadB.max()
            
            self.createObjects(xmin,xmax,ymin,ymax,zmin,zmax,titleList,xlabelList,ylabelList)
            self.__isPlotIni = True
        
        thisDatetime = datetime.datetime.fromtimestamp(self.spectraObj.m_BasicHeader.utc)
        pltitle = "Self Spectra - Date: %s" %(thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
        
        self.colorplotObj.setFigure(indexPlot)
        self.colorplotObj.setNewPage(pltitle)
        self.initPlot()
        
        for channel in range(nsubplots):
            data = datadB[channel,:,:]
            subtitle = "Channel: %d - Noise: %.2f" %(channel, noise[channel])
            self.colorplotObj.plot(channel+1, x, y, data, subtitle)
        
        self.colorplotObj.refresh()
        
        if save:
            self.colorplotObj.setFigure(indexPlot)
            path = "/home/roj-idl71/tmp/"
            now = datetime.datetime.now().timetuple()
            file = "spc_img%02d_%03d_%02d%02d%02d.png"%(indexPlot,now[7],now[3],now[4],now[5])
            filename = os.path.join(path,file)
            self.colorplotObj.savePlot(indexPlot, filename)
        
        self.colorplotObj.closePage()