def extract_x_y(matfilename, xfieldname, yfieldname, exclude_last_line=True):
m, fieldnames = smartReadMat(matfilename)
xpos = fieldnames.index(xfieldname)
ypos = fieldnames.index(yfieldname)
x = m[:,xpos]
y = m[:,ypos]
if exclude_last_line:
x = x[0:-1]
y = y[0:-1]
return x,y
def __getattr__(self, name):
if name in self.subdirs:
subdir = ExperimentDirectory(os.path.join(self.expdir, name))
setattr(self, name, subdir)
return subdir
elif name in self.pmats or name in self.amats or name in self.csv:
# minorly ugly, but will do for now...
if name in self.pmats: filename = os.path.join(self.expdir, name+'.pmat')
if name in self.amats: filename = os.path.join(self.expdir, name+'.amat')
if name in self.csv : filename = os.path.join(self.expdir, name+'.csv' )
arr, fieldnames = smartReadMat(filename)
return self._add_array(name, arr, fieldnames)
else:
raise ValueError, "ExperimentDirectory '%s' does not contain a component '%s'" \
% (self.expdir, name)
def __getattr__(self, name):
if name in self.subdirs:
subdir = ExperimentDirectory(os.path.join(self.expdir, name))
setattr(self, name, subdir)
return subdir
elif name in self.pmats or name in self.amats or name in self.csv:
# minorly ugly, but will do for now...
if name in self.pmats:
filename = os.path.join(self.expdir, name + '.pmat')
if name in self.amats:
filename = os.path.join(self.expdir, name + '.amat')
if name in self.csv:
filename = os.path.join(self.expdir, name + '.csv')
arr, fieldnames = smartReadMat(filename)
return self._add_array(name, arr, fieldnames)
else:
raise ValueError, "ExperimentDirectory '%s' does not contain a component '%s'" \
% (self.expdir, name)
