Home

Collecting thoughts on a "Uproot 3 → Uproot 4" cheat sheet/migration guide

Everyone is welcome to contribute!

(I'll write that documentation, but anything you want to make sure gets into such a page, dump it here!)

Custom interpretations

Whenever the automatic serialisation of uproot fails for whatever reason, a custom interpretation comes to the rescue. The concept remained the same but a few details have been changed. Here is an example how a custom jagged array interpretation was utilised in uproot3:

uproot3

import uproot  # version 3
from skhep_testdata import data_path

f = uproot.open(data_path("uproot-issue124.root"))

tree = f["KM3NET_EVENT"]

snapshot_hits = tree["snapshotHits"].array(
    uproot.asjagged(
        uproot.astable(
            uproot.asdtype(
                [
                    ("dom_id", ">i4"),
                    ("channel_id", "u1"),
                    ("time", "<u4"),
                    ("tot", "u1"),
                ]
            )
        ),
        skipbytes=10,
    )
)

This will return a JaggedArray:

>>> snapshot_hits
<JaggedArray [[<Row 0> <Row 1> <Row 2> ... <Row 50> <Row 51> <Row 52>]  ...  [<Row 849> <Row 850> <Row 851> ... <Row 887> <Row 888> <Row 889>] [<Row 890> <Row 891> <Row 892> ... <Row 920> <Row 921> <Row 922>]] at 0x7f9b8e6c89d0>

>>> snapshot_hits.dom_id
<JaggedArray [[808432835 808432835 808432835 ... 809526097 809526097 809526097]   ...  [808432835 808488997 808488997 ... 809526097 809526097 809544061] [808432835 808432835 808432835 ... 809526097 809526097 809544061]] at 0x7f9bc99b05e0>

uproot4

import uproot4 as uproot  # version 4
from skhep_testdata import data_path

f = uproot.open(data_path("uproot-issue124.root"))

tree = f["KM3NET_EVENT"]

snapshot_hits = tree["snapshotHits"].array(
    uproot4.interpretation.jagged.AsJagged(
        uproot4.interpretation.numerical.AsDtype(
            [
                ("dom_id", ">i4"),
                ("channel_id", "u1"),
                ("time", "<u4"),
                ("tot", "u1"),
            ]
        ), header_bytes=10,
    )
)

Which will return an awkward1.Array:

>>> snapshot_hits
<Array [[{dom_id: 808432835, ... tot: 30}]] type='23 * var * {"dom_id": int32, "...'>