##// END OF EJS Templates
schain-jc
RSS
La opcion nombre de graficos ha sido añadida a los modulos de VoltagePlot y SpectraPlot
Miguel Valdez -
r57:141fba166d2d
parent child
Show More
Show file before | Show file after | Hide comments
@@ -1,174 +1,174
1 '''
1 '''
2 Created on Feb 7, 2012
2 Created on Feb 7, 2012
3
3
4 @author $Author$
4 @author $Author$
5 @version $Id$
5 @version $Id$
6 '''
6 '''
7
7
8 import os, sys
8 import os, sys
9 import numpy
9 import numpy
10 import datetime
10 import datetime
11 import plplot
11 import plplot
12
12
13 path = os.path.split(os.getcwd())[0]
13 path = os.path.split(os.getcwd())[0]
14 sys.path.append(path)
14 sys.path.append(path)
15
15
16 from Graphics.BaseGraph import *
16 from Graphics.BaseGraph import *
17 from Model.Spectra import Spectra
17 from Model.Spectra import Spectra
18
18
19 class Spectrum():
19 class Spectrum():
20
20
21 def __init__(self, Spectra, index=0):
21 def __init__(self, Spectra, index=0):
22
22
23 """
23 """
24
24
25 Inputs:
25 Inputs:
26
26
27 type: "power" ->> Potencia
27 type: "power" ->> Potencia
28 "iq" ->> Real + Imaginario
28 "iq" ->> Real + Imaginario
29 """
29 """
30
30
31 self.__isPlotConfig = False
31 self.__isPlotConfig = False
32
32
33 self.__isPlotIni = False
33 self.__isPlotIni = False
34
34
35 self.__xrange = None
35 self.__xrange = None
36
36
37 self.__yrange = None
37 self.__yrange = None
38
38
39 self.nGraphs = 0
39 self.nGraphs = 0
40
40
41 self.indexPlot = index
41 self.indexPlot = index
42
42
43 self.graphObjList = []
43 self.graphObjList = []
44
44
45 self.m_Spectra = Spectra
45 self.m_Spectra = Spectra
46
46
47
47
48 def __addGraph(self, subpage, title="", xlabel="", ylabel="", showColorbar=False, showPowerProfile=True, XAxisAsTime=False):
48 def __addGraph(self, subpage, title="", xlabel="", ylabel="", showColorbar=False, showPowerProfile=True, XAxisAsTime=False):
49
49
50 graphObj = ColorPlot()
50 graphObj = ColorPlot()
51 graphObj.setup(subpage,
51 graphObj.setup(subpage,
52 title,
52 title,
53 xlabel,
53 xlabel,
54 ylabel,
54 ylabel,
55 showColorbar=showColorbar,
55 showColorbar=showColorbar,
56 showPowerProfile=showPowerProfile,
56 showPowerProfile=showPowerProfile,
57 XAxisAsTime=XAxisAsTime)
57 XAxisAsTime=XAxisAsTime)
58
58
59 self.graphObjList.append(graphObj)
59 self.graphObjList.append(graphObj)
60
60
61
61
62 def setup(self, titleList=None, xlabelList=None, ylabelList=None, showColorbar=False, showPowerProfile=True, XAxisAsTime=False):
62 def setup(self, titleList=None, xlabelList=None, ylabelList=None, showColorbar=False, showPowerProfile=True, XAxisAsTime=False):
63
63
64 nChan = int(self.m_Spectra.m_SystemHeader.numChannels)
64 nChan = int(self.m_Spectra.m_SystemHeader.numChannels)
65 channels = range(nChan)
65 channels = range(nChan)
66
66
67 myXlabel = "Radial Velocity (m/s)"
67 myXlabel = "Radial Velocity (m/s)"
68 myYlabel = "Range (km)"
68 myYlabel = "Range (km)"
69
69
70 for i in channels:
70 for i in channels:
71 if titleList != None:
71 if titleList != None:
72 myTitle = titleList[i]
72 myTitle = titleList[i]
73 myXlabel = xlabelList[i]
73 myXlabel = xlabelList[i]
74 myYlabel = ylabelList[i]
74 myYlabel = ylabelList[i]
75
75
76 # if self.m_Spectra.m_NoiseObj != None:
76 # if self.m_Spectra.m_NoiseObj != None:
77 # noise = '%4.2fdB' %(self.m_Spectra.m_NoiseObj[i])
77 # noise = '%4.2fdB' %(self.m_Spectra.m_NoiseObj[i])
78 # else:
78 # else:
79 noise = '--'
79 noise = '--'
80
80
81 myTitle = "Channel: %d - Noise: %s" %(i, noise)
81 myTitle = "Channel: %d - Noise: %s" %(i, noise)
82
82
83 self.__addGraph(i+1,
83 self.__addGraph(i+1,
84 title=myTitle,
84 title=myTitle,
85 xlabel=myXlabel,
85 xlabel=myXlabel,
86 ylabel=myYlabel,
86 ylabel=myYlabel,
87 showColorbar=showColorbar,
87 showColorbar=showColorbar,
88 showPowerProfile=showPowerProfile,
88 showPowerProfile=showPowerProfile,
89 XAxisAsTime=XAxisAsTime)
89 XAxisAsTime=XAxisAsTime)
90
90
91 self.nGraphs = nChan
91 self.nGraphs = nChan
92 self.__isPlotConfig = True
92 self.__isPlotConfig = True
93
93
94 def iniPlot(self, winTitle=""):
94 def iniPlot(self, winTitle=""):
95
95
96 nx = int(numpy.sqrt(self.nGraphs)+1)
96 nx = int(numpy.sqrt(self.nGraphs)+1)
97 #ny = int(self.nGraphs/nx)
97 #ny = int(self.nGraphs/nx)
98
98
99 plplot.plsstrm(self.indexPlot)
99 plplot.plsstrm(self.indexPlot)
100 plplot.plparseopts([winTitle], plplot.PL_PARSE_FULL)
100 plplot.plparseopts([winTitle], plplot.PL_PARSE_FULL)
101 plplot.plsetopt("geometry", "%dx%d" %(300*nx, 240*nx))
101 plplot.plsetopt("geometry", "%dx%d" %(300*nx, 240*nx))
102 plplot.plsdev("xwin")
102 plplot.plsdev("xwin")
103 plplot.plscolbg(255,255,255)
103 plplot.plscolbg(255,255,255)
104 plplot.plscol0(1,0,0,0)
104 plplot.plscol0(1,0,0,0)
105 plplot.plinit()
105 plplot.plinit()
106 plplot.plspause(False)
106 plplot.plspause(False)
107 plplot.pladv(0)
107 plplot.pladv(0)
108 plplot.plssub(nx, nx)
108 plplot.plssub(nx, nx)
109
109
110 self.__nx = nx
110 self.__nx = nx
111 self.__ny = nx
111 self.__ny = nx
112 self.__isPlotIni = True
112 self.__isPlotIni = True
113
113
114
114
115 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, winTitle="Spectra"):
115 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, winTitle="Spectra"):
116
116
117 if not(self.__isPlotConfig):
117 if not(self.__isPlotConfig):
118 self.setup(titleList,
118 self.setup(titleList,
119 xlabelList,
119 xlabelList,
120 ylabelList,
120 ylabelList,
121 showColorbar,
121 showColorbar,
122 showPowerProfile,
122 showPowerProfile,
123 XAxisAsTime)
123 XAxisAsTime)
124
124
125 if not(self.__isPlotIni):
125 if not(self.__isPlotIni):
126 self.iniPlot()
126 self.iniPlot(winTitle)
127
127
128 plplot.plsstrm(self.indexPlot)
128 plplot.plsstrm(self.indexPlot)
129
129
130 data = 10.*numpy.log10(self.m_Spectra.data_spc)
130 data = 10.*numpy.log10(self.m_Spectra.data_spc)
131
131
132 nX, nY, nChan = numpy.shape(data)
132 nX, nY, nChan = numpy.shape(data)
133
133
134 x = numpy.arange(nX)
134 x = numpy.arange(nX)
135 y = self.m_Spectra.heights
135 y = self.m_Spectra.heights
136
136
137 thisDatetime = datetime.datetime.fromtimestamp(self.m_Spectra.m_BasicHeader.utc)
137 thisDatetime = datetime.datetime.fromtimestamp(self.m_Spectra.m_BasicHeader.utc)
138 txtDate = "Self Spectra - Date: %s" %(thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
138 txtDate = "Self Spectra - Date: %s" %(thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
139
139
140 if xmin == None: xmin = x[0]
140 if xmin == None: xmin = x[0]
141 if xmax == None: xmax = x[-1]
141 if xmax == None: xmax = x[-1]
142 if ymin == None: ymin = y[0]
142 if ymin == None: ymin = y[0]
143 if ymax == None: ymax = y[-1]
143 if ymax == None: ymax = y[-1]
144 if zmin == None: zmin = numpy.nanmin(abs(data))
144 if zmin == None: zmin = numpy.nanmin(abs(data))
145 if zmax == None: zmax = numpy.nanmax(abs(data))
145 if zmax == None: zmax = numpy.nanmax(abs(data))
146
146
147 plplot.plbop()
147 plplot.plbop()
148
148
149 plplot.plssub(self.__nx, self.__ny)
149 plplot.plssub(self.__nx, self.__ny)
150 for i in range(self.nGraphs):
150 for i in range(self.nGraphs):
151 self.graphObjList[i].iniSubpage()
151 self.graphObjList[i].iniSubpage()
152 self.graphObjList[i].plotData(data[i,:,:],
152 self.graphObjList[i].plotData(data[i,:,:],
153 x,
153 x,
154 y,
154 y,
155 xmin=xmin,
155 xmin=xmin,
156 xmax=xmax,
156 xmax=xmax,
157 ymin=ymin,
157 ymin=ymin,
158 ymax=ymax,
158 ymax=ymax,
159 zmin=zmin,
159 zmin=zmin,
160 zmax=zmax)
160 zmax=zmax)
161
161
162 plplot.plssub(1,0)
162 plplot.plssub(1,0)
163 plplot.pladv(0)
163 plplot.pladv(0)
164 plplot.plvpor(0., 1., 0., 1.)
164 plplot.plvpor(0., 1., 0., 1.)
165 plplot.plmtex("t",-1., 0.5, 0.5, txtDate)
165 plplot.plmtex("t",-1., 0.5, 0.5, txtDate)
166 plplot.plflush()
166 plplot.plflush()
167 plplot.pleop()
167 plplot.pleop()
168
168
169 def end(self):
169 def end(self):
170 plplot.plend()
170 plplot.plend()
171
171
172
172
173 if __name__ == '__main__':
173 if __name__ == '__main__':
174 pass No newline at end of file
174 pass
Show file before | Show file after | Hide comments
@@ -1,182 +1,182
1 '''
1 '''
2 Created on Feb 7, 2012
2 Created on Feb 7, 2012
3
3
4 @author $Author$
4 @author $Author$
5 @version $Id$
5 @version $Id$
6 '''
6 '''
7 import os, sys
7 import os, sys
8 import numpy
8 import numpy
9 import plplot
9 import plplot
10
10
11 path = os.path.split(os.getcwd())[0]
11 path = os.path.split(os.getcwd())[0]
12 sys.path.append(path)
12 sys.path.append(path)
13
13
14 from Graphics.BaseGraph import *
14 from Graphics.BaseGraph import *
15 from Model.Voltage import Voltage
15 from Model.Voltage import Voltage
16
16
17 class Osciloscope():
17 class Osciloscope():
18
18
19 def __init__(self, Voltage, index=0):
19 def __init__(self, Voltage, index=0):
20
20
21 """
21 """
22
22
23 Inputs:
23 Inputs:
24
24
25 type: "power" ->> Potencia
25 type: "power" ->> Potencia
26 "iq" ->> Real + Imaginario
26 "iq" ->> Real + Imaginario
27 """
27 """
28
28
29 self.__isPlotConfig = False
29 self.__isPlotConfig = False
30
30
31 self.__isPlotIni = False
31 self.__isPlotIni = False
32
32
33 self.__xrange = None
33 self.__xrange = None
34
34
35 self.__yrange = None
35 self.__yrange = None
36
36
37 self.m_Voltage = None
37 self.m_Voltage = None
38
38
39 self.nGraphs = 0
39 self.nGraphs = 0
40
40
41 self.indexPlot = index
41 self.indexPlot = index
42
42
43 self.graphObjList = []
43 self.graphObjList = []
44
44
45 self.m_Voltage = Voltage
45 self.m_Voltage = Voltage
46
46
47
47
48 def __addGraph(self, subpage, title="", xlabel="", ylabel="", XAxisAsTime=False):
48 def __addGraph(self, subpage, title="", xlabel="", ylabel="", XAxisAsTime=False):
49
49
50 graphObj = LinearPlot()
50 graphObj = LinearPlot()
51 graphObj.setup(subpage, title="", xlabel="", ylabel="", XAxisAsTime=False)
51 graphObj.setup(subpage, title="", xlabel="", ylabel="", XAxisAsTime=False)
52 #graphObj.setScreenPos()
52 #graphObj.setScreenPos()
53
53
54 self.graphObjList.append(graphObj)
54 self.graphObjList.append(graphObj)
55
55
56 del graphObj
56 del graphObj
57
57
58 # def setXRange(self, xmin, xmax):
58 # def setXRange(self, xmin, xmax):
59 # self.__xrange = (xmin, xmax)
59 # self.__xrange = (xmin, xmax)
60 #
60 #
61 # def setYRange(self, ymin, ymax):
61 # def setYRange(self, ymin, ymax):
62 # self.__yrange = (ymin, ymax)
62 # self.__yrange = (ymin, ymax)
63
63
64
64
65 def setup(self, titleList=None, xlabelList=None, ylabelList=None, XAxisAsTime=False):
65 def setup(self, titleList=None, xlabelList=None, ylabelList=None, XAxisAsTime=False):
66
66
67 nChan = int(self.m_Voltage.m_SystemHeader.numChannels)
67 nChan = int(self.m_Voltage.m_SystemHeader.numChannels)
68
68
69 myTitle = ""
69 myTitle = ""
70 myXlabel = ""
70 myXlabel = ""
71 myYlabel = ""
71 myYlabel = ""
72
72
73 for i in range(nChan):
73 for i in range(nChan):
74 if titleList != None:
74 if titleList != None:
75 myTitle = titleList[i]
75 myTitle = titleList[i]
76 myXlabel = xlabelList[i]
76 myXlabel = xlabelList[i]
77 myYlabel = ylabelList[i]
77 myYlabel = ylabelList[i]
78
78
79 self.__addGraph(i+1, title=myTitle, xlabel=myXlabel, ylabel=myYlabel, XAxisAsTime=XAxisAsTime)
79 self.__addGraph(i+1, title=myTitle, xlabel=myXlabel, ylabel=myYlabel, XAxisAsTime=XAxisAsTime)
80
80
81 self.nGraphs = nChan
81 self.nGraphs = nChan
82 self.__isPlotConfig = True
82 self.__isPlotConfig = True
83
83
84 def iniPlot(self, winTitle=""):
84 def iniPlot(self, winTitle=""):
85
85
86 plplot.plsstrm(self.indexPlot)
86 plplot.plsstrm(self.indexPlot)
87 plplot.plparseopts([winTitle], plplot.PL_PARSE_FULL)
87 plplot.plparseopts([winTitle], plplot.PL_PARSE_FULL)
88 plplot.plsetopt("geometry", "%dx%d" %(700, 115*self.nGraphs))
88 plplot.plsetopt("geometry", "%dx%d" %(700, 115*self.nGraphs))
89 plplot.plsdev("xwin")
89 plplot.plsdev("xwin")
90 plplot.plscolbg(255,255,255)
90 plplot.plscolbg(255,255,255)
91 plplot.plscol0(1,0,0,0)
91 plplot.plscol0(1,0,0,0)
92 plplot.plinit()
92 plplot.plinit()
93 plplot.plspause(False)
93 plplot.plspause(False)
94 plplot.plssub(1, self.nGraphs)
94 plplot.plssub(1, self.nGraphs)
95
95
96 self.__isPlotIni = True
96 self.__isPlotIni = True
97
97
98 def plotData(self, xmin=None, xmax=None, ymin=None, ymax=None, idProfile=None, titleList=None, xlabelList=None, ylabelList=None, XAxisAsTime=False, type='iq', winTitle="Voltage"):
98 def plotData(self, xmin=None, xmax=None, ymin=None, ymax=None, idProfile=None, titleList=None, xlabelList=None, ylabelList=None, XAxisAsTime=False, type='iq', winTitle="Voltage"):
99
99
100 if idProfile != None and idProfile != self.m_Voltage.idProfile:
100 if idProfile != None and idProfile != self.m_Voltage.idProfile:
101 return
101 return
102
102
103 if not(self.__isPlotConfig):
103 if not(self.__isPlotConfig):
104 self.setup(titleList, xlabelList, ylabelList, XAxisAsTime)
104 self.setup(titleList, xlabelList, ylabelList, XAxisAsTime)
105
105
106 if not(self.__isPlotIni):
106 if not(self.__isPlotIni):
107 self.iniPlot()
107 self.iniPlot(winTitle)
108
108
109 plplot.plsstrm(self.indexPlot)
109 plplot.plsstrm(self.indexPlot)
110
110
111 data = self.m_Voltage.data
111 data = self.m_Voltage.data
112
112
113 x = self.m_Voltage.heights
113 x = self.m_Voltage.heights
114
114
115 if xmin == None: xmin = x[0]
115 if xmin == None: xmin = x[0]
116 if xmax == None: xmax = x[-1]
116 if xmax == None: xmax = x[-1]
117 if ymin == None: ymin = numpy.nanmin(abs(data))
117 if ymin == None: ymin = numpy.nanmin(abs(data))
118 if ymax == None: ymax = numpy.nanmax(abs(data))
118 if ymax == None: ymax = numpy.nanmax(abs(data))
119
119
120 plplot.plbop()
120 plplot.plbop()
121 for i in range(self.nGraphs):
121 for i in range(self.nGraphs):
122 y = data[:,i]
122 y = data[:,i]
123
123
124 self.graphObjList[i].iniSubpage()
124 self.graphObjList[i].iniSubpage()
125 self.graphObjList[i].plotComplexData(x, y, xmin, xmax, ymin, ymax, 8, type)
125 self.graphObjList[i].plotComplexData(x, y, xmin, xmax, ymin, ymax, 8, type)
126
126
127 plplot.plflush()
127 plplot.plflush()
128 plplot.pleop()
128 plplot.pleop()
129
129
130 def end(self):
130 def end(self):
131 plplot.plend()
131 plplot.plend()
132
132
133 class VoltagePlot(object):
133 class VoltagePlot(object):
134 '''
134 '''
135 classdocs
135 classdocs
136 '''
136 '''
137
137
138 __m_Voltage = None
138 __m_Voltage = None
139
139
140 def __init__(self, m_Voltage):
140 def __init__(self, m_Voltage):
141 '''
141 '''
142 Constructor
142 Constructor
143 '''
143 '''
144 self.__m_Voltage = m_Voltage
144 self.__m_Voltage = m_Voltage
145
145
146 def setup(self):
146 def setup(self):
147 pass
147 pass
148
148
149 def addGraph(self, type, xrange=None, yrange=None, zrange=None):
149 def addGraph(self, type, xrange=None, yrange=None, zrange=None):
150 pass
150 pass
151
151
152 def plotData(self):
152 def plotData(self):
153 pass
153 pass
154
154
155 if __name__ == '__main__':
155 if __name__ == '__main__':
156
156
157 import numpy
157 import numpy
158
158
159 plplot.plsetopt("geometry", "%dx%d" %(450*2, 200*2))
159 plplot.plsetopt("geometry", "%dx%d" %(450*2, 200*2))
160 plplot.plsdev("xcairo")
160 plplot.plsdev("xcairo")
161 plplot.plscolbg(255,255,255)
161 plplot.plscolbg(255,255,255)
162 plplot.plscol0(1,0,0,0)
162 plplot.plscol0(1,0,0,0)
163 plplot.plinit()
163 plplot.plinit()
164 plplot.plssub(1, 2)
164 plplot.plssub(1, 2)
165
165
166 nx = 64
166 nx = 64
167 ny = 100
167 ny = 100
168
168
169 data = numpy.random.uniform(-50,50,(nx,ny))
169 data = numpy.random.uniform(-50,50,(nx,ny))
170
170
171 baseObj = RTI()
171 baseObj = RTI()
172 baseObj.setup(1, "Spectrum", "Frequency", "Range", "br_green", False, False)
172 baseObj.setup(1, "Spectrum", "Frequency", "Range", "br_green", False, False)
173 baseObj.plotData(data)
173 baseObj.plotData(data)
174
174
175 data = numpy.random.uniform(-50,50,(nx,ny))
175 data = numpy.random.uniform(-50,50,(nx,ny))
176
176
177 base2Obj = RTI()
177 base2Obj = RTI()
178 base2Obj.setup(2, "Spectrum", "Frequency", "Range", "br_green", True, True)
178 base2Obj.setup(2, "Spectrum", "Frequency", "Range", "br_green", True, True)
179 base2Obj.plotData(data)
179 base2Obj.plotData(data)
180
180
181 plplot.plend()
181 plplot.plend()
182 exit(0) No newline at end of file
182 exit(0)
General Comments 0
You need to be logged in to leave comments. Login now