readMilleBinary.py

Go to the documentation of this file.

#!/usr/bin32/python
 
 
 
from __future__ import print_function
import array, sys
 
# ############### read millepede binary file #################
#  
 
Cfiles = 1  # Cfiles
#Cfiles = 0 # Fortran files
# 
 
intfmt = 'i'  # SL5, gcc-4
#intfmt = 'l' # SL4, gcc-3
#
 
fname = "milleBinaryISN.dat"
#
 
mrec = 10
 
skiprec = 0
 
minval = None  # allows for NaNs
#
# ## C. Kleinwort - DESY ########################
 
# ## use command line arguments ?
narg = len(sys.argv)
if narg > 1:
  if narg < 3:
    print(" usage: readMilleBinary.py <file name> <number of records> [<number of records to skip> <minimum value to print derivative>]")
    sys.exit(2)
  else:
    fname = sys.argv[1]
    mrec = int(sys.argv[2])
    if narg > 3:
      skiprec = int(sys.argv[3])
    if narg > 4:
      minval = float(sys.argv[4])
 
#print " input ", fname, mrec, skiprec
 
f = open(fname, "rb")
 
nrec = 0
try:
    while (nrec < mrec + skiprec) or (mrec < 0):
# read 1 record
        nr = 0
        if (Cfiles == 0):
            lenf = array.array(intfmt)
            lenf.fromfile(f, 2)
 
        length = array.array(intfmt)
        length.fromfile(f, 1)
        nr = abs(length[0] / 2)
        nrec += 1
 
        if length[0] > 0:
            glder = array.array('f')
        else:
            glder = array.array('d')
        glder.fromfile(f, nr)
 
        inder = array.array(intfmt)
        inder.fromfile(f, nr)
 
        if (Cfiles == 0):
            lenf = array.array(intfmt)
            lenf.fromfile(f, 2)
 
        if (nrec <= skiprec):  # must be after last fromfile
            continue
 
        print(" === NR ", nrec, length[0] / 2)
 
        # no details, only header
        if (mrec < -1):
            continue
 
        i = 0
        nh = 0
        ja = 0
        jb = 0
        jsp = 0
        nsp = 0
        while (i < (nr - 1)):
            i += 1
            while (i < nr) and (inder[i] != 0): i += 1
            ja = i
            i += 1
            while (i < nr) and (inder[i] != 0): i += 1
            jb = i
            i += 1
            # special data ?
            if (ja + 1 == jb) and (glder[jb] < 0.):
                jsp = jb
                nsp = int(-glder[jb])
                i += nsp - 1
                print(' ### spec. ', nsp, inder[jsp + 1:i + 1], glder[jsp + 1:i + 1])
                continue
            while (i < nr) and (inder[i] != 0): i += 1
            i -= 1
            nh += 1
            if (jb < i):
# measurement with global derivatives
                print(' -g- meas. ', nh, inder[jb + 1], jb - ja - 1, i - jb, glder[ja], glder[jb])
            else:
# measurement without global derivatives
                print(' -l- meas. ', nh, inder[ja + 1], jb - ja - 1, i - jb, glder[ja], glder[jb])
            if (ja + 1 < jb):
                lab = []
                val = []
                for k in range(ja + 1, jb):
                    if minval is None:
                        lab.append(inder[k])
                        val.append(glder[k])
                    elif abs(glder[k]) >= minval:
                        lab.append(inder[k])
                        val.append(glder[k])
                print(" local  ", lab)
                print(" local  ", val)
            if (jb + 1 < i + 1):
                lab = []
                val = []
                for k in range(jb + 1, i + 1):
                    if minval is None:
                        lab.append(inder[k])
                        val.append(glder[k]) 
                    elif abs(glder[k]) >= minval:
                        lab.append(inder[k])
                        val.append(glder[k])
                print(" global ", lab)
                print(" global ", val)
 
except EOFError:
    print()
    if (nr > 0):
        print(" >>> error: end of file before end of record", nrec)
    else:
        print(" end of file after", nrec, "records")
 
f.close()