data plotting; waveform filtering and convenience imports in order to get a overview window displaying data; new data structure available -> PyLoT should be able to read PILOT data

QtPyLoT.py

@@ -166,7 +166,20 @@ class MainWindow(QMainWindow):
             self.data = Data(evtdata=self.fname)
 
     def getWFFnames(self):
-        pass
+        try:
+            evt = self.getData().getEvtData()
+            if evt.picks:
+                for pick in evt.picks:
+                    if pick.waveform_id is not None:
+                        fname = pick.waveform_id.getSEEDstring()
+                        if fname not in self.fnames:
+                            self.fnames.append(fname)
+            else:
+                if self.dataStructure:
+                    searchPath = self.dataStructure.expandDataPath()
+        except:
+            return None
+
 
     def saveData(self):
         settings = QSettings()
@@ -177,7 +190,7 @@ class MainWindow(QMainWindow):
             return False
         return True
 
-    def getComponent(self):
+    def getDispComponent(self):
         return self
 
     def getData(self):
@@ -294,20 +307,20 @@ class MainWindow(QMainWindow):
         return True
 
     def plotData(self):
-        pass #self.data.plotData(self.DataPlot)
+        self.getData().plotData(self.getDataWidget())
 
     def filterData(self):
         if self.getData():
             kwargs = {}
-            freq = self.filteroptions.getFreq()
+            freq = self.getFilterOptions().getFreq()
             if len(freq) > 1:
                 kwargs['freqmin'] = freq[0]
                 kwargs['freqmax'] = freq[1]
             else:
                 kwargs['freq'] = freq
-            kwargs['type'] = self.filteroptions.getFilterType()
+            kwargs['type'] = self.getFilterOptions().getFilterType()
-            #kwargs['order'] = self.filteroptions.getOrder()
+            kwargs['corners'] = self.filteroptions.getOrder()
-            self.data.filter(**kwargs)
+            self.getData().filter(kwargs)
 
     def adjustFilterOptions(self):
         filteroptions = None

pylot/core/read/__init__.py

@@ -1,6 +1,8 @@
 from pylot.core.read.inputs import AutoPickParameter
 from pylot.core.read.inputs import FilterOptions
-from pylot.core.read.data import Data
 from pylot.core.read.data import GenericDataStructure
 from pylot.core.read.data import SeiscompDataStructure
+from pylot.core.read.data import PilotDataStructure
+from pylot.core.read.data import Data
+
 

pylot/core/read/data.py

@@ -94,8 +94,10 @@ class Data(object):
                               not implemented: {1}'''.format(evtformat, e))
 
     def plotData(self, widget):
-        time_ax = np.arange(0, len(self.wfdata[0].data)/self.wfdata[0].stats.sampling_rate, self.wfdata[0].stats.delta)
+        wfst = self.getWFData()
-        widget.axes.plot(time_ax, self.wfdata[0].data)
+        time_ax = np.arange(0, len(wfst[0].data) / wfst[0].stats.sampling_rate,
+                            wfst[0].stats.delta)
+        widget.axes.plot(time_ax, wfst[0].data)
 
     def getID(self):
         try:
@@ -103,12 +105,22 @@ class Data(object):
         except:
             return 'smi:bug/pylot/1234'
 
+    def filter(self, kwargs):
+        self.getWFData().filter(**kwargs)
+
+    def getWFData(self):
+        return self.wfdata
+
+    def getEvtData(self):
+        return self.evtdata
+
 
 class GenericDataStructure(object):
     '''
     GenericDataBase type holds all information about the current data-
     base working on.
     '''
+
     def __init__(self, stexp=None, folderdepth=4, **kwargs):
         structExpression = []
         depth = 0
@@ -130,6 +142,53 @@ class GenericDataStructure(object):
         self.dataBaseDict = {}
 
 
+class PilotDataStructure(object):
+    '''
+    Object containing the data access information for the old PILOT data
+    structure.
+    '''
+
+    def __init__(self, dataformat='GSE2', fsuffix='gse',
+                 root='/data/Egelados/EVENT_DATA/LOCAL/', database='2006.01',
+                 **kwargs):
+        self.dataType = dataformat
+        self.__pdsFields = {'ROOT': root,
+                          'DATABASE': database,
+                          'SUFFIX': fsuffix
+        }
+
+        self.modifiyFields(**kwargs)
+
+    def modifiyFields(self, **kwargs):
+        if kwargs and isinstance(kwargs, dict):
+            for key, value in kwargs.iteritems():
+                key = str(key)
+                if type(value) not in (str, int, float):
+                    for n, val in enumerate(value):
+                        value[n] = str(val)
+                else:
+                    value = str(value)
+                try:
+                    if key in self.getFields().keys():
+                        self.getFields()[key] = value
+                    else:
+                        raise KeyError('unknown PDS wildcard: %s.' % key)
+                except KeyError, e:
+                    errmsg = ''
+                    errmsg += 'WARNING:\n'
+                    errmsg += 'unable to set values for PDS fields\n'
+                    errmsg += '%s; desired value was: %s\n' % (e, value)
+                    print errmsg
+
+    def getType(self):
+        return self.dataType
+
+    def getFields(self):
+        return self.__pdsFields
+
+    def expandDataPath(self):
+        return
+
 class SeiscompDataStructure(object):
     '''
     Dictionary containing the data access information for an SDS data archive:
@@ -211,8 +270,8 @@ class SeiscompDataStructure(object):
                 else:
                     value = str(value)
                 try:
-                    if key in self.getSDSFields().keys():
+                    if key in self.getFields().keys():
-                        self.getSDSFields()[key] = value
+                        self.getFields()[key] = value
                     else:
                         raise KeyError('unknown SDS wildcard: %s.' % key)
                 except KeyError, e:
@@ -225,7 +284,7 @@ class SeiscompDataStructure(object):
     def getType(self):
         return self.__typeOptions[self.dataType]
 
-    def getSDSFields(self):
+    def getFields(self):
         return self.__sdsFields
 
     def expandDataPath(self):

pylot/core/util/widgets.py

@@ -134,16 +134,26 @@ class InputsTab(PropTab):
         fulluser = settings.value("user/FullName")
         login = settings.value("user/Login")
 
-        fullNameLabel = QLabel("Full name for user '{0}'".format(login))
+        fullNameLabel = QLabel("Full name for user '{0}': ".format(login))
 
+        # get the full name of the actual user
         self.fullNameEdit = QLineEdit()
         self.fullNameEdit.setText(fulluser)
 
+        # information about data structure
         dataroot = settings.value("data/dataRoot")
-        dataDirLabel = QLabel("data directory:")
+        dataDirLabel = QLabel("data root directory: ")
         self.dataDirEdit = QLineEdit()
         self.dataDirEdit.setText(dataroot)
         self.dataDirEdit.selectAll()
+        structureLabel = QLabel("data structure: ")
+        self.structureSelect = QComboBox()
+
+        from pylot.core.util.structure import DATASTRUCTURE
+
+        datastruct = DATASTRUCTURE.keys()
+        self.structureSelect.addItems(datastruct)
+        self.updateWidget(DATASTRUCTURE)
 
         layout = QGridLayout()
         layout.addWidget(dataDirLabel, 0, 0)
@@ -159,6 +169,13 @@ class InputsTab(PropTab):
         values["user/FullName"] = self.fullNameEdit.text()
         return values
 
+    def updateWidget(self, structure):
+        key = self.structureSelect.currentText()
+        structure = structure[key]
+        structure().getFields().keys()
+
+
+
 
 class OutputsTab(PropTab):