Add hooks into BoxHI.set_zdir_grid to save and resume computation, which default to doing nothing