diff --git a/boxtree/traversal.py b/boxtree/traversal.py index c42189d833d2243048b0fac62aec88fb7f38e3bc..f7e7045ede4f1b3e29738b7012c0f7e49863e2ff 100644 --- a/boxtree/traversal.py +++ b/boxtree/traversal.py @@ -1364,6 +1364,14 @@ class FMMTraversalInfo(DeviceDataRecord): An instance of :class:`boxtree.Tree`. + .. attribute:: nboxes + + Number of boxe in the tree. + + .. attribute:: nlevels + + Number of levels in the tree. + .. attribute:: well_sep_is_n_away The distance (measured in target box diameters in the :math:`l^\infty` @@ -1388,6 +1396,10 @@ class FMMTraversalInfo(DeviceDataRecord): If :attr:`boxtree.Tree.sources_are_targets`, then ``target_boxes is source_boxes``. + .. attribute:: ntarget_boxes + + Number of :attr:`target_boxes`. + .. attribute:: source_parent_boxes ``box_id_t [*]`` @@ -1667,6 +1679,18 @@ class FMMTraversalInfo(DeviceDataRecord): # }}} + @property + def nboxes(self): + return self.tree.nboxes + + @property + def nlevels(self): + return self.tree.nlevels + + @property + def ntarget_boxes(self): + return len(self.target_boxes) + @property def ntarget_or_target_parent_boxes(self): return len(self.target_or_target_parent_boxes)