'''
Created on Feb 7, 2012

@author $Author$
@version $Id$
'''
import numpy
import os
import sys

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

from Graphics.BaseGraph import *
from Model.Voltage import Voltage

class Osciloscope:
    linearplotObj = None
    
    def __init__(self, Voltage, index):
        self.__isPlotConfig = False
        self.__isPlotIni = False
        self.__xrange = None
        self.__yrange = None
        self.indexPlot = index
        self.voltageObj = Voltage
    
    def setup(self,indexPlot,nsubplot,winTitle=''):
        self.linearplotObj = LinearPlot(indexPlot,nsubplot,winTitle)
    
    def initPlot(self,xmin,xmax,ymin,ymax,titleList,xlabelList,ylabelList):
        nsubplot = self.voltageObj.nChannels
        
        for index in range(nsubplot):
            title = titleList[index]
            xlabel = xlabelList[index]
            ylabel = ylabelList[index]
            subplot = index
            self.linearplotObj.setup(subplot+1,xmin,xmax,ymin,ymax,title,xlabel,ylabel)
    
    def plotData(self,
                 xmin=None,
                 xmax=None,
                 ymin=None,
                 ymax=None,
                 titleList=None,
                 xlabelList=None,
                 ylabelList=None,
                 winTitle='',
                 type="power"):
        
        databuffer = self.voltageObj.data
        
        height = self.voltageObj.heightList
        nsubplot = self.voltageObj.nChannels
        indexPlot = self.indexPlot
        
        
        if not(self.__isPlotConfig):
            self.setup(indexPlot,nsubplot,winTitle)
            self.__isPlotConfig = True
        
        if not(self.__isPlotIni):
            if titleList == None:
                titleList = []
                thisDatetime = datetime.datetime.fromtimestamp(self.voltageObj.m_BasicHeader.utc)
                txtdate = "Date: %s" %(thisDatetime.strftime("%d-%b-%Y"))
                for i in range(nsubplot):
                    titleList.append("Channel: %d %s" %(i, txtdate))

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

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

            if xmin == None: xmin = height[0]
            if xmax == None: xmax = height[-1]
            if ymin == None: ymin = numpy.nanmin(abs(databuffer))
            if ymax == None: ymax = numpy.nanmax(abs(databuffer))                
            
            self.initPlot(xmin,xmax,ymin,ymax,titleList,xlabelList,ylabelList)
            self.__isPlotIni = True
        
        self.linearplotObj.setFigure(indexPlot)
        
        for index in range(nsubplot):
            data = databuffer[index,:]
            self.linearplotObj.plot(subplot=index+1,x=height,y=data,type=type)
        
        self.linearplotObj.refresh()
        
class RTI:
    colorplotObj = None
    
    def __init__(self, Voltage, index):
        self.__isPlotConfig = False
        self.__isPlotIni = False
        self.__xrange = None
        self.__yrange = None
        self.indexPlot = index
        self.voltageObj = Voltage

    def setup(self,indexPlot,nsubplot,winTitle='',colormap="br_green",showColorbar=False,showPowerProfile=False,XAxisAsTime=False):
        self.colorplotObj = RtiPlot(indexPlot,nsubplot,winTitle,colormap,showColorbar,showPowerProfile,XAxisAsTime)

    def initPlot(self,xmin,xmax,ymin,ymax,zmin,zmax,titleList,xlabelList,ylabelList,timezone,npoints):

        nsubplot = self.voltageObj.nChannels
        timedata = self.voltageObj.m_BasicHeader.utc

        for index in range(nsubplot):
            title = titleList[index]
            xlabel = xlabelList[index]
            ylabel = ylabelList[index]
            subplot = index
            self.colorplotObj.setup(subplot+1,xmin,xmax,ymin,ymax,zmin,zmax,title,xlabel,ylabel,timedata,timezone,npoints)
            
    def plotData(self,
                 xmin=None,
                 xmax=None,
                 ymin=None,
                 ymax=None,
                 zmin=None,
                 zmax=None,
                 titleList=None,
                 xlabelList=None,
                 ylabelList=None,
                 winTitle='',
                 timezone='lt',
                 npoints=1000.0,
                 colormap="br_green",
                 showColorbar=True,
                 showPowerProfile=True,
                 XAxisAsTime=True,
                 save = False):
        
        databuffer = self.voltageObj.data
        timedata = self.voltageObj.m_BasicHeader.utc
        height = self.voltageObj.heightList
        nsubplot = self.voltageObj.nChannels
        indexPlot = self.indexPlot

        if not(self.__isPlotConfig):
            self.setup(indexPlot,nsubplot,winTitle,colormap,showColorbar,showPowerProfile,XAxisAsTime)
            self.__isPlotConfig = True

        if not(self.__isPlotIni):
            if titleList == None:
                titleList = []
                thisDatetime = datetime.datetime.fromtimestamp(timedata)
                txtdate = "Date: %s" %(thisDatetime.strftime("%d-%b-%Y"))
                for i in range(nsubplot):
                    titleList.append("Channel: %d %s" %(i, txtdate))

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

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

            if xmin == None: xmin = 0
            if xmax == None: xmax = 23
            if ymin == None: ymin = min(self.voltageObj.heightList)
            if ymax == None: ymax = max(self.voltageObj.heightList)                
            if zmin == None: zmin = 0
            if zmax == None: zmax = 50


            self.initPlot(xmin,xmax,ymin,ymax,zmin,zmax,titleList,xlabelList,ylabelList,timezone,npoints)
            self.__isPlotIni = True
        

        self.colorplotObj.setFigure(indexPlot)
        
        if timezone == 'lt':
            timedata = timedata - time.timezone
        
        for index in range(nsubplot):
            data = databuffer[index,:]
            self.colorplotObj.plot(subplot=index+1,x=timedata,y=height,z=data)
        
        self.colorplotObj.refresh()
        
        if save:
            self.colorplotObj.setFigure(indexPlot)
            path = "/Users/jro/Pictures"
            now = datetime.datetime.now().timetuple()
            file = "rti_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)