from schainpy.model.data.jrodata import * from schainpy.model.proc.jroproc_base import ProcessingUnit, Operation from schainpy.model.io.jroIO_base import * import scipy.io as sio import pprint import numpy as np from os import listdir from os.path import isfile, join import datetime import cmath class matoffReader(ProcessingUnit): index=None list=None firsttime=True utccounter=None utcfiletime=None utcmatcounter=0 utcfirst=None utclist=None def __init__(self): self.dataOut = Spectra() self.online = False return def __setHeader(self, datastuff): self.dataOut.pairsList=[(0,1)] self.dataOut.channelList = range(np.array(datastuff.get('power')).shape[1]) self.dataOut.nProfiles = len(np.array(datastuff.get('vel')).flatten()) #this! self.dataOut.nIncohInt = 20 self.dataOut.nCohInt = 1 #this! self.dataOut.ippSeconds = 0.004 #this! self.dataOut.nFFTPoints = len(np.array(datastuff.get('vel')).flatten()) self.dataOut.timeZone = 0 self.dataOut.heightList = np.array(datastuff.get('hts')).flatten() def __readFile(self, currentfile): print "Reading from this file:" + currentfile #filesplit=currentfile.split("\\") filesplit=currentfile.split("/") newsplit=filesplit[-2] newnewsplit=newsplit.split(".") newnewsplit=[int(i) for i in newnewsplit] gooblist=datetime.datetime(newnewsplit[0],newnewsplit[1],newnewsplit[2],newnewsplit[3],newnewsplit[4],newnewsplit[5]) self.utcfirst=(gooblist-datetime.datetime(1970,1,1)).total_seconds() newsplit=filesplit[-1] newnewsplit=newsplit.split(".") goobnum=newnewsplit[0] goobnum=int(goobnum) self.utcfirst=self.utcfirst+goobnum*2 # if (currentfile[43:]=='0.mat'): # self.utcmatcounter=0 # self.utcfirst=self.utclist[self.index] # if (self.utcmatcounter>60): # self.utcmatcounter=0 # print self.utcmatcounter print self.utcfirst try: datastuff=sio.loadmat(currentfile) except: return None, None dataphase=datastuff.get('phase') data3=datastuff.get('doppler0') data4=datastuff.get('doppler1') data3= np.array(data3) data4 = np.array(data4) datacoh=datastuff.get('coherence2') datacohphase=datacoh*np.exp(-dataphase*1j) # data31 = np.fliplr(data3) # data41 = np.fliplr(data4) data31 = data3.reshape((1,data3.shape[0],data3.shape[1])) data41 = data4.reshape((1,data4.shape[0],data4.shape[1])) datacohphase1 = datacohphase.reshape((1,datacoh.shape[0],datacoh.shape[1])) datastack = np.vstack((data31,data41)) self.__setHeader(datastuff) spc = datastack cspc = datacohphase1 return spc, cspc def __findFiles(self, path, startDate=None, endDate=None,startTime=datetime.time(0,0,0), endTime=datetime.time(23,59,59)): if startDate == None: startDate = datetime.date(1970,1,1) if endDate == None: endDate = datetime.date(2050,1,1) startsearch1=datetime.datetime.combine(startDate,startTime) startsearch2=(startsearch1-datetime.datetime(1970,1,1)).total_seconds() endsearch1=datetime.datetime.combine(endDate,endTime) endsearch2=(endsearch1-datetime.datetime(1970,1,1)).total_seconds() dirList = listdir(path) dirList = sorted(dirList) dirListFiltered=[] fileListFiltered=[] utclist=[] if not dirList: print "No directories found" return [] #if self.online: # dirList= [dirList[-1]] if self.online: currentdate = datetime.datetime.now() strsplit1=currentdate.strftime('%Y.%m.%d') dirList = fnmatch.filter(dirList,strsplit1+'*') for thisDir in dirList: if not os.path.isdir(os.path.join(path, thisDir)): continue strsplit=thisDir.split('.') timeints=[int(i) for i in strsplit] timelist=datetime.datetime(timeints[0],timeints[1],timeints[2],timeints[3],timeints[4],timeints[5]) utctime=(timelist-datetime.datetime(1970,1,1)).total_seconds() if not self.online: if (utctime > endsearch2): continue if (utctime < startsearch2): continue dirListFiltered.append(thisDir) utclist.append(utctime) if not dirListFiltered: print "filtro" return [] for thisDir in dirListFiltered: pathFile = os.path.join(self.path, thisDir) fileList = os.listdir(pathFile) if not fileList: continue for k in range(len(fileList)): thisFile = str(k)+'.mat' if not os.path.isfile(os.path.join(pathFile, thisFile)): continue fileListFiltered.append(os.path.join(pathFile, thisFile)) return fileListFiltered def __getNextOnlineFile(self, seconds = 40): filename = self.__getNextOfflineFile() if filename: return filename ncurrentfiles = len(self.fileList) nTries = 0 while (True): filelist = self.__findFiles(self.path) if len(filelist) > ncurrentfiles: break nTries += 1 if nTries > 3: break print "Waiting %d seconds ..." %seconds time.sleep(40) if not (len(filelist) > ncurrentfiles): return None self.fileList = filelist filename = self.__getNextOfflineFile() return filename def __getNextOfflineFile(self): if self.index >= len(self.fileList): return None filename=self.fileList[self.index] self.index += 1 return filename def __getNextFile(self): if self.online: filename = self.__getNextOnlineFile() else: filename = self.__getNextOfflineFile() return filename def setup(self, path, startDate=None, endDate=None,startTime=datetime.time(0,0,0), endTime=datetime.time(23,59,59)): fileList = self.__findFiles(path, startDate, endDate, startTime, endTime) if self.online: self.index = len(fileList) -1 else: self.index = 0 self.fileList = fileList print "fin setup" def run(self,path=None,startDate=None, endDate=None, startTime=datetime.time(0,0,0), endTime=datetime.time(23,59,59), walk=True,timezone='ut', all=0,online=False,ext=None,**kwargs): self.path=path self.ext=ext self.startDate=startDate self.endDate=endDate self.startTime=startTime self.endTime=endTime self.online = online self.dataOut.flagNoData = True if (self.firsttime==True): self.setup(path, startDate, endDate, startTime, endTime) self.firsttime=False if not self.fileList: self.dataOut.flagNoData = True print "lista vacia" return currentfile = self.__getNextFile() if not currentfile: "no file" return spc, cspc = self.__readFile(currentfile) if spc!=None: self.dataOut.data_spc = spc self.dataOut.data_cspc = cspc self.dataOut.utctime = self.utcfirst self.dataOut.flagNoData = False return 1