Skip to content

tp: add tree_propagate_down operator for tree transformations#5257

Merged
LalitMaganti merged 63 commits intomainfrom
dev/lalitm/tree-propagate-down
Apr 1, 2026
Merged

tp: add tree_propagate_down operator for tree transformations#5257
LalitMaganti merged 63 commits intomainfrom
dev/lalitm/tree-propagate-down

Conversation

@LalitMaganti
Copy link
Copy Markdown
Member

@LalitMaganti LalitMaganti commented Mar 24, 2026

Summary

  • Add PropagateDown() method to TreeTransformer for root-to-leaf BFS propagation
  • Support SUM, MIN, MAX, FIRST, LAST aggregate operations
  • Propagated columns become real columns enabling filter→propagate→filter chains
  • PropagateTreeDown bytecode takes spec_start/spec_count with separate source/dest columns

Stack

Test plan

  • 13 new propagate diff tests including chained propagation and filter→propagate→filter
  • All 23 tree tests pass

@LalitMaganti LalitMaganti requested a review from a team as a code owner March 24, 2026 19:29
@LalitMaganti LalitMaganti marked this pull request as draft March 24, 2026 19:55
@LalitMaganti LalitMaganti force-pushed the dev/lalitm/tree-propagate-down branch from e7a3b5d to d3ea725 Compare March 24, 2026 21:44
@LalitMaganti LalitMaganti changed the base branch from main to dev/lalitm/tree-columns-refactor March 24, 2026 21:46
@LalitMaganti LalitMaganti force-pushed the dev/lalitm/tree-columns-refactor branch from 8530a03 to 9dd318b Compare March 24, 2026 21:49
@LalitMaganti LalitMaganti force-pushed the dev/lalitm/tree-propagate-down branch from d3ea725 to 3e443d7 Compare March 24, 2026 21:49
Add TreeColumns, a simple columnar storage type for tree data with
a normalized parent array and dense null bitvectors. Add
TreeColumnsBuilder which wraps RuntimeDataframeBuilder for type
inference and null tracking, then converts the result to TreeColumns
via a new BuildRaw() short-circuit on AdhocDataframeBuilder that
returns raw FlexVectors + BitVectors without sort analysis, type
downcasting, or Dataframe creation.

TreeTransformer now takes TreeColumns instead of Dataframe, and
TreeFromTable builds TreeColumns directly. This eliminates the need
to destructure a Dataframe in the TreeTransformer constructor and
provides a cleaner, tree-specific data representation.
Add a PropagateDown() method to TreeTransformer that propagates
column values from root toward leaves via BFS. Each spec copies a
source column into a new output column and applies an aggregate
operation (SUM, MIN, MAX, FIRST, LAST) downward through the tree.

Propagated columns become real columns in the transformer, so
subsequent FilterTree() calls can reference them, enabling
filter→propagate→filter chains.

The PropagateTreeDown bytecode now takes spec_start/spec_count to
process a contiguous range of specs, and each spec has separate
source/dest column indices to support the copy-then-propagate
pattern.
@LalitMaganti LalitMaganti force-pushed the dev/lalitm/tree-columns-refactor branch from 9dd318b to 1566928 Compare March 24, 2026 21:56
@LalitMaganti LalitMaganti force-pushed the dev/lalitm/tree-propagate-down branch from 3e443d7 to f76fd52 Compare March 24, 2026 21:56
@LalitMaganti LalitMaganti marked this pull request as ready for review March 30, 2026 18:14
LalitMaganti added a commit that referenced this pull request Mar 30, 2026
## Summary
- Add `TreeColumns`, a simple columnar storage type for tree data with
normalized parent array and dense null bitvectors
- Add `TreeColumnsBuilder` which wraps `RuntimeDataframeBuilder` for
type inference/null tracking, then converts via a new `BuildRaw()`
short-circuit that skips sort analysis, type downcasting, and Dataframe
creation
- `TreeTransformer` now takes `TreeColumns` instead of `Dataframe`

## Stack
- **#5262 tp: introduce TreeColumns and refactor TreeTransformer to use
it** (this PR)
- #5257 tp: add tree_propagate_down operator for tree transformations

## Test plan
- Existing tree filter diff tests pass (10 tests)
Base automatically changed from dev/lalitm/tree-columns-refactor to main March 30, 2026 22:12
@LalitMaganti LalitMaganti enabled auto-merge (squash) April 1, 2026 13:14
@LalitMaganti LalitMaganti merged commit 0a9d51e into main Apr 1, 2026
24 checks passed
@LalitMaganti LalitMaganti deleted the dev/lalitm/tree-propagate-down branch April 1, 2026 13:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants