コード例 #1
0
ファイル: mad_ang_mod.py プロジェクト: saramuel/MadGraph
import numpy as np
from matplotlib import pyplot as plt
import sys
import lhe_reader_mod

#arr=np.loadtxt("test3_events3.txt")
arr = lhe_reader_mod.read("run5_events.lhe")
arr = arr.reshape(arr.shape[0]/4, 4, arr.shape[1])

#particles - decay with PT_H, one line too much in file
h = arr[:,0,:]
b = arr[:,2,:]
bb = arr[:,3,:]

#transverse momentum (beam axis : z)
PT_H = np.sqrt((h[:,0:2]**2).sum(axis=1))
PT_b = np.sqrt((b[:,0:2]**2).sum(axis=1))
PT_bb = np.sqrt((bb[:,0:2]**2).sum(axis=1))

#Theta
TH_H = np.arccos(h[:,2]/np.sqrt((h[:,0:3]**2).sum(axis=1)))
TH_b = np.arccos(b[:,2]/np.sqrt((b[:,0:3]**2).sum(axis=1)))
TH_bb = np.arccos(bb[:,2]/np.sqrt((bb[:,0:3]**2).sum(axis=1)))

#Eta
Et_H = -np.log(np.tan(TH_H/2))
Et_b = -np.log(np.tan(TH_b/2))
Et_bb = -np.log(np.tan(TH_bb/2))

#Phi
Phi_H = np.arcsin(h[:,1]/(np.sqrt((h[:,0:3]**2).sum(axis=1))*np.sin(TH_H)))
コード例 #2
0
ファイル: cheatfastjet.py プロジェクト: saramuel/areas
import numpy as np
from matplotlib import pyplot as plt
import sys
#import lhe_reader_mod_yotam
import lhe_reader_mod
from matplotlib.font_manager import FontProperties 

#arr = lhe_reader_mod.read("unweighted_events_LHC.lhe")
arr = lhe_reader_mod.read("events_10E4.lhe")
#arr = lhe_reader_mod_yotam.read("pythia_events_yotam.lhe")
arr = arr.reshape(arr.shape[0]/4, 4, arr.shape[1])

#particles - decay with PT_H, one line too much in file
h = arr[:,0,:]
b = arr[:,2,:]
bb = arr[:,3,:]
print h.shape

#select events that have been selected by fastjet
#event = np.loadtxt('eventnb.txt', dtype=(int))
#event = np.loadtxt('eventnb10t.txt', dtype=(int))
#h=h[event,:]
#b=b[event,:]
#bb=bb[event,:]

#select only first events 
h=h[:1000,:]
b=b[:1000,:]
bb=bb[:1000,:]
#h=h[:250,:]
#b=b[:250,:]
コード例 #3
0
ファイル: mad_ang_mod06.py プロジェクト: saramuel/areas
import numpy as np
from matplotlib import pyplot as plt
import sys
import lhe_reader_mod
from matplotlib.font_manager import FontProperties

# arr = lhe_reader_mod.read("unweighted_events_LHC.lhe")
arr = lhe_reader_mod.read("unweighted_events.lhe")
arr = arr.reshape(arr.shape[0] / 4, 4, arr.shape[1])

# particles - decay with PT_H, one line too much in file
h = arr[:, 0, :]
b = arr[:, 2, :]
bb = arr[:, 3, :]

P_H = h[:, 0:3]
P_Hx = P_H[:, 0]
P_Hy = P_H[:, 1]
P_Hz = P_H[:, 2]

P_b = b[:, 0:3]
P_bx = P_b[:, 0]
P_by = P_b[:, 1]
P_bz = P_b[:, 2]

P_bb = bb[:, 0:3]
P_bbx = P_bb[:, 0]
P_bby = P_bb[:, 1]
P_bbz = P_bb[:, 2]

# transverse momentum (beam axis : z)