def com(filename, frame):
if filename == 'dh_D0':
df = dh_D0
if filename == 'dh_Dst':
df = dh_Dst
if filename == 'dh_2460':
df = dh_2460
if filename == 'dh_2420':
df = dh_2420
if frame == 'B' or frame == 'W':
COM = LorentzVector(df[frame + '_PX_TRUE'], df[frame + '_PY_TRUE'],
df[frame + '_PZ_TRUE'], df[frame + '_E_TRUE'])
elif frame == 'D0':
if filename == 'dh_D0':
COM = LorentzVector(df['D0_PX_TRUE'], df['D0_PY_TRUE'],
df['D0_PZ_TRUE'], df['D0_E_TRUE'])
elif filename == 'dh_Dst':
COM = LorentzVector(df['Dst_PX_TRUE'], df['Dst_PY_TRUE'],
df['Dst_PZ_TRUE'], df['Dst_E_TRUE'])
elif filename == 'dh_2460':
COM = LorentzVector(df['Dstst_PX_TRUE'], df['Dstst_PY_TRUE'],
df['Dstst_PZ_TRUE'], df['Dstst_E_TRUE'])
elif filename == 'dh_2420':
COM = LorentzVector(df['Dstst_PX_TRUE'], df['Dstst_PY_TRUE'],
df['Dstst_PZ_TRUE'], df['Dstst_E_TRUE'])
return COM
######
external_stylesheets = ['https://codepen.io/chriddyp/pen/bWLwgP.css']
app = dash.Dash(__name__, external_stylesheets=external_stylesheets)
server = app.server
dft=root_pandas.read_root('model_tree.root',key='DecayTree')
df=dft.head(100)
B=LorentzVector(df['B_PX_TRUE'],df['B_PY_TRUE'],df['B_PZ_TRUE'],df['B_E_TRUE'])
Dst=LorentzVector(df['Dst_PX_TRUE'],df['Dst_PY_TRUE'],df['Dst_PZ_TRUE'],df['Dst_E_TRUE'])
tau=LorentzVector(df['Tau_PX_TRUE'],df['Tau_PY_TRUE'],df['Tau_PZ_TRUE'],df['Tau_E_TRUE'])
D0=LorentzVector(df['D0_PX_TRUE'],df['D0_PY_TRUE'],df['D0_PZ_TRUE'],df['D0_E_TRUE'])
nuB=LorentzVector(df['B_nu_PX_TRUE'],df['B_nu_PY_TRUE'],df['B_nu_PZ_TRUE'],df['B_nu_E_TRUE'])
K=LorentzVector(df['D0_K_PX_TRUE'],df['D0_K_PY_TRUE'],df['D0_K_PZ_TRUE'],df['D0_K_E_TRUE'])
piDst=LorentzVector(df['Dst_Pi_PX_TRUE'],df['Dst_Pi_PY_TRUE'],df['Dst_Pi_PZ_TRUE'],df['Dst_Pi_E_TRUE'])
piK=LorentzVector(df['D0_Pi_PX_TRUE'],df['D0_Pi_PY_TRUE'],df['D0_Pi_PZ_TRUE'],df['D0_Pi_E_TRUE'])
pitau1=LorentzVector(df['Tau_Pi1_PX_TRUE'],df['Tau_Pi1_PY_TRUE'],df['Tau_Pi1_PZ_TRUE'],df['Tau_Pi1_E_TRUE'])
pitau2=LorentzVector(df['Tau_Pi2_PX_TRUE'],df['Tau_Pi2_PY_TRUE'],df['Tau_Pi2_PZ_TRUE'],df['Tau_Pi2_E_TRUE'])
pitau3=LorentzVector(df['Tau_Pi3_PX_TRUE'],df['Tau_Pi3_PY_TRUE'],df['Tau_Pi3_PZ_TRUE'],df['Tau_Pi3_E_TRUE'])
nutau=LorentzVector(df['Tau_nu_PX_TRUE'],df['Tau_nu_PY_TRUE'],df['Tau_nu_PZ_TRUE'],df['Tau_nu_E_TRUE'])
import os
import root_pandas
import pandas as pd
import hepvector
from hepvector.numpyvector import Vector3D,LorentzVector
import plotly
import plotly.graph_objs as go
df=root_pandas.read_root('model_tree.root',key='DecayTree') #Read the data ->download root in git or copy root in putty
########
B=LorentzVector(df['B_PX_TRUE'],df['B_PY_TRUE'],df['B_PZ_TRUE'],df['B_E_TRUE'])
Dst=LorentzVector(df['Dst_PX_TRUE'],df['Dst_PY_TRUE'],df['Dst_PZ_TRUE'],df['Dst_E_TRUE'])
tau=LorentzVector(df['Tau_PX_TRUE'],df['Tau_PY_TRUE'],df['Tau_PZ_TRUE'],df['Tau_E_TRUE'])
D0=LorentzVector(df['D0_PX_TRUE'],df['D0_PY_TRUE'],df['D0_PZ_TRUE'],df['D0_E_TRUE'])
nuB=LorentzVector(df['B_nu_PX_TRUE'],df['B_nu_PY_TRUE'],df['B_nu_PZ_TRUE'],df['B_nu_E_TRUE'])
K=LorentzVector(df['D0_K_PX_TRUE'],df['D0_K_PY_TRUE'],df['D0_K_PZ_TRUE'],df['D0_K_E_TRUE'])
piDst=LorentzVector(df['Dst_Pi_PX_TRUE'],df['Dst_Pi_PY_TRUE'],df['Dst_Pi_PZ_TRUE'],df['Dst_Pi_E_TRUE'])
piK=LorentzVector(df['D0_Pi_PX_TRUE'],df['D0_Pi_PY_TRUE'],df['D0_Pi_PZ_TRUE'],df['D0_Pi_E_TRUE'])
def drawevent_2420(selection,radio,frame):
if selection is None:
return {}
else:
i=selection['points'][0]['pointNumber']
dis=(df['B_FD_TRUE'][i]+df['D0_FD_TRUE'][i]+df['Tau_FD_TRUE'][i]+df['Dstst_FD_TRUE'][i]+df['Dst_FD_TRUE'][i])/5 #the average flight distance
cmax = B.p[i]
def colorval(vec): #determine the color
val=vec.p[i]
if val==cmax:
return 'rgb(0,0,0)'
if val<cmax and val>cmax/4:
return 'rgb(0,0,83)'
if val<=cmax/4 and val>cmax/16 :
return 'rgb(0,0,128)'
if val<=cmax/16 and val>cmax/64:
return 'rgb(0,0,196)'
else :
return 'rgb(0,0,255)'
if frame=='lab':
PV_X,PV_Y,PV_Z=(df['B_Ori_z_TRUE'][i],df['B_Ori_x_TRUE'][i],df['B_Ori_y_TRUE'][i])
B_X,B_Y,B_Z=(df['B_End_z_TRUE'][i],df['B_End_x_TRUE'][i],df['B_End_y_TRUE'][i])
tau_X,tau_Y,tau_Z=[df['Tau_End_z_TRUE'][i],df['Tau_End_x_TRUE'][i],df['Tau_End_y_TRUE'][i]]
pitau1_X=df['Tau_Pi1_PZ_TRUE'][i]*dis/df['Tau_Pi1_P_TRUE'][i]+tau_X
pitau1_Y=df['Tau_Pi1_PX_TRUE'][i]*dis/df['Tau_Pi1_P_TRUE'][i]+tau_Y
pitau1_Z=df['Tau_Pi1_PY_TRUE'][i]*dis/df['Tau_Pi1_P_TRUE'][i]+tau_Z
pitau2_X=df['Tau_Pi2_PZ_TRUE'][i]*dis/df['Tau_Pi2_P_TRUE'][i]+tau_X
pitau2_Y=df['Tau_Pi2_PX_TRUE'][i]*dis/df['Tau_Pi2_P_TRUE'][i]+tau_Y
pitau2_Z=df['Tau_Pi2_PY_TRUE'][i]*dis/df['Tau_Pi2_P_TRUE'][i]+tau_Z
pitau3_X=df['Tau_Pi3_PZ_TRUE'][i]*dis/df['Tau_Pi3_P_TRUE'][i]+tau_X
pitau3_Y=df['Tau_Pi3_PX_TRUE'][i]*dis/df['Tau_Pi3_P_TRUE'][i]+tau_Y
pitau3_Z=df['Tau_Pi3_PY_TRUE'][i]*dis/df['Tau_Pi3_P_TRUE'][i]+tau_Z
nutau_X=df['Tau_nu_PZ_TRUE'][i]*dis/df['Tau_nu_P_TRUE'][i]+tau_X
nutau_Y=df['Tau_nu_PX_TRUE'][i]*dis/df['Tau_nu_P_TRUE'][i]+tau_Y
nutau_Z=df['Tau_nu_PY_TRUE'][i]*dis/df['Tau_nu_P_TRUE'][i]+tau_Z
nu_X=df['B_nu_PZ_TRUE'][i]*dis/df['B_nu_P_TRUE'][i]+B_X
nu_Y=df['B_nu_PX_TRUE'][i]*dis/df['B_nu_P_TRUE'][i]+B_Y
nu_Z=df['B_nu_PY_TRUE'][i]*dis/df['B_nu_P_TRUE'][i]+B_Z
Dstst_X,Dstst_Y,Dstst_Z=(df['Dstst_End_z_TRUE'][i],df['Dstst_End_x_TRUE'][i],df['Dstst_End_y_TRUE'][i])
Dst_X,Dst_Y,Dst_Z=(df['Dst_End_z_TRUE'][i],df['Dst_End_x_TRUE'][i],df['Dst_End_y_TRUE'][i])
D0_X,D0_Y,D0_Z=(df['D0_End_z_TRUE'][i],df['D0_End_x_TRUE'][i],df['D0_End_y_TRUE'][i])
K_X=df['D0_K_PZ_TRUE'][i]*dis/df['D0_K_P_TRUE'][i]+D0_X
K_Y=df['D0_K_PX_TRUE'][i]*dis/df['D0_K_P_TRUE'][i]+D0_Y
K_Z=df['D0_K_PY_TRUE'][i]*dis/df['D0_K_P_TRUE'][i]+D0_Z
piD0_X=df['D0_Pi_PZ_TRUE'][i]*dis/df['D0_Pi_P_TRUE'][i]+D0_X
piD0_Y=df['D0_Pi_PX_TRUE'][i]*dis/df['D0_Pi_P_TRUE'][i]+D0_Y
piD0_Z=df['D0_Pi_PY_TRUE'][i]*dis/df['D0_Pi_P_TRUE'][i]+D0_Z
piDst_X=df['Dst_Pi_PZ_TRUE'][i]*dis/df['Dst_Pi_P_TRUE'][i]+Dst_X
piDst_Y=df['Dst_Pi_PX_TRUE'][i]*dis/df['Dst_Pi_P_TRUE'][i]+Dst_Y
piDst_Z=df['Dst_Pi_PY_TRUE'][i]*dis/df['Dst_Pi_P_TRUE'][i]+Dst_Z
piDstst_X=df['Dstst_Pi_PZ_TRUE'][i]*dis/df['Dstst_Pi_P_TRUE'][i]+Dstst_X
piDstst_Y=df['Dstst_Pi_PX_TRUE'][i]*dis/df['Dstst_Pi_P_TRUE'][i]+Dstst_Y
piDstst_Z=df['Dstst_Pi_PY_TRUE'][i]*dis/df['Dstst_Pi_P_TRUE'][i]+Dstst_Z
else:
COM=LorentzVector(df[frame+'_PX_TRUE'],df[frame+'_PY_TRUE'],df[frame+'_PZ_TRUE'],df[frame+'_E_TRUE'])
liste_part=change_frame(COM)
[newB,newW,newDst,newDstst,newD0,newtau,newnuB,newK,newpiDst,newpiDstst,newpiD0,newpitau1,newpitau2,newpitau3,newnutau]=liste_part
PV_X,PV_Y,PV_Z=(0,0,0)
if frame=='B':
B_X,B_Y,B_Z=(0,0,0)
else:
B_X,B_Y,B_Z=[newB.z[i]*df['B_FD_TRUE'][i]/newB.p[i],newB.x[i]*df['B_FD_TRUE'][i]/newB.p[i],newB.y[i]*df['B_FD_TRUE'][i]/newB.p[i]]
tau_X,tau_Y,tau_Z=[newtau.z[i]*df['Tau_FD_TRUE'][i]/newtau.p[i]+B_X,newtau.x[i]*df['Tau_FD_TRUE'][i]/newtau.p[i]+B_Y,newtau.y[i]*df['Tau_FD_TRUE'][i]/newtau.p[i]+B_Z]
nu_X=newnuB.z[i]*dis/newnuB.p[i]+B_X
nu_Y=newnuB.x[i]*dis/newnuB.p[i]+B_Y
nu_Z=newnuB.y[i]*dis/newnuB.p[i]+B_Z
Dstst_X,Dstst_Y,Dstst_Z=[newDstst.z[i]*df['Dstst_FD_TRUE'][i]/newDstst.p[i]+B_X,newDstst.x[i]*df['Dstst_FD_TRUE'][i]/newDstst.p[i]+B_Y,newDstst.y[i]*df['Dstst_FD_TRUE'][i]/newDstst.p[i]+B_Z]
Dst_X,Dst_Y,Dst_Z=[newDst.z[i]*df['Dst_FD_TRUE'][i]/newDst.p[i]+Dstst_X,newDst.x[i]*df['Dst_FD_TRUE'][i]/newDst.p[i]+Dstst_Y,newDst.y[i]*df['Dst_FD_TRUE'][i]/newDst.p[i]+Dstst_Z]
D0_X,D0_Y,D0_Z=[newD0.z[i]*df['D0_FD_TRUE'][i]/newD0.p[i]+Dst_X,newD0.x[i]*df['D0_FD_TRUE'][i]/newD0.p[i]+Dst_Y,newD0.y[i]*df['D0_FD_TRUE'][i]/newD0.p[i]+Dst_Z]
pitau1_X=newpitau1.z[i]*dis/newpitau1.p[i]+tau_X
pitau1_Y=newpitau1.x[i]*dis/newpitau1.p[i]+tau_Y
pitau1_Z=newpitau1.y[i]*dis/newpitau1.p[i]+tau_Z
pitau2_X=newpitau2.z[i]*dis/newpitau2.p[i]+tau_X
pitau2_Y=newpitau2.x[i]*dis/newpitau2.p[i]+tau_Y
pitau2_Z=newpitau2.y[i]*dis/newpitau2.p[i]+tau_Z
pitau3_X=newpitau3.z[i]*dis/newpitau3.p[i]+tau_X
pitau3_Y=newpitau3.x[i]*dis/newpitau3.p[i]+tau_Y
pitau3_Z=newpitau3.y[i]*dis/newpitau3.p[i]+tau_Z
nutau_X=newnutau.z[i]*dis/newnutau.p[i]+tau_X
nutau_Y=newnutau.x[i]*dis/newnutau.p[i]+tau_Y
nutau_Z=newnutau.y[i]*dis/newnutau.p[i]+tau_Z
K_X=newK.z[i]*dis/newK.p[i]+D0_X
K_Y=newK.x[i]*dis/newK.p[i]+D0_Y
K_Z=newK.y[i]*dis/newK.p[i]+D0_Z
piD0_X=newpiD0.z[i]*dis/newpiD0.p[i]+D0_X
piD0_Y=newpiD0.x[i]*dis/newpiD0.p[i]+D0_Y
piD0_Z=newpiD0.y[i]*dis/newpiD0.p[i]+D0_Z
piDst_X=newpiDst.z[i]*dis/newpiDst.p[i]+Dst_X
piDst_Y=newpiDst.x[i]*dis/newpiDst.p[i]+Dst_Y
piDst_Z=newpiDst.y[i]*dis/newpiDst.p[i]+Dst_Z
piDstst_X=newpiDstst.z[i]*dis/newpiDstst.p[i]+Dstst_X
piDstst_Y=newpiDstst.x[i]*dis/newpiDstst.p[i]+Dstst_Y
piDstst_Z=newpiDstst.y[i]*dis/newpiDstst.p[i]+Dstst_Z
xrange=max(abs([Dst_X,Dstst_X,D0_X,tau_X,nu_X,K_X,piDst_X,piDstst_X,piD0_X,pitau1_X,
pitau2_X,pitau3_X,nutau_X]))
yrange=max(abs([Dst_Y,Dstst_Y,D0_Y,tau_Y,nu_Y,K_Y,piDst_Y,piDstst_Y,piD0_Y,pitau1_Y,
pitau2_Y,pitau3_Y,nutau_Y]))
zrange=max(abs([Dst_Z,Dstst_Z,D0_Z,tau_Z,nu_Z,K_Z,piDst_Z,piDstst_Z,piD0_Z,pitau1_Z,
pitau2_Z,pitau3_Z,nutau_Z]))
if radio=='3D':
traceB=go.Scatter3d(x=[PV_X,B_X],y=[PV_Y,B_Y],z=[PV_Z,B_Z],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['PV', ''],textposition='top left',line = dict(color =colorval(B),width=3))
tracetau=go.Scatter3d(x=[B_X,tau_X],y=[B_Y,tau_Y],z=[B_Z,tau_Z],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'tau'],textposition='top left',line = dict(color = colorval(tau),width=3))
tracenuB=go.Scatter3d(x=[B_X,nu_X],y=[B_Y,nu_Y],z=[B_Z,nu_Z],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)",opacity=0.8),text=['', 'nuB'],textposition='top left',line = dict(color = colorval(nuB),width=3))
tracepitau1=go.Scatter3d(x=[tau_X,pitau1_X],y=[tau_Y,pitau1_Y],z=[tau_Z,pitau1_Z],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'pi'],textposition='top left',line = dict(color = colorval(pitau1),width = 3))
tracepitau2=go.Scatter3d(x=[tau_X,pitau2_X],y=[tau_Y,pitau2_Y],z=[tau_Z,pitau2_Z],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'pi'],textposition='top left',line = dict(color = colorval(pitau2),width = 3))
tracepitau3=go.Scatter3d(x=[tau_X,pitau3_X],y=[tau_Y,pitau3_Y],z=[tau_Z,pitau3_Z],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'pi'],textposition='top left',line = dict(color = colorval(pitau3),width = 3))
tracenutau=go.Scatter3d(x=[tau_X,nutau_X],y=[tau_Y,nutau_Y],z=[tau_Z,nutau_Z],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'nu'],textposition='top left',line = dict(color = colorval(nutau),width = 3))
traceK=go.Scatter3d(x=[D0_X,K_X],y=[D0_Y,K_Y],z=[D0_Z,K_Z],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'K'],textposition='top left',line = dict(color =colorval(K),width=3))
tracepiD0=go.Scatter3d(x=[D0_X,piD0_X],y=[D0_Y,piD0_Y],z=[D0_Z,piD0_Z],mode='lines+markers+text',marker=dict(size=5,color="rgb(5,200,5)", opacity=0.8),text=['', 'pi'],textposition='top left',line = dict(color=colorval(piD0),width=3))
traceD0=go.Scatter3d(x=[Dst_X,D0_X],y=[Dst_Y,D0_Y],z=[Dst_Z,D0_Z],mode='lines+markers+text',marker=dict(size=5,color="rgb(5,200,5)", opacity=0.8),text=['', 'D0'],textposition='top left',line = dict(color=colorval(D0),width=3))
tracepiDstst=go.Scatter3d(x=[Dstst_X,piDstst_X],y=[Dstst_Y,piDstst_Y],z=[Dstst_Z,piDstst_Z],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'pi'],textposition='top left',line = dict(color = colorval(piDst),width = 3))
traceDstst=go.Scatter3d(x=[B_X,Dstst_X],y=[B_Y,Dstst_Y],z=[B_Z,Dstst_Z],mode='lines+markers+text',marker=dict(size=5,color="rgb(5,200,5)",opacity=0.8,cmin=cmin,cmax=cmax),text=['', 'D**'],textposition='top left',line = dict(width=3,color=colorval(D0)))
tracepiDst=go.Scatter3d(x=[Dst_X,piDst_X],y=[Dst_Y,piDst_Y],z=[Dst_Z,piDst_Z],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'pi'],textposition='top left',line = dict(color = colorval(piDst),width = 3))
traceDst=go.Scatter3d(x=[Dstst_X,Dst_X],y=[Dstst_Y,Dst_Y],z=[Dstst_Z,Dst_Z],mode='lines+markers+text',marker=dict(size=5,color="rgb(5,200,5)",opacity=0.8,cmin=cmin,cmax=cmax),text=['', 'D*'],textposition='top left',line = dict(width=3,color=colorval(D0)))
layout_event = go.Layout(
showlegend=False,
#width=400,
#height=400,
paper_bgcolor = '#F4F4F8',
plot_bgcolor = '#F4F4F8',
autosize=True,
margin=dict(t=10, b=10, l=20, r=10),
scene=dict(
camera=dict(eye=dict(x=1.5,y=1.5,z=1.5)),
xaxis=dict(
range=[-abs(xrange), abs(xrange)],
title='Z direction [nm]',
titlefont=dict(
family='Arial, sans-serif',
size=18,
color='black'
),
gridcolor='#bdbdbd',
gridwidth=2,
zerolinecolor='#969696',
zerolinewidth=4,
linecolor='#636363',
linewidth=4,
showbackground=True,
backgroundcolor='rgb(230, 230,230)'
),
yaxis=dict(
range=[-abs(yrange), abs(yrange)],
title='X direction [nm]',
titlefont=dict(
family='Arial, sans-serif',
size=18,
color='black'
),
gridcolor='#bdbdbd',
gridwidth=2,
zerolinecolor='#969696',
zerolinewidth=4,
linecolor='#636363',
linewidth=4,
showbackground=True,
backgroundcolor='rgb(230, 230, 230)'
),
zaxis=dict(
range=[-abs(zrange), abs(zrange)],
title='Y direction [nm]',
titlefont=dict(
family='Arial, sans-serif',
size=18,
color='black'
),
gridcolor='#bdbdbd',
gridwidth=2,
zerolinecolor='#969696',
zerolinewidth=4,
linecolor='#636363',
linewidth=4,
showbackground=True,
backgroundcolor='rgb(230, 230,230)'
),
aspectratio = dict(x=1, y=1, z=0.7),
aspectmode = 'manual'
)
)
elif radio=='ZX':
traceB=go.Scatter(x=[PV_X,B_X],y=[PV_Y,B_Y],mode='lines+markers+text',text=['PV','B'],textposition='top center',line=dict(color='darkblue',width=2))
tracetau=go.Scatter(x=[B_X,tau_X],y=[B_Y,tau_Y],mode='lines+markers+text',text=['','tau'],textposition='top center',line=dict(color='darkblue',width=2))
tracenuB=go.Scatter(x=[B_X,nu_X],y=[B_Y,nu_Y],mode='lines+markers+text',text=['','nu'],textposition='top center',line=dict(color='darkblue',width=2))
tracepitau1=go.Scatter(x=[tau_X,pitau1_X],y=[tau_Y,pitau1_Y],mode='lines+markers+text',text=['', 'pi'],textposition='top center',line=dict(color='darkblue',width=2))
tracepitau2=go.Scatter(x=[tau_X,pitau2_X],y=[tau_Y,pitau2_Y],mode='lines+markers+text',text=['', 'pi'],textposition='top center',line=dict(color='darkblue',width=2))
tracepitau3=go.Scatter(x=[tau_X,pitau3_X],y=[tau_Y,pitau3_Y],mode='lines+markers+text',text=['', 'pi'],textposition='top center',line=dict(color='darkblue',width=2))
traceK=go.Scatter(x=[D0_X,K_X],y=[D0_Y,K_Y],mode='lines+markers+text',text=['','K'],textposition='top center',line=dict(color='darkblue',width=2))
tracepiD0=go.Scatter(x=[D0_X,piD0_X],y=[D0_Y,piD0_Y],mode='lines+markers+text',text=['', 'pi'],textposition='top center',line=dict(color='darkblue',width=2))
tracenutau=go.Scatter(x=[tau_X,nutau_X],y=[tau_Y,nutau_Y],mode='lines+markers+text',text=['', 'nu'],textposition='top center',line=dict(color='darkblue',width=2))
tracepiDstst=go.Scatter(x=[Dstst_X,piDstst_X],y=[Dstst_Y,piDstst_Y],mode='lines+markers+text',text=['','pi'],textposition='top center',line=dict(color='darkblue',width=2))
traceDstst=go.Scatter(x=[B_X,Dstst_X],y=[B_Y,Dstst_Y],mode='lines+markers+text',text=['','D**'],textposition='top center',line=dict(color='darkblue',width=2))
tracepiDst=go.Scatter(x=[Dst_X,piDst_X],y=[Dst_Y,piDst_Y],mode='lines+markers+text',text=['','pi'],textposition='top center',line=dict(color='darkblue',width=2))
traceD0=go.Scatter(x=[Dst_X,D0_X],y=[Dst_Y,D0_Y],mode='lines+markers+text',text=['','D0'],textposition='top center',line=dict(color='darkblue',width=2))
traceDst=go.Scatter(x=[Dstst_X,Dst_X],y=[Dstst_Y,Dst_Y],mode='lines+markers+text',text=['','D*'],textposition='top center',line=dict(color='darkblue',width=2))
layout_event = go.Layout(
showlegend=False,
paper_bgcolor = '#F4F4F8',
plot_bgcolor = '#F4F4F8',
xaxis=dict(
title='Z direction [nm]',
showgrid=True,
zeroline=True,
showline=True,
mirror='ticks',
gridcolor='#bdbdbd',
gridwidth=2,
zerolinecolor='#969696',
zerolinewidth=4,
linecolor='#636363',
linewidth=6
),
yaxis=dict(
title='X direction [nm]',
showgrid=True,
zeroline=True,
showline=True,
mirror='ticks',
gridcolor='#bdbdbd',
gridwidth=2,
zerolinecolor='#969696',
zerolinewidth=4,
linecolor='#636363',
linewidth=6
)
)
elif radio=='XY':
traceB=go.Scatter(x=[PV_Y,B_Y],y=[PV_Z,B_Z],mode='lines+markers+text',text=['PV','B'],textposition='top center',line=dict(color='darkblue',width=2))
tracetau=go.Scatter(x=[B_Y,tau_Y],y=[B_Z,tau_Z],mode='lines+markers+text',text=['','tau'],textposition='top center',line=dict(color='darkblue',width=2))
tracenuB=go.Scatter(x=[B_Y,nu_Y],y=[B_Z,nu_Z],mode='lines+markers+text',text=['','nu'],textposition='top center',line=dict(color='darkblue',width=2))
tracepitau1=go.Scatter(x=[tau_Y,pitau1_Y],y=[tau_Z,pitau1_Z],mode='lines+markers+text',text=['', 'pi'],textposition='top center',line=dict(color='darkblue',width=2))
tracepitau2=go.Scatter(x=[tau_Y,pitau2_Y],y=[tau_Z,pitau2_Z],mode='lines+markers+text',text=['', 'pi'],textposition='top center',line=dict(color='darkblue',width=2))
tracepitau3=go.Scatter(x=[tau_Y,pitau3_Y],y=[tau_Z,pitau3_Z],mode='lines+markers+text',text=['', 'pi'],textposition='top center',line=dict(color='darkblue',width=2))
traceK=go.Scatter(x=[D0_Y,K_Y],y=[D0_Z,K_Z],mode='lines+markers+text',text=['','K'],textposition='top center',line=dict(color='darkblue',width=2))
tracepiD0=go.Scatter(x=[D0_Y,piD0_Y],y=[D0_Z,piD0_Z],mode='lines+markers+text',text=['', 'pi'],textposition='top center',line=dict(color='darkblue',width=2))
tracenutau=go.Scatter(x=[tau_Y,nutau_Y],y=[tau_Z,nutau_Z],mode='lines+markers+text',text=['', 'nu'],textposition='top center',line=dict(color='darkblue',width=2))
tracepiDstst=go.Scatter(x=[Dstst_Y,piDstst_Y],y=[Dstst_Z,piDstst_Z],mode='lines+markers+text',text=['','pi'],textposition='top center',line=dict(color='darkblue',width=2))
traceDstst=go.Scatter(x=[B_Y,Dstst_Y],y=[B_Z,Dstst_Z],mode='lines+markers+text',text=['','D**'],textposition='top center',line=dict(color='darkblue',width=2))
tracepiDst=go.Scatter(x=[Dst_Y,piDst_Y],y=[Dst_Z,piDst_Z],mode='lines+markers+text',text=['','pi'],textposition='top center',line=dict(color='darkblue',width=2))
traceD0=go.Scatter(x=[Dst_Y,D0_Y],y=[Dst_Z,D0_Z],mode='lines+markers+text',text=['','D0'],textposition='top center',line=dict(color='darkblue',width=2))
traceDst=go.Scatter(x=[Dstst_Y,Dst_Y],y=[Dstst_Z,Dst_Z],mode='lines+markers+text',text=['','D*'],textposition='top center',line=dict(color='darkblue',width=2))
layout_event = go.Layout(
showlegend=False,
paper_bgcolor = '#F4F4F8',
plot_bgcolor = '#F4F4F8',
xaxis=dict(
title='X direction [nm]',
showgrid=True,
zeroline=True,
showline=True,
mirror='ticks',
gridcolor='#bdbdbd',
gridwidth=2,
zerolinecolor='#969696',
zerolinewidth=4,
linecolor='#636363',
linewidth=6
),
yaxis=dict(
title='Y direction [nm]',
showgrid=True,
zeroline=True,
showline=True,
mirror='ticks',
gridcolor='#bdbdbd',
gridwidth=2,
zerolinecolor='#969696',
zerolinewidth=4,
linecolor='#636363',
linewidth=6
)
)
elif radio=='YZ':
traceB=go.Scatter(x=[PV_Z,B_Z],y=[PV_X,B_X],mode='lines+markers+text',text=['PV','B'],textposition='top center',line=dict(color='darkblue',width=2))
tracetau=go.Scatter(x=[B_Z,tau_Z],y=[B_X,tau_X],mode='lines+markers+text',text=['','tau'],textposition='top center',line=dict(color='darkblue',width=2))
tracenuB=go.Scatter(x=[B_Z,nu_Z],y=[B_X,nu_X],mode='lines+markers+text',text=['','nu'],textposition='top center',line=dict(color='darkblue',width=2))
tracepitau1=go.Scatter(x=[tau_Z,pitau1_Z],y=[tau_X,pitau1_X],mode='lines+markers+text',text=['', 'pi'],textposition='top center',line=dict(color='darkblue',width=2))
tracepitau2=go.Scatter(x=[tau_Z,pitau2_Z],y=[tau_X,pitau2_X],mode='lines+markers+text',text=['', 'pi'],textposition='top center',line=dict(color='darkblue',width=2))
tracepitau3=go.Scatter(x=[tau_Z,pitau3_Z],y=[tau_X,pitau3_X],mode='lines+markers+text',text=['', 'pi'],textposition='top center',line=dict(color='darkblue',width=2))
traceK=go.Scatter(x=[D0_Z,K_Z],y=[D0_X,K_X],mode='lines+markers+text',text=['','K'],textposition='top center',line=dict(color='darkblue',width=2))
tracepiD0=go.Scatter(x=[D0_Z,piD0_Z],y=[D0_X,piD0_X],mode='lines+markers+text',text=['', 'pi'],textposition='top center',line=dict(color='darkblue',width=2))
tracenutau=go.Scatter(x=[tau_Z,nutau_Z],y=[tau_X,nutau_X],mode='lines+markers+text',text=['', 'nu'],textposition='top center',line=dict(color='darkblue',width=2))
traceD0=go.Scatter(x=[Dst_Z,D0_Z],y=[Dst_X,D0_X],mode='lines+markers+text',text=['','D0'],textposition='top center',line=dict(color='darkblue',width=2))
traceDst=go.Scatter(x=[Dstst_Z,Dst_Z],y=[Dstst_X,Dst_X],mode='lines+markers+text',text=['','D*'],textposition='top center',line=dict(color='darkblue',width=2))
tracepiDstst=go.Scatter(x=[Dstst_Z,piDstst_Z],y=[Dstst_X,piDstst_X],mode='lines+markers+text',text=['','pi'],textposition='top center',line=dict(color='darkblue',width=2))
traceDstst=go.Scatter(x=[B_Z,Dstst_Z],y=[B_X,Dstst_X],mode='lines+markers+text',text=['','D**'],textposition='top center',line=dict(color='darkblue',width=2))
tracepiDst=go.Scatter(x=[Dst_Z,piDst_Z],y=[Dst_X,piDst_X],mode='lines+markers+text',text=['','pi'],textposition='top center',line=dict(color='darkblue',width=2))
layout_event = go.Layout(
showlegend=False,
paper_bgcolor = '#F4F4F8',
plot_bgcolor = '#F4F4F8',
xaxis=dict(
title='Y direction [nm]',
showgrid=True,
zeroline=True,
showline=True,
mirror='ticks',
gridcolor='#bdbdbd',
gridwidth=2,
zerolinecolor='#969696',
zerolinewidth=4,
linecolor='#636363',
linewidth=6
),
yaxis=dict(
title='Z direction [nm]',
showgrid=True,
zeroline=True,
showline=True,
mirror='ticks',
gridcolor='#bdbdbd',
gridwidth=2,
zerolinecolor='#969696',
zerolinewidth=4,
linecolor='#636363',
linewidth=6
)
)
data_event=[traceB,tracetau,traceD0,traceDst,traceDstst,tracepiDstst,tracepiDst,tracenuB,traceK,tracepiD0,tracepitau1,tracepitau2,tracepitau3,tracenutau]
return {'data': data_event, 'layout':layout_event
}
def drawangle_2420(selection,choice):
if selection is None:
return {}
else:
i=selection['points'][0]['pointNumber']
if choice=='two':
angleBx,angleBy,angleBz=(0,0,0)
angleDststx,angleDststy,angleDststz=(nouvDstst.z[i]/nouvDstst.p[i],nouvDstst.x[i]/nouvDstst.p[i],nouvDstst.y[i]/nouvDstst.p[i])
angleDstx,angleDsty,angleDstz=(nouvDst.z[i]/nouvDst.p[i]+angleDststx,nouvDst.x[i]/nouvDst.p[i]+angleDststy,nouvDst.y[i]/nouvDst.p[i]+angleDststz)
anglepix,anglepiy,anglepiz=(nouvpi.z[i]/nouvpi.p[i]+angleDststx,nouvpi.x[i]/nouvpi.p[i]+angleDststy,nouvpi.y[i]/nouvpi.p[i]+angleDststz)
angleQx,angleQy,angleQz=(-angleDststx,-angleDststy,-angleDststz)
angletaux,angletauy,angletauz=(angleQx+nouvtau.z[i]/nouvtau.p[i],angleQy+nouvtau.x[i]/nouvtau.p[i],angleQz+nouvtau.y[i]/nouvtau.p[i])
anglenux,anglenuy,anglenuz=(angleQx-nouvtau.z[i]/nouvtau.p[i],angleQy-nouvtau.x[i]/nouvtau.p[i],angleQz-nouvtau.y[i]/nouvtau.p[i])
d1x,d1y,d1z=(2*angleQx,2*angleQy,2*angleQz)
d2x,d2y,d2z=(2*angleDststx,2*angleDststy,2*angleDststz)
dash1=go.Scatter3d(x=[angleQx,d1x],y=[angleQy,d1y],z=[angleQz,d1z],mode='lines',line = dict(color = ('rgb(20, 20, 255)'),width = 3,dash='dot'))
dash2=go.Scatter3d(x=[angleDststx,d2x],y=[angleDststy,d2y],z=[angleDststz,d2z],mode='lines',line = dict(color = ('rgb(20, 20, 255)'),width = 3,dash='dot'))
mesh1 = go.Mesh3d(x=[d1x,angletaux,angleQx],y=[d1y,angletauy,angleQy],z=[d1z,angletauz,angleQz],
opacity=0.4,
color='#3E3A3A')
mesh2 = go.Mesh3d(x=[d2x,angleDstx,angleDststx],y=[d2y,angleDsty,angleDststy],z=[d2z,angleDstz,angleDststz],
opacity=0.4,
color='#3E3A3A')
traceQ=go.Scatter3d(x=[angleBx,angleQx],y=[angleBy,angleQy],z=[angleBz,angleQz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['B', 'W'],textposition='top right',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
traceDstst=go.Scatter3d(x=[angleBx,angleDststx],y=[angleBy,angleDststy],z=[angleBz,angleDststz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'D**'],textposition='top left',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
tracenu=go.Scatter3d(x=[angleQx,anglenux],y=[angleQy,anglenuy],z=[angleQz,anglenuz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'nu'],textposition='top left',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
tracetau=go.Scatter3d(x=[angleQx,angletaux],y=[angleQy,angletauy],z=[angleQz,angletauz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'tau'],textposition='top left',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
traceDst=go.Scatter3d(x=[angleDststx,angleDstx],y=[angleDststy,angleDsty],z=[angleDststz,angleDstz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'D*'],textposition='top left',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
tracepi=go.Scatter3d(x=[angleDststx,anglepix],y=[angleDststy,anglepiy],z=[angleDststz,anglepiz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'pi'],textposition='top left',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
dataangle=[traceQ,traceDstst,tracenu,tracetau,traceDst,tracepi,dash1,dash2,mesh1,mesh2]
else:
COM=LorentzVector(df[choice+'_PX_TRUE'],df[choice+'_PY_TRUE'],df[choice+'_PZ_TRUE'],df[choice+'_E_TRUE'])
liste_part=change_frame(COM)
[newB,newW,newDst,newDstst,newD0,newtau,newnuB,newK,newpiDst,newpiDstst,newpiD0,newpitau1,newpitau2,newpitau3,newnutau]=liste_part
if choice=='Dstst':
angleDststx,angleDststy,angleDststz=(0,0,0)
angleBx,angleBy,angleBz=(newB.z[i]/newB.p[i],newB.x[i]/newB.p[i],newB.y[i]/newB.p[i])
angleQx,angleQy,angleQz=(newW.z[i]/newW.p[i]+angleBx,newW.x[i]/newW.p[i]+angleBy,newW.y[i]/newW.p[i]+angleBz)
d1x,d1y,d1z=(-angleBx,-angleBy,-angleBz)
angleDstx,angleDsty,angleDstz=(newDst.z[i]/newDst.p[i]+angleDststx,newDst.x[i]/newDst.p[i]+angleDststy,newDst.y[i]/newDst.p[i]+angleDststz)
anglepix,anglepiy,anglepiz=(newpiDstst.z[i]/newpiDstst.p[i]+angleDststx,newpiDstst.x[i]/newpiDstst.p[i]+angleDststy,newpiDstst.y[i]/newpiDstst.p[i]+angleDststz)
angletaux,angletauy,angletauz=(angleQx+newtau.z[i]/newtau.p[i],angleQy+newtau.x[i]/newtau.p[i],angleQz+newtau.y[i]/newtau.p[i])
anglenux,anglenuy,anglenuz=(angleQx+newnuB.z[i]/newnuB.p[i],angleQy+newnuB.x[i]/newnuB.p[i],angleQz+newnuB.y[i]/newnuB.p[i])
traceQ=go.Scatter3d(x=[angleBx,angleQx],y=[angleBy,angleQy],z=[angleBz,angleQz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['B', 'W'],textposition='top right',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
traceDstst=go.Scatter3d(x=[angleBx,angleDststx],y=[angleBy,angleDststy],z=[angleBz,angleDststz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'D**'],textposition='top left',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
tracenu=go.Scatter3d(x=[angleQx,anglenux],y=[angleQy,anglenuy],z=[angleQz,anglenuz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'nu'],textposition='top left',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
tracetau=go.Scatter3d(x=[angleQx,angletaux],y=[angleQy,angletauy],z=[angleQz,angletauz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'tau'],textposition='top left',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
traceDst=go.Scatter3d(x=[angleDststx,angleDstx],y=[angleDststy,angleDsty],z=[angleDststz,angleDstz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'D*'],textposition='top left',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
tracepi=go.Scatter3d(x=[angleDststx,anglepix],y=[angleDststy,anglepiy],z=[angleDststz,anglepiz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'pi'],textposition='top left',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
dash1=go.Scatter3d(x=[angleDststx,d1x],y=[angleDststy,d1y],z=[angleDststz,d1z],mode='lines',line = dict(color = ('rgb(20, 20, 255)'),width = 3,dash='dot'))
mesh1 = go.Mesh3d(x=[d1x,0,angleDstx],y=[d1y,0,angleDsty],z=[d1z,0,angleDstz],
opacity=0.4,
color='#3E3A3A')
dataangle=[traceQ,traceDstst,tracenu,tracetau,traceDst,tracepi,dash1,mesh1]
if choice=='W':
angleQx,angleQy,angleQz=(0,0,0)
angleBx,angleBy,angleBz=(newB.z[i]/newB.p[i],newB.x[i]/newB.p[i],newB.y[i]/newB.p[i])
angleDststx,angleDststy,angleDststz=(newDstst.z[i]/newDstst.p[i]+angleBx,newDstst.x[i]/newDstst.p[i]+angleBy,newDstst.y[i]/newDstst.p[i]+angleBz)
angleDstx,angleDsty,angleDstz=(newDst.z[i]/newDst.p[i]+angleDststx,newDst.x[i]/newDst.p[i]+angleDststy,newDst.y[i]/newDst.p[i]+angleDststz)
anglepix,anglepiy,anglepiz=(newpiDstst.z[i]/newpiDstst.p[i]+angleDststx,newpiDstst.x[i]/newpiDstst.p[i]+angleDststy,newpiDstst.y[i]/newpiDstst.p[i]+angleDststz)
angletaux,angletauy,angletauz=(angleQx+newtau.z[i]/newtau.p[i],angleQy+newtau.x[i]/newtau.p[i],angleQz+newtau.y[i]/newtau.p[i])
anglenux,anglenuy,anglenuz=(angleQx+newnuB.z[i]/newnuB.p[i],angleQy+newnuB.x[i]/newnuB.p[i],angleQz+newnuB.y[i]/newnuB.p[i])
d1x,d1y,d1z=(-angleBx,-angleBy,-angleBz)
dash1=go.Scatter3d(x=[angleQx,d1x],y=[angleQy,d1y],z=[angleQz,d1z],mode='lines',line = dict(color = ('rgb(20, 20, 255)'),width = 3,dash='dot'))
traceQ=go.Scatter3d(x=[angleBx,angleQx],y=[angleBy,angleQy],z=[angleBz,angleQz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['B', 'W'],textposition='top right',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
traceDst=go.Scatter3d(x=[angleDststx,angleDstx],y=[angleDststy,angleDsty],z=[angleDststz,angleDstz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'D*'],textposition='top left',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
traceDstst=go.Scatter3d(x=[angleBx,angleDststx],y=[angleBy,angleDststy],z=[angleBz,angleDststz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'D**'],textposition='top left',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
tracenu=go.Scatter3d(x=[angleQx,anglenux],y=[angleQy,anglenuy],z=[angleQz,anglenuz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'nu'],textposition='top left',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
tracetau=go.Scatter3d(x=[angleQx,angletaux],y=[angleQy,angletauy],z=[angleQz,angletauz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'tau'],textposition='top left',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
traceK=go.Scatter3d(x=[angleDststx,angleDstx],y=[angleDststy,angleDsty],z=[angleDststz,angleDstz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'D*'],textposition='top left',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
tracepi=go.Scatter3d(x=[angleDststx,anglepix],y=[angleDststy,anglepiy],z=[angleDststz,anglepiz],mode='lines+markers+text',marker=dict(size=5,color= "rgb(5,200,5)", opacity=0.8),text=['', 'pi'],textposition='top left',line = dict(color = ('rgb(0, 0, 255)'),width = 3))
mesh1 = go.Mesh3d(x=[d1x,0,angletaux],y=[d1y,0,angletauy],z=[d1z,0,angletauz],
opacity=0.4,
color='#3E3A3A')
dataangle=[traceQ,traceDstst,tracenu,tracetau,traceDst,tracepi,dash1,mesh1]
layoutangle = go.Layout(
showlegend=False,
paper_bgcolor = '#F4F4F8',
plot_bgcolor = '#F4F4F8',
autosize=True,
margin=dict(t=10, b=10, l=20, r=10),
scene=dict(
camera=dict(eye=dict(x=1.5,y=1.5,z=1.5)),
xaxis=dict(
title='Z',
titlefont=dict(
family='Arial, sans-serif',
size=18,
color='black'
),
gridcolor='#bdbdbd',
gridwidth=2,
zerolinecolor='#969696',
zerolinewidth=4,
linecolor='#636363',
linewidth=4,
showbackground=True,
backgroundcolor='rgb(230, 230,230)'
),
yaxis=dict(
title='X',
titlefont=dict(
family='Arial, sans-serif',
size=18,
color='black'
),
gridcolor='#bdbdbd',
gridwidth=2,
zerolinecolor='#969696',
zerolinewidth=4,
linecolor='#636363',
linewidth=4,
showbackground=True,
backgroundcolor='rgb(230, 230, 230)'
),
zaxis=dict(
title='Y',
titlefont=dict(
family='Arial, sans-serif',
size=18,
color='black'
),
gridcolor='#bdbdbd',
gridwidth=2,
zerolinecolor='#969696',
zerolinewidth=4,
linecolor='#636363',
linewidth=4,
showbackground=True,
backgroundcolor='rgb(230, 230,230)'
),
aspectratio = dict(x=1, y=1, z=0.7),
aspectmode = 'manual'
)
)
return {'data':dataangle,'layout':layoutangle}
dht_2420['W_E_TRUE']=dht_2420['B_E_TRUE']-dht_2420['Dstst_E_TRUE']
df=dht_2420.head(30300)
"""
external_stylesheets = ['https://codepen.io/chriddyp/pen/bWLwgP.css']
app = dash.Dash(__name__)
app.css.append_css({'external_url': 'https://codepen.io/plotly/pen/EQZeaW.css'})
server = app.server
"""
B=LorentzVector(df['B_PX_TRUE'],df['B_PY_TRUE'],df['B_PZ_TRUE'],df['B_E_TRUE'])
W=LorentzVector(df['W_PX_TRUE'],df['W_PY_TRUE'],df['W_PZ_TRUE'],df['W_E_TRUE'])
Dstst=LorentzVector(df['Dstst_PX_TRUE'],df['Dstst_PY_TRUE'],df['Dstst_PZ_TRUE'],df['Dstst_E_TRUE'])
Dst=LorentzVector(df['Dst_PX_TRUE'],df['Dst_PY_TRUE'],df['Dst_PZ_TRUE'],df['Dst_E_TRUE'])
tau=LorentzVector(df['Tau_PX_TRUE'],df['Tau_PY_TRUE'],df['Tau_PZ_TRUE'],df['Tau_E_TRUE'])
D0=LorentzVector(df['D0_PX_TRUE'],df['D0_PY_TRUE'],df['D0_PZ_TRUE'],df['D0_E_TRUE'])
nuB=LorentzVector(df['B_nu_PX_TRUE'],df['B_nu_PY_TRUE'],df['B_nu_PZ_TRUE'],df['B_nu_E_TRUE'])
K=LorentzVector(df['D0_K_PX_TRUE'],df['D0_K_PY_TRUE'],df['D0_K_PZ_TRUE'],df['D0_K_E_TRUE'])
piDst=LorentzVector(df['Dst_Pi_PX_TRUE'],df['Dst_Pi_PY_TRUE'],df['Dst_Pi_PZ_TRUE'],df['Dst_Pi_E_TRUE'])
piDstst=LorentzVector(df['Dstst_Pi_PX_TRUE'],df['Dstst_Pi_PY_TRUE'],df['Dstst_Pi_PZ_TRUE'],df['Dstst_Pi_E_TRUE'])
piD0=LorentzVector(df['D0_Pi_PX_TRUE'],df['D0_Pi_PY_TRUE'],df['D0_Pi_PZ_TRUE'],df['D0_Pi_E_TRUE'])
pitau1=LorentzVector(df['Tau_Pi1_PX_TRUE'],df['Tau_Pi1_PY_TRUE'],df['Tau_Pi1_PZ_TRUE'],df['Tau_Pi1_E_TRUE'])
pitau2=LorentzVector(df['Tau_Pi2_PX_TRUE'],df['Tau_Pi2_PY_TRUE'],df['Tau_Pi2_PZ_TRUE'],df['Tau_Pi2_E_TRUE'])
pitau3=LorentzVector(df['Tau_Pi3_PX_TRUE'],df['Tau_Pi3_PY_TRUE'],df['Tau_Pi3_PZ_TRUE'],df['Tau_Pi3_E_TRUE'])
def particles(filename):
if filename == 'dh_D0':
df = dh_D0
if filename == 'dh_Dst':
df = dh_Dst
if filename == 'dh_2460':
df = dh_2460
if filename == 'dh_2420':
df = dh_2420
if filename == 'dh_D0':
B = LorentzVector(df['B_PX_TRUE'], df['B_PY_TRUE'], df['B_PZ_TRUE'],
df['B_E_TRUE'])
W = LorentzVector(df['W_PX_TRUE'], df['W_PY_TRUE'], df['W_PZ_TRUE'],
df['W_E_TRUE'])
D0 = LorentzVector(df['D0_PX_TRUE'], df['D0_PY_TRUE'],
df['D0_PZ_TRUE'], df['D0_E_TRUE'])
tau = LorentzVector(df['Tau_PX_TRUE'], df['Tau_PY_TRUE'],
df['Tau_PZ_TRUE'], df['Tau_E_TRUE'])
nuB = LorentzVector(df['B_nu_PX_TRUE'], df['B_nu_PY_TRUE'],
df['B_nu_PZ_TRUE'], df['B_nu_E_TRUE'])
K = LorentzVector(df['D0_K_PX_TRUE'], df['D0_K_PY_TRUE'],
df['D0_K_PZ_TRUE'], df['D0_K_E_TRUE'])
piD0 = LorentzVector(df['D0_Pi_PX_TRUE'], df['D0_Pi_PY_TRUE'],
df['D0_Pi_PZ_TRUE'], df['D0_Pi_E_TRUE'])
pitau1 = LorentzVector(df['Tau_Pi1_PX_TRUE'], df['Tau_Pi1_PY_TRUE'],
df['Tau_Pi1_PZ_TRUE'], df['Tau_Pi1_E_TRUE'])
pitau2 = LorentzVector(df['Tau_Pi2_PX_TRUE'], df['Tau_Pi2_PY_TRUE'],
df['Tau_Pi2_PZ_TRUE'], df['Tau_Pi2_E_TRUE'])
pitau3 = LorentzVector(df['Tau_Pi3_PX_TRUE'], df['Tau_Pi3_PY_TRUE'],
df['Tau_Pi3_PZ_TRUE'], df['Tau_Pi3_E_TRUE'])
nutau = LorentzVector(df['Tau_nu_PX_TRUE'], df['Tau_nu_PY_TRUE'],
df['Tau_nu_PZ_TRUE'], df['Tau_nu_E_TRUE'])
particle_list = [
B, W, D0, tau, nuB, K, piD0, pitau1, pitau2, pitau3, nutau
]
elif filename == 'dh_Dst':
B = LorentzVector(df['B_PX_TRUE'], df['B_PY_TRUE'], df['B_PZ_TRUE'],
df['B_E_TRUE'])
W = LorentzVector(df['W_PX_TRUE'], df['W_PY_TRUE'], df['W_PZ_TRUE'],
df['W_E_TRUE'])
Dst = LorentzVector(df['Dst_PX_TRUE'], df['Dst_PY_TRUE'],
df['Dst_PZ_TRUE'], df['Dst_E_TRUE'])
tau = LorentzVector(df['Tau_PX_TRUE'], df['Tau_PY_TRUE'],
df['Tau_PZ_TRUE'], df['Tau_E_TRUE'])
D0 = LorentzVector(df['D0_PX_TRUE'], df['D0_PY_TRUE'],
df['D0_PZ_TRUE'], df['D0_E_TRUE'])
nuB = LorentzVector(df['B_nu_PX_TRUE'], df['B_nu_PY_TRUE'],
df['B_nu_PZ_TRUE'], df['B_nu_E_TRUE'])
K = LorentzVector(df['D0_K_PX_TRUE'], df['D0_K_PY_TRUE'],
df['D0_K_PZ_TRUE'], df['D0_K_E_TRUE'])
piDst = LorentzVector(df['Dst_Pi_PX_TRUE'], df['Dst_Pi_PY_TRUE'],
df['Dst_Pi_PZ_TRUE'], df['Dst_Pi_E_TRUE'])
piK = LorentzVector(df['D0_Pi_PX_TRUE'], df['D0_Pi_PY_TRUE'],
df['D0_Pi_PZ_TRUE'], df['D0_Pi_E_TRUE'])
pitau1 = LorentzVector(df['Tau_Pi1_PX_TRUE'], df['Tau_Pi1_PY_TRUE'],
df['Tau_Pi1_PZ_TRUE'], df['Tau_Pi1_E_TRUE'])
pitau2 = LorentzVector(df['Tau_Pi2_PX_TRUE'], df['Tau_Pi2_PY_TRUE'],
df['Tau_Pi2_PZ_TRUE'], df['Tau_Pi2_E_TRUE'])
pitau3 = LorentzVector(df['Tau_Pi3_PX_TRUE'], df['Tau_Pi3_PY_TRUE'],
df['Tau_Pi3_PZ_TRUE'], df['Tau_Pi3_E_TRUE'])
nutau = LorentzVector(df['Tau_nu_PX_TRUE'], df['Tau_nu_PY_TRUE'],
df['Tau_nu_PZ_TRUE'], df['Tau_nu_E_TRUE'])
particle_list = [
B, W, Dst, D0, tau, nuB, K, piDst, piK, pitau1, pitau2, pitau3,
nutau
]
elif filename == 'dh_2460':
B = LorentzVector(df['B_PX_TRUE'], df['B_PY_TRUE'], df['B_PZ_TRUE'],
df['B_E_TRUE'])
W = LorentzVector(df['W_PX_TRUE'], df['W_PY_TRUE'], df['W_PZ_TRUE'],
df['W_E_TRUE'])
Dstst = LorentzVector(df['Dstst_PX_TRUE'], df['Dstst_PY_TRUE'],
df['Dstst_PZ_TRUE'], df['Dstst_E_TRUE'])
Dst = LorentzVector(df['Dst_PX_TRUE'], df['Dst_PY_TRUE'],
df['Dst_PZ_TRUE'], df['Dst_E_TRUE'])
tau = LorentzVector(df['Tau_PX_TRUE'], df['Tau_PY_TRUE'],
df['Tau_PZ_TRUE'], df['Tau_E_TRUE'])
D0 = LorentzVector(df['D0_PX_TRUE'], df['D0_PY_TRUE'],
df['D0_PZ_TRUE'], df['D0_E_TRUE'])
nuB = LorentzVector(df['B_nu_PX_TRUE'], df['B_nu_PY_TRUE'],
df['B_nu_PZ_TRUE'], df['B_nu_E_TRUE'])
K = LorentzVector(df['D0_K_PX_TRUE'], df['D0_K_PY_TRUE'],
df['D0_K_PZ_TRUE'], df['D0_K_E_TRUE'])
piDst = LorentzVector(df['Dst_Pi_PX_TRUE'], df['Dst_Pi_PY_TRUE'],
df['Dst_Pi_PZ_TRUE'], df['Dst_Pi_E_TRUE'])
piDstst = LorentzVector(df['Dstst_Pi_PX_TRUE'], df['Dstst_Pi_PY_TRUE'],
df['Dstst_Pi_PZ_TRUE'], df['Dstst_Pi_E_TRUE'])
piD0 = LorentzVector(df['D0_Pi_PX_TRUE'], df['D0_Pi_PY_TRUE'],
df['D0_Pi_PZ_TRUE'], df['D0_Pi_E_TRUE'])
pitau1 = LorentzVector(df['Tau_Pi1_PX_TRUE'], df['Tau_Pi1_PY_TRUE'],
df['Tau_Pi1_PZ_TRUE'], df['Tau_Pi1_E_TRUE'])
pitau2 = LorentzVector(df['Tau_Pi2_PX_TRUE'], df['Tau_Pi2_PY_TRUE'],
df['Tau_Pi2_PZ_TRUE'], df['Tau_Pi2_E_TRUE'])
pitau3 = LorentzVector(df['Tau_Pi3_PX_TRUE'], df['Tau_Pi3_PY_TRUE'],
df['Tau_Pi3_PZ_TRUE'], df['Tau_Pi3_E_TRUE'])
nutau = LorentzVector(df['Tau_nu_PX_TRUE'], df['Tau_nu_PY_TRUE'],
df['Tau_nu_PZ_TRUE'], df['Tau_nu_E_TRUE'])
particle_list = [
B, W, Dst, Dstst, D0, tau, nuB, K, piDst, piDstst, piD0, pitau1,
pitau2, pitau3, nutau
]
elif filename == 'dh_2420':
B = LorentzVector(df['B_PX_TRUE'], df['B_PY_TRUE'], df['B_PZ_TRUE'],
df['B_E_TRUE'])
W = LorentzVector(df['W_PX_TRUE'], df['W_PY_TRUE'], df['W_PZ_TRUE'],
df['W_E_TRUE'])
Dstst = LorentzVector(df['Dstst_PX_TRUE'], df['Dstst_PY_TRUE'],
df['Dstst_PZ_TRUE'], df['Dstst_E_TRUE'])
Dst = LorentzVector(df['Dst_PX_TRUE'], df['Dst_PY_TRUE'],
df['Dst_PZ_TRUE'], df['Dst_E_TRUE'])
tau = LorentzVector(df['Tau_PX_TRUE'], df['Tau_PY_TRUE'],
df['Tau_PZ_TRUE'], df['Tau_E_TRUE'])
D0 = LorentzVector(df['D0_PX_TRUE'], df['D0_PY_TRUE'],
df['D0_PZ_TRUE'], df['D0_E_TRUE'])
nuB = LorentzVector(df['B_nu_PX_TRUE'], df['B_nu_PY_TRUE'],
df['B_nu_PZ_TRUE'], df['B_nu_E_TRUE'])
K = LorentzVector(df['D0_K_PX_TRUE'], df['D0_K_PY_TRUE'],
df['D0_K_PZ_TRUE'], df['D0_K_E_TRUE'])
piDst = LorentzVector(df['Dst_Pi_PX_TRUE'], df['Dst_Pi_PY_TRUE'],
df['Dst_Pi_PZ_TRUE'], df['Dst_Pi_E_TRUE'])
piDstst = LorentzVector(df['Dstst_Pi_PX_TRUE'], df['Dstst_Pi_PY_TRUE'],
df['Dstst_Pi_PZ_TRUE'], df['Dstst_Pi_E_TRUE'])
piD0 = LorentzVector(df['D0_Pi_PX_TRUE'], df['D0_Pi_PY_TRUE'],
df['D0_Pi_PZ_TRUE'], df['D0_Pi_E_TRUE'])
pitau1 = LorentzVector(df['Tau_Pi1_PX_TRUE'], df['Tau_Pi1_PY_TRUE'],
df['Tau_Pi1_PZ_TRUE'], df['Tau_Pi1_E_TRUE'])
pitau2 = LorentzVector(df['Tau_Pi2_PX_TRUE'], df['Tau_Pi2_PY_TRUE'],
df['Tau_Pi2_PZ_TRUE'], df['Tau_Pi2_E_TRUE'])
pitau3 = LorentzVector(df['Tau_Pi3_PX_TRUE'], df['Tau_Pi3_PY_TRUE'],
df['Tau_Pi3_PZ_TRUE'], df['Tau_Pi3_E_TRUE'])
nutau = LorentzVector(df['Tau_nu_PX_TRUE'], df['Tau_nu_PY_TRUE'],
df['Tau_nu_PZ_TRUE'], df['Tau_nu_E_TRUE'])
particle_list = [
B, W, Dst, Dstst, D0, tau, nuB, K, piDst, piDstst, piD0, pitau1,
pitau2, pitau3, nutau
]
return particle_list