hard code TLeafB

JuliaHEP · Jul 2, 2021 · 5474dc6 · 5474dc6
1 parent c40da32
commit 5474dc6
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 0 deletions.
diff --git a/src/bootstrap.jl b/src/bootstrap.jl
@@ -264,6 +264,41 @@ end
 
 primitivetype(l::TLeafF) = Float32
 
+# FIXME this should be generated and inherited from TLeaf
+# https://root.cern/doc/master/TLeafB_8h_source.html#l00026
+@with_kw struct TLeafB
+    # from TNamed
+    fName
+    fTitle
+
+    # from TLeaf
+    fLen
+    fLenType
+    fOffset
+    fIsRange
+    fIsUnsigned
+    fLeafCount
+
+    # own fields
+    fMinimum
+    fMaximum
+end
+
+function parsefields!(io, fields, T::Type{TLeafB})
+    preamble = Preamble(io, T)
+    parsefields!(io, fields, TLeaf)
+    fields[:fMinimum] = readtype(io, UInt8)
+    fields[:fMaximum] = readtype(io, UInt8)
+    endcheck(io, preamble)
+end
+
+function unpack(io, tkey::TKey, refs::Dict{Int32, Any}, T::Type{TLeafB})
+    @initparse
+    parsefields!(io, fields, T)
+    T(;fields...)
+end
+
+primitivetype(l::TLeafB) = UInt8
 # FIXME this should be generated and inherited from TLeaf
 @with_kw struct TLeafD
     # from TNamed

diff --git a/test/samples/NanoAODv5_sample.root b/test/samples/NanoAODv5_sample.root