Ejemplo n.º 1
0
 def display(self,item):
     self.currentFile=self.listShots.currentItem().text()
     signal1=self.signalChoice1.currentText()
     signal2=self.signalChoice2.currentText()
     time1,data1,sampling1=readHdf5.getData(self.currentFile,signal1,self.env)
     time2,data2,sampling2=readHdf5.getData(self.currentFile,signal2,self.env)
     if sampling2>=sampling1:
          self.data2=interp1d(time2,data2)(time1)
          self.timei=time1
          self.data1=data1
     else:
          self.data1=interp1d(time1,data1)(time2)
          self.timei=time2
          self.data2=data2
     self.p1.clear()
     self.p1.plot(self.timei,data1)
     self.p2.clear()
     self.p2.plot(self.timei,self.data2)
     #self.p2.linkXAxis(self.p1)
     self.lr1=pg.LinearRegionItem([self.timei[0],self.timei[-1]])
     self.lr1.setZValue(-10)
     self.lr2=pg.LinearRegionItem([self.timei[0],self.timei[-1]])
     self.lr2.setZValue(-10)
     self.p1.addItem(self.lr1)
     self.p2.addItem(self.lr2)
     self.lr1.sigRegionChanged.connect(self.updatePlot1)
     self.lr2.sigRegionChanged.connect(self.updatePlot2)
Ejemplo n.º 2
0
 def plotSignal(self,item):
     indexColor=0
     self.p1.clear()
     self.p2.clear()
     for fileName in self.fileName:
         print fileName
         self.Back=False
         self.signalName=item
         time,data,self.sampling=readHdf5.getData(fileName,item,self.env)
         self.time=np.array(time)
         self.data=np.array(data)
         self.frqlabel.setText('sampling rate: '+str(self.sampling))
         pencil=pg.mkPen(color=pg.intColor(indexColor))
         self.p1.plot(self.time,self.data,pen=pencil)
         self.p2.plot(self.time,self.data,pen=pencil)
         indexColor=indexColor+1
         
     if self.existingData:
         self.lr=pg.LinearRegionItem([self.left,self.right])
     else:
         self.lr=pg.LinearRegionItem([self.time[0],self.time[-1]])
         self.left=self.time[0]
         self.right=self.time[-1]
     self.lr.setZValue(-10)
     self.p1.addItem(self.lr)
     self.lr.sigRegionChanged.connect(self.updatePlot)
Ejemplo n.º 3
0
 def __init__(self,filename,env,parent=None):
   super(positionARM,self).__init__(parent)
   pg.setConfigOption('background', 'w')
   pg.setConfigOption('foreground', 'k')
   self.fileName=filename[0]
   self.setWindowTitle("Correlate attributes")
   #self.resize(1000,1000)
   self.env=env       
   view=pg.PlotWidget()
   item='PXI M6251/manip_pos'
   time,data,self.sampling=readHdf5.getData(self.fileName,item,self.env)
   data=self.movingaverage(data,5000)
   view.plot(time,data)
   vbox1=QtGui.QVBoxLayout()
   vbox1.addStretch(1)
   vbox1.addWidget(view)
   self.setLayout(vbox1)
   self.show()
Ejemplo n.º 4
0
    def get(self):
#        self.write('List of discharges\n')        
#        
#        conn=sqlite3.connect('ishtar')
#        curs=conn.cursor()
#        curs.execute('select * from shots')
#        for row,form in enumerate(curs):
#            for column, item in enumerate(form):
#                #print str(item)
#                self.write(str(item)+'\n')
        env=Environment()
        time,data,sampling=readHdf5.getData('00857_Data','Generator/Fpower',env)
        #print time[100],data[100]
        plot = figure()
        plot.line(np.array(time), np.array(data))

        html = file_html(plot, CDN, "my plot") 
        self.write(html)
Ejemplo n.º 5
0
    def calculate(self):
        self.results=dict()        
        #item=self.signalChoice.currentText()
        for file in self.fileName:

            for item in self.env.getSignals():
                try:
                    print item
                    time,data,self.sampling=readHdf5.getData(file,item,self.env) 
                    rang=self.p2.getPlotItem().getViewBox().viewRange()
                    rang2=rang[0]
                    average=np.mean(data[(time>=rang2[0]) & (time<=rang2[1])])
                    maxi=np.max(data[(time>=rang2[0]) & (time<=rang2[1])])
                    mini=np.min(data[(time>=rang2[0]) & (time<=rang2[1])])
               #     self.results[file]=[average,maxi,mini]
                #    self.textResult.append(file+':\n'+'Mean: '+str(average)+'\n'+'Max: '+str(maxi)+'\n'+'Min: '+str(mini)+'\n')
                    nameAttr=self.env.attr(item)+self.attribut.text()
                    readHdf5.saveAttr(file,nameAttr,np.array([average,maxi,mini]),self.env)
                except:
                    pass
        pickle.dump([self.left,self.right,self.attribut.text(),self.signalChoice.currentText()], open( "User.dat", "wb" ) )       
