Evacuation: extract the inner fold function

This makes the code more readable, which will help with the
multi-group evacuation.

Signed-off-by: Iustin Pop <iustin@google.com>
Reviewed-by: Balazs Lecz <leczb@google.com>