From 902679195aef653c7d8601dfcf75ffd790d80cc6 Mon Sep 17 00:00:00 2001
From: Yngve Inntjore Levinsen <Yngve.Levinsen@esss.se>
Date: Wed, 4 Feb 2015 11:28:32 +0100
Subject: [PATCH] added partran file reader (ascii)
---
ess/TraceWin.py | 33 +++++++++++++++++++++++++++++++++
1 file changed, 33 insertions(+)
diff --git a/ess/TraceWin.py b/ess/TraceWin.py
index 2e0020d..73f9268 100644
--- a/ess/TraceWin.py
+++ b/ess/TraceWin.py
@@ -1,4 +1,5 @@
+
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]
+
+
--
GitLab