/
plotting.py
74 lines (67 loc) · 1.88 KB
/
plotting.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
import re
import json
import glob
import subprocess
import shlex
def make_harvest_from_result(result, masses):
return {
"CLs": result["CLs_obs"],
"CLsexp": result["CLs_exp"][2],
"clsd1s": result["CLs_exp"][1],
"clsd2s": result["CLs_exp"][0],
"clsu1s": result["CLs_exp"][3],
"clsu2s": result["CLs_exp"][4],
"covqual": 3,
"dodgycov": 0,
"excludedXsec": -999007,
"expectedUpperLimit": -1,
"expectedUpperLimitMinus1Sig": -1,
"expectedUpperLimitMinus2Sig": -1,
"expectedUpperLimitPlus1Sig": -1,
"expectedUpperLimitPlus2Sig": -1,
"fID": -1,
"failedcov": 0,
"failedfit": 0,
"failedp0": 0,
"failedstatus": 0,
"fitstatus": 0,
"mn1": masses[2],
"mn2": masses[1],
"mode": -1,
"msb": masses[0],
"nexp": -1,
"nofit": 0,
"p0": 0,
"p0d1s": -1,
"p0d2s": -1,
"p0exp": -1,
"p0u1s": -1,
"p0u2s": -1,
"p1": 0,
"seed": 0,
"sigma0": -1,
"sigma1": -1,
"upperLimit": -1,
"upperLimitEstimatedError": -1,
"xsec": -999007,
}
def harvest_results(regions):
pattern = re.compile(r"sbottom_(\d+)_(\d+)_(\d+)")
dataList = []
for region in regions:
harvest = []
files = "results/region{region}.result.sbottom_*_*_*.json".format(
region=region,
)
for fname in glob.glob(files):
result = json.load(open(fname))
m = pattern.search(fname)
masses = list(map(int, m.groups()))
# only use 60 GeV
if masses[2] != 60:
continue
harvest.append(make_harvest_from_result(result, masses))
dataList.append((f'region{region}', harvest))
return dataList
if __name__ == '__main__':
main()