ROOT Map() showing errors for any written TTrees with TBaskets

[ioanaif]

An issue was introduced in between uproot 4.1.1 and 4.1.5, where ROOT .Map() shows errors for any TTrees with TBaskets (where the TBaskets can be empty or not).

>>> import uproot
>>> import ROOT
>>> import numpy as np
>>> uproot.__version__
5.0.10
>>> file = uproot.recreate("simple_reproducer.root")
>>> file["tree"] = {"some_array": np.array([1,2,3])}
>>> rf = ROOT.TFile("simple_reproducer.root")
>>> rf.Map()
20230810/162540  At:100     N=142       TFile                     
20230810/162557  At:242     N=101       TBasket                   
Address = 343	Nbytes = 0	=====E R R O R=======
0/000000  At:343     N=1         END           

[ioanaif]

I looked into this error and attempted to fix it. I didn’t find the fix, but I did isolate where the issue is coming from. Thus, I will write here my findings. 




The problem is in writable.py:

uproot5/src/uproot/writing/writable.py

Lines 1323 to 1334 in c128e4b

	for model in (
	uproot.models.TLeaf.Model_TLeafB_v1,
	uproot.models.TLeaf.Model_TLeafS_v1,
	uproot.models.TLeaf.Model_TLeafI_v1,
	uproot.models.TLeaf.Model_TLeafL_v1,
	uproot.models.TLeaf.Model_TLeafF_v1,
	uproot.models.TLeaf.Model_TLeafD_v1,
	uproot.models.TLeaf.Model_TLeafC_v1,
	uproot.models.TLeaf.Model_TLeafO_v1,
	uproot.models.TBranch.Model_TBranch_v13,
	uproot.models.TTree.Model_TTree_v20,
	):

If we only have L1333, i.e. uproot.models.TTree.Model_TTree_v20 in the list of models, then the Map() issue disappears.

Thus, the issue comes from some of the streamers in TLeaf and TBranch. In each TLeaf type the issue goes away if we comment out this part:

uproot5/src/uproot/models/TLeaf.py

Lines 185 to 192 in c128e4b

	_rawstreamer_TLeaf_v2,
	(
	None,
	b"@\x00\x02\x14\xff\xff\xff\xffTStreamerInfo\x00@\x00\x01\xfe\x00\t@\x00\x00\x14\x00\x01\x00\x01\x00\x00\x00\x00\x03\x01\x00\x00\x06TLeafB\x00\x0f\x1eK^\x00\x00\x00\x01@\x00\x01\xd8\xff\xff\xff\xffTObjArray\x00@\x00\x01\xc6\x00\x03\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x03\x00\x00\x00\x00@\x00\x00\x89\xff\xff\xff\xffTStreamerBase\x00@\x00\x00s\x00\x03@\x00\x00i\x00\x04@\x00\x00:\x00\x01\x00\x01\x00\x00\x00\x00\x03\x00\x00\x00\x05TLeaf'Leaf: description of a Branch data type\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00m\x1e\x81R\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x04BASE\x00\x00\x00\x02@\x00\x00\x8e\xff\xff\xff\xffTStreamerBasicType\x00@\x00\x00s\x00\x02@\x00\x00m\x00\x04@\x00\x00>\x00\x01\x00\x01\x00\x00\x00\x00\x03\x00\x00\x00\x08fMinimum(Minimum value if leaf range is specified\x00\x00\x00\x01\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x04char@\x00\x00\x8e\xff\xff\xff\xffTStreamerBasicType\x00@\x00\x00s\x00\x02@\x00\x00m\x00\x04@\x00\x00>\x00\x01\x00\x01\x00\x00\x00\x00\x03\x00\x00\x00\x08fMaximum(Maximum value if leaf range is specified\x00\x00\x00\x01\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x04char\x00",
	"TLeafB",
	1,
	),
	)

Where _rawstreamer_TLeaf_v2 is defined here:

uproot5/src/uproot/models/TLeaf.py

