mille.py

Go to the documentation of this file.

'''
Input/output of MP-II binary records.
 
Created on Aug 1, 2011
 
@author: kleinwrt
'''
 
 
 
import array, math
 
 
 
class MilleRecord(object):
 
  
  def __init__(self, doublePrec=False):
    
    self.__doublePrecision = doublePrec
    
    self.__position = 1 
    
    self.__numData = 0  
    
    self.__recLen = 0 
    
    self.__iMeas = 0    
    
    self.__iErr = 0     
    
    self.__inder = array.array('i') 
    
    self.__glder = array.array('d' if doublePrec else 'f') 
 
  
  def addData(self, dataList):
    if (self.__numData == 0):  # first word is error counter
      self.__inder.append(0)
      self.__glder.append(0.) 
    self.__numData += 1     
 
    aMeas, aPrec, indLocal, derLocal, labGlobal, derGlobal = dataList
    self.__inder.append(0)
    self.__glder.append(aMeas)
    self.__inder.fromlist(indLocal)
    self.__glder.fromlist(derLocal)    
    self.__inder.append(0)
    self.__glder.append(1.0 / math.sqrt(aPrec))  # convert to error
    self.__inder.fromlist(labGlobal)
    self.__glder.fromlist(derGlobal)
 
  
  def getData(self):
    aMeas = self.__glder[self.__iMeas]
    indLocal = []
    derLocal = []
    for i in range(self.__iMeas + 1, self.__iErr):
      indLocal.append(self.__inder[i])
      derLocal.append(self.__glder[i])
    aPrec = 1.0 / self.__glder[self.__iErr] ** 2  # convert to precision 
    indGlobal = []
    derGlobal = []
    for i in range(self.__iErr + 1, self.__position):
      indGlobal.append(self.__inder[i])
      derGlobal.append(self.__glder[i])    
    return aMeas, aPrec, indLocal, derLocal, indGlobal, derGlobal   
 
  
  def printRecord(self):
    print " MilleRecord, len: ", len(self.__inder)
    print self.__inder
    print self.__glder
 
  
  def writeRecord(self, aFile):
    header = array.array('i')  # header with number of words
    header.append(-len(self.__inder) * 2 if self.__doublePrecision else len(self.__inder) * 2)
    header.tofile(aFile)
    self.__glder.tofile(aFile)
    self.__inder.tofile(aFile)
 
  
  def readRecord(self, aFile):
    header = array.array('i')  # header with number of words
    header.fromfile(aFile, 1)
    self.__recLen = abs(header[0] / 2)
    if header[0] < 0:
      self.__glder = array.array('d')
    self.__glder.fromfile(aFile, self.__recLen)
    self.__inder.fromfile(aFile, self.__recLen)
 
  
  def moreData(self):
    if (self.__position < self.__recLen):
      while (self.__position < self.__recLen and self.__inder[self.__position] != 0):
        self.__position += 1
      self.__iMeas = self.__position 
      self.__position += 1
      while (self.__position < self.__recLen and self.__inder[self.__position] != 0):
        self.__position += 1
      self.__iErr = self.__position 
      self.__position += 1
      # special data?
      if (self.__iMeas + 1 == self.__iErr and self.__glder[self.__iErr] < 0):
        self.__position += int(-self.__glder[self.__iErr])
      else:  
        while (self.__position < self.__recLen and self.__inder[self.__position] != 0):
          self.__position += 1
      self.__numData += 1  
      return True
    else:
      return False
 
  
  def specialDataTag(self):
    aTag = -1
    if (self.__iMeas + 1 == self.__iErr and self.__glder[self.__iErr] < 0):
      aTag = int(-self.__glder[self.__iErr] * 10. + 0.5) % 10
    return aTag