From 18feaeaea3963d8d8c880739abed0015e9a2644c 2012-02-09 14:36:24
From: Miguel Valdez <miguel.urco@jro.igp.gob.pe>
Date: 2012-02-09 14:36:24
Subject: [PATCH] El módulo BasicGraph fue cambiado de nombre a BaseGraph.

---

diff --git a/schainpy/Graphics/BaseGraph.py b/schainpy/Graphics/BaseGraph.py
new file mode 100644
index 0000000..12a8628
--- /dev/null
+++ b/schainpy/Graphics/BaseGraph.py
@@ -0,0 +1,606 @@
+"""
+Created on Feb 7, 2012
+
+@autor $Author$
+@version $Id$
+
+"""
+
+import numpy
+import plplot
+
+class BaseGraph:
+    """
+    
+    """
+    
+    hasNotRange = True
+    
+    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 = False
+    
+    #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
+        
+        self.hasNotRange = False
+    
+    def setColormap(self, colormap=None):
+        
+        if colormap == None:
+            colormap = self.__colormap
+            
+        cmap1_init(colormap)
+    
+    def setXAxisAsTime(self):
+        self.__xrangeIsTime = True
+    
+    def plotBox(self):
+        """
+        
+        """
+        plplot.plvpor(self.__xpos[0], self.__xpos[1], self.__ypos[0], self.__ypos[1])
+        plplot.plwind(float(self.xrange[0]),
+                      float(self.xrange[1]),
+                      float(self.yrange[0]),
+                      float(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.]
+        
+        self.plotBox()
+    
+    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, x, y, xmin=None, xmax=None, ymin=None, ymax=None, zmin=None, zmax=None):
+        """
+        """
+        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 = numpy.nanmin(data)
+        if zmax == None: zmax = numpy.nanmax(data)
+        
+        self.plotBox()
+        plplot.plimage(data,
+                       float(x[0]),
+                       float(x[-1]),
+                       float(y[0]),
+                       float(y[-1]),
+                       float(zmin),
+                       float(zmax),
+                       float(xmin),
+                       float(xmax),
+                       float(ymin),
+                       float(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)
+
+
+class ColorPlot():
+
+    
+    graphObjDict = {}
+    showColorbar = False
+    showPowerProfile = True
+    
+    __szchar = 0.7
+    __xrange = None
+    __yrange = None
+    __zrange = None
+    
+    m_BaseGraph = BaseGraph()
+    
+    def __init__(self):
+        
+        key = "colorplot"
+        self.m_BaseGraph.setName(key)
+        
+        self.graphObjDict[key] = self.m_BaseGraph
+    
+    def setup(self, subpage, title="", xlabel="", ylabel="", colormap="jet", showColorbar=False, showPowerProfile=False, XAxisAsTime=False):
+        """
+        """
+        
+        self.m_BaseGraph.setSubpage(subpage)
+        self.m_BaseGraph.setSzchar(self.__szchar)
+        self.m_BaseGraph.setOpt("bcnts","bcnts")
+        self.m_BaseGraph.setup(title,
+                      xlabel,
+                      ylabel,
+                      colormap)
+        
+        if showColorbar:
+            key = "colorbar"
+            
+            cmapObj = BaseGraph()
+            cmapObj.setName(key)
+            cmapObj.setSubpage(subpage)
+            cmapObj.setSzchar(self.__szchar)
+            cmapObj.setOpt("bc","bcmt")
+            cmapObj.setup(title="dBs",
+                          xlabel="",
+                          ylabel="",
+                          colormap=colormap)
+            
+            self.graphObjDict[key] = cmapObj
+            
+        
+        if showPowerProfile:
+            key = "powerprof"
+            
+            powObj = BaseGraph()
+            powObj.setName(key)
+            powObj.setSubpage(subpage)
+            powObj.setSzchar(self.__szchar)
+            plplot.pllsty(2)
+            powObj.setOpt("bcntg","bc")
+            plplot.pllsty(1)
+            powObj.setup(title="Power Profile",
+                         xlabel="dBs",
+                         ylabel="")
+            
+            self.graphObjDict[key] = powObj
+            
+        self.showColorbar = showColorbar
+        self.showPowerProfile = showPowerProfile
+        self.setPos()
+        
+        if XAxisAsTime:
+            self.m_BaseGraph.setXAxisAsTime()
+        #self.setPos(xi = 0.05, yi = 0.18, xw = 0.92, yw = 0.74, xcmapw = 0.015, xpoww = 0.14, deltaxcmap = 0.01, deltaxpow = 0.02)
+        
+    
+    def setPos(self, xi = 0.12, yi = 0.14, xw = 0.78, yw = 0.80, xcmapw = 0.05, xpoww = 0.24, deltaxcmap = 0.02, deltaxpow = 0.06):
+        
+        if self.showColorbar:
+            xw -= xcmapw + deltaxcmap
+        
+        if self.showPowerProfile:
+            xw -= xpoww + deltaxpow
+        
+        xf = xi + xw
+        yf = yi + yw
+        xcmapf = xf
+        
+        self.m_BaseGraph.setScreenPos([xi, xf], [yi, yf])
+        
+        if self.showColorbar:
+            xcmapi = xf + deltaxcmap
+            xcmapf = xcmapi + xcmapw
+            
+            key = "colorbar"
+            cmapObj = self.graphObjDict[key]
+            cmapObj.setScreenPos([xcmapi, xcmapf], [yi, yf])
+        
+        if self.showPowerProfile:
+            
+            xpowi = xcmapf + deltaxpow
+            xpowf = xpowi + xpoww
+            
+            key = "powerprof"
+            powObj = self.graphObjDict[key]
+            powObj.setScreenPos([xpowi, xpowf], [yi, yf])
+    
+    def setRanges(self, xrange, yrange, zrange):
+        
+        self.m_BaseGraph.setRanges(xrange, yrange, zrange)
+        
+        keyList = self.graphObjDict.keys()
+        
+        key = "colorbar"
+        if key in keyList:
+            cmapObj = self.graphObjDict[key]
+            cmapObj.setRanges([0., 1.], zrange)
+        
+        key = "powerprof"
+        if key in keyList:
+            powObj = self.graphObjDict[key]
+            powObj.setRanges(zrange, yrange)
+    
+    def plotData(self, data, x=None, y=None, xmin=None, xmax=None, ymin=None, ymax=None, zmin=None, zmax=None):
+        """
+        """
+        
+        try:
+            nX, nY = numpy.shape(data)
+        except:
+            raise ValueError, "data is not a numpy array"
+        
+        if x == None: x = numpy.arange(nX)
+        if y == None: y = numpy.arange(nY)
+        
+        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 = numpy.nanmin(data)
+        if zmax == None: zmax = numpy.nanmax(data)
+           
+        if self.m_BaseGraph.hasNotRange:
+            self.setRanges([xmin, xmax], [ymin,ymax], [zmin,zmax])
+        
+        self.m_BaseGraph.initSubpage()
+        self.m_BaseGraph.basicPcolorPlot(data, x, y, xmin, xmax, ymin, ymax, self.m_BaseGraph.zrange[0], self.m_BaseGraph.zrange[1])
+        
+        if self.showColorbar:
+            key = "colorbar"
+            cmapObj = self.graphObjDict[key]
+            cmapObj.colorbarPlot()
+        
+        if self.showPowerProfile:
+            power = numpy.average(data, axis=1)
+            
+            step = (ymax - ymin)/(nY-1)
+            heis = numpy.arange(ymin, ymax + step, step)
+            
+            key = "powerprof"
+            powObj = self.graphObjDict[key]
+            powObj.basicXYPlot(power, heis)
+
+class LinearPlot():
+    
+    __szchar = 0.7
+    __xrange = None
+    __yrange = None
+    
+    m_BaseGraph = BaseGraph()
+    
+    def __init__(self):
+        
+        key = "linearplot"
+        self.m_BaseGraph.setName(key)
+        
+        self.graphObjDict[key] = self.m_BaseGraph
+    
+    def setup(self, subpage, title="", xlabel="", ylabel="", colormap="jet", XAxisAsTime=False):
+        """
+        """
+        
+        self.m_BaseGraph.setSubpage(subpage)
+        self.m_BaseGraph.setSzchar(self.__szchar)
+        self.m_BaseGraph.setOpt("bcnts","bcnts")
+        self.m_BaseGraph.setup(title,
+                               xlabel,
+                               ylabel,
+                               colormap)
+        
+        self.setPos()
+        
+        if XAxisAsTime:
+            self.m_BaseGraph.setXAxisAsTime()
+        #self.setPos(xi = 0.05, yi = 0.18, xw = 0.92, yw = 0.74, xcmapw = 0.015, xpoww = 0.14, deltaxcmap = 0.01, deltaxpow = 0.02)
+        
+    
+    def setPos(self, xi = 0.12, yi = 0.14, xw = 0.78, yw = 0.80):
+        
+        xf = xi + xw
+        yf = yi + yw
+        
+        self.m_BaseGraph.setScreenPos([xi, xf], [yi, yf])
+    
+    def setRanges(self, xrange, yrange, zrange):
+        
+        self.m_BaseGraph.setRanges(xrange, yrange, zrange)
+    
+    def plotData(self, x, y):
+        """
+        """
+        xmin = x[0]
+        xmax = x[-1]
+
+        ymin = y[0]
+        ymax = y[-1]
+           
+        if self.m_BaseGraph.hasNotRange:
+            self.setRanges([xmin, xmax], [ymin,ymax])
+        
+        self.m_BaseGraph.initSubpage()
+        self.m_BaseGraph.basicLineTimePlot(x, y)
+
+
+def cmap1_init(colormap="gray"):
+    
+    ncolor = None
+    rgb_lvl = None
+    
+    # Routine for defining a specific color map 1 in HLS space.
+    # if gray is true, use basic grayscale variation from half-dark to light.
+    # otherwise use false color variation from blue (240 deg) to red (360 deg).
+
+    # Independent variable of control points.
+    i = numpy.array((0., 1.))
+    if colormap=="gray":
+        ncolor = 256
+        # Hue for control points.  Doesn't matter since saturation is zero.
+        h = numpy.array((0., 0.))
+        # Lightness ranging from half-dark (for interest) to light.
+        l = numpy.array((0.5, 1.))
+        # Gray scale has zero saturation
+        s = numpy.array((0., 0.))
+        
+        # number of cmap1 colours is 256 in this case.
+        plplot.plscmap1n(ncolor)
+        # Interpolate between control points to set up cmap1.
+        plplot.plscmap1l(0, i, h, l, s)
+        
+        return None
+        
+    if colormap=="br_green":
+        ncolor = 256
+        # Hue ranges from blue (240 deg) to red (0 or 360 deg)
+        h = numpy.array((240., 0.))
+        # Lightness and saturation are constant (values taken from C example).
+        l = numpy.array((0.6, 0.6))
+        s = numpy.array((0.8, 0.8))
+
+        # number of cmap1 colours is 256 in this case.
+        plplot.plscmap1n(ncolor)
+        # Interpolate between control points to set up cmap1.
+        plplot.plscmap1l(0, i, h, l, s)
+        
+        return None
+    
+    if colormap=="tricolor":
+        ncolor = 3
+        # Hue ranges from blue (240 deg) to red (0 or 360 deg)
+        h = numpy.array((240., 0.))
+        # Lightness and saturation are constant (values taken from C example).
+        l = numpy.array((0.6, 0.6))
+        s = numpy.array((0.8, 0.8))
+
+        # number of cmap1 colours is 256 in this case.
+        plplot.plscmap1n(ncolor)
+        # Interpolate between control points to set up cmap1.
+        plplot.plscmap1l(0, i, h, l, s)
+        
+        return None
+    
+    if colormap == 'rgb' or colormap == 'rgb666':
+        
+        color_sz = 6
+        ncolor = color_sz*color_sz*color_sz
+        pos = numpy.zeros((ncolor))
+        r = numpy.zeros((ncolor))
+        g = numpy.zeros((ncolor))
+        b = numpy.zeros((ncolor))
+        ind = 0
+        for ri in range(color_sz):
+            for gi in range(color_sz):
+                for bi in range(color_sz):
+                    r[ind] = ri/(color_sz-1.0)
+                    g[ind] = gi/(color_sz-1.0)
+                    b[ind] = bi/(color_sz-1.0)
+                    pos[ind] = ind/(ncolor-1.0)
+                    ind += 1
+        rgb_lvl = [6,6,6]  #Levels for RGB colors
+        
+    if colormap == 'rgb676':
+        ncolor = 6*7*6
+        pos = numpy.zeros((ncolor))
+        r = numpy.zeros((ncolor))
+        g = numpy.zeros((ncolor))
+        b = numpy.zeros((ncolor))
+        ind = 0
+        for ri in range(8):
+            for gi in range(8):
+                for bi in range(4):
+                    r[ind] = ri/(6-1.0)
+                    g[ind] = gi/(7-1.0)
+                    b[ind] = bi/(6-1.0)
+                    pos[ind] = ind/(ncolor-1.0)
+                    ind += 1
+        rgb_lvl = [6,7,6]  #Levels for RGB colors
+    
+    if colormap == 'rgb685':
+        ncolor = 6*8*5
+        pos = numpy.zeros((ncolor))
+        r = numpy.zeros((ncolor))
+        g = numpy.zeros((ncolor))
+        b = numpy.zeros((ncolor))
+        ind = 0
+        for ri in range(8):
+            for gi in range(8):
+                for bi in range(4):
+                    r[ind] = ri/(6-1.0)
+                    g[ind] = gi/(8-1.0)
+                    b[ind] = bi/(5-1.0)
+                    pos[ind] = ind/(ncolor-1.0)
+                    ind += 1
+        rgb_lvl = [6,8,5]  #Levels for RGB colors
+                    
+    if colormap == 'rgb884':
+        ncolor = 8*8*4
+        pos = numpy.zeros((ncolor))
+        r = numpy.zeros((ncolor))
+        g = numpy.zeros((ncolor))
+        b = numpy.zeros((ncolor))
+        ind = 0
+        for ri in range(8):
+            for gi in range(8):
+                for bi in range(4):
+                    r[ind] = ri/(8-1.0)
+                    g[ind] = gi/(8-1.0)
+                    b[ind] = bi/(4-1.0)
+                    pos[ind] = ind/(ncolor-1.0)
+                    ind += 1
+        rgb_lvl = [8,8,4]  #Levels for RGB colors
+    
+    if ncolor == None:
+        raise ValueError, "The colormap selected is not valid"
+    
+    plplot.plscmap1n(ncolor)
+    plplot.plscmap1l(1, pos, r, g, b)
+    
+    return rgb_lvl
\ No newline at end of file