source: sans/utils/bt5/bt5plot2/usans.py @ 462

Last change on this file since 462 was 462, checked in by ajj, 14 years ago
File size: 4.0 KB
Line 
1#!/usr/bin/python
2
3#import Tkinter as Tk
4
5def numeric_compare(x, y):
6    x = float(x)
7    y = float(y)   
8
9    if x < y:
10        return - 1
11    elif x == y:
12        return 0
13    else: # x>y
14        return 1 
15
16def getBT5DataFromFile(fileName):
17    '''
18    Takes a filename and returns a dictionary of the detector values
19    keyed by varying value (ususally A2 or A5)
20    '''
21    detdata = {}
22    metadata = {}
23    motlist = []
24
25    #print "File: ",fileName   
26    inputfile = open(fileName, "r")
27
28    inputdata = inputfile.readlines()
29
30    if len(inputdata) < 2:
31        inputfile.close()
32        return 0,0
33
34    if inputdata[1].find('Filename') > 0:
35
36        mdtmp = inputdata[0].replace("'","")
37        mdtmp = mdtmp.split()
38   
39        #Sundry metadata about run settings
40        (metadata['filename'], metadata['datetime'],
41        metadata['mon'],metadata['prefactor'],
42        metadata['base'],metadata['numpnts'],
43        metadata['type']) = (mdtmp[0],' '.join(mdtmp[1:5]),float(mdtmp[6]),int(mdtmp[7]),mdtmp[8],int(mdtmp[9]),mdtmp[10])
44   
45        #Comment string
46        metadata['title'] = inputdata[2].strip()
47   
48        #Start, step and end values for motors 1-6
49        motlist.append(inputdata[5].split()[1:])
50        motlist.append(inputdata[6].split()[1:])
51        motlist.append(inputdata[7].split()[1:])
52        motlist.append(inputdata[8].split()[1:])
53        motlist.append(inputdata[9].split()[1:])
54        motlist.append(inputdata[10].split()[1:]) 
55        metadata['motorvals'] = motlist
56   
57        for index in range(13, len(inputdata), 2):
58            detdata[float(inputdata[index].split()[0])] = inputdata[index + 1].split(',')   
59
60        for key in detdata.keys():
61            for val in range(0, len(detdata[key])):
62                detdata[key][val] = int(detdata[key][val])
63
64        inputfile.close()
65        return detdata,metadata
66
67    else:
68        inputfile.close()
69        return 0,0
70
71def printBT5DetData(detdata):
72    '''
73    Print the contents of the file in a formatted fashion
74
75    Takes a dictionary of data as provided by getBT5DataFromFile() and prints out the contents
76    in a formatted fashion
77    '''
78    motorvals = detdata.keys()
79    motorvals.sort(cmp=numeric_compare)
80
81    for motorval in motorvals:
82        str = repr(motorval) + ":"
83        str += "\tMon: " + repr(detdata[motorval][0])
84        str += "\tDet 1-5: " + repr(detdata[motorval][2])
85        str += "\t" + repr(detdata[motorval][1])
86        str += "\t" + repr(detdata[motorval][4])
87        str += "\t" + repr(detdata[motorval][5])
88        str += "\t" + repr(detdata[motorval][6])
89        str += "\tTrans: " + repr(detdata[motorval][3])
90        print str
91
92    return 0
93
94def getAlignVals(data, motorval):
95    '''
96    Return the values we record in the logbook for a given motor position
97
98    Takes a dictionary as provided by getBT5DataFromFile and returns a dictionary with
99    keys Central, Trans and Sum
100    '''
101    alignvals = {}
102
103    alignvals['Central'] = data[motorval][1]
104    alignvals['Trans'] = data[motorval][3]
105    alignvals['Sum'] = data[motorval][1] + data[motorval][2] + data[motorval][4] + data[motorval][5] + data[motorval][6]     
106    return alignvals
107
108def maxDetCount(data, detector):
109    '''
110    Return the maximum value and corresponding motor position for a given detector
111   
112    Takes a dictionary as provided by getBT5DataFromFile() and returns a dictionary with
113    keys Position and Value
114    '''   
115    maxpos = ''
116    maxval = 0
117    result = {}
118
119    mvals = data.keys()
120    det = {'1':2, '2':1, '3':4, '4':5, '5':6}[repr(detector)]
121
122    for mval in mvals:
123        if data[mval][det] > maxval:
124            maxval = data[mval][det]       
125            maxpos = mval
126   
127    result['Position'] = maxpos
128    result['Value'] = maxval
129
130    return result   
131
132   
133   
134
135if __name__ == '__main__':
136    import sys
137    data,metadata = getBT5DataFromFile(sys.argv[1])
138    printBT5DetData(data)
139
140    maxinfo = maxDetCount(data, 2)
141    print maxinfo
142    avals = getAlignVals(data, maxinfo['Position'])
143    print avals
Note: See TracBrowser for help on using the repository browser.