FGolf/Tools/mergeJSON.py

#!/usr/bin/env python

import sys,os,json

def convertRange(runs):
    result = {}
    for run in runs.keys():
        tmp = []
        for lumirange in runs[run]:
            tmp.extend(range(lumirange[0],lumirange[1]+1))
        result[run] = tmp
    return result
    
def convertArray(runs):
    result = {}
    for run in runs.keys():
        lumis = runs[run]
        lumis.sort()
        endRange = lumis[0]
        startRange = lumis[0]
        lumiRanges = []
        for lumi in range(1,len(lumis)) :
            if endRange+1 == lumis[lumi] :
                endRange = lumis[lumi]
            else :
                lumiRanges.append([startRange,endRange])
                endRange = lumis[lumi]
                startRange = lumis[lumi]
        lumiRanges.append([startRange,endRange])
        result[run] = lumiRanges
    return result

input_arrays = []
for argv in sys.argv[1:]:
    input_arrays.append(convertRange(json.load(open(argv))))
    
result_array = {}
for array in input_arrays:
    for run in array.keys():
        if run in result_array.keys():
            for lumi in array[run]:
                if lumi not in result_array[run]:
                    result_array[run].append(lumi)
        else :
            result_array[run] = array[run]            

result_range = convertArray(result_array)

output_handle = open("merged.json",'w')
json.dump(result_range,output_handle)
output_handle.close()


Revision:	1.1
Committed:	Sun Aug 29 18:25:19 2010 UTC (14 years, 8 months ago) by fgolf
Content type:	text/x-python
Branch:	MAIN
CVS Tags:	ss_summer2011approvalV2, ss_summer2011approval, synchMay2011v1, ss20May2011, ss2010_10_06, fkwsynch3pbv2, fkw_3pb_samesign_synch, HEAD
Log Message:	some tools useful for a variety of studies
#	Content
1	#!/usr/bin/env python
2
3	import sys,os,json
4
5	def convertRange(runs):
6	result = {}
7	for run in runs.keys():
8	tmp = []
9	for lumirange in runs[run]:
10	tmp.extend(range(lumirange[0],lumirange[1]+1))
11	result[run] = tmp
12	return result
13
14	def convertArray(runs):
15	result = {}
16	for run in runs.keys():
17	lumis = runs[run]
18	lumis.sort()
19	endRange = lumis[0]
20	startRange = lumis[0]
21	lumiRanges = []
22	for lumi in range(1,len(lumis)) :
23	if endRange+1 == lumis[lumi] :
24	endRange = lumis[lumi]
25	else :
26	lumiRanges.append([startRange,endRange])
27	endRange = lumis[lumi]
28	startRange = lumis[lumi]
29	lumiRanges.append([startRange,endRange])
30	result[run] = lumiRanges
31	return result
32
33	input_arrays = []
34	for argv in sys.argv[1:]:
35	input_arrays.append(convertRange(json.load(open(argv))))
36
37	result_array = {}
38	for array in input_arrays:
39	for run in array.keys():
40	if run in result_array.keys():
41	for lumi in array[run]:
42	if lumi not in result_array[run]:
43	result_array[run].append(lumi)
44	else :
45	result_array[run] = array[run]
46
47	result_range = convertArray(result_array)
48
49	output_handle = open("merged.json",'w')
50	json.dump(result_range,output_handle)
51	output_handle.close()
52
53