Floating group: base current head on window

Instead of always returning the first head on the screen as the current
one, it makes more sense to instead return the head on which the current
window resides.