Matplotlib_jetmass/jetmassfilter.py at main · monamelop/Matplotlib_jetmass · GitHub

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
# Script to filter events with absurd jetmass values

import numpy as np
import yoda
import glob
import os
import sys
import jetmass


code = str(sys.argv[1])

# The following should be changed for the correct energy
#folder = '/sampa/archive/monalisa/Mestrado/results/yoda/0-10/'
folder = '/home/monalisa/Desktop/Mestrado/JetMass/Testfilter'
#rejectedfolder = '/sampa/archive/monalisa/Mestrado/results/yoda/0-10/rejected/'
rejectedfolder = '/home/monalisa/Desktop/Mestrado/JetMass/Testfilter/rejected'


jmmax = 0.7

files = glob.glob(folder + '/' + code + f'*_jetmass_R0.4.yoda')

print(f'Calculating JetMass of {len(files)} files...')


obslist = ['/JET_MASS/Jet_Mass_60_80','/JET_MASS/Jet_Mass_80_100','/JET_MASS/Jet_Mass_100_120']


for l in obslist:
    jetm = []
    for f in files:
        jetm.append(jetmass.JM(f, l))


    dists = []

    for i in range(len(jetm[0][1])):
        values = []
        for j in range(len(jetm)):
            v = jetm[j][1][i]
            values.append(v)
        dists.append(values)

    print(f'Comparing JetMass with threshold of {jmmax}...')
    rejected = []

    for p in dists:
        positions = np.where(np.asarray(p) > jmmax)[0]
        for pos in positions:
            rej = files[pos]
            print(f'\nReject file: {rej}, Jet_Mass = {p[pos]}')
            rejected.append(rej)
            os.system(f'cp {rej.replace("0.4", "*")} {rejectedfolder}/0-10/')
            #os.system(f'cp {rej.replace("jetmass").replace(0.4, "*")} rejected/0-10')
    print(f'''\n\nRemoved weird JetMass files.
    {code} 0-10 0.4: {100*len(set(rejected))/len(files)}%\n''')