Skip to content

Commit 61d9724

Browse files
committed
fmom: make PtEtaPhiM a struct instead of [4]float64
1 parent b7028fa commit 61d9724

File tree

1 file changed

+12
-10
lines changed

1 file changed

+12
-10
lines changed

fmom/ptetaphim.go

Lines changed: 12 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -8,10 +8,12 @@ import (
88
"math"
99
)
1010

11-
type PtEtaPhiM [4]float64
11+
type PtEtaPhiM struct {
12+
P4 Vec4
13+
}
1214

1315
func NewPtEtaPhiM(pt, eta, phi, m float64) PtEtaPhiM {
14-
return PtEtaPhiM([4]float64{pt, eta, phi, m})
16+
return PtEtaPhiM{P4: Vec4{X: pt, Y: eta, Z: phi, T: m}}
1517
}
1618

1719
func (p4 *PtEtaPhiM) Clone() P4 {
@@ -20,19 +22,19 @@ func (p4 *PtEtaPhiM) Clone() P4 {
2022
}
2123

2224
func (p4 *PtEtaPhiM) Pt() float64 {
23-
return p4[0]
25+
return p4.P4.X
2426
}
2527

2628
func (p4 *PtEtaPhiM) Eta() float64 {
27-
return p4[1]
29+
return p4.P4.Y
2830
}
2931

3032
func (p4 *PtEtaPhiM) Phi() float64 {
31-
return p4[2]
33+
return p4.P4.Z
3234
}
3335

3436
func (p4 *PtEtaPhiM) M() float64 {
35-
return p4[3]
37+
return p4.P4.T
3638
}
3739

3840
func (p4 *PtEtaPhiM) E() float64 {
@@ -150,8 +152,8 @@ func (p4 *PtEtaPhiM) Pz() float64 {
150152
}
151153

152154
func (p4 *PtEtaPhiM) Set(p P4) {
153-
p4[0] = p.Pt()
154-
p4[1] = p.Eta()
155-
p4[2] = p.Phi()
156-
p4[3] = p.M()
155+
p4.P4.X = p.Pt()
156+
p4.P4.Y = p.Eta()
157+
p4.P4.Z = p.Phi()
158+
p4.P4.T = p.M()
157159
}

0 commit comments

Comments
 (0)