added partran file reader (ascii)

ess/TraceWin.py

 
+
 class dst:
     '''
     Simple class to read in a 
@@ -309,3 +310,35 @@ class remote_data_merger:
 
             return data
 
+class partran:
+    '''
+    Read partran1.out files..
+    '''
+    def __init__(self,filename):
+        self.filename=filename
+        self._readAsciiFile()
+
+    def _readAsciiFile(self):
+
+        import numpy
+
+        stream=file(self.filename,'r')
+        for i in xrange(10):
+            line=stream.readline()
+        self.columns=['NUM']+line.split()
+        self.data=numpy.loadtxt(stream)
+
+        # seems we have some inconsistencies with columns...
+        #self._dict={}
+        #print self.data.shape,len(self.columns)
+        #for i in xrange(len(self.columns)):
+        #    self._dict[self.columns[i]]=self.data[:,i]
+
+    def get(self,column):
+        if column not in self.columns:
+            print "WARNING, could not find",column
+            return
+        i=self.columns.index(column)
+        return self.data[:,i]
+
+