root_mcp.core.tools.data_access module

Data access tools for reading TTree or RNTuple data.

class root_mcp.core.tools.data_access.DataAccessTools(config, file_manager, path_validator, tree_reader)

Bases: object

Tools for accessing TTree or RNTuple data.

Parameters:

• config (Config)

• file_manager (FileManager)

• path_validator (PathValidator)

• tree_reader (TreeReader)

__init__(config, file_manager, path_validator, tree_reader)

Initialize data access tools.

Parameters:

• config (Config) – Server configuration

• file_manager (FileManager) – File manager instance

• path_validator (PathValidator) – Path validator instance

• tree_reader (TreeReader) – Tree reader instance

get_branch_stats(path, tree, branches, selection=None)

Compute statistics for branches.

Parameters:

• path (str) – File path

• tree (str) – Tree name

• branches (list[str]) – Branches to analyze

• selection (str | None) – Optional cut expression

Returns:

Branch statistics

Return type:

dict[str, Any]

read_branches(path, tree_name, branches, selection=None, limit=None, offset=0, entry_start=None, entry_stop=None, flatten=False, defines=None)

Read branch data from a TTree or RNTuple.

Parameters:

• path (str) – File path

• tree_name (str) – Tree name

• branches (list[str]) – List of branch names (can include derived branches from defines)

• selection (str | None) – Optional cut expression

• limit (int | None) – Maximum entries to return (alternative to entry_stop)

• offset (int) – Number of entries to skip (alternative to entry_start)

• entry_start (int | None) – Start entry index (alternative to offset)

• entry_stop (int | None) – Stop entry index (alternative to limit)

• flatten (bool) – Flatten jagged arrays

• defines (dict[str, str] | None) – Optional derived variable definitions {name: expression}

Returns:

Branch data and metadata

Return type:

dict[str, Any]

sample_tree(path, tree, size=100, method='first', branches=None, seed=None)

Get a sample from a tree.

Parameters:

• path (str) – File path

• tree (str) – Tree name

• size (int) – Sample size

• method (str) – “first” or “random”

• branches (list[str] | None) – Branches to include (None = all)

• seed (int | None) – Random seed

Returns:

Sample data and metadata

Return type:

dict[str, Any]