Return branch element if end of braching reached

JuliaHEP · Dec 23, 2022 · 0fead00 · 0fead00
1 parent ec5c883
commit 0fead00
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/displays.jl b/src/displays.jl
@@ -47,7 +47,13 @@ function children(t::TTree)
     end
 end
 children(t::Union{TTree, TBranchElement}) = t.fBranches
-Base.show(io::IO, ::MIME"text/plain", b::Union{TTree, TBranchElement}) = print_tree(io, b)
+function Base.show(io::IO, ::MIME"text/plain", b::Union{TTree, TBranchElement})
+    if isempty(b.fBranches)
+        print(io, b)
+    else
+        print_tree(io, b)
+    end
+end
 printnode(io::IO, t::TBranchElement) = print(io, "$(t.fName)")
 printnode(io::IO, t::TTree) = print(io, "$(t.fName) (TTree)")
 printnode(io::IO, f::ROOTFile) = print(io, f.filename)