schain-jc Files · schainpy/Graphics/SpectraPlot.py

Modulo de lectura de espectros corregido y testeado....

Modulo de lectura de espectros corregido y testeado. Modulo de graficos de self-espectras agregado y probado.

Miguel Valdez - - Load All Authors

File last commit:

r25:7eaaf56ff468


                r25:7eaaf56ff468

Download file

             SpectraPlot.py
        
                    148 lines
            
             | 4.4 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / schainpy / Graphics / SpectraPlot.py
          
                    History
                
                 |
                  Source
                 | Raw
                 |Copy content
                 |Copy permalink

        Daniel Valdez
    
Fijando variables svn:keywords Author Id

              r16
            
      '''

      Created on Feb 7, 2012

      @author $Author$

      @version $Id$

      '''

        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
      import os, sys

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
      import numpy

      import plplot

        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
      path = os.path.split(os.getcwd())[0]

      sys.path.append(path)

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
      from Graphics.BaseGraph import *

      from Model.Spectra import Spectra

        Miguel Valdez
    
              r11
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
      class Spectrum():

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
          def __init__(self, Spectra):

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
              """

        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
              Inputs:

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
                  type:   "power" ->> Potencia

                          "iq"    ->> Real + Imaginario

              """

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
              self.__isPlotConfig = False

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
              self.__isPlotIni = False

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
              self.__xrange = None

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
              self.__yrange = None

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
              self.nGraphs = 0

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
              self.graphObjList = [] 

              self.m_Spectra = Spectra

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
          def __addGraph(self, subpage, title="", xlabel="", ylabel="", showColorbar=False, showPowerProfile=True, XAxisAsTime=False):

              graphObj = ColorPlot()

              graphObj.setup(subpage,

                             title,

                             xlabel,

                             ylabel,

                             showColorbar=showColorbar,

                             showPowerProfile=showPowerProfile,

                             XAxisAsTime=XAxisAsTime)

              self.graphObjList.append(graphObj)

          def setup(self, titleList=None, xlabelList=None, ylabelList=None, showColorbar=False, showPowerProfile=True, XAxisAsTime=False):

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
              nChan = int(self.m_Spectra.m_SystemHeader.numChannels)

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
              myTitle = ""

              myXlabel = ""

              myYlabel = ""

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
              for i in range(nChan):

                  if titleList != None:

                      myTitle = titleList[i]

                      myXlabel = xlabelList[i]

                      myYlabel = ylabelList[i]

                  self.__addGraph(i+1,

                                  title=myTitle,

                                  xlabel=myXlabel,

                                  ylabel=myYlabel,

                                  showColorbar=showColorbar,

                                  showPowerProfile=showPowerProfile,

                                  XAxisAsTime=XAxisAsTime)

              self.nGraphs = nChan

              self.__isPlotConfig = True

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
          def iniPlot(self):

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
              nx = int(numpy.sqrt(self.nGraphs)+1)

              #ny = int(self.nGraphs/nx)

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
              plplot.plsetopt("geometry", "%dx%d" %(400*nx, 300*nx))

              plplot.plsdev("xcairo")

              plplot.plscolbg(255,255,255)

              plplot.plscol0(1,0,0,0)

              plplot.plinit()

              plplot.plspause(False)

              plplot.pladv(0)

              plplot.plssub(nx, nx)

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
              self.__isPlotIni = True     

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
          def plotData(self, xmin=None, xmax=None, ymin=None, ymax=None, zmin=None, zmax=None, titleList=None, xlabelList=None, ylabelList=None, showColorbar=False, showPowerProfile=True, XAxisAsTime=False):

              if not(self.__isPlotConfig):

                  self.setup(titleList,

                             xlabelList,

                             ylabelList,

                             showColorbar,

                             showPowerProfile,

                             XAxisAsTime)

              if not(self.__isPlotIni):

                  self.iniPlot()

              data = numpy.log10(self.m_Spectra.data_spc)

              nX, nY, nChan = numpy.shape(data)

              x = numpy.arange(nX)        

              y = self.m_Spectra.heights

              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(abs(data))

              if zmax == None: zmax = numpy.nanmax(abs(data))

              plplot.plbop()

              for i in range(self.nGraphs):

                  self.graphObjList[i].iniSubpage()

                  self.graphObjList[i].plotData(data[:,:,i],

                                                x,

                                                y,

                                                xmin=xmin,

                                                xmax=xmax,

                                                ymin=ymin,

                                                ymax=ymax,

                                                zmin=zmin,

                                                zmax=zmax)

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
              plplot.plflush()

              plplot.pleop()

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
          def end(self):

              plplot.plend()

        Miguel Valdez
    