Ejemplo n.º 6
0
 def calculatefft(self):
     indexColor=0
     self.p3.clear()
     self.p3.addLegend()
     for fileName in self.fileName:
         time,data,self.sampling=readHdf5.getData(fileName,self.signalName,self.env)
         self.time=np.array(time)
         self.data=np.array(data)
         rang=self.p2.getPlotItem().getViewBox().viewRange()
         rang2=rang[0]
         # Number of samplepoints
         dataslice=self.data[(self.time>=rang2[0]) & (self.time<=rang2[1])]
         N=len(dataslice)
         # sample spacing
         T = 1.0 / self.sampling
         yf = scipy.fftpack.fft(dataslice)
         xf = np.linspace(0.0, 1.0/(2.0*T), N/2)
         pencil=pg.mkPen(color=pg.intColor(indexColor))
         self.p3.plot(xf, 2.0/N * np.abs(yf[0:N/2]),pen=pencil,name=fileName[0:-5])
         indexColor=indexColor+1
     self.p3.setLabel('left', "Spectral power (SI)")
     self.p3.setLabel('bottom', "Frequency (Hz)")
Ejemplo n.º 7
0
    def __init__(self,filename,env,parent=None):
         super(overview,self).__init__(parent)
         pg.setConfigOption('background', 'w')
         pg.setConfigOption('foreground', 'k')
         self.fileName=filename
         self.setWindowTitle("Overview Signal")
         self.showMaximized()
         self.env=env
         p1=self.addPlot(title='Current Big coils')
         self.nextRow()
         p2=self.addPlot(title='Pressure')
         self.nextRow()
         p3=self.addPlot(title='Helicon power')         
         self.nextRow()
         p4=self.addPlot(title='ICRF power')      
         self.nextRow()
         p5=self.addPlot(title='Density Langmuir probe')         
         self.nextRow()
         p6=self.addPlot(title='Amplitude B-dot probes')         
         self.nextRow()
         p7=self.addPlot(title='Phase B-dot probes')         
         indexColor=0
         for x in self.fileName:
             try:
                 timeCoils,dataCoils,sampling=readHdf5.getData(x,'S7/Big_coil_I [A]',env)
             except:
                 pass

             try:
                 timeP,dataP,sampling=readHdf5.getData(x,'S7/vacuum [mbar]',env)
             except: 
                 pass
             try:
                 timeHF,dataHF,sampling=readHdf5.getData(x,'Generator/Fpower',env)
             except: 
                 pass
             try:
                 timeHR,dataHR,sampling=readHdf5.getData(x,'Generator/Rpower',env)
             except: 
                 pass
             try:
                 timeIF,dataIF,sampling=readHdf5.getData(x,'PXI M6251/IC_fwd',env)
             except: 
                 pass
             try:
                 timeIR,dataIR,sampling=readHdf5.getData(x,'PXI M6251/IC_ref',env)
             except: 
                 pass
             try:
                 dataL=np.array(readHdf5.getDataProcess(x,'Process/Langmuir_dens',env))
                 timeL=np.array(readHdf5.getDataProcess(x,'Process/Langmuir_time',env))
             except: 
                 pass
             try:
                 timeFW,dataFW,sampling=readHdf5.getData(x,'PXI M6251/U_fast',env)
             except: 
                 pass
             try:
                 timeSW,dataSW,sampling=readHdf5.getData(x,'PXI M6251/U_slow',env)
             except: 
                 pass
             try:
                 timePhase1,dataPhase1,sampling=readHdf5.getData(x,'PXI M6251/Phi_fast',env)
             except: 
                 pass
             try:
                 timePhase2,dataPhase2,sampling=readHdf5.getData(x,'PXI M6251/Phi_slow',env)
             except: 
                 pass
             
             pencil=pg.mkPen(color=pg.intColor(indexColor)) 
             try:
                 p1.plot(timeCoils,dataCoils,pen=pencil)
             except:    
                 pass
             try:
                p2.plot(timeP,dataP,pen=pencil)
             except:    
                pass
             try:
                p3.plot(timeHF,dataHF,pen=pencil)
                p3.plot(timeHR,dataHR,pen=pencil)
             except:    
                pass
             try:
                p4.plot(timeIF,dataIF,pen=pencil)
                p4.plot(timeIR,dataIR,pen=pencil)
             except:    
                pass
             try:
                p5.plot(timeL,dataL,pen=pencil)
             except:    
                pass
             try:
                p6.plot(timeFW,dataFW,pen=pencil)
                p6.plot(timeSW,dataSW,pen=pencil)
             except:    
                pass
             try:
                p7.plot(timePhase1,dataPhase1-dataPhase2,pen=pencil)

             except:
                 pass
             indexColor=indexColor+1
             
         p7.setXLink(p1)
         p6.setXLink(p1)
         p5.setXLink(p1)        
         p4.setXLink(p1)         
         p3.setXLink(p1)         
         p2.setXLink(p1)