source: sans/utils/bt5/combine/bt5reader.py @ 922

#!/usr/bin/python

#import Tkinter as Tk

def numeric_compare(x,y):
        x = float(x)
        y = float(y)    

        if x < y:
                return -1
        elif x==y:
                return 0
        else: # x>y
                return 1 

def getBT5DataFromFile(fileName):
        '''
        Takes a filename and returns a dictionary of the detector values
        keyed by varying value (ususally A2 or A5)
        '''
        detdata = {}
        summary = {}

        inputfile = open(fileName, "r")

        inputdata = inputfile.readlines()

        header = inputdata[0:13]

        for index in range(13,len(inputdata),2):
                summary[inputdata[index].split()[0]] = inputdata[index].split()
                detdata[inputdata[index].split()[0]] = inputdata[index+1].split(',')    

        for key in detdata.keys():
                for val in range(0,len(detdata[key])):
                        detdata[key][val] = int(detdata[key][val])

        inputfile.close()
        return (header,summary,detdata)

def printBT5FormatData(header,summarydata,scalardata):
        '''
        Print out header and data in .bt5 format
        '''
        for line in header:
                print line,
        
        summarykeys = summarydata.keys()
        summarykeys.sort(numeric_compare)

        for key in summarykeys:
                print '\t%s\t%s\t%s\t%s\t%s' % tuple(summarydata[key])
                print '%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d' % tuple(scalardata[key])
                
        return (0)

def addBT5Data(filenamelist):
        '''
        Takes a list of filenames, loads the data and sums it.
        '''
        headers = {}
        scalars = {}
        summarys = {}

        for i  in range (0,len(filenamelist)):
                (headers[i],summarys[i],scalars[i]) = getBT5DataFromFile(filenamelist[i])

        #for key in headers:
        #       printBT5FormatData(headers[key],summarys[key],scalars[key])     

        outheader = headers[0]
        outscalars = scalars[0]
        outsummary = summarys[0]
        
        for key in summarys[0]:
                for i in range(1,len(filenamelist)):
                        #Fix the summary data
                        outsummary[key][1] = float(outsummary[key][1])+float(summarys[i][key][1])
                        outsummary[key][2] = int(outsummary[key][2])+int(summarys[i][key][2])
                        outsummary[key][3] = int(outsummary[key][3])+int(summarys[i][key][3])
                        outsummary[key][4] = int(outsummary[key][4])+int(summarys[i][key][4])
                        #Fix the scalar values
                        for j in range(0,16):
                                outscalars[key][j] = outscalars[key][j]+scalars[i][key][j]      


        return (outheader,outsummary,outscalars)



def printBT5DetData(detdata):
        '''
        Print the contents of the file in a formatted fashion

        Takes a dictionary of data as provided by getBT5DataFromFile() and prints out the contents
        in a formatted fashion
        '''
        motorvals = detdata.keys()
        motorvals.sort(cmp=numeric_compare)

        for motorval in motorvals:
                str = motorval+":"
                str += "\tMon: "+repr(detdata[motorval][0])
                str += "\tDet 1-5: "+repr(detdata[motorval][2])
                str += "\t"+repr(detdata[motorval][1])
                str += "\t"+repr(detdata[motorval][4])
                str += "\t"+repr(detdata[motorval][5])
                str += "\t"+repr(detdata[motorval][6])
                str += "\tTrans: "+repr(detdata[motorval][3])
                print str

        return 0

def getAlignVals(data,motorval):
        '''
        Return the values we record in the logbook for a given motor position

        Takes a dictionary as provided by getBT5DataFromFile and returns a dictionary with 
        keys Central, Trans and Sum
        '''
        alignvals = {}

        alignvals['Central'] = data[motorval][1]
        alignvals['Trans'] = data[motorval][3]
        alignvals['Sum'] = data[motorval][1]+data[motorval][2]+data[motorval][4]+data[motorval][5]+data[motorval][6]    
        return alignvals

def maxDetCount(data,detector):
        '''
        Return the maximum value and corresponding motor position for a given detector
        
        Takes a dictionary as provided by getBT5DataFromFile() and returns a dictionary with
        keys Position and Value
        '''     
        maxpos = ''
        maxval = 0
        result = {}

        mvals = data.keys()
        det = {'1':2, '2':1, '3':4, '4':5, '5':6}[repr(detector)]

        for mval in mvals:
                if data[mval][det] > maxval:
                        maxval = data[mval][det]                
                        maxpos = mval
        
        result['Position'] = maxpos
        result['Value'] = maxval

        return result   

        
        

if __name__ == '__main__':
        import sys
        (header,summary,data) = getBT5DataFromFile(sys.argv[1])
        for line in header:
                print line,
        
        printBT5FormatData(header,summary,data)

        maxinfo =  maxDetCount(data,2)
        print maxinfo
        avals = getAlignVals(data,maxinfo['Position'])
        print avals