groot/rphys: make LorentzVector implement fmt.Stringer

go-hep · Dec 16, 2021 · 7851644 · 7851644
1 parent 07e2951
commit 7851644
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/groot/rphys/lorentzvector.go b/groot/rphys/lorentzvector.go
@@ -5,6 +5,7 @@
 package rphys
 
 import (
+	"fmt"
 	"reflect"
 
 	"go-hep.org/x/hep/groot/rbase"
@@ -90,6 +91,14 @@ func (vec *LorentzVector) UnmarshalROOT(r *rbytes.RBuffer) error {
 	return r.Err()
 }
 
+func (vec *LorentzVector) String() string {
+	return fmt.Sprintf(
+		"TLorentzVector{P: {%v, %v, %v}, E: %v}",
+		vec.p.x, vec.p.y, vec.p.z,
+		vec.e,
+	)
+}
+
 func init() {
 	{
 		f := func() reflect.Value {