root/rbytes: add WriteObject

go-hep · Feb 1, 2022 · e488e92 · e488e92
1 parent fa6827e
commit e488e92
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/groot/rbytes/wbuffer.go b/groot/rbytes/wbuffer.go
@@ -154,6 +154,20 @@ func (w *WBuffer) WriteVersion(vers int16) int64 {
 	return pos
 }
 
+func (w *WBuffer) WriteObject(obj root.Object) error {
+	if w.err != nil {
+		return w.err
+	}
+
+	if v := reflect.ValueOf(obj); (v == reflect.Value{}) || v.IsNil() {
+		w.WriteU32(0) // NULL pointer
+		return w.err
+	}
+
+	_, w.err = obj.(Marshaler).MarshalROOT(w)
+	return w.err
+}
+
 func (w *WBuffer) WriteObjectAny(obj root.Object) error {
 	if w.err != nil {
 		return w.err