source: nappy/trunk/naFile1020.py @ 343

"""
naFile1020.py
=============

Container module for NAFile1020 class.

"""

# Imports from python standard library

# Imports from local package
from textParser import *
import naFile1010

class NAFile1020(naFile1010.NAFile1010):
    """
    Class to read, write and interact with NASA Ames files conforming to the
    File Format Index (FFI) 1020.
    """

    def readHeader(self):
        """
        Reads FFI-specifc header section.
        """    
        self._normalizedX="no"
        self._readCommonHeader()
        self.DX=readItemsFromLine(self.file.readline(), self.NIV, float)
        if self.DX==0:
            raise "DX found to be zero (0). Not allowed for FFI 1020."

        self.NVPM=readItemFromLine(self.file.readline(), int)
        self.XNAME=readItemsFromLines(self._readLines(self.NIV), self.NIV, str)
        self._readVariablesHeaderSection()
        self._readAuxVariablesHeaderSection()
        self._readComments()
  
    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*self.NVPM, float)              
        count=0
        for n in range(self.NV):
            self.V[n].append([])
            for i in range(self.NVPM):   # Number of steps where independent variable is implied
                self.V[n][ivar_count].append(v[count])
            count=count+1
        return rtlines

    def _normalizeIndVars(self):
        """
        Normalizes the values in the unbounded independent variable for FFIs
        that store an abbreviated version of this axis.
        """
        newX=[]
        for x in self.X[0]:
            for i in range(self.NVPM):
                newX.append(x+(i*self.DX))
        self.X[0]=newX
        self._normalizedX="yes"