groot/rbytes: add RBuffer.ReadStdVectorI32

go-hep · Jan 31, 2022 · d8d3eb5 · d8d3eb5
1 parent 1737316
commit d8d3eb5
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/groot/rbytes/rbuffer.go b/groot/rbytes/rbuffer.go
@@ -456,6 +456,28 @@ func (r *RBuffer) ReadArrayString(arr []string) {
 	}
 }
 
+func (r *RBuffer) ReadStdVectorI32(sli *[]int32) {
+	if r.err != nil {
+		return
+	}
+	const typename = "vector<int>"
+	beg := r.Pos()
+	vers, pos, bcnt := r.ReadVersion(typename)
+	if vers != rvers.StreamerInfo {
+		r.err = fmt.Errorf(
+			"rbytes: invalid %s version: got=%d, want=%d",
+			typename, vers, rvers.StreamerInfo,
+		)
+		return
+	}
+	n := int(r.ReadI32())
+	*sli = ResizeI32(*sli, n)
+	for i := range *sli {
+		(*sli)[i] = r.ReadI32()
+	}
+	r.CheckByteCount(pos, bcnt, beg, typename)
+}
+
 func (r *RBuffer) ReadStdVectorF64(sli *[]float64) {
 	if r.err != nil {
 		return