source: nappy/trunk/naFile1010.py @ 345

"""
naFile1010.py
=============

Container module for NAFile1010 class.

"""
#$Log$
#Revision 1.2  2004/11/03 16:23:41  selatham
#updated by selatham for bug fixes and new write methods
#
# Imports from python standard library

# Imports from local package
import naFile1001
from textParser import *

class NAFile1010(naFile1001.NAFile1001):
    """
    Class to read, write and interact with NASA Ames files conforming to the
    File Format Index (FFI) 1010.
    """
#$Log$
#Revision 1.2  2004/11/03 16:23:41  selatham
#updated by selatham for bug fixes and new write methods
#
#

    def readHeader(self):
        """
        Reads FFI-specifc header section.
        """
        self._readCommonHeader()
        self.DX=readItemsFromLine(self.file.readline(), self.NIV, float)
        self.XNAME=readItemsFromLines(self._readLines(self.NIV), self.NIV, str)
        self._readVariablesHeaderSection()
        self._readAuxVariablesHeaderSection()
        self._readComments()

    def writeHeader(self):
        """
        Writes FFI-specifc header section.
        """
        self._writeCommonHeader()
        self.file.write(("%s "*self.NIV+"\n") % tuple(self.DX))
        self.file.write("%s\n"*self.NIV % tuple(self.XNAME))
        self._writeVariablesHeaderSection()
        self._writeAuxVariablesHeaderSection()
        self._writeComments()

    def _setupArrays(self):
        """
        Sets up FFI-specific arrays to fill with data (lists of lists).
        """        
        self.X=[]
        self.V=[]
        self.A=[]
        for n in range(self.NV):
            self.V.append([])
        for a in range(self.NAUXV):
            self.A.append([])

    def _readData1(self, datalines, ivar_count):
        """
        Reads first line/section of current block of data.
        """
        # Start with independent and Auxilliary vars
        #print "NAUXV = %s" %self.NAUXV                 testing
        (x2_and_a, rtlines)=readItemsFromUnknownLines(datalines, 1+self.NAUXV, float)
        #print "First line of block = %s"  %x2_and_a            testing
        (x, aux)=(x2_and_a[0], x2_and_a[1:])
        self.X.append(x)
        count=0
        for a in range(self.NAUXV):
            self.A[a].append(aux[count])
            count=count+1
        return rtlines
   
    def _readData2(self, datalines, ivar_count):
        """
        Reads second line/section (if used) of current block of data.
        """        
        # Now get the dependent variables
        (v, rtlines)=readItemsFromUnknownLines(datalines, self.NV, float)              
        # Set up mth list in self.V                              removed by selatham
        #self.V.append([])                                      removed by selatham
        count=0
        for n in range(self.NV):
            #self.V[n].append([])                                       removed by selatham
            #self.V[ivar_count].append(v[count])                removed by selatham
            self.V[n].append(v[count])
            count=count+1
        return rtlines

    def writeData(self):
        """
        Writes the data section of the file.
        This method can be called directly by the user.
        """
        for m in range(len(self.X)):
            # Write Independent variable mark and auxiliary variables
            var_string="%s    " % self.X[m]
            for a in range(self.NAUXV):
                var_string=var_string+("%s    " % self.A[a][m])
            self.file.write("%s\n" % var_string.rstrip())
            # Write dependant variables
            var_string=""
            for n in range(self.NV):
                 var_string=var_string+("%s    " %self.V[n][m])
            self.file.write("%s    \n" %var_string)