[layouts] improve tree_manager ergonomics