import numpy
import plplot

class BasicGraph:
    """
    
    """
    
    hasRange = False
    
    xrange = None
    yrange = None
    zrange = None
    
    xlabel = None
    ylabel = None
    title = None
    
    legends = None
    
    __name = None
    __subpage = None
    __szchar = None
    
    __colormap = None
    __colbox  = None
    __colleg = None    
     
    __xpos = None
    __ypos = None
    
    __xopt = None #"bcnst"
    __yopt = None #"bcnstv"
      
    __xlpos = None
    __ylpos = None
    
    __xrangeIsTime = None
    
    #Advanced
    __xg = None
    __yg = None
    
    def __init__(self):
        """
        
        """
        pass
     
    def hasNotXrange(self):
        
        if self.xrange == None:
            return 1
        
        return 0

    def hasNotYrange(self):
        
        if self.yrange == None:
            return 1
        
        return 0

    def hasNotZrange(self):
        
        if self.zrange == None:
            return 1
        
        return 0
    def setName(self, name):
        self.__name = name
        
    def setScreenPos(self, xpos, ypos):
        self.__xpos = xpos
        self.__ypos = ypos
    
    def setScreenPosbyWidth(self, xoff, yoff, xw, yw):
        self.__xpos = [xoff, xoff + xw]
        self.__ypos = [yoff, yoff + yw]
        
    def setSubpage(self, subpage):
        self.__subpage = subpage
        
    def setSzchar(self, szchar):
        self.__szchar = szchar
        
    def setOpt(self, xopt, yopt):
        self.__xopt = xopt
        self.__yopt = yopt

    def setRanges(self, xrange, yrange, zrange=None):
        """
        """
        self.xrange = xrange
        
        self.yrange = yrange
        
        if zrange != None:
            self.zrange = zrange
    
    def setColormap(self, colormap=None):
        
        if colormap == None:
            colormap = self.__colormap
            
        cmap1_init(colormap)
        
    def plotBox(self):
        """
        
        """
        plplot.plvpor(self.__xpos[0], self.__xpos[1], self.__ypos[0], self.__ypos[1])
        plplot.plwind(self.xrange[0], self.xrange[1], self.yrange[0], self.yrange[1])
        plplot.plbox(self.__xopt, 0.0, 0, self.__yopt, 0.0, 0)
        plplot.pllab(self.xlabel, self.ylabel, self.title)
      
    def setup(self, title=None, xlabel=None, ylabel=None, colormap=None):
        """
        """
        self.title = title
        self.xlabel = xlabel
        self.ylabel = ylabel
        self.__colormap = colormap
    
    def initSubpage(self):
        
        if plplot.plgdev() == '':
            raise ValueError, "Plot device has not been initialize"
        
        plplot.pladv(self.__subpage)
        plplot.plschr(0.0, self.__szchar)
        
        if self.__xrangeIsTime:
            plplot.pltimefmt("%H:%M")
            
        self.setColormap()
        self.initPlot()
    
    def initPlot(self):
        """
        
        """
        if plplot.plgdev() == '':
            raise ValueError, "Plot device has not been initialize"
        
        xrange = self.xrange
        if xrange == None:
            xrange = [0., 1.]
        
        yrange = self.yrange
        if yrange == None:
            yrange = [0., 1.]
        
        plplot.plvpor(self.__xpos[0], self.__xpos[1], self.__ypos[0], self.__ypos[1])
        plplot.plwind(xrange[0], xrange[1], yrange[0], yrange[1])
        plplot.plbox(self.__xopt, 0.0, 0, self.__yopt, 0.0, 0)
        plplot.pllab(self.xlabel, self.ylabel, self.title)
    
    def colorbarPlot(self):
        data = numpy.arange(256)
        data = numpy.reshape(data, (1,-1))
        
        self.plotBox()
        plplot.plimage(data,
                       self.xrange[0],
                       self.xrange[1],
                       self.yrange[0],
                       self.yrange[1],
                       0.,
                       255.,
                       self.xrange[0],
                       self.xrange[1],
                       self.yrange[0],
                       self.yrange[1],)
        
    def basicXYPlot(self, x, y):
        self.plotBox()
        plplot.plline(x, y)
    
    def basicXYwithErrorPlot(self):
        pass
    
    def basicLineTimePlot(self):
        pass
    
    def basicPcolorPlot(self, data, xmin, xmax, ymin, ymax, zmin, zmax):
        """
        """
        
        self.plotBox()
        plplot.plimage(data, xmin, xmax, ymin, ymax, zmin, zmax, xmin, xmax, ymin, ymax)

        
    def __getBoxpltr(self, x, y, deltax=None, deltay=None):        
        
        if not(len(x)>1 and len(y)>1):
            raise ValueError, "x axis and y axis are empty"
        
        if deltax == None: deltax = x[-1] - x[-2]
        if deltay == None: deltay = y[-1] - y[-2]
        
        x1 = numpy.append(x, x[-1] + deltax)
        y1 = numpy.append(y, y[-1] + deltay)
        
        xg = (numpy.multiply.outer(x1, numpy.ones(len(y1))))
        yg = (numpy.multiply.outer(numpy.ones(len(x1)), y1))
        
        self.__xg = xg
        self.__yg = yg
        
    def advPcolorPlot(self, data, x, y, zmin=0., zmax=0.):
        """
        """
        
        if self.__xg == None and self.__yg == None:
            self.__getBoxpltr(x, y)
        
        plplot.plimagefr(data, x[0], x[-1], y[0], y[-1], 0., 0., zmin, zmax, plplot.pltr2, self.__xg, self.__yg)