all: introduce fmom.Vec4, make fmom.PxPyPzE a struct instead of [4]float64

cmd/lhef2hepmc/main.go

@@ -132,21 +132,21 @@ func main() {
 			Barcode: -1,
 		}
 		p1 := hepmc.Particle{
-			Momentum: fmom.PxPyPzE{
+			Momentum: fmom.NewPxPyPzE(
 				0, 0,
 				dec.Run.EBMUP[0],
 				dec.Run.EBMUP[0],
-			},
+			),
 			PdgID:   dec.Run.IDBMUP[0],
 			Status:  4,
 			Barcode: 1,
 		}
 		p2 := hepmc.Particle{
-			Momentum: fmom.PxPyPzE{
+			Momentum: fmom.NewPxPyPzE(
 				0, 0,
 				dec.Run.EBMUP[1],
 				dec.Run.EBMUP[1],
-			},
+			),
 			PdgID:   dec.Run.IDBMUP[1],
 			Status:  4,
 			Barcode: 2,
@@ -170,12 +170,12 @@ func main() {
 			}
 			nmax += 1
 			vtx.AddParticleOut(&hepmc.Particle{
-				Momentum: fmom.PxPyPzE{
+				Momentum: fmom.NewPxPyPzE(
 					lhevt.PUP[i][0],
 					lhevt.PUP[i][1],
 					lhevt.PUP[i][2],
 					lhevt.PUP[i][3],
-				},
+				),
 				GeneratedMass: lhevt.PUP[i][4],
 				PdgID:         lhevt.IDUP[i],
 				Status:        1,

fads/fastjet_finder.go

@@ -149,8 +149,10 @@ func (tsk *FastJetFinder) Process(ctx fwk.Context) error {
 	dphiMax := 0.0
 	output = make([]Candidate, 0, len(outjets))
 	for i := range outjets {
-		jet := &outjets[i]
-		area := fmom.PxPyPzE{0, 0, 0, 0}
+		var (
+			jet  = &outjets[i]
+			area fmom.PxPyPzE
+		)
 		if tsk.areaDef != nil {
 			// FIXME
 			panic("not implemented")
@@ -187,7 +189,7 @@ func (tsk *FastJetFinder) Process(ctx fwk.Context) error {
 			cand.Add(cst)
 		}
 
-		cand.Pos[3] = time / wtime
+		cand.Pos.P4.T = time / wtime
 		cand.Area = area
 		cand.DEta = detaMax
 		cand.DPhi = dphiMax

fads/utils.go

@@ -46,5 +46,5 @@ func newPtEtaPhiE(pt, eta, phi, ene float64) fmom.PxPyPzE {
 	py := pt * math.Sin(phi)
 	pz := pt * math.Sinh(eta)
 
-	return fmom.PxPyPzE{px, py, pz, ene}
+	return fmom.NewPxPyPzE(px, py, pz, ene)
 }

fmom/eetaphim.go

@@ -8,10 +8,12 @@ import (
 	"math"
 )
 
-type EEtaPhiM [4]float64
+type EEtaPhiM struct {
+	P4 Vec4
+}
 
-func NewEEtaPhiM(et, eta, phi, m float64) EEtaPhiM {
-	return EEtaPhiM([4]float64{et, eta, phi, m})
+func NewEEtaPhiM(e, eta, phi, m float64) EEtaPhiM {
+	return EEtaPhiM{P4: Vec4{X: e, Y: eta, Z: phi, T: m}}
 }
 
 func (p4 *EEtaPhiM) Clone() P4 {
@@ -20,19 +22,19 @@ func (p4 *EEtaPhiM) Clone() P4 {
 }
 
 func (p4 *EEtaPhiM) E() float64 {
-	return p4[0]
+	return p4.P4.X
 }
 
 func (p4 *EEtaPhiM) Eta() float64 {
-	return p4[1]
+	return p4.P4.Y
 }
 
 func (p4 *EEtaPhiM) Phi() float64 {
-	return p4[2]
+	return p4.P4.Z
 }
 
 func (p4 *EEtaPhiM) M() float64 {
-	return p4[3]
+	return p4.P4.T
 }
 
 func (p4 *EEtaPhiM) M2() float64 {
@@ -143,8 +145,8 @@ func (p4 *EEtaPhiM) Pz() float64 {
 }
 
 func (p4 *EEtaPhiM) Set(p P4) {
-	p4[0] = p.E()
-	p4[1] = p.Eta()
-	p4[2] = p.Phi()
-	p4[3] = p.M()
+	p4.P4.X = p.E()
+	p4.P4.Y = p.Eta()
+	p4.P4.Z = p.Phi()
+	p4.P4.T = p.M()
 }

fmom/etetaphim.go

@@ -8,10 +8,12 @@ import (
 	"math"
 )
 
-type EtEtaPhiM [4]float64
+type EtEtaPhiM struct {
+	P4 Vec4
+}
 
 func NewEtEtaPhiM(et, eta, phi, m float64) EtEtaPhiM {
-	return EtEtaPhiM([4]float64{et, eta, phi, m})
+	return EtEtaPhiM{P4: Vec4{X: et, Y: eta, Z: phi, T: m}}
 }
 
 func (p4 *EtEtaPhiM) Clone() P4 {
@@ -20,19 +22,19 @@ func (p4 *EtEtaPhiM) Clone() P4 {
 }
 
 func (p4 *EtEtaPhiM) Et() float64 {
-	return p4[0]
+	return p4.P4.X
 }
 
 func (p4 *EtEtaPhiM) Eta() float64 {
-	return p4[1]
+	return p4.P4.Y
 }
 
 func (p4 *EtEtaPhiM) Phi() float64 {
-	return p4[2]
+	return p4.P4.Z
 }
 
 func (p4 *EtEtaPhiM) M() float64 {
-	return p4[3]
+	return p4.P4.T
 }
 
 func (p4 *EtEtaPhiM) M2() float64 {
@@ -143,8 +145,8 @@ func (p4 *EtEtaPhiM) Pz() float64 {
 }
 
 func (p4 *EtEtaPhiM) Set(p P4) {
-	p4[0] = p.Et()
-	p4[1] = p.Eta()
-	p4[2] = p.Phi()
-	p4[3] = p.M()
+	p4.P4.X = p.Et()
+	p4.P4.Y = p.Eta()
+	p4.P4.Z = p.Phi()
+	p4.P4.T = p.M()
 }

fmom/iptcotthphim.go

@@ -8,10 +8,12 @@ import (
 	"math"
 )
 
-type IPtCotThPhiM [4]float64
+type IPtCotThPhiM struct {
+	P4 Vec4
+}
 
 func NewIPtCotThPhiM(pt, eta, phi, m float64) IPtCotThPhiM {
-	return IPtCotThPhiM([4]float64{pt, eta, phi, m})
+	return IPtCotThPhiM{P4: Vec4{X: pt, Y: eta, Z: phi, T: m}}
 }
 
 func (p4 *IPtCotThPhiM) Clone() P4 {
@@ -20,19 +22,19 @@ func (p4 *IPtCotThPhiM) Clone() P4 {
 }
 
 func (p4 *IPtCotThPhiM) IPt() float64 {
-	return p4[0]
+	return p4.P4.X
 }
 
 func (p4 *IPtCotThPhiM) CotTh() float64 {
-	return p4[1]
+	return p4.P4.Y
 }
 
 func (p4 *IPtCotThPhiM) Phi() float64 {
-	return p4[2]
+	return p4.P4.Z
 }
 
 func (p4 *IPtCotThPhiM) M() float64 {
-	return p4[3]
+	return p4.P4.T
 }
 
 func (p4 *IPtCotThPhiM) Pt() float64 {
@@ -137,8 +139,8 @@ func (p4 *IPtCotThPhiM) SinPhi() float64 {
 }
 
 func (p4 *IPtCotThPhiM) Set(p P4) {
-	p4[0] = p.IPt()
-	p4[1] = p.CotTh()
-	p4[2] = p.Phi()
-	p4[3] = p.M()
+	p4.P4.X = p.IPt()
+	p4.P4.Y = p.CotTh()
+	p4.P4.Z = p.Phi()
+	p4.P4.T = p.M()
 }

fmom/ops.go

@@ -110,10 +110,10 @@ func IAdd(dst, src P4) P4 {
 	default:
 		panic(fmt.Errorf("fmom: invalid P4 concrete value: %#v", dst))
 	}
-	p4[0] += src.Px()
-	p4[1] += src.Py()
-	p4[2] += src.Pz()
-	p4[3] += src.E()
+	p4.P4.X += src.Px()
+	p4.P4.Y += src.Py()
+	p4.P4.Z += src.Pz()
+	p4.P4.T += src.E()
 	sum.Set(p4)
 	return sum
 }

fmom/p4.go

@@ -30,3 +30,8 @@ type P4 interface {
 	Set(p4 P4)
 	Clone() P4
 }
+
+// Vec4 holds the four components of a Lorentz vector.
+type Vec4 struct {
+	X, Y, Z, T float64
+}

fmom/ptetaphim.go

@@ -8,10 +8,12 @@ import (
 	"math"
 )
 
-type PtEtaPhiM [4]float64
+type PtEtaPhiM struct {
+	P4 Vec4
+}
 
 func NewPtEtaPhiM(pt, eta, phi, m float64) PtEtaPhiM {
-	return PtEtaPhiM([4]float64{pt, eta, phi, m})
+	return PtEtaPhiM{P4: Vec4{X: pt, Y: eta, Z: phi, T: m}}
 }
 
 func (p4 *PtEtaPhiM) Clone() P4 {
@@ -20,19 +22,19 @@ func (p4 *PtEtaPhiM) Clone() P4 {
 }
 
 func (p4 *PtEtaPhiM) Pt() float64 {
-	return p4[0]
+	return p4.P4.X
 }
 
 func (p4 *PtEtaPhiM) Eta() float64 {
-	return p4[1]
+	return p4.P4.Y
 }
 
 func (p4 *PtEtaPhiM) Phi() float64 {
-	return p4[2]
+	return p4.P4.Z
 }
 
 func (p4 *PtEtaPhiM) M() float64 {
-	return p4[3]
+	return p4.P4.T
 }
 
 func (p4 *PtEtaPhiM) E() float64 {
@@ -150,8 +152,8 @@ func (p4 *PtEtaPhiM) Pz() float64 {
 }
 
 func (p4 *PtEtaPhiM) Set(p P4) {
-	p4[0] = p.Pt()
-	p4[1] = p.Eta()
-	p4[2] = p.Phi()
-	p4[3] = p.M()
+	p4.P4.X = p.Pt()
+	p4.P4.Y = p.Eta()
+	p4.P4.Z = p.Phi()
+	p4.P4.T = p.M()
 }

fmom/pxpypze.go

@@ -8,48 +8,28 @@ import (
 	"math"
 )
 
-type PxPyPzE [4]float64
+type PxPyPzE struct {
+	P4 Vec4
+}
 
 func NewPxPyPzE(px, py, pz, e float64) PxPyPzE {
-	return PxPyPzE([4]float64{px, py, pz, e})
+	return PxPyPzE{P4: Vec4{X: px, Y: py, Z: pz, T: e}}
 }
 
 func (p4 *PxPyPzE) Clone() P4 {
 	pp := *p4
 	return &pp
 }
 
-func (p4 *PxPyPzE) Px() float64 {
-	return p4[0]
-}
-
-func (p4 *PxPyPzE) Py() float64 {
-	return p4[1]
-}
-
-func (p4 *PxPyPzE) Pz() float64 {
-	return p4[2]
-}
-
-func (p4 *PxPyPzE) E() float64 {
-	return p4[3]
-}
+func (p4 *PxPyPzE) Px() float64 { return p4.P4.X }
+func (p4 *PxPyPzE) Py() float64 { return p4.P4.Y }
+func (p4 *PxPyPzE) Pz() float64 { return p4.P4.Z }
+func (p4 *PxPyPzE) E() float64  { return p4.P4.T }
 
-func (p4 *PxPyPzE) X() float64 {
-	return p4[0]
-}
-
-func (p4 *PxPyPzE) Y() float64 {
-	return p4[1]
-}
-
-func (p4 *PxPyPzE) Z() float64 {
-	return p4[2]
-}
-
-func (p4 *PxPyPzE) T() float64 {
-	return p4[3]
-}
+func (p4 *PxPyPzE) X() float64 { return p4.P4.X }
+func (p4 *PxPyPzE) Y() float64 { return p4.P4.Y }
+func (p4 *PxPyPzE) Z() float64 { return p4.P4.Z }
+func (p4 *PxPyPzE) T() float64 { return p4.P4.T }
 
 func (p4 *PxPyPzE) M2() float64 {
 	px := p4.Px()
@@ -206,8 +186,8 @@ func (p4 *PxPyPzE) Rapidity() float64 {
 }
 
 func (p4 *PxPyPzE) Set(p P4) {
-	p4[0] = p.Px()
-	p4[1] = p.Py()
-	p4[2] = p.Pz()
-	p4[3] = p.E()
+	p4.P4.X = p.Px()
+	p4.P4.Y = p.Py()
+	p4.P4.Z = p.Pz()
+	p4.P4.T = p.E()
 }