Lines 14 to 19 in c128e4b

	_rawstreamer_TLeaf_v2 = (
	None,
	b"@\x00\x05\x04\xff\xff\xff\xffTStreamerInfo\x00@\x00\x04\xee\x00\t@\x00\x00\x13\x00\x01\x00\x01\x00\x00\x00\x00\x03\x01\x00\x00\x05TLeaf\x00m\x1e\x81R\x00\x00\x00\x02@\x00\x04\xc9\xff\xff\xff\xffTObjArray\x00@\x00\x04\xb7\x00\x03\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x07\x00\x00\x00\x00@\x00\x00\x8d\xff\xff\xff\xffTStreamerBase\x00@\x00\x00w\x00\x03@\x00\x00m\x00\x04@\x00\x00>\x00\x01\x00\x01\x00\x00\x00\x00\x03\x00\x00\x00\x06TNamed*The basis for a named object (name, title)\x00\x00\x00C\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xdf\xb7J<\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x04BASE\x00\x00\x00\x01@\x00\x00\x94\xff\xff\xff\xffTStreamerBasicType\x00@\x00\x00y\x00\x02@\x00\x00s\x00\x04@\x00\x00E\x00\x01\x00\x01\x00\x00\x00\x00\x03\x00\x00\x00\x04fLen3Number of fixed length elements in the leaf's data.\x00\x00\x00\x03\x00\x00\x00\x04\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x03int@\x00\x00\x87\xff\xff\xff\xffTStreamerBasicType\x00@\x00\x00l\x00\x02@\x00\x00f\x00\x04@\x00\x008\x00\x01\x00\x01\x00\x00\x00\x00\x03\x00\x00\x00\x08fLenType\"Number of bytes for this data type\x00\x00\x00\x03\x00\x00\x00\x04\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x03int@\x00\x00\x89\xff\xff\xff\xffTStreamerBasicType\x00@\x00\x00n\x00\x02@\x00\x00h\x00\x04@\x00\x00:\x00\x01\x00\x01\x00\x00\x00\x00\x03\x00\x00\x00\x07fOffset%Offset in ClonesArray object (if one)\x00\x00\x00\x03\x00\x00\x00\x04\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x03int@\x00\x01\x14\xff\xff\xff\xffTStreamerBasicType\x00@\x00\x00\xf9\x00\x02@\x00\x00\xf3\x00\x04@\x00\x00\xc4\x00\x01\x00\x01\x00\x00\x00\x00\x03\x00\x00\x00\x08fIsRange\xae(=kTRUE if leaf has a range, kFALSE otherwise). This is equivalent to being a 'leafcount'. For a TLeafElement the range information is actually store in the TBranchElement.\x00\x00\x00\x12\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x04bool@\x00\x00\x8f\xff\xff\xff\xffTStreamerBasicType\x00@\x00\x00t\x00\x02@\x00\x00n\x00\x04@\x00\x00?\x00\x01\x00\x01\x00\x00\x00\x00\x03\x00\x00\x00\x0bfIsUnsigned&(=kTRUE if unsigned, kFALSE otherwise)\x00\x00\x00\x12\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x04bool@\x00\x00\xb2\xff\xff\xff\xffTStreamerObjectPointer\x00@\x00\x00\x93\x00\x02@\x00\x00\x8d\x00\x04@\x00\x00\\\x00\x01\x00\x01\x00\x00\x00\x00\x03\x00\x00\x00\nfLeafCountDPointer to Leaf count if variable length (we do not own the counter)\x00\x00\x00@\x00\x00\x00\x08\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x06TLeaf*\x00",
	"TLeaf",
	2,
	)

As for TBranch, the issue is fixed when commenting out lines 562 to 568:

uproot5/src/uproot/models/TBranch.py

Lines 559 to 569 in c128e4b

	class_rawstreamers = (
	uproot.models.TH._rawstreamer_TCollection_v3,
	uproot.models.TH._rawstreamer_TSeqCollection_v0,
	uproot.models.TObjArray._rawstreamer_TObjArray_v3,
	_rawstreamer_ROOT_3a3a_TIOFeatures_v1,
	uproot.models.TH._rawstreamer_TAttFill_v2,
	uproot.models.TH._rawstreamer_TString_v2,
	uproot.models.TH._rawstreamer_TObject_v1,
	uproot.models.TH._rawstreamer_TNamed_v1,
	_rawstreamer_TBranch_v13,
	)