def select_blocks(self, selector): mask = self.oct_handler.mask(selector, domain_id = self.domain_id) slicer = OctreeSubsetBlockSlice(self) for i, sl in slicer: yield sl, np.atleast_3d(mask[i,...])