Make undo restore selection
Imagine I have a document open and I have carefully selected a region that I want to cut and paste somewhere else. I press ctrl-x to cut it, but then I notice that I've selected one too many bytes. I undo the operation, which restores the deleted bytes, but now my selection is cleared. I have to carefully select the region again and hopefully not make the same mistake.
Instead, I would prefer that the selection is restored so that I can easily modify the selection with shift-left or shift-right, then try again.