hard code TLeafB

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

Hide file tree

Showing 5 changed files with 36 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -6,3 +6,4 @@
 /dev/
 /docs/build/
 /docs/site/
+__pycache__/
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
diff --git a/test/samples/__pycache__/lz4.cpython-39.pyc b/test/samples/__pycache__/lz4.cpython-39.pyc
diff --git a/test/samples/tree_with_large_array_lz4.root b/test/samples/tree_with_large_array_lz4.root