Adicion de directorios de procesamiento y utilitarios

              r10
            
      if __name__ == '__main__':

        Miguel Valdez
    
Modulo de lectura de espectros corregido y testeado....

              r25
            
          pass

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

Daniel Valdez Fijando variables svn:keywords Author Id	r16	'''
		Created on Feb 7, 2012

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

Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	import os, sys
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10	import numpy
		import plplot

Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	path = os.path.split(os.getcwd())[0]
		sys.path.append(path)
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	from Graphics.BaseGraph import *
		from Model.Spectra import Spectra
Miguel Valdez	r11
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	class Spectrum():
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	def __init__(self, Spectra):
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
		"""

Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	Inputs:
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	type: "power" ->> Potencia
		"iq" ->> Real + Imaginario
		"""
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	self.__isPlotConfig = False
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	self.__isPlotIni = False
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	self.__xrange = None
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	self.__yrange = None
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	self.nGraphs = 0
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	self.graphObjList = []

		self.m_Spectra = Spectra
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10

Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	def __addGraph(self, subpage, title="", xlabel="", ylabel="", showColorbar=False, showPowerProfile=True, XAxisAsTime=False):

		graphObj = ColorPlot()
		graphObj.setup(subpage,
		title,
		xlabel,
		ylabel,
		showColorbar=showColorbar,
		showPowerProfile=showPowerProfile,
		XAxisAsTime=XAxisAsTime)

		self.graphObjList.append(graphObj)


		def setup(self, titleList=None, xlabelList=None, ylabelList=None, showColorbar=False, showPowerProfile=True, XAxisAsTime=False):
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	nChan = int(self.m_Spectra.m_SystemHeader.numChannels)
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	myTitle = ""
		myXlabel = ""
		myYlabel = ""
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	for i in range(nChan):
		if titleList != None:
		myTitle = titleList[i]
		myXlabel = xlabelList[i]
		myYlabel = ylabelList[i]

		self.__addGraph(i+1,
		title=myTitle,
		xlabel=myXlabel,
		ylabel=myYlabel,
		showColorbar=showColorbar,
		showPowerProfile=showPowerProfile,
		XAxisAsTime=XAxisAsTime)

		self.nGraphs = nChan
		self.__isPlotConfig = True
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	def iniPlot(self):
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	nx = int(numpy.sqrt(self.nGraphs)+1)
		#ny = int(self.nGraphs/nx)
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	plplot.plsetopt("geometry", "%dx%d" %(400nx, 300nx))
		plplot.plsdev("xcairo")
		plplot.plscolbg(255,255,255)
		plplot.plscol0(1,0,0,0)
		plplot.plinit()
		plplot.plspause(False)
		plplot.pladv(0)
		plplot.plssub(nx, nx)
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	self.__isPlotIni = True
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	def plotData(self, xmin=None, xmax=None, ymin=None, ymax=None, zmin=None, zmax=None, titleList=None, xlabelList=None, ylabelList=None, showColorbar=False, showPowerProfile=True, XAxisAsTime=False):

		if not(self.__isPlotConfig):
		self.setup(titleList,
		xlabelList,
		ylabelList,
		showColorbar,
		showPowerProfile,
		XAxisAsTime)

		if not(self.__isPlotIni):
		self.iniPlot()

		data = numpy.log10(self.m_Spectra.data_spc)

		nX, nY, nChan = numpy.shape(data)

		x = numpy.arange(nX)
		y = self.m_Spectra.heights

		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(abs(data))
		if zmax == None: zmax = numpy.nanmax(abs(data))

		plplot.plbop()
		for i in range(self.nGraphs):
		self.graphObjList[i].iniSubpage()
		self.graphObjList[i].plotData(data[:,:,i],
		x,
		y,
		xmin=xmin,
		xmax=xmax,
		ymin=ymin,
		ymax=ymax,
		zmin=zmin,
		zmax=zmax)
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25
		plplot.plflush()
		plplot.pleop()
Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	def end(self):
		plplot.plend()

Miguel Valdez Adicion de directorios de procesamiento y utilitarios	r10
		if __name__ == '__main__':
Miguel Valdez Modulo de lectura de espectros corregido y testeado....	r25	